package cn.iocoder.yudao.module.pms.dal.redis;

import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateUtil;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Repository;

import javax.annotation.Resource;
import java.time.Duration;
import java.time.LocalDateTime;


/**
 * pms 业务单据的 Redis DAO
 *
 * @author ruiqi
 */
@Repository
public class BizNoRedisDAO {

    /**
     * 保养计划
     */
    public static final String MAIN_PLAN_NO_PREFIX = "BYJH";

    /**
     * 保养工单
     */
    public static final String MAIN_WORK_ORDER_NO_PREFIX = "BYGD";

    /**
     * 物料领用单
     */
    public static final String MATERIAL_REQ_ORDER_NO_PREFIX = "WLLY";

    @Resource
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 生成序号，使用当前日期，格式为 {PREFIX} + yyyyMMdd + 6 位自增
     * 例如说：BYJH 202109 000001 （没有中间空格）
     *
     * @param prefix 前缀
     * @return 序号
     */
    public String generate(String prefix) {
        // 递增序号
        String noPrefix = prefix + DateUtil.format(LocalDateTime.now(), DatePattern.PURE_DATE_PATTERN);
        String key = RedisKeyConstants.NO + noPrefix;
        Long no = stringRedisTemplate.opsForValue().increment(key);
        // 设置过期时间
        stringRedisTemplate.expire(key, Duration.ofDays(1L));
        return noPrefix + String.format("%06d", no);
    }

}
