fxkh-txl-service/src/main/java/com/whdc/service/impl/AdinfoServiceImpl.java

package com.whdc.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.whdc.mapper.AdinfoMapper;
import com.whdc.model.entity.Adinfo;
import com.whdc.model.vo.AdcdTree;
import com.whdc.service.IAdinfoService;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * <p>
 * 服务实现类
 * </p>
 * @author xusan
 * @date 2024-05-11
 */
@Service
public class AdinfoServiceImpl extends ServiceImpl<AdinfoMapper, Adinfo> implements IAdinfoService {


    @Override
    public IPage<Adinfo> page(Adinfo dto) {
        return baseMapper.page(new Page<>(), dto);
    }


    @Override
    public List<Adinfo> find(Adinfo dto) {
        return baseMapper.find(dto);
    }




    /**
     * 行政区划树型查询
     * 420116,001,002,000
     * 6位区县，9位街道，12位组
     * 省  市 县
     * 42 28 22 100 011 100
     *
     * @param adcd
     * @return
     */
//    @Cacheable(cacheNames = {REDIS_KEY}, key = "getMethodName()", condition = "#adcd", unless = "false", cacheResolver = "redisExpireCacheResolver")
    public List<AdcdTree> tree(String adcd, String adnm) {

        List<Adinfo> list = treeList(adcd, adnm);

        if (CollectionUtils.isNotEmpty(list)){
            list = list.stream().sorted(Comparator.comparing(Adinfo::getAdcd)
                    .reversed()
            ).collect(Collectors.toList());
        }

        // 省 42 2000000000000
        Map<String, AdcdTree> province = new HashMap<>();

        // 市 4205 00000000000
        Map<String, List<AdcdTree>> city = new HashMap<>();

        // 县 421122 000000000
        Map<String, List<AdcdTree>> county = new HashMap<>();

        // 镇 421122100 000000
        Map<String, List<AdcdTree>> town = new HashMap<>();

        // 村 421122100201 000
        Map<String, List<AdcdTree>> village = new HashMap<>();

        // 组 421122100201100
        Map<String, List<AdcdTree>> group = new HashMap<>();

        List<AdcdTree> tree = new ArrayList<>();

        for (Adinfo adinfo : list) {
            String ad = adinfo.getAdcd();
            String nm = adinfo.getAdnm();
            AdcdTree adcdTree = new AdcdTree().setAdcd(ad).setAdnm(nm).setAdcdChildren(new ArrayList<>());

            // 判断行政区划编码
            String provinceStr = ad.substring(0, 2);
            String cityStr = ad.substring(0, 4);
            String countyStr = ad.substring(0, 6);
            String townStr = ad.substring(0, 9);
            String villageStr = ad.substring(0, 12);

            if ("0000000000000".equals(ad.substring(2))) { // 省

                adcdTree.setAdlevel(0);
                adcdTree.setAdcdChildren(city.get(provinceStr));
//                province.put(provinceStr, adcdTree);
                tree.add(adcdTree);

            } else if ("00000000000".equals(ad.substring(4))) { // 市
                adcdTree.setAdlevel(1);
                adcdTree.setAdcdChildren(county.get(cityStr));
                addTree(city, provinceStr, adcdTree);

            } else if ("000000000".equals(ad.substring(6))) { // 县
                adcdTree.setAdlevel(2);
                adcdTree.setAdcdChildren(town.get(countyStr));
                addTree(county, cityStr, adcdTree);

            } else if ("000000".equals(ad.substring(9))) { // 镇
                adcdTree.setAdlevel(3);
                adcdTree.setAdcdChildren(village.get(townStr));
                addTree(town, countyStr, adcdTree);

            } else if ("000".equals(ad.substring(12))) { // 村
                adcdTree.setAdlevel(4);
                adcdTree.setAdcdChildren(group.get(villageStr));
                addTree(village, townStr, adcdTree);

            } else { // 组
                adcdTree.setAdlevel(5);
                adcdTree.setAdcdChildren(null);
                addTree(group, villageStr, adcdTree);
            }

        }

        // 排序

        return sorted(tree);
    }


    public List<Adinfo> treeList(String adcd, String adnm) {

        if(adcd.endsWith("0000000000000")){
            adcd = adcd.substring(0,2);
        }
        if (StringUtils.isNotBlank(adcd) && StringUtils.isNotBlank(adnm)) {

            // 需要查出层级数据
            return baseMapper.selectByCdOrNm(adcd, adnm);

        } else {
            LambdaQueryWrapper<Adinfo> queryWrapper = new LambdaQueryWrapper<>();
            if (StringUtils.isNotBlank(adcd)) {
                queryWrapper.like(Adinfo::getAdcd, adcd);
            }
            List<Adinfo> adinfos = baseMapper.selectList(queryWrapper);

            return adinfos; // 查所有
        }

    }

    private List<AdcdTree>  sorted( List<AdcdTree> tree) {

        List<AdcdTree> sorteds = null;

        if (CollectionUtils.isNotEmpty(tree)){
            sorteds = tree.stream().sorted(Comparator.comparing(AdcdTree::getAdcd)
            ).collect(Collectors.toList());
            sorteds.forEach(o->{
                o.setAdcdChildren(sorted(o.getAdcdChildren()));
            });
        }

        return sorteds;
    }

    private void addTree(Map<String, List<AdcdTree>> data, String key, AdcdTree tree) {
        List<AdcdTree> trees = data.get(key);
        if (CollectionUtils.isEmpty(trees)) {
            trees = new ArrayList<>();
        }
        trees.add(tree);

        data.put(key, trees);
    }

}