smartmate/backend/dao/task.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
}

// 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
}