96be3e2a02
✨ feat(task,agent): 新增任务完成接口，并打通聊天全链路 Token 记账 - ✅ 新增“标记任务为完成”接口，并补充幂等保护，避免重复完成导致状态污染 - 📊 为聊天链路补充 Token 统计能力： - 流式主对话链路直接读取模型 `usage` - Agent 链路通过 `Eino callback + ctx` 聚合 `Generate usage` - 在流式场景下补齐缺失的 `usage` 数据 - 🧾 按口径 B 完成 Token 落库： - 用户消息 `token` 记为 `0` - 助手消息记录本轮总 `token` - 持久化时同步更新 `chat_histories.tokens_consumed`、`agent_chats.tokens_total`、`users.token_usage` - 🔄 异步标题生成产生的 Token 通过 Outbox 事件完成账本增量调整，保证统计口径一致 - 📝 同步更新 `AGENTS.md` 与 `.gitignore` - 📚 小幅更新 README 说明文档
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package api
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net/http"
|
||
"time"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"github.com/LoveLosita/smartflow/backend/respond"
|
||
"github.com/LoveLosita/smartflow/backend/service"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
type TaskHandler struct {
|
||
// 伸出手：准备接住 Service
|
||
svc *service.TaskService
|
||
}
|
||
|
||
// NewTaskHandler 创建 TaskHandler 实例
|
||
func NewTaskHandler(svc *service.TaskService) *TaskHandler {
|
||
return &TaskHandler{
|
||
svc: svc,
|
||
}
|
||
}
|
||
|
||
func (th *TaskHandler) AddTask(c *gin.Context) {
|
||
//1. 绑定请求参数
|
||
var req model.UserAddTaskRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
c.JSON(http.StatusBadRequest, respond.WrongParamType)
|
||
fmt.Println(err)
|
||
return
|
||
}
|
||
// 用户ID从上下文中获取
|
||
userID := c.GetInt("user_id")
|
||
//2. 调用 Service 层处理业务逻辑
|
||
// 创建一个带 1 秒超时的上下文
|
||
ctx, cancel := context.WithTimeout(c.Request.Context(), 1*time.Second)
|
||
defer cancel() // 记得释放资源
|
||
resp, err := th.svc.AddTask(ctx, &req, userID)
|
||
if err != nil {
|
||
respond.DealWithError(c, err)
|
||
return
|
||
}
|
||
//3. 返回响应
|
||
c.JSON(http.StatusOK, respond.RespWithData(respond.Ok, resp))
|
||
}
|
||
|
||
func (th *TaskHandler) GetUserTasks(c *gin.Context) {
|
||
// 用户ID从上下文中获取
|
||
userID := c.GetInt("user_id")
|
||
//2. 调用 Service 层处理业务逻辑
|
||
// 创建一个带 1 秒超时的上下文
|
||
ctx, cancel := context.WithTimeout(c.Request.Context(), 1*time.Second)
|
||
defer cancel() // 记得释放资源
|
||
resp, err := th.svc.GetUserTasks(ctx, userID)
|
||
if err != nil {
|
||
respond.DealWithError(c, err)
|
||
return
|
||
}
|
||
//3. 返回响应
|
||
c.JSON(http.StatusOK, respond.RespWithData(respond.Ok, resp))
|
||
}
|
||
|
||
// CompleteTask 标记任务为已完成。
|
||
//
|
||
// 职责边界：
|
||
// 1. 负责解析请求与读取 user_id；
|
||
// 2. 负责调用 Service 执行业务；
|
||
// 3. 不负责幂等校验（幂等由路由中间件处理）。
|
||
func (th *TaskHandler) CompleteTask(c *gin.Context) {
|
||
// 1. 绑定请求参数。
|
||
var req model.UserCompleteTaskRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
c.JSON(http.StatusBadRequest, respond.WrongParamType)
|
||
fmt.Println(err)
|
||
return
|
||
}
|
||
|
||
// 2. 从鉴权上下文获取 user_id，保证只能操作自己的任务。
|
||
userID := c.GetInt("user_id")
|
||
|
||
// 3. 设置短超时，避免该写接口长期占用连接。
|
||
ctx, cancel := context.WithTimeout(c.Request.Context(), 1*time.Second)
|
||
defer cancel()
|
||
|
||
// 4. 调用 Service 执行“标记完成”逻辑。
|
||
resp, err := th.svc.CompleteTask(ctx, &req, userID)
|
||
if err != nil {
|
||
respond.DealWithError(c, err)
|
||
return
|
||
}
|
||
|
||
// 5. 返回统一响应结构。
|
||
c.JSON(http.StatusOK, respond.RespWithData(respond.Ok, resp))
|
||
}
|