package org.jeecg.modules.eam.base.rule; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import io.netty.util.internal.StringUtil; import org.jeecg.common.handler.IFillRuleHandler; import org.jeecg.common.util.SpringContextUtils; import org.jeecg.common.util.YouBianCodeUtil; import org.jeecg.modules.eam.base.entity.EamBaseRepairDepart; import org.jeecg.modules.eam.base.service.IEamBaseRepairDepartService; import java.util.ArrayList; import java.util.List; /** * @author: LiuS * @create: 2023-07-04 16:15 * @Description: 车间机构编码生成规则 */ public class OrgCodeRepDepRule implements IFillRuleHandler { @Override public Object execute(JSONObject params, JSONObject formData) { IEamBaseRepairDepartService eamBaseRepairDepartService = (IEamBaseRepairDepartService) SpringContextUtils.getBean("eamBaseRepairDepartImpl"); LambdaQueryWrapper query = new LambdaQueryWrapper(); LambdaQueryWrapper query1 = new LambdaQueryWrapper(); // 创建一个List集合,存储查询返回的所有EamBaseRepairDepart对象 List eamBaseRepairDepartList = new ArrayList<>(); String[] strArray = new String[2]; //定义车间类型 String orgType = ""; // 定义新编码字符串 String newOrgCode = ""; // 定义旧编码字符串 String oldOrgCode = ""; String parentId = null; if (formData != null && formData.size() > 0) { Object obj = formData.get("parentId"); if (obj != null) { parentId = obj.toString(); } } else { if (params != null) { Object obj = params.get("parentId"); if (obj != null) { parentId = obj.toString(); } } } //如果是最高级,则查询出同级的org_code, 调用工具类生成编码并返回 if (StringUtil.isNullOrEmpty(parentId)) { // 线判断数据库中的表是否为空,空则直接返回初始编码 query1.eq(EamBaseRepairDepart::getParentId, "").or().isNull(EamBaseRepairDepart::getParentId); query1.orderByDesc(EamBaseRepairDepart::getOrgCode); eamBaseRepairDepartList = eamBaseRepairDepartService.list(query1); if (eamBaseRepairDepartList == null || eamBaseRepairDepartList.size() == 0) { strArray[0] = YouBianCodeUtil.getNextYouBianCode(null); strArray[1] = "1"; return strArray; } else { EamBaseRepairDepart eamBaseRepairDepart = eamBaseRepairDepartList.get(0); oldOrgCode = eamBaseRepairDepart.getOrgCode(); orgType = eamBaseRepairDepart.getOrgType(); newOrgCode = YouBianCodeUtil.getNextYouBianCode(oldOrgCode); } } else {//反之则查询出所有同级的车间,获取结果后有两种情况,有同级和没有同级 // 封装查询同级的条件 query.eq(EamBaseRepairDepart::getParentId, parentId); // 降序排序 query.orderByDesc(EamBaseRepairDepart::getOrgCode); // 查询出同级维修部门班组的集合 List parentList = eamBaseRepairDepartService.list(query); // 查询出父级维修部门班组 EamBaseRepairDepart repairDepart = eamBaseRepairDepartService.getById(parentId); // 获取父级维修部门班组的code String parentCode = repairDepart.getOrgCode(); // 根据父级维修部门班组类型算出当前维修部门班组的类型 orgType = String.valueOf(Integer.valueOf(repairDepart.getOrgType()) + 1); // 处理同级维修部门班组为null的情况 if (parentList == null || parentList.size() == 0) { // 直接生成当前的维修部门班组编码并返回 newOrgCode = YouBianCodeUtil.getSubYouBianCode(parentCode, null); } else { //处理有同级维修部门班组的情况 // 获取同级维修部门班组的编码,利用工具类 String subCode = parentList.get(0).getOrgCode(); // 返回生成的当前维修部门班组编码 newOrgCode = YouBianCodeUtil.getSubYouBianCode(parentCode, subCode); } } // 返回最终封装了维修部门班组编码和维修部门班组类型的数组 strArray[0] = newOrgCode; strArray[1] = orgType; return strArray; } }