248 lines
8.5 KiB
Go
248 lines
8.5 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"time"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"github.com/LoveLosita/smartflow/backend/respond"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type TaskDAO struct {
|
||
// 这是一个口袋，用来装数据库连接实例
|
||
db *gorm.DB
|
||
}
|
||
|
||
// NewTaskDAO 创建TaskDAO实例
|
||
// NewTaskDAO 接收一个 *gorm.DB，并把它塞进结构体的口袋里
|
||
func NewTaskDAO(db *gorm.DB) *TaskDAO {
|
||
return &TaskDAO{
|
||
db: db,
|
||
}
|
||
}
|
||
|
||
func (r *TaskDAO) WithTx(tx *gorm.DB) *TaskDAO {
|
||
return &TaskDAO{db: tx}
|
||
}
|
||
|
||
// AddTask 为指定用户添加任务
|
||
func (dao *TaskDAO) AddTask(req *model.Task) (*model.Task, error) {
|
||
if err := dao.db.Create(req).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
return req, nil
|
||
}
|
||
|
||
func (dao *TaskDAO) GetTasksByUserID(userID int) ([]model.Task, error) {
|
||
var tasks []model.Task
|
||
if err := dao.db.Where("user_id = ?", userID).Find(&tasks).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
if len(tasks) == 0 { // 如果没有任务，返回自定义错误
|
||
return nil, respond.UserTasksEmpty
|
||
}
|
||
return tasks, nil
|
||
}
|
||
|
||
// CompleteTaskByID 将指定任务标记为“已完成”。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只负责“当前用户 + 指定 task_id”的完成状态更新；
|
||
// 2. 不负责幂等中间件（由路由层统一挂载）；
|
||
// 3. 不负责业务层响应包装（由 Service 层处理）。
|
||
//
|
||
// 返回语义：
|
||
// 1. 第一个返回值 *model.Task：返回更新后的任务快照（至少含 ID/UserID/IsCompleted）；
|
||
// 2. 第二个返回值 bool：
|
||
// 2.1 true：任务原本就已完成，本次属于幂等命中；
|
||
// 2.2 false：本次从未完成成功更新为已完成；
|
||
// 3. error：
|
||
// 3.1 gorm.ErrRecordNotFound：任务不存在或不属于当前用户；
|
||
// 3.2 其他 error：数据库异常。
|
||
func (dao *TaskDAO) CompleteTaskByID(ctx context.Context, userID int, taskID int) (*model.Task, bool, error) {
|
||
// 1. 基础兜底：非法参数直接返回“记录不存在”语义，避免下游误写。
|
||
if userID <= 0 || taskID <= 0 {
|
||
return nil, false, gorm.ErrRecordNotFound
|
||
}
|
||
|
||
// 2. 先查询目标任务，明确区分“已完成”与“不存在”。
|
||
var target model.Task
|
||
findErr := dao.db.WithContext(ctx).
|
||
Where("id = ? AND user_id = ?", taskID, userID).
|
||
First(&target).Error
|
||
if findErr != nil {
|
||
return nil, false, findErr
|
||
}
|
||
|
||
// 3. 若任务已完成，直接按幂等成功返回，不再写库。
|
||
if target.IsCompleted {
|
||
return &target, true, nil
|
||
}
|
||
|
||
// 4. 若任务未完成，执行状态更新。
|
||
//
|
||
// 4.1 使用 Model(&model.Task{UserID:userID}) 的目的：
|
||
// 让 cache_deleter 在 GORM Update 回调里拿到 user_id，从而正确删除任务缓存。
|
||
// 4.2 更新条件继续限定 user_id + id，避免误更新其他用户数据。
|
||
updateResult := dao.db.WithContext(ctx).
|
||
Model(&model.Task{UserID: userID}).
|
||
Where("id = ? AND user_id = ?", taskID, userID).
|
||
Update("is_completed", true)
|
||
if updateResult.Error != nil {
|
||
return nil, false, updateResult.Error
|
||
}
|
||
|
||
// 5. 极端并发兜底：
|
||
// 5.1 若 RowsAffected=0，可能是并发请求已先一步更新；
|
||
// 5.2 此时二次读取任务状态，若已完成则按幂等成功返回，否则视为不存在/异常。
|
||
if updateResult.RowsAffected == 0 {
|
||
var check model.Task
|
||
checkErr := dao.db.WithContext(ctx).
|
||
Where("id = ? AND user_id = ?", taskID, userID).
|
||
First(&check).Error
|
||
if checkErr != nil {
|
||
return nil, false, checkErr
|
||
}
|
||
if check.IsCompleted {
|
||
return &check, true, nil
|
||
}
|
||
return nil, false, errors.New("任务状态更新失败")
|
||
}
|
||
|
||
// 6. 返回更新后的快照给 Service 层组装响应。
|
||
target.IsCompleted = true
|
||
return &target, false, nil
|
||
}
|
||
|
||
// UndoCompleteTaskByID 将指定任务从“已完成”恢复为“未完成”。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只负责当前用户(user_id)下指定 task_id 的状态恢复；
|
||
// 2. 若任务本就未完成，按业务要求返回明确错误，不做幂等成功；
|
||
// 3. 不负责响应文案拼装（由 Service 层处理）。
|
||
//
|
||
// 返回语义：
|
||
// 1. *model.Task：恢复后的任务快照；
|
||
// 2. error：
|
||
// 2.1 gorm.ErrRecordNotFound：任务不存在或不属于当前用户；
|
||
// 2.2 respond.TaskNotCompleted：任务当前不是“已完成”状态，不能执行取消勾选；
|
||
// 2.3 其他 error：数据库异常。
|
||
func (dao *TaskDAO) UndoCompleteTaskByID(ctx context.Context, userID int, taskID int) (*model.Task, error) {
|
||
// 1. 参数兜底：非法 user/task 参数统一按“记录不存在”处理，避免误写。
|
||
if userID <= 0 || taskID <= 0 {
|
||
return nil, gorm.ErrRecordNotFound
|
||
}
|
||
|
||
// 2. 先读取目标任务，明确区分“不存在”和“状态不允许恢复”。
|
||
var target model.Task
|
||
findErr := dao.db.WithContext(ctx).
|
||
Where("id = ? AND user_id = ?", taskID, userID).
|
||
First(&target).Error
|
||
if findErr != nil {
|
||
return nil, findErr
|
||
}
|
||
|
||
// 3. 严格业务约束：若任务当前未完成，直接返回业务错误。
|
||
// 3.1 这是本接口和“标记完成”接口的关键差异：这里不做幂等成功。
|
||
if !target.IsCompleted {
|
||
return nil, respond.TaskNotCompleted
|
||
}
|
||
|
||
// 4. 执行状态恢复（is_completed=true -> false）。
|
||
//
|
||
// 4.1 使用 Model(&model.Task{UserID:userID}) 的目的是让 cache_deleter 拿到 user_id，
|
||
// 从而在回调中正确删除该用户任务缓存。
|
||
updateResult := dao.db.WithContext(ctx).
|
||
Model(&model.Task{UserID: userID}).
|
||
Where("id = ? AND user_id = ?", taskID, userID).
|
||
Update("is_completed", false)
|
||
if updateResult.Error != nil {
|
||
return nil, updateResult.Error
|
||
}
|
||
|
||
// 5. 并发兜底：
|
||
// 5.1 若 RowsAffected=0，说明可能被并发请求先一步恢复；
|
||
// 5.2 重新读取当前状态，若已是未完成则按业务规则返回“任务未完成”错误。
|
||
if updateResult.RowsAffected == 0 {
|
||
var check model.Task
|
||
checkErr := dao.db.WithContext(ctx).
|
||
Where("id = ? AND user_id = ?", taskID, userID).
|
||
First(&check).Error
|
||
if checkErr != nil {
|
||
return nil, checkErr
|
||
}
|
||
if !check.IsCompleted {
|
||
return nil, respond.TaskNotCompleted
|
||
}
|
||
return nil, errors.New("取消任务完成状态失败")
|
||
}
|
||
|
||
// 6. 回填恢复后状态并返回。
|
||
target.IsCompleted = false
|
||
return &target, nil
|
||
}
|
||
|
||
// PromoteTaskUrgencyByIDs 批量执行“任务紧急性平移”。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只负责把满足条件的任务从“不紧急象限”平移到“紧急象限”：
|
||
// 1.1 priority=2 -> 1（重要不紧急 -> 重要且紧急）；
|
||
// 1.2 priority=4 -> 3（不简单不重要 -> 简单不重要）；
|
||
// 2. 只更新本次指定 user_id + task_ids 范围内的数据；
|
||
// 3. 不负责事件发布、重试去重和缓存策略（由 Service/Outbox 负责）。
|
||
//
|
||
// 幂等与一致性说明：
|
||
// 1. SQL 条件会限制 `is_completed=0`、`urgency_threshold_at<=now`、`priority IN (2,4)`；
|
||
// 2. 同一批任务重复调用时，已经平移过的记录不会再次更新（幂等）；
|
||
// 3. 使用 `Model(&model.Task{UserID:userID})` 是为了让 GORM 回调拿到 user_id，从而触发 cache_deleter 删除任务缓存。
|
||
func (dao *TaskDAO) PromoteTaskUrgencyByIDs(ctx context.Context, userID int, taskIDs []int, now time.Time) (int64, error) {
|
||
// 1. 基础兜底：非法 user 或空任务列表直接无操作返回。
|
||
if userID <= 0 || len(taskIDs) == 0 {
|
||
return 0, nil
|
||
}
|
||
|
||
// 2. 去重并过滤非正数 ID，避免无效 where in 条件放大 SQL 噪音。
|
||
validTaskIDs := compactPositiveIntIDs(taskIDs)
|
||
if len(validTaskIDs) == 0 {
|
||
return 0, nil
|
||
}
|
||
|
||
// 3. 条件更新：只更新“已到紧急分界线且仍处于非紧急象限”的任务。
|
||
result := dao.db.WithContext(ctx).
|
||
Model(&model.Task{UserID: userID}).
|
||
Where("user_id = ?", userID).
|
||
Where("id IN ?", validTaskIDs).
|
||
Where("is_completed = ?", false).
|
||
Where("urgency_threshold_at IS NOT NULL AND urgency_threshold_at <= ?", now).
|
||
Where("priority IN ?", []int{2, 4}).
|
||
Update("priority", gorm.Expr("CASE WHEN priority = 2 THEN 1 WHEN priority = 4 THEN 3 ELSE priority END"))
|
||
|
||
if result.Error != nil {
|
||
return 0, result.Error
|
||
}
|
||
return result.RowsAffected, nil
|
||
}
|
||
|
||
// compactPositiveIntIDs 对 int 切片做“去重 + 过滤非正数”。
|
||
//
|
||
// 说明：
|
||
// 1. 该函数是 DAO 内部参数清洗工具，不参与任何业务判定；
|
||
// 2. 返回结果不保证稳定顺序，对当前 SQL where in 场景无影响。
|
||
func compactPositiveIntIDs(ids []int) []int {
|
||
seen := make(map[int]struct{}, len(ids))
|
||
result := make([]int, 0, len(ids))
|
||
for _, id := range ids {
|
||
if id <= 0 {
|
||
continue
|
||
}
|
||
if _, exists := seen[id]; exists {
|
||
continue
|
||
}
|
||
seen[id] = struct{}{}
|
||
result = append(result, id)
|
||
}
|
||
return result
|
||
}
|