Merge remote-tracking branch 'origin/dev_business' into dev_business

云

10 天以前 106aaa378126774a2547b6c0d559c62114579f4c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package cn.iocoder.yudao.module.mes.service.md.autocode.strategy.impl;
 
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.autocode.MesMdAutoCodePartDO;
import cn.iocoder.yudao.module.mes.dal.redis.md.autocode.MesMdAutoCodeRedisDAO;
import cn.iocoder.yudao.module.mes.enums.md.autocode.MesMdAutoCodeCycleMethodEnum;
import cn.iocoder.yudao.module.mes.enums.md.autocode.MesMdAutoCodePartTypeEnum;
import cn.iocoder.yudao.module.mes.service.md.autocode.strategy.MesMdAutoCodeContext;
import cn.iocoder.yudao.module.mes.service.md.autocode.strategy.MesMdAutoCodePartStrategy;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Component;
 
import java.time.Duration;
import java.time.LocalDateTime;
 
/**
 * MES 编码规则 - 流水号策略
 *
 * @author 芋道源码
 */
@Component
public class MesMdAutoCodeSerialNumberPartStrategy implements MesMdAutoCodePartStrategy {
 
    @Resource
    private MesMdAutoCodeRedisDAO redisDAO;
 
    @Override
    public Integer getType() {
        return MesMdAutoCodePartTypeEnum.SERIAL_NUMBER.getType();
    }
 
    @Override
    public String generate(MesMdAutoCodePartDO part, MesMdAutoCodeContext context) {
        // 1.1 构建 Redis key 后缀
        String keySuffix = buildRedisKeySuffix(part, context);
        // 1.2 获取过期时间
        Duration duration = Boolean.TRUE.equals(part.getCycleFlag()) ? getExpireDuration(part.getCycleMethod()) : null;
 
        // 2. 获取流水号
        Integer step = part.getSerialStep() != null ? part.getSerialStep() : 1;
        Integer startNo = part.getSerialStartNo() != null ? part.getSerialStartNo() : 1;
        Long serialNo = redisDAO.increment(keySuffix, duration, startNo, step);
 
        // 3. 格式化输出（补零）
        String serialStr = String.valueOf(serialNo);
        if (serialStr.length() < part.getLength()) {
            serialStr = StrUtil.padPre(serialStr, part.getLength(), '0');
        }
        context.setSerialNo(serialNo);
        return serialStr;
    }
 
    /**
     * 构建 Redis key 后缀
     */
    private String buildRedisKeySuffix(MesMdAutoCodePartDO part, MesMdAutoCodeContext context) {
        StringBuilder keySuffix = new StringBuilder();
        keySuffix.append(context.getRule().getId());
 
        // 如果不循环，直接返回
        if (!Boolean.TRUE.equals(part.getCycleFlag()) || part.getCycleMethod() == null) {
            return keySuffix.toString();
        }
 
        // 根据循环方式构建 key 后缀
        keySuffix.append(":");
        LocalDateTime time = LocalDateTime.now();
        if (part.getCycleMethod().equals(MesMdAutoCodeCycleMethodEnum.YEAR.getMethod())) {
            keySuffix.append(DateUtil.format(time, "yyyy"));
        } else if (part.getCycleMethod().equals(MesMdAutoCodeCycleMethodEnum.MONTH.getMethod())) {
            keySuffix.append(DateUtil.format(time, "yyyyMM"));
        } else if (part.getCycleMethod().equals(MesMdAutoCodeCycleMethodEnum.DAY.getMethod())) {
            keySuffix.append(DateUtil.format(time, "yyyyMMdd"));
        } else if (part.getCycleMethod().equals(MesMdAutoCodeCycleMethodEnum.HOUR.getMethod())) {
            keySuffix.append(DateUtil.format(time, "yyyyMMddHH"));
        } else if (part.getCycleMethod().equals(MesMdAutoCodeCycleMethodEnum.MINUTE.getMethod())) {
            keySuffix.append(DateUtil.format(time, "yyyyMMddHHmm"));
        } else if (part.getCycleMethod().equals(MesMdAutoCodeCycleMethodEnum.INPUT_CHAR.getMethod())) {
            if (StrUtil.isNotEmpty(context.getInputChar())) {
                keySuffix.append(context.getInputChar());
            }
        } else {
            throw new IllegalArgumentException("未知的循环方式：" + part.getCycleMethod());
        }
        return keySuffix.toString();
    }
 
    /**
     * 获取过期时间
     */
    private Duration getExpireDuration(Integer cycleMethod) {
        if (cycleMethod == null) {
            return Duration.ofDays(1);
        }
        if (cycleMethod.equals(MesMdAutoCodeCycleMethodEnum.YEAR.getMethod())) {
            return Duration.ofDays(366);
        } else if (cycleMethod.equals(MesMdAutoCodeCycleMethodEnum.MONTH.getMethod())) {
            return Duration.ofDays(32);
        } else if (cycleMethod.equals(MesMdAutoCodeCycleMethodEnum.DAY.getMethod())) {
            return Duration.ofDays(2);
        } else if (cycleMethod.equals(MesMdAutoCodeCycleMethodEnum.HOUR.getMethod())) {
            return Duration.ofHours(2);
        } else if (cycleMethod.equals(MesMdAutoCodeCycleMethodEnum.MINUTE.getMethod())) {
            return Duration.ofMinutes(2);
        } else if (cycleMethod.equals(MesMdAutoCodeCycleMethodEnum.INPUT_CHAR.getMethod())) {
            // 永不过期
            return null;
        } else {
            throw new IllegalArgumentException("未知的循环方式：" + cycleMethod);
        }
    }
 
}