zhang
2025-05-20 1313906bb1eb983d3beece810035e7fc28d6a92f
zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/CodeServiceImpl.java
@@ -3,6 +3,7 @@
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zy.acs.common.utils.CodeUtils;
import com.zy.acs.framework.common.Cools;
import com.zy.acs.manager.common.domain.CodeExcel;
import com.zy.acs.manager.common.exception.BusinessException;
import com.zy.acs.manager.manager.entity.Code;
@@ -11,14 +12,18 @@
import com.zy.acs.manager.manager.service.CodeGapService;
import com.zy.acs.manager.manager.service.CodeService;
import com.zy.acs.manager.manager.service.RouteService;
import com.zy.acs.framework.common.Cools;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import javax.annotation.PostConstruct;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Slf4j
@Service("codeService")
public class CodeServiceImpl extends ServiceImpl<CodeMapper, Code> implements CodeService {
@@ -27,9 +32,45 @@
    @Autowired
    private RouteService routeService;
    public static final Map<Long, Code> CODE_ID_CACHE = new HashMap<>();
    public static final Map<String, Code> CODE_DATA_CACHE = new HashMap<>();
    @PostConstruct
    public void init() {
        for (Code code : this.list()) {
            CODE_ID_CACHE.put(code.getId(), code);
            CODE_DATA_CACHE.put(code.getData(), code);
        }
        log.info("The code cache was initialized...");
    }
    @Override
    public Code selectByData(String data) {
        return this.getOne(new LambdaQueryWrapper<Code>().eq(Code::getData, data));
    }
    @Override
    public Code getCacheById(Long id) {
        Code code = CODE_ID_CACHE.get(id);
        if (code == null) {
            code = this.getById(id);
            if (code != null) {
                CODE_ID_CACHE.put(id, code);
            }
        }
        return code;
    }
    @Override
    public Code getCacheByData(String data) {
        Code code = CODE_DATA_CACHE.get(data);
        if (code == null) {
            code = this.selectByData(data);
            if (null != code) {
                CODE_DATA_CACHE.put(data, code);
            }
        }
        return code;
    }
    @Override
@@ -167,4 +208,14 @@
        return this.baseMapper.selectDistinctCountFromY();
    }
    @Override
    public List<Code> getAllLocCode() {
        List<Long> ids = this.baseMapper.selectAllLocCode();
        List<Code> codeList = new ArrayList<>();
        for (Long id : ids) {
            codeList.add(this.getById(id));
        }
        return codeList;
    }
}