初始化代码迁移

cuikaidong

2025-06-12 44e283b774bb1168d0c17dfe5070a1ca8e2274cd

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package org.jeecg.modules.system.service.impl;
 
import java.util.List;
import java.util.Map;
 
import com.alibaba.fastjson.JSONObject;
import org.jeecg.common.constant.FillRuleConstant;
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(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 && !"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);
    }
 
}