Enhance smoking tracking features with AI next smoke time suggestions
- Added new API endpoint `GET /api/v1/smoke/next_smoke_time` to provide AI-generated suggestions for the next smoking time based on user data. - Introduced a new database table `fa_smoke_ai_next_smoke` to store structured AI time node suggestions. - Updated smoke handler and service to integrate the new AI next smoke time functionality. - Enhanced documentation to reflect the new API endpoint and its usage, including details on how to generate AI time nodes.
This commit is contained in:
nepiedg
@@ -0,0 +1,12 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func (h *SmokeHandler) GetAINextSmokeTime(c *gin.Context) {
|
||||
q := c.Request.URL.Query()
|
||||
q.Set("mode", "ai")
|
||||
c.Request.URL.RawQuery = q.Encode()
|
||||
h.GetNextSmokeTime(c)
|
||||
}
|
||||
@@ -2,6 +2,7 @@ package handler
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"time"
|
||||
@@ -17,13 +18,23 @@ type SmokeHandler struct {
|
||||
smokeLogService *smokeservice.SmokeLogService
|
||||
smokeAIAdviceService *smokeservice.SmokeAIAdviceService
|
||||
smokeProfileService *smokeservice.SmokeProfileService
|
||||
smokeNextService *smokeservice.SmokeNextService
|
||||
smokeAINextService *smokeservice.SmokeAINextSmokeService
|
||||
}
|
||||
|
||||
func NewSmokeHandler(smokeLogService *smokeservice.SmokeLogService, smokeAIAdviceService *smokeservice.SmokeAIAdviceService, smokeProfileService *smokeservice.SmokeProfileService) *SmokeHandler {
|
||||
func NewSmokeHandler(
|
||||
smokeLogService *smokeservice.SmokeLogService,
|
||||
smokeAIAdviceService *smokeservice.SmokeAIAdviceService,
|
||||
smokeProfileService *smokeservice.SmokeProfileService,
|
||||
smokeNextService *smokeservice.SmokeNextService,
|
||||
smokeAINextService *smokeservice.SmokeAINextSmokeService,
|
||||
) *SmokeHandler {
|
||||
return &SmokeHandler{
|
||||
smokeLogService: smokeLogService,
|
||||
smokeAIAdviceService: smokeAIAdviceService,
|
||||
smokeProfileService: smokeProfileService,
|
||||
smokeNextService: smokeNextService,
|
||||
smokeAINextService: smokeAINextService,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -37,8 +48,8 @@ type createSmokeLogRequest struct {
|
||||
// 真实抽烟时间（精确到时分秒，可补录）
|
||||
SmokeAt string `json:"smoke_at"`
|
||||
Remark string `json:"remark"`
|
||||
Level int64 `json:"level"`
|
||||
Num int `json:"num"`
|
||||
Level *int64 `json:"level"`
|
||||
Num *int `json:"num"`
|
||||
}
|
||||
|
||||
func (h *SmokeHandler) Create(c *gin.Context) {
|
||||
@@ -54,6 +65,83 @@ func (h *SmokeHandler) Create(c *gin.Context) {
|
||||
return
|
||||
}
|
||||
|
||||
level := int64(1)
|
||||
if req.Level != nil {
|
||||
if *req.Level < 0 {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "level 不能为负数"))
|
||||
return
|
||||
}
|
||||
level = *req.Level
|
||||
}
|
||||
num := 1
|
||||
if req.Num != nil {
|
||||
if *req.Num < 0 {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "num 不能为负数"))
|
||||
return
|
||||
}
|
||||
num = *req.Num
|
||||
}
|
||||
|
||||
if level < 0 || num < 0 {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "level/num 不能为负数"))
|
||||
return
|
||||
}
|
||||
|
||||
var smokeTime *time.Time
|
||||
if req.SmokeTime != "" {
|
||||
parsed, err := time.ParseInLocation(dateLayout, req.SmokeTime, time.Local)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "smoke_time 格式错误，应为 YYYY-MM-DD"))
|
||||
return
|
||||
}
|
||||
smokeTime = &parsed
|
||||
}
|
||||
|
||||
var smokeAt *time.Time
|
||||
if req.SmokeAt != "" {
|
||||
parsed, err := time.ParseInLocation(dateTimeLayout, req.SmokeAt, time.Local)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "smoke_at 格式错误，应为 YYYY-MM-DD HH:MM:SS"))
|
||||
return
|
||||
}
|
||||
smokeAt = &parsed
|
||||
}
|
||||
|
||||
record, err := h.smokeLogService.Create(c.Request.Context(), int(user.ID), smokeservice.CreateSmokeLogRequest{
|
||||
SmokeTime: smokeTime,
|
||||
SmokeAt: smokeAt,
|
||||
Remark: req.Remark,
|
||||
Level: level,
|
||||
Num: num,
|
||||
})
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "创建记录失败，请稍后重试"))
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.Success(record))
|
||||
}
|
||||
|
||||
type resistedSmokeLogRequest struct {
|
||||
SmokeTime string `json:"smoke_time"`
|
||||
SmokeAt string `json:"smoke_at"`
|
||||
Remark string `json:"remark"`
|
||||
}
|
||||
|
||||
// Resist 表示“想抽但忍住了”：在 fa_smoke_log 中写入 level=0,num=0。
|
||||
func (h *SmokeHandler) Resist(c *gin.Context) {
|
||||
user, ok := middleware.CurrentUser(c)
|
||||
if !ok {
|
||||
c.JSON(http.StatusUnauthorized, model.Error(http.StatusUnauthorized, "未登录或登录已过期"))
|
||||
return
|
||||
}
|
||||
|
||||
var req resistedSmokeLogRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil && !errors.Is(err, io.EOF) {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请求参数错误"))
|
||||
return
|
||||
}
|
||||
|
||||
var smokeTime *time.Time
|
||||
if req.SmokeTime != "" {
|
||||
parsed, err := time.ParseInLocation(dateLayout, req.SmokeTime, time.Local)
|
||||
@@ -78,8 +166,8 @@ func (h *SmokeHandler) Create(c *gin.Context) {
|
||||
SmokeTime: smokeTime,
|
||||
SmokeAt: smokeAt,
|
||||
Remark: req.Remark,
|
||||
Level: req.Level,
|
||||
Num: req.Num,
|
||||
Level: 0,
|
||||
Num: 0,
|
||||
})
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "创建记录失败，请稍后重试"))
|
||||
@@ -271,6 +359,15 @@ func (h *SmokeHandler) Update(c *gin.Context) {
|
||||
return
|
||||
}
|
||||
|
||||
if req.Level != nil && *req.Level < 0 {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "level 不能为负数"))
|
||||
return
|
||||
}
|
||||
if req.Num != nil && *req.Num < 0 {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "num 不能为负数"))
|
||||
return
|
||||
}
|
||||
|
||||
smokeTimeProvided := req.SmokeTime != nil
|
||||
var smokeTime *time.Time
|
||||
if req.SmokeTime != nil {
|
||||
|
||||
@@ -0,0 +1,140 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"wx_service/internal/middleware"
|
||||
"wx_service/internal/model"
|
||||
smokeservice "wx_service/internal/smoke/service"
|
||||
)
|
||||
|
||||
type nextSmokeTimeUnifiedResponse struct {
|
||||
Source string `json:"source"`
|
||||
NotBeforeAt string `json:"not_before_at"`
|
||||
SuggestedAt string `json:"suggested_at"`
|
||||
TimeNodes []string `json:"time_nodes,omitempty"`
|
||||
Advice string `json:"advice,omitempty"`
|
||||
Default smokeservice.NextSmokeSuggestion `json:"default"`
|
||||
AI *smokeservice.AINextSmokeSuggestion `json:"ai,omitempty"`
|
||||
}
|
||||
|
||||
func (h *SmokeHandler) GetNextSmokeTime(c *gin.Context) {
|
||||
user, ok := middleware.CurrentUser(c)
|
||||
if !ok {
|
||||
c.JSON(http.StatusUnauthorized, model.Error(http.StatusUnauthorized, "未登录或登录已过期"))
|
||||
return
|
||||
}
|
||||
|
||||
asOf := time.Now().In(time.Local)
|
||||
planDate := dateOnlyLocal(asOf)
|
||||
if v := strings.TrimSpace(c.Query("date")); v != "" {
|
||||
switch strings.ToLower(v) {
|
||||
case "today":
|
||||
planDate = dateOnlyLocal(asOf)
|
||||
case "tomorrow":
|
||||
planDate = dateOnlyLocal(asOf).AddDate(0, 0, 1)
|
||||
default:
|
||||
parsed, err := time.ParseInLocation(dateLayout, v, time.Local)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "date 格式错误，应为 YYYY-MM-DD 或 today/tomorrow"))
|
||||
return
|
||||
}
|
||||
planDate = parsed
|
||||
}
|
||||
}
|
||||
if planDate.Before(dateOnlyLocal(asOf)) {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "date 不能早于今天"))
|
||||
return
|
||||
}
|
||||
|
||||
view, err := h.smokeProfileService.GetView(c.Request.Context(), int(user.ID))
|
||||
if err != nil {
|
||||
if errors.Is(err, smokeservice.ErrSmokeProfileInvalidTime) {
|
||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "作息时间格式错误，应为 HH:MM"))
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取基础信息失败，请稍后重试"))
|
||||
return
|
||||
}
|
||||
|
||||
defaultSuggestion, err := h.smokeNextService.GetDefaultSuggestion(c.Request.Context(), int(user.ID), asOf, planDate, view)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "计算失败，请稍后重试"))
|
||||
return
|
||||
}
|
||||
|
||||
mode := strings.ToLower(strings.TrimSpace(c.DefaultQuery("mode", "auto")))
|
||||
|
||||
formatPtr := func(t *time.Time) string {
|
||||
if t == nil {
|
||||
return ""
|
||||
}
|
||||
return t.In(time.Local).Format(time.RFC3339)
|
||||
}
|
||||
|
||||
resp := nextSmokeTimeUnifiedResponse{
|
||||
Source: "default",
|
||||
NotBeforeAt: formatPtr(defaultSuggestion.NextSmokeAt),
|
||||
SuggestedAt: formatPtr(defaultSuggestion.NextSmokeAt),
|
||||
Default: defaultSuggestion,
|
||||
}
|
||||
|
||||
// mode=default: 永远返回默认建议
|
||||
if mode == "default" {
|
||||
c.JSON(http.StatusOK, model.Success(resp))
|
||||
return
|
||||
}
|
||||
|
||||
// mode=auto: 仅在“已存在 AI 时间节点”时使用 AI（不主动生成）
|
||||
// mode=ai: 尝试生成/刷新 AI，再优先使用 AI；失败则回落到默认
|
||||
var ai smokeservice.AINextSmokeSuggestion
|
||||
var hasAI bool
|
||||
|
||||
if mode == "ai" {
|
||||
v, err := h.smokeAINextService.GetOrGenerate(c.Request.Context(), user, asOf, planDate, "v1", defaultSuggestion)
|
||||
if err != nil {
|
||||
switch {
|
||||
case errors.Is(err, smokeservice.ErrAINextLocked):
|
||||
c.JSON(http.StatusForbidden, model.Error(http.StatusForbidden, "需要观看广告解锁后才可生成"))
|
||||
return
|
||||
case errors.Is(err, smokeservice.ErrAINextServiceDisabled):
|
||||
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "AI 服务暂不可用，请联系管理员"))
|
||||
return
|
||||
default:
|
||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "生成 AI 建议失败，请稍后重试"))
|
||||
return
|
||||
}
|
||||
}
|
||||
ai = v
|
||||
hasAI = len(ai.TimeNodes) > 0
|
||||
} else {
|
||||
v, ok, err := h.smokeAINextService.GetCached(c.Request.Context(), user, planDate, "v1")
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取 AI 建议失败，请稍后重试"))
|
||||
return
|
||||
}
|
||||
ai = v
|
||||
hasAI = ok && len(ai.TimeNodes) > 0
|
||||
}
|
||||
|
||||
if hasAI {
|
||||
resp.Source = "ai"
|
||||
resp.NotBeforeAt = ai.NotBeforeAt
|
||||
resp.SuggestedAt = ai.SuggestedAt
|
||||
resp.TimeNodes = ai.TimeNodes
|
||||
resp.Advice = ai.Advice
|
||||
resp.AI = &ai
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, model.Success(resp))
|
||||
}
|
||||
|
||||
func dateOnlyLocal(t time.Time) time.Time {
|
||||
local := t.In(time.Local)
|
||||
return time.Date(local.Year(), local.Month(), local.Day(), 0, 0, 0, 0, time.Local)
|
||||
}
|
||||
Reference in New Issue
Block a user