package cn.iocoder.yudao.module.mes.dal.redis.pro.task;

import cn.iocoder.yudao.module.mes.dal.redis.RedisKeyConstants;
import jakarta.annotation.Resource;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Repository;

import java.time.Duration;
import java.time.LocalDateTime;

/**
 * MES 生产任务报工倒计时提醒 Redis DAO
 *
 * @author 超级管理员
 */
@Repository
public class MesProTaskWaitRemindRedisDAO {

    /**
     * 提醒去重 KEY 的过期时间（24 小时），同一任务同一等级同一轮倒计时 24 小时内只提醒一次
     */
    private static final Duration REMIND_TIMEOUT = Duration.ofHours(24);
    /**
     * 扫描门闩的过期时间（2 分钟），避免多实例/多次并发重复扫描
     */
    private static final Duration SCAN_LOCK_TIMEOUT = Duration.ofMinutes(2);

    @Resource
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 尝试标记已提醒（同一任务同一等级同一轮倒计时只提醒一次）
     *
     * @param taskId 生产任务编号
     * @param level 提醒等级（1=剩余10分钟、2=剩余5分钟、3=倒计时结束）
     * @param waitStartTime 倒计时开始时间，参与 KEY 保证重新开始倒计时后会重新提醒
     * @return true 表示首次标记成功，需要发送提醒
     */
    public boolean tryMarkReminded(Long taskId, Integer level, LocalDateTime waitStartTime) {
        String key = RedisKeyConstants.TASK_WAIT_REMIND + taskId + ":" + level + ":" + waitStartTime;
        return Boolean.TRUE.equals(stringRedisTemplate.opsForValue().setIfAbsent(key, "1", REMIND_TIMEOUT));
    }

    /**
     * 尝试获取扫描门闩，避免多实例并发重复扫描
     *
     * @return true 表示获取成功（可以执行扫描）
     */
    public boolean tryAcquireScanLock() {
        String key = RedisKeyConstants.TASK_WAIT_REMIND_SCAN;
        return Boolean.TRUE.equals(stringRedisTemplate.opsForValue().setIfAbsent(key, "1", SCAN_LOCK_TIMEOUT));
    }

}