jnpf-java-datareport-univer-v6x/jnpf-datareport-univer-biz/target/classes/jnpf/service/impl/CodeNumServiceImpl.java

package jnpf.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import jnpf.base.service.SuperServiceImpl;
import jnpf.entity.CodeNumEntity;
import jnpf.mapper.CodeNumMapper;
import jnpf.service.CodeNumService;
import jnpf.util.DateUtil;
import jnpf.util.RandomUtil;
import lombok.Synchronized;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.function.Predicate;
import java.util.function.Supplier;
import java.util.stream.Stream;

/**
 * 编码获取服务
 *
 * @author JNPF开发平台组
 * @version v6.0.0
 * @copyright 引迈信息技术有限公司
 * @date 2025/2/28 11:16:06
 */
@Service
public class CodeNumServiceImpl extends SuperServiceImpl<CodeNumMapper, CodeNumEntity> implements CodeNumService {

    @Synchronized
    @Override
    public Integer getNumByType(String type, Integer times) {
        CodeNumEntity codeNumEntity;
        Integer num = 1;
        Integer dateValue = Integer.parseInt(DateUtil.nowDateTime().substring(0, 8));

        QueryWrapper<CodeNumEntity> wrapper = new QueryWrapper();
        wrapper.lambda().eq(CodeNumEntity::getType, type);
        List<CodeNumEntity> list = this.list(wrapper);
        if (list.size() > 0) {
            codeNumEntity = list.get(0);
            if (Objects.equals(dateValue, codeNumEntity.getDateValue())) {
                num = codeNumEntity.getNum();
            }
        } else {
            codeNumEntity = new CodeNumEntity();
            codeNumEntity.setId(RandomUtil.uuId());


        }
        codeNumEntity.setType(type);
        codeNumEntity.setDateValue(dateValue);
        codeNumEntity.setNum(num + times);
        this.saveOrUpdate(codeNumEntity);
        return num;
    }

    @Override
    public List<String> getCode(String type, Integer num) {
        Integer value = this.getNumByType(type, num);
        List<String> list = new ArrayList<>();
        for (int n = 0; n < num; n++) {
            String numStr = String.format("%06d", value + n);
            StringBuilder sb = new StringBuilder();
            sb.append(type);
            sb.append(DateUtil.nowDateTime().substring(0, 8));
            sb.append(numStr);
            list.add(sb.toString());
        }
        return list;
    }


    @Override
    public String getCodeOnce(String type) {
        return this.getCode(type, 1).get(0);
    }

    @Override
    @Synchronized
    public String getCodeFunction(Supplier<String> getCode, Predicate<String> existCode) {
        return Stream.generate(getCode)
                .filter(code -> !existCode.test(code))
                .findFirst()
                .orElseThrow(() -> new RuntimeException("无法获取唯一编码"));
    }
}