package com.ruoyi.system.utils;

import java.text.SimpleDateFormat;
import java.util.Date;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.system.mapper.SysTaskMapper;

/**
 * 任务编号生成器
 * 
 * @author ruoyi
 */
@Component
public class TaskCodeGenerator {
    
    @Autowired
    private SysTaskMapper sysTaskMapper;
    
    /**
     * 生成任务编号
     * 格式：TASK + YYYYMMDD + 4位序号
     * 
     * @return 任务编号
     */
    public String generateTaskCode() {
        String dateStr = DateUtils.dateTimeNow("yyyyMMdd");
        return generateTaskCode(dateStr);
    }
    
    /**
     * 根据日期生成任务编号
     * 
     * @param date 日期
     * @return 任务编号
     */
    public String generateTaskCode(Date date) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        String dateStr = sdf.format(date);
        return generateTaskCode(dateStr);
    }
    
    /**
     * 根据日期字符串生成任务编号
     * 
     * @param dateStr 日期字符串（格式：yyyyMMdd）
     * @return 任务编号
     */
    private String generateTaskCode(String dateStr) {
        String datePrefix = "TASK" + dateStr;
        
        // 查询当日最大编号
        String maxTaskCode = sysTaskMapper.selectMaxTaskCodeByDatePrefix(datePrefix);
        
        int sequence = 1;
        if (maxTaskCode != null && maxTaskCode.startsWith(datePrefix)) {
            // 提取序号部分
            String sequenceStr = maxTaskCode.substring(datePrefix.length());
            try {
                sequence = Integer.parseInt(sequenceStr) + 1;
            } catch (NumberFormatException e) {
                sequence = 1;
            }
        }
        
        // 格式化为4位序号
        return datePrefix + String.format("%04d", sequence);
    }
}