Blizzard
5b2f8e50bc
## 首页添加的任务不显示（根因比表象严重）
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>
111 lines
3.3 KiB
Go
111 lines
3.3 KiB
Go
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
|
|
}
|