art: 系统管理-分类字典功能改造，导入功能修改

zhangherong

2025-05-12 7e254edced5f0361882471ebffbad572c181235d

 lxzn-module-system/lxzn-system-biz/src/main/java/org/jeecg/modules/system/service/impl/SysCategoryServiceImpl.java

@@ -1,15 +1,14 @@
package org.jeecg.modules.system.service.impl;

import com.alibaba.fastjson.JSONObject;
import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.constant.FillRuleConstant;
import org.apache.commons.lang3.StringUtils;
import org.jeecg.common.constant.SymbolConstant;
import org.jeecg.common.exception.JeecgBootException;
import org.jeecg.common.util.FillRuleUtil;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.system.entity.SysCategory;
import org.jeecg.modules.system.mapper.SysCategoryMapper;
@@ -18,221 +17,328 @@
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import javax.annotation.Resource;
import java.util.*;
import java.util.stream.Collectors;

/**
 * @Description: 分类字典
 * @Author: jeecg-boot
 * @Date:   2019-05-29
 * @Date: 2019-05-29
 * @Version: V1.0
 */
@Service
public class SysCategoryServiceImpl extends ServiceImpl<SysCategoryMapper, SysCategory> implements ISysCategoryService {

   @Override
   public void addSysCategory(SysCategory sysCategory) {
      String categoryCode = "";
      String categoryPid = ISysCategoryService.ROOT_PID_VALUE;
      String parentCode = null;
      if(oConvertUtils.isNotEmpty(sysCategory.getPid())){
         categoryPid = sysCategory.getPid();
    @Resource
    private SysCategoryMapper sysCategoryMapper;

         //PID 不是根节点 说明需要设置父节点 hasChild 为1
         if(!ISysCategoryService.ROOT_PID_VALUE.equals(categoryPid)){
            SysCategory parent = baseMapper.selectById(categoryPid);
            parentCode = parent.getCode();
            if(parent!=null && !ISysCategoryService.HAS_CHILD.equals(parent.getHasChild())){
               parent.setHasChild(ISysCategoryService.HAS_CHILD);
               baseMapper.updateById(parent);
            }
         }
      }
      //update-begin--Author:baihailong  Date:20191209 for：分类字典编码规则生成器做成公用配置
      JSONObject formData = new JSONObject();
      formData.put("pid",categoryPid);
      categoryCode = (String) FillRuleUtil.executeRule(FillRuleConstant.CATEGORY,formData);
      //update-end--Author:baihailong  Date:20191209 for：分类字典编码规则生成器做成公用配置
      sysCategory.setCode(categoryCode);
      sysCategory.setPid(categoryPid);
      baseMapper.insert(sysCategory);
   }
	
   @Override
   public void updateSysCategory(SysCategory sysCategory) {
      if(oConvertUtils.isEmpty(sysCategory.getPid())){
         sysCategory.setPid(ISysCategoryService.ROOT_PID_VALUE);
      }else{
         //如果当前节点父ID不为空 则设置父节点的hasChild 为1
         SysCategory parent = baseMapper.selectById(sysCategory.getPid());
         if(parent!=null && !ISysCategoryService.HAS_CHILD.equals(parent.getHasChild())){
            parent.setHasChild(ISysCategoryService.HAS_CHILD);
            baseMapper.updateById(parent);
         }
      }
      baseMapper.updateById(sysCategory);
   }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void addSysCategory(SysCategory sysCategory) {
        //修改字典分类  编码为手动输入
        if (StringUtils.isNotBlank(sysCategory.getPid())) {
            SysCategory parent = sysCategoryMapper.selectById(sysCategory.getPid());
            if (parent == null) {
                //父节点不存在 默认为根节点
                throw new JeecgBootException("父节点不存在，请重新添加！");
            } else {
                //校验编码唯一
                SysCategory original = getByCodeAndRootCode(sysCategory.getCode(), parent.getRootCode());
                if (original != null) {
                    throw new JeecgBootException("分类编码已存在，请修改后重试！");
                }
                if (ISysCategoryService.HAS_CHILD.equals(parent.getHasChild())) {
                    parent.setHasChild(ISysCategoryService.HAS_CHILD);
                    sysCategoryMapper.updateById(parent);
                }
                sysCategory.setRootCode(parent.getRootCode());
                String parentIds = parent.getParentIds() == null ? parent.getId() : (parent.getParentIds() + "," + parent.getId());
                sysCategory.setParentIds(parentIds);
            }
        } else {
            //校验编码唯一
            SysCategory original = getByCodeAndRootCode(sysCategory.getCode(), null);
            if (original != null) {
                throw new JeecgBootException("分类编码已存在，请修改后重试！");
            }
            //父节点不存在 默认为根节点
            sysCategory.setPid(ISysCategoryService.ROOT_PID_VALUE);
            sysCategory.setRootCode(sysCategory.getCode());
            sysCategory.setParentIds(null);
        }
        sysCategoryMapper.insert(sysCategory);
    }

   @Override
   public List<TreeSelectModel> queryListByCode(String pcode) throws JeecgBootException{
      String pid = ROOT_PID_VALUE;
      if(oConvertUtils.isNotEmpty(pcode)) {
         List<SysCategory> list = baseMapper.selectList(new LambdaQueryWrapper<SysCategory>().eq(SysCategory::getCode, pcode));
         if(list==null || list.size() ==0) {
            throw new JeecgBootException("该编码【"+pcode+"】不存在，请核实!");
         }
         if(list.size()>1) {
            throw new JeecgBootException("该编码【"+pcode+"】存在多个，请核实!");
         }
         pid = list.get(0).getId();
      }
      return baseMapper.queryListByPid(pid,null);
   }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateSysCategory(SysCategory sysCategory) {
        SysCategory entity = sysCategoryMapper.selectById(sysCategory.getId());
        if (entity == null) {
            throw new JeecgBootException("编辑的数据不存在，请刷新重试！");
        }
        if (oConvertUtils.isEmpty(sysCategory.getPid())) {
            sysCategory.setPid(ISysCategoryService.ROOT_PID_VALUE);
        } else {
            //如果当前节点父ID不为空 则设置父节点的hasChild 为1
            SysCategory parent = sysCategoryMapper.selectById(sysCategory.getPid());
            if (parent != null && !ISysCategoryService.HAS_CHILD.equals(parent.getHasChild())) {
                parent.setHasChild(ISysCategoryService.HAS_CHILD);
                sysCategoryMapper.updateById(parent);
            } else {
                sysCategory.setPid(ISysCategoryService.ROOT_PID_VALUE);
            }
        }
        sysCategoryMapper.updateById(sysCategory);
    }

   @Override
   public List<TreeSelectModel> queryListByPid(String pid) {
      if(oConvertUtils.isEmpty(pid)) {
         pid = ROOT_PID_VALUE;
      }
      return baseMapper.queryListByPid(pid,null);
   }
    @Override
    public List<TreeSelectModel> queryListByCode(String pcode) throws JeecgBootException {
        String pid = ROOT_PID_VALUE;
        if (oConvertUtils.isNotEmpty(pcode)) {
            List<SysCategory> list = baseMapper.selectList(new LambdaQueryWrapper<SysCategory>().eq(SysCategory::getCode, pcode));
            if (list == null || list.size() == 0) {
                throw new JeecgBootException("该编码【" + pcode + "】不存在，请核实!");
            }
            if (list.size() > 1) {
                throw new JeecgBootException("该编码【" + pcode + "】存在多个，请核实!");
            }
            pid = list.get(0).getId();
        }
        return baseMapper.queryListByPid(pid, null);
    }

   @Override
   public List<TreeSelectModel> queryListByPid(String pid, Map<String, String> condition) {
      if(oConvertUtils.isEmpty(pid)) {
         pid = ROOT_PID_VALUE;
      }
      return baseMapper.queryListByPid(pid,condition);
   }
    @Override
    public List<TreeSelectModel> queryListByPid(String pid) {
        if (oConvertUtils.isEmpty(pid)) {
            pid = ROOT_PID_VALUE;
        }
        return baseMapper.queryListByPid(pid, null);
    }

   @Override
   public String queryIdByCode(String code) {
      return baseMapper.queryIdByCode(code);
   }
    @Override
    public List<TreeSelectModel> queryListByPid(String pid, Map<String, String> condition) {
        if (oConvertUtils.isEmpty(pid)) {
            pid = ROOT_PID_VALUE;
        }
        return baseMapper.queryListByPid(pid, condition);
    }

   @Override
   @Transactional(rollbackFor = Exception.class)
   public void deleteSysCategory(String ids) {
      String allIds = this.queryTreeChildIds(ids);
      String pids = this.queryTreePids(ids);
      //1.删除时将节点下所有子节点一并删除
      this.baseMapper.deleteBatchIds(Arrays.asList(allIds.split(",")));
      //2.将父节点中已经没有下级的节点，修改为没有子节点
      if(oConvertUtils.isNotEmpty(pids)){
         LambdaUpdateWrapper<SysCategory> updateWrapper = new UpdateWrapper<SysCategory>()
               .lambda()
               .in(SysCategory::getId,Arrays.asList(pids.split(",")))
               .set(SysCategory::getHasChild,"0");
         this.update(updateWrapper);
      }
   }
    @Override
    public String queryIdByCode(String code) {
        return baseMapper.queryIdByCode(code);
    }

   /**
    * 查询节点下所有子节点
    * @param ids
    * @return
    */
   private String queryTreeChildIds(String ids) {
      //获取id数组
      String[] idArr = ids.split(",");
      StringBuffer sb = new StringBuffer();
      for (String pidVal : idArr) {
         if(pidVal != null){
            if(!sb.toString().contains(pidVal)){
               if(sb.toString().length() > 0){
                  sb.append(",");
               }
               sb.append(pidVal);
               this.getTreeChildIds(pidVal,sb);
            }
         }
      }
      return sb.toString();
   }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteSysCategory(String ids) {
        String allIds = this.queryTreeChildIds(ids);
        String pids = this.queryTreePids(ids);
        //1.删除时将节点下所有子节点一并删除
        this.baseMapper.deleteBatchIds(Arrays.asList(allIds.split(",")));
        //2.将父节点中已经没有下级的节点，修改为没有子节点
        if (oConvertUtils.isNotEmpty(pids)) {
            LambdaUpdateWrapper<SysCategory> updateWrapper = new UpdateWrapper<SysCategory>()
                    .lambda()
                    .in(SysCategory::getId, Arrays.asList(pids.split(",")))
                    .set(SysCategory::getHasChild, "0");
            this.update(updateWrapper);
        }
    }

   /**
    * 查询需修改标识的父节点ids
    * @param ids
    * @return
    */
   private String queryTreePids(String ids) {
      StringBuffer sb = new StringBuffer();
      //获取id数组
      String[] idArr = ids.split(",");
      for (String id : idArr) {
         if(id != null){
            SysCategory category = this.baseMapper.selectById(id);
            //根据id查询pid值
            String metaPid = category.getPid();
            //查询此节点上一级是否还有其他子节点
            LambdaQueryWrapper<SysCategory> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(SysCategory::getPid,metaPid);
            queryWrapper.notIn(SysCategory::getId,Arrays.asList(idArr));
            List<SysCategory> dataList = this.baseMapper.selectList(queryWrapper);
                boolean flag = (dataList == null || dataList.size()==0) && !Arrays.asList(idArr).contains(metaPid)
    /**
     * 查询节点下所有子节点
     *
     * @param ids
     * @return
     */
    private String queryTreeChildIds(String ids) {
        //获取id数组
        String[] idArr = ids.split(",");
        StringBuffer sb = new StringBuffer();
        for (String pidVal : idArr) {
            if (pidVal != null) {
                if (!sb.toString().contains(pidVal)) {
                    if (sb.toString().length() > 0) {
                        sb.append(",");
                    }
                    sb.append(pidVal);
                    this.getTreeChildIds(pidVal, sb);
                }
            }
        }
        return sb.toString();
    }

    /**
     * 查询需修改标识的父节点ids
     *
     * @param ids
     * @return
     */
    private String queryTreePids(String ids) {
        StringBuffer sb = new StringBuffer();
        //获取id数组
        String[] idArr = ids.split(",");
        for (String id : idArr) {
            if (id != null) {
                SysCategory category = this.baseMapper.selectById(id);
                //根据id查询pid值
                String metaPid = category.getPid();
                //查询此节点上一级是否还有其他子节点
                LambdaQueryWrapper<SysCategory> queryWrapper = new LambdaQueryWrapper<>();
                queryWrapper.eq(SysCategory::getPid, metaPid);
                queryWrapper.notIn(SysCategory::getId, Arrays.asList(idArr));
                List<SysCategory> dataList = this.baseMapper.selectList(queryWrapper);
                boolean flag = (dataList == null || dataList.size() == 0) && !Arrays.asList(idArr).contains(metaPid)
                        && !sb.toString().contains(metaPid);
                if(flag){
               //如果当前节点原本有子节点 现在木有了，更新状态
               sb.append(metaPid).append(",");
            }
         }
      }
      if(sb.toString().endsWith(SymbolConstant.COMMA)){
         sb = sb.deleteCharAt(sb.length() - 1);
      }
      return sb.toString();
   }
                if (flag) {
                    //如果当前节点原本有子节点 现在木有了，更新状态
                    sb.append(metaPid).append(",");
                }
            }
        }
        if (sb.toString().endsWith(SymbolConstant.COMMA)) {
            sb = sb.deleteCharAt(sb.length() - 1);
        }
        return sb.toString();
    }

   /**
    * 递归 根据父id获取子节点id
    * @param pidVal
    * @param sb
    * @return
    */
   private StringBuffer getTreeChildIds(String pidVal,StringBuffer sb){
      LambdaQueryWrapper<SysCategory> queryWrapper = new LambdaQueryWrapper<>();
      queryWrapper.eq(SysCategory::getPid,pidVal);
      List<SysCategory> dataList = baseMapper.selectList(queryWrapper);
      if(dataList != null && dataList.size()>0){
         for(SysCategory category : dataList) {
            if(!sb.toString().contains(category.getId())){
               sb.append(",").append(category.getId());
            }
            this.getTreeChildIds(category.getId(), sb);
         }
      }
      return sb;
   }
    /**
     * 递归 根据父id获取子节点id
     *
     * @param pidVal
     * @param sb
     * @return
     */
    private StringBuffer getTreeChildIds(String pidVal, StringBuffer sb) {
        LambdaQueryWrapper<SysCategory> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(SysCategory::getPid, pidVal);
        List<SysCategory> dataList = baseMapper.selectList(queryWrapper);
        if (dataList != null && dataList.size() > 0) {
            for (SysCategory category : dataList) {
                if (!sb.toString().contains(category.getId())) {
                    sb.append(",").append(category.getId());
                }
                this.getTreeChildIds(category.getId(), sb);
            }
        }
        return sb;
    }

   @Override
   public List<String> loadDictItem(String ids) {
      return this.loadDictItem(ids, true);
   }
    @Override
    public List<String> loadDictItem(String ids) {
        return this.loadDictItem(ids, true);
    }

   @Override
   public List<String> loadDictItem(String ids, boolean delNotExist) {
      String[] idArray = ids.split(",");
      LambdaQueryWrapper<SysCategory> query = new LambdaQueryWrapper<>();
      query.in(SysCategory::getId, Arrays.asList(idArray));
      // 查询数据
      List<SysCategory> list = super.list(query);
      // 取出name并返回
      List<String> textList;
      // update-begin--author:sunjianlei--date:20210514--for：新增delNotExist参数，设为false不删除数据库里不存在的key ----
      if (delNotExist) {
         textList = list.stream().map(SysCategory::getName).collect(Collectors.toList());
      } else {
         textList = new ArrayList<>();
         for (String id : idArray) {
            List<SysCategory> res = list.stream().filter(i -> id.equals(i.getId())).collect(Collectors.toList());
            textList.add(res.size() > 0 ? res.get(0).getName() : id);
         }
      }
      // update-end--author:sunjianlei--date:20210514--for：新增delNotExist参数，设为false不删除数据库里不存在的key ----
      return textList;
   }
    @Override
    public List<String> loadDictItem(String ids, boolean delNotExist) {
        String[] idArray = ids.split(",");
        LambdaQueryWrapper<SysCategory> query = new LambdaQueryWrapper<>();
        query.in(SysCategory::getId, Arrays.asList(idArray));
        // 查询数据
        List<SysCategory> list = super.list(query);
        // 取出name并返回
        List<String> textList;
        // update-begin--author:sunjianlei--date:20210514--for：新增delNotExist参数，设为false不删除数据库里不存在的key ----
        if (delNotExist) {
            textList = list.stream().map(SysCategory::getName).collect(Collectors.toList());
        } else {
            textList = new ArrayList<>();
            for (String id : idArray) {
                List<SysCategory> res = list.stream().filter(i -> id.equals(i.getId())).collect(Collectors.toList());
                textList.add(res.size() > 0 ? res.get(0).getName() : id);
            }
        }
        // update-end--author:sunjianlei--date:20210514--for：新增delNotExist参数，设为false不删除数据库里不存在的key ----
        return textList;
    }

    @Override
    public List<SysCategory> listByParams(SysCategory query) {
        QueryWrapper<SysCategory> queryWrapper = new QueryWrapper<>(query);
        if (query != null) {
            if (StringUtils.isNotBlank(query.getName())) {
                queryWrapper.like("t.name", query.getName());
            }
            if (StringUtils.isNotBlank(query.getCode())) {
                queryWrapper.like("t.code", query.getCode());
            }
        }
        queryWrapper.orderByDesc("t.id");
        return sysCategoryMapper.listByParams(queryWrapper);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deepAdd(SysCategory sysCategory, List<String> errorMessage) {
        String parenCode = (StringUtils.isBlank(sysCategory.getParentCode()) ? "0" : sysCategory.getParentCode());
        if (ROOT_PID_VALUE.equals(parenCode)) {
            sysCategory.setPid(ROOT_PID_VALUE);
            sysCategory.setRootCode(sysCategory.getCode());
            sysCategory.setParentIds(null);
            //根节点
            SysCategory entity = this.getByCodeAndRootCode(sysCategory.getCode(), null);
            if (entity != null) {
                sysCategory.setId(entity.getId());
                sysCategoryMapper.updateById(sysCategory);
            } else {
                sysCategoryMapper.insert(sysCategory);
            }
        } else {
            //获取上级节点
            String pid = sysCategory.getPid();
            SysCategory parent = this.getById(pid);
            if (parent == null) {
                String rootCode = sysCategory.getRootCode();
                parent = this.getByCodeAndRootCode(parenCode, rootCode);
            }
            if (parent != null) {
                sysCategory.setPid(parent.getId());
                sysCategory.setRootCode(parent.getRootCode());
                String parentIds = (parent.getParentIds() == null ? parent.getId() : (parent.getParentIds() + "," + parent.getId()));
                sysCategory.setParentIds(parentIds);
                if (!ISysCategoryService.HAS_CHILD.equals(parent.getHasChild())) {
                    //更新根节点
                    parent.setHasChild(HAS_CHILD);
                    sysCategoryMapper.updateById(parent);
                }
                SysCategory entity = this.getByCodeAndRootCode(sysCategory.getCode(), parent.getRootCode());
                if (entity != null) {
                    sysCategory.setId(entity.getId());
                    sysCategoryMapper.updateById(sysCategory);
                } else {
                    sysCategoryMapper.insert(sysCategory);
                }
            } else {
                errorMessage.add(String.format("编码[%s]的父节点没有找到,", sysCategory.getCode()));
                return;
            }
        }
        if (CollectionUtil.isNotEmpty(sysCategory.getChildren())) {
            sysCategory.getChildren().forEach(child -> {
                child.setPid(sysCategory.getId());
                this.deepAdd(child, errorMessage);
            });
        }

    }

    /**
     * 查询编码是否已存在
     *
     * @param code     分类编码
     * @param rootCode 根节点编码 为空，添加根节点， 不为空，添加子节点
     * @return
     */
    @Override
    public SysCategory getByCodeAndRootCode(String code, String rootCode) {
        LambdaQueryWrapper<SysCategory> queryWrapper = new LambdaQueryWrapper<>();
        if (StringUtils.isNotBlank(rootCode)) {
            queryWrapper.eq(SysCategory::getRootCode, rootCode);
        }
        queryWrapper.eq(SysCategory::getCode, code);
        return sysCategoryMapper.selectOne(queryWrapper);
    }

}