fix: 带 date 查任务时区错位导致重复生成 + 评论弹窗重做 + AI 每日次数上限
## 首页添加的任务不显示(根因比表象严重)
time.Parse("2006-01-02", q) 返回的是 UTC 时间,dayStart 又保留了
t.Location(),于是带 ?date= 查询时:
查询区间 = 07-29 00:00 UTC ~ 次日 = CST 的 08:00 ~ 次日 08:00
已有任务的 task_date 是 07-29 00:00 CST,落在区间外
后果不只是「新任务不显示」——ensureDayTasks 因此认为今天没有任务,
每刷一次首页就重新生成一批。用户手动加的那条(00:00 CST)永远不在
那个错位窗口里,所以管理页看得到、首页看不到。
全项目 7 处 time.Parse 日期解析统一换成 ParseInLocation + time.Local
(任务、日计划、生日、到家日期、提醒到期日都受影响)。
实测:加一条后带 date 查得到 4 条,连查 4 次仍是 4 条不再增长。
## 评论弹窗
- 改成居中弹出。评论以输入为主,贴底弹层会被键盘顶掉大半屏
- 单行 input 换成自动撑高的 textarea,最多 500 字,带字数
- 支持配图,最多 3 张(评论表加 images 字段)
- 空评论原来会直接把弹层关掉,看起来像发成功了,改成明确提示
- 发完就地刷新列表,不再关闭弹层——连着回复更顺
## AI 每日次数上限
AI 调用是真金白银,不设上限等于把钱包交给用户。新增按「用户 + 自然日
+ 功能」计的额度,后台「社区运营」页可配问问 AI / 异常评估 / AI 计划
三档,改完立即生效。
两个刻意的设计:
- 扣额度在请求大模型之前,失败也算用掉一次。否则刷接口空转照样烧钱
- 配置读不到时回落到保守默认值,绝不「读不到就不限制」——那正是配置
出问题时最不该发生的事
超限返回业务码 42900,小程序端明确说明「明天 0 点恢复」,不当成网络
错误让用户反复重试。
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
@@ -8,6 +8,10 @@ import (
|
||||
|
||||
// AIChat 记录一问一答:启用模型则真调(注入宠物档案),否则规则化文案
|
||||
func (s *Service) AIChat(userID string, petID *string, session, text string) (string, error) {
|
||||
// 先扣额度再请求大模型:失败也算用掉一次,否则刷接口空转照样烧钱
|
||||
if err := s.consumeAIQuota(userID, aiKindChat); err != nil {
|
||||
return "", err
|
||||
}
|
||||
reply := "我会先判断风险等级,再建议你记录关键观察项。若出现频繁呕吐、便血、精神明显变差或持续超过 24 小时,建议尽快就医。"
|
||||
if s.ai != nil && s.ai.Enabled() {
|
||||
if r, err := s.llmChat(petID, text); err == nil && r != "" {
|
||||
|
||||
@@ -68,6 +68,9 @@ type SymptomResult struct {
|
||||
|
||||
// AssessSymptom 异常风险评估:启用模型则结构化输出,否则规则化
|
||||
func (s *Service) AssessSymptom(userID, petID string, in SymptomInput) (*SymptomResult, error) {
|
||||
if err := s.consumeAIQuota(userID, aiKindSymptom); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if _, err := s.ownedPet(userID, petID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
@@ -0,0 +1,110 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/sundynix/pets-be/internal/model"
|
||||
)
|
||||
|
||||
// ErrAIQuotaExceeded 今日 AI 次数用完
|
||||
var ErrAIQuotaExceeded = errors.New("今日 AI 次数已用完")
|
||||
|
||||
// AI 调用是真金白银,不设上限等于把钱包交给用户。
|
||||
// 额度按「用户 + 自然日」计,从后台配置读,改完立刻生效。
|
||||
const (
|
||||
aiKindChat = "chat" // 问问 AI
|
||||
aiKindSymptom = "symptom" // 异常观察评估
|
||||
aiKindPlan = "plan" // AI 生成计划
|
||||
)
|
||||
|
||||
// aiQuotaConfig 从后台配置取每日额度。取不到就用保守默认值,
|
||||
// 绝不「取不到就不限制」——那正是配置出问题时最不该发生的事。
|
||||
func (s *Service) aiQuotaConfig() model.AIQuotaConfig {
|
||||
var c model.AIQuotaConfig
|
||||
if err := s.db.First(&c, "id = ?", "1").Error; err != nil {
|
||||
return model.AIQuotaConfig{DailyChat: 20, DailySymptom: 10, DailyPlan: 5, Enabled: true}
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
// aiDailyLimit 某类 AI 功能的每日上限;<=0 表示不限
|
||||
func (s *Service) aiDailyLimit(kind string) (int, bool) {
|
||||
c := s.aiQuotaConfig()
|
||||
if !c.Enabled {
|
||||
return 0, false
|
||||
}
|
||||
switch kind {
|
||||
case aiKindChat:
|
||||
return c.DailyChat, true
|
||||
case aiKindSymptom:
|
||||
return c.DailySymptom, true
|
||||
case aiKindPlan:
|
||||
return c.DailyPlan, true
|
||||
}
|
||||
return 0, false
|
||||
}
|
||||
|
||||
// consumeAIQuota 扣一次额度。超限返回 ErrAIQuotaExceeded,调用方要在真正
|
||||
// 请求大模型之前调用它——扣完再调,失败也算用掉一次,避免刷接口空转烧钱。
|
||||
func (s *Service) consumeAIQuota(userID, kind string) error {
|
||||
limit, on := s.aiDailyLimit(kind)
|
||||
if !on || limit <= 0 {
|
||||
return nil
|
||||
}
|
||||
day := time.Now().Format("2006-01-02")
|
||||
var u model.AIUsage
|
||||
err := s.db.Where("user_id = ? AND day = ? AND kind = ?", userID, day, kind).First(&u).Error
|
||||
if err == nil {
|
||||
if u.Count >= limit {
|
||||
return fmt.Errorf("%w(每天 %d 次,明天恢复)", ErrAIQuotaExceeded, limit)
|
||||
}
|
||||
return s.db.Model(&model.AIUsage{}).Where("id = ?", u.ID).
|
||||
UpdateColumn("count", u.Count+1).Error
|
||||
}
|
||||
return s.db.Create(&model.AIUsage{UserID: userID, Day: day, Kind: kind, Count: 1}).Error
|
||||
}
|
||||
|
||||
// AIQuotaLeft 返回各类今日剩余次数,给小程序显示
|
||||
func (s *Service) AIQuotaLeft(userID string) map[string]any {
|
||||
c := s.aiQuotaConfig()
|
||||
day := time.Now().Format("2006-01-02")
|
||||
var rows []model.AIUsage
|
||||
s.db.Where("user_id = ? AND day = ?", userID, day).Find(&rows)
|
||||
used := map[string]int{}
|
||||
for _, r := range rows {
|
||||
used[r.Kind] = r.Count
|
||||
}
|
||||
left := func(limit int, kind string) int {
|
||||
if !c.Enabled || limit <= 0 {
|
||||
return -1 // -1 表示不限
|
||||
}
|
||||
if n := limit - used[kind]; n > 0 {
|
||||
return n
|
||||
}
|
||||
return 0
|
||||
}
|
||||
return map[string]any{
|
||||
"enabled": c.Enabled,
|
||||
"chat": left(c.DailyChat, aiKindChat),
|
||||
"symptom": left(c.DailySymptom, aiKindSymptom),
|
||||
"plan": left(c.DailyPlan, aiKindPlan),
|
||||
}
|
||||
}
|
||||
|
||||
// GetAIQuotaConfig / SaveAIQuotaConfig 后台读写
|
||||
func (s *Service) GetAIQuotaConfig() model.AIQuotaConfig { return s.aiQuotaConfig() }
|
||||
|
||||
func (s *Service) SaveAIQuotaConfig(in model.AIQuotaConfig) (model.AIQuotaConfig, error) {
|
||||
c := s.aiQuotaConfig()
|
||||
c.ID = "1"
|
||||
c.Enabled = in.Enabled
|
||||
c.DailyChat = maxInt(in.DailyChat, 0)
|
||||
c.DailySymptom = maxInt(in.DailySymptom, 0)
|
||||
c.DailyPlan = maxInt(in.DailyPlan, 0)
|
||||
if err := s.db.Save(&c).Error; err != nil {
|
||||
return c, err
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
@@ -138,7 +138,7 @@ func (s *Service) Calendar(userID, petID string, year, month int) (*CalendarResu
|
||||
set := s.datesWithActivity(petID, start, end)
|
||||
days := make([]int, 0, len(set))
|
||||
for k := range set {
|
||||
if t, err := time.Parse("2006-01-02", k); err == nil {
|
||||
if t, err := time.ParseInLocation("2006-01-02", k, time.Local); err == nil {
|
||||
days = append(days, t.Day())
|
||||
}
|
||||
}
|
||||
@@ -156,7 +156,7 @@ func (s *Service) DayPlan(userID, petID, dateStr string) (map[string]any, error)
|
||||
if _, err := s.ownedPet(userID, petID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
day, err := time.ParseInLocation("2006-01-02", dateStr, time.Now().Location())
|
||||
day, err := time.ParseInLocation("2006-01-02", dateStr, time.Local)
|
||||
if err != nil {
|
||||
return nil, ErrNotFound
|
||||
}
|
||||
@@ -192,6 +192,9 @@ func (s *Service) DayPlan(userID, petID, dateStr string) (map[string]any, error)
|
||||
|
||||
// CreateAIPlan 基于用户描述做规则化提取,生成待确认的 AI 计划
|
||||
func (s *Service) CreateAIPlan(userID, petID string, input string) (*model.Plan, error) {
|
||||
if err := s.consumeAIQuota(userID, aiKindPlan); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
pet, err := s.ownedPet(userID, petID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
||||
Reference in New Issue
Block a user