package org.jeecg.modules.system.service.impl; import java.util.List; import java.util.Map; import com.alibaba.fastjson.JSONObject; import org.jeecg.common.exception.JeecgBootException; import org.jeecg.common.util.FillRuleUtil; import org.jeecg.common.util.YouBianCodeUtil; import org.jeecg.common.util.oConvertUtils; import org.jeecg.modules.system.entity.SysCategory; import org.jeecg.modules.system.mapper.SysCategoryMapper; import org.jeecg.modules.system.model.TreeSelectModel; import org.jeecg.modules.system.service.ISysCategoryService; import org.springframework.stereotype.Service; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; /** * @Description: 分类字典 * @Author: jeecg-boot * @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(); //PID 不是根节点 说明需要设置父节点 hasChild 为1 if(!ISysCategoryService.ROOT_PID_VALUE.equals(categoryPid)){ SysCategory parent = baseMapper.selectById(categoryPid); parentCode = parent.getCode(); if(parent!=null && !"1".equals(parent.getHasChild())){ parent.setHasChild("1"); baseMapper.updateById(parent); } } } //update-begin--Author:baihailong Date:20191209 for:分类字典编码规则生成器做成公用配置 JSONObject formData = new JSONObject(); formData.put("pid",categoryPid); categoryCode = (String) FillRuleUtil.executeRule("category_code_rule",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 && !"1".equals(parent.getHasChild())){ parent.setHasChild("1"); baseMapper.updateById(parent); } } baseMapper.updateById(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 public List<TreeSelectModel> queryListByPid(String pid) { if(oConvertUtils.isEmpty(pid)) { pid = ROOT_PID_VALUE; } 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 String queryIdByCode(String code) { return baseMapper.queryIdByCode(code); } }