Version:0.0.6.dev.260204
feat: 🏗️ 在 sv/dao 层引入 ctx 支持链路追踪与超时控制(预留扩展) - 架构改造:在 sv 与 dao 层统一引入 ctx 🧩 - 为后续链路追踪、超时控制等能力提供支持 ⏱️(未来开发) perf: 🚀 为获取任务类列表接口引入 Redis 缓存并保证一致性 - 获取任务类列表接口新增 Redis 缓存提升访问速度 ⚡ - 通过新增任务类时主动删除缓存,确保主从一致性 ✅ - 防止缓存与数据库列表不一致问题 🛡️
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/LoveLosita/smartflow/backend/dao"
|
||||
@@ -33,7 +34,7 @@ func CheckSingleCourse(req model.UserCheckCourseRequest) bool {
|
||||
}
|
||||
|
||||
// AddUserCourses 添加用户课程表
|
||||
func (ss *ScheduleService) AddUserCourses(req model.UserImportCoursesRequest, userID int) error {
|
||||
func (ss *ScheduleService) AddUserCourses(ctx context.Context, req model.UserImportCoursesRequest, userID int) error {
|
||||
//1.先校验参数是否正确
|
||||
for _, course := range req.Courses {
|
||||
result := CheckSingleCourse(course)
|
||||
|
||||
@@ -1,51 +1,81 @@
|
||||
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) *TaskClassService {
|
||||
func NewTaskClassService(taskClassRepo *dao.TaskClassDAO, cacheRepo *dao.CacheDAO) *TaskClassService {
|
||||
return &TaskClassService{
|
||||
taskClassRepo: taskClassRepo,
|
||||
cacheRepo: cacheRepo,
|
||||
}
|
||||
}
|
||||
|
||||
// AddTaskClass 为指定用户添加任务类
|
||||
func (sv *TaskClassService) AddTaskClass(req *model.UserAddTaskClassRequest, userID int) error {
|
||||
return sv.taskClassRepo.Transaction(func(txDAO *dao.TaskClassDAO) error {
|
||||
//1.先转换
|
||||
func (sv *TaskClassService) AddTaskClass(ctx context.Context, req *model.UserAddTaskClassRequest, userID 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
|
||||
}
|
||||
//2.插入 task class 并获取 ID
|
||||
|
||||
taskClassID, err := txDAO.AddTaskClass(taskClass)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
//3.插入 items
|
||||
|
||||
for i := range items {
|
||||
items[i].CategoryID = &taskClassID // 关联 task class ID
|
||||
items[i].CategoryID = &taskClassID
|
||||
}
|
||||
if err := txDAO.AddTaskClassItems(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(userID int) (*model.UserGetTaskClassesResponse, error) {
|
||||
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
|
||||
}
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/LoveLosita/smartflow/backend/conv"
|
||||
"github.com/LoveLosita/smartflow/backend/dao"
|
||||
"github.com/LoveLosita/smartflow/backend/model"
|
||||
@@ -19,7 +21,7 @@ func NewTaskService(dao *dao.TaskDAO) *TaskService {
|
||||
}
|
||||
}
|
||||
|
||||
func (ts *TaskService) AddTask(req *model.UserAddTaskRequest, userID int) (*model.UserAddTaskResponse, error) {
|
||||
func (ts *TaskService) AddTask(ctx context.Context, req *model.UserAddTaskRequest, userID int) (*model.UserAddTaskResponse, error) {
|
||||
//1. 调用 conv 层进行转换
|
||||
taskModel := conv.UserAddTaskRequestToModel(req, userID)
|
||||
//2.检查优先级是否合法
|
||||
@@ -36,7 +38,7 @@ func (ts *TaskService) AddTask(req *model.UserAddTaskRequest, userID int) (*mode
|
||||
return response, nil
|
||||
}
|
||||
|
||||
func (ts *TaskService) GetUserTasks(userID int) ([]model.GetUserTaskResp, error) {
|
||||
func (ts *TaskService) GetUserTasks(ctx context.Context, userID int) ([]model.GetUserTaskResp, error) {
|
||||
//1. 调用 dao 层获取数据
|
||||
tasks, err := ts.dao.GetTasksByUserID(userID)
|
||||
if err != nil {
|
||||
|
||||
@@ -6,6 +6,8 @@ import (
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"context"
|
||||
|
||||
"github.com/LoveLosita/smartflow/backend/auth"
|
||||
"github.com/LoveLosita/smartflow/backend/dao"
|
||||
"github.com/LoveLosita/smartflow/backend/model"
|
||||
@@ -26,7 +28,7 @@ func NewUserService(userRepo *dao.UserDAO, cacheRepo *dao.CacheDAO) *UserService
|
||||
}
|
||||
}
|
||||
|
||||
func (sv *UserService) UserRegister(user model.UserRegisterRequest) (*model.UserRegisterResponse, error) {
|
||||
func (sv *UserService) UserRegister(ctx context.Context, user model.UserRegisterRequest) (*model.UserRegisterResponse, error) {
|
||||
//检查是否有空字段
|
||||
if user.Username == "" || user.Password == "" ||
|
||||
user.PhoneNumber == "" {
|
||||
@@ -57,7 +59,7 @@ func (sv *UserService) UserRegister(user model.UserRegisterRequest) (*model.User
|
||||
return &model.UserRegisterResponse{ID: newUser.ID}, nil
|
||||
}
|
||||
|
||||
func (sv *UserService) UserLogin(req *model.UserLoginRequest) (*model.Tokens, error) {
|
||||
func (sv *UserService) UserLogin(ctx context.Context, req *model.UserLoginRequest) (*model.Tokens, error) {
|
||||
var tokens model.Tokens
|
||||
hashedPwd, err := sv.userRepo.GetUserHashedPasswordByName(req.Username) //调用dao层的方法
|
||||
if err != nil {
|
||||
@@ -86,29 +88,7 @@ func (sv *UserService) UserLogin(req *model.UserLoginRequest) (*model.Tokens, er
|
||||
return &tokens, nil
|
||||
}
|
||||
|
||||
/*func (sv *UserService) RefreshTokenHandler(refreshToken string) (*model.Tokens, error) {
|
||||
// 验证刷新令牌
|
||||
token, err := auth.ValidateRefreshToken(refreshToken, sv.cacheRepo)
|
||||
if err != nil || !token.Valid { // 刷新令牌无效
|
||||
return nil, respond.InvalidRefreshToken
|
||||
}
|
||||
|
||||
// 生成新的访问令牌和刷新令牌
|
||||
if claims, ok := token.Claims.(jwt.MapClaims); ok {
|
||||
userID := int(claims["user_id"].(float64))
|
||||
newAccessToken, newRefreshToken, err := auth.GenerateTokens(userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 返回新的访问令牌和刷新令牌
|
||||
return &model.Tokens{AccessToken: newAccessToken, RefreshToken: newRefreshToken}, nil
|
||||
} else {
|
||||
return nil, respond.InvalidClaims
|
||||
}
|
||||
}*/
|
||||
|
||||
func (sv *UserService) RefreshTokenHandler(refreshToken string) (*model.Tokens, error) {
|
||||
func (sv *UserService) RefreshTokenHandler(ctx context.Context, refreshToken string) (*model.Tokens, error) {
|
||||
// 1. 验证刷新令牌 (这里已经包含了 Redis 黑名单检查)
|
||||
token, err := auth.ValidateRefreshToken(refreshToken, sv.cacheRepo)
|
||||
if err != nil {
|
||||
@@ -132,7 +112,7 @@ func (sv *UserService) RefreshTokenHandler(refreshToken string) (*model.Tokens,
|
||||
return nil, respond.InvalidClaims
|
||||
}
|
||||
|
||||
func (sv *UserService) UserLogout(jti string, expireTime time.Time) error {
|
||||
func (sv *UserService) UserLogout(ctx context.Context, jti string, expireTime time.Time) error {
|
||||
//1.直接把 jti 扔进黑名单
|
||||
expiration := time.Until(expireTime)
|
||||
err := sv.cacheRepo.SetBlacklist(jti, expiration)
|
||||
|
||||
Reference in New Issue
Block a user