wlzboy
2025-10-26 2c86a8bd60deed0dd0e044bad6fb83f75d19a332
ruoyi-system/src/main/java/com/ruoyi/system/utils/TaskCodeGenerator.java
New file
@@ -0,0 +1,70 @@
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);
    }
}