eb521a4c35
feat: 🏗️ 在 sv/dao 层引入 ctx 支持链路追踪与超时控制（预留扩展） - 架构改造：在 sv 与 dao 层统一引入 ctx 🧩 - 为后续链路追踪、超时控制等能力提供支持 ⏱️（未来开发） perf: 🚀 为获取任务类列表接口引入 Redis 缓存并保证一致性 - 获取任务类列表接口新增 Redis 缓存提升访问速度 ⚡ - 通过新增任务类时主动删除缓存，确保主从一致性 ✅ - 防止缓存与数据库列表不一致问题 🛡️
77 lines
2.2 KiB
Go
77 lines
2.2 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"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 {
|
|
switch {
|
|
case errors.Is(err, respond.InvalidPriority): //如果是无效刷新令牌或者无效claims或者无效签名方法
|
|
c.JSON(http.StatusBadRequest, err)
|
|
return
|
|
default:
|
|
c.JSON(http.StatusInternalServerError, respond.InternalError(err))
|
|
}
|
|
}
|
|
//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 {
|
|
switch {
|
|
case errors.Is(err, respond.UserTasksEmpty): //如果任务列表为空
|
|
c.JSON(http.StatusOK, respond.RespWithData(respond.UserTasksEmpty, []model.Task{})) //确实没错误，但是任务列表为空，返回自定义响应
|
|
return
|
|
default:
|
|
c.JSON(http.StatusInternalServerError, respond.InternalError(err))
|
|
return
|
|
}
|
|
}
|
|
//3. 返回响应
|
|
c.JSON(http.StatusOK, respond.RespWithData(respond.Ok, resp))
|
|
}
|