69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package api
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"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 层处理业务逻辑
|
|
resp, err := th.svc.AddTask(&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 层处理业务逻辑
|
|
resp, err := th.svc.GetUserTasks(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))
|
|
}
|