wlzboy
2025-09-24 62a079a15b46925283581f6caaf631b5a4558927
ruoyi-system/src/main/java/com/ruoyi/system/utils/TaskCodeGenerator.java
@@ -2,8 +2,10 @@
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;
/**
 * 任务编号生成器
@@ -13,6 +15,9 @@
@Component
public class TaskCodeGenerator {
    
    @Autowired
    private SysTaskMapper sysTaskMapper;
    /**
     * 生成任务编号
     * 格式:TASK + YYYYMMDD + 4位序号
@@ -21,7 +26,7 @@
     */
    public String generateTaskCode() {
        String dateStr = DateUtils.dateTimeNow("yyyyMMdd");
        return "TASK" + dateStr + "0001";
        return generateTaskCode(dateStr);
    }
    
    /**
@@ -33,6 +38,33 @@
    public String generateTaskCode(Date date) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        String dateStr = sdf.format(date);
        return "TASK" + dateStr + "0001";
        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);
    }
}