1bcbd41bec
feat: ✅ 新增获取完整任务类与修改任务类接口并完成测试 - 新增获取完整任务类接口 📋 - 实现创建任务类的逆向数据转换函数 🔄 - 工程量较大，涉及完整数据结构还原 🏗️ - 新增修改任务类接口 ✏️ - 调整 service 层 AddOrUpdateTaskClass 函数逻辑 - 复用创建任务类的大部分实现，并通过 method 区分创建/更新操作 ♻️ - 更新 dao 层操作逻辑 🗄️ - 增加防止越权修改其它用户任务类的机制 🔒 - 两个接口代码量巨大，但均已测试通过 🧪💪
99 lines
2.8 KiB
Go
99 lines
2.8 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"log"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/conv"
|
||
"github.com/LoveLosita/smartflow/backend/dao"
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"github.com/go-redis/redis/v8"
|
||
)
|
||
|
||
type TaskClassService struct {
|
||
// 这里可以添加数据库连接或其他依赖
|
||
taskClassRepo *dao.TaskClassDAO
|
||
cacheRepo *dao.CacheDAO
|
||
}
|
||
|
||
func NewTaskClassService(taskClassRepo *dao.TaskClassDAO, cacheRepo *dao.CacheDAO) *TaskClassService {
|
||
return &TaskClassService{
|
||
taskClassRepo: taskClassRepo,
|
||
cacheRepo: cacheRepo,
|
||
}
|
||
}
|
||
|
||
// AddOrUpdateTaskClass 为指定用户添加任务类
|
||
func (sv *TaskClassService) AddOrUpdateTaskClass(ctx context.Context, req *model.UserAddTaskClassRequest, userID int, method int, targetTaskClassID int) error {
|
||
// 1) 先写数据库（事务内）
|
||
if err := sv.taskClassRepo.Transaction(func(txDAO *dao.TaskClassDAO) error {
|
||
taskClass, items, err := conv.ProcessUserAddTaskClassRequest(req, userID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if method == 1 { // 更新操作
|
||
taskClass.ID = targetTaskClassID
|
||
}
|
||
|
||
taskClassID, err := txDAO.AddOrUpdateTaskClass(userID, taskClass)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
for i := range items {
|
||
items[i].CategoryID = &taskClassID
|
||
}
|
||
if err := txDAO.AddOrUpdateTaskClassItems(userID, items); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 2) 事务提交成功后删除该用户缓存(如果有)
|
||
err := sv.cacheRepo.DeleteTaskClassList(ctx, userID)
|
||
if err != nil {
|
||
log.Printf("redis删除任务分类列表缓存失败: userID=%d err=%v", userID, err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (sv *TaskClassService) GetUserTaskClassInfos(ctx context.Context, userID int) (*model.UserGetTaskClassesResponse, error) {
|
||
//1.先查询redis
|
||
list, err := sv.cacheRepo.GetTaskClassList(ctx, userID)
|
||
if err == nil {
|
||
//命中缓存
|
||
return &list, nil
|
||
} else if !errors.Is(err, redis.Nil) { //不是缓存未命中错误，说明redis可能炸了，照常放行
|
||
log.Println("redis获取任务分类列表失败:", err)
|
||
}
|
||
//2.缓存未命中，查询数据库
|
||
taskClasses, err := sv.taskClassRepo.GetUserTaskClasses(userID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
resp := conv.TaskClassModelToResponse(taskClasses)
|
||
//3.写入缓存
|
||
err = sv.cacheRepo.AddTaskClassList(ctx, userID, resp)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return resp, nil
|
||
}
|
||
|
||
func (sv *TaskClassService) GetUserCompleteTaskClass(ctx context.Context, userID int, taskClassID int) (*model.UserAddTaskClassRequest, error) {
|
||
//1.查询数据库
|
||
taskClass, err := sv.taskClassRepo.GetCompleteTaskClassByID(ctx, taskClassID, userID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
//2.转换为响应结构体
|
||
resp, err := conv.ProcessUserGetCompleteTaskClassRequest(taskClass)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return resp, nil
|
||
}
|