属性结构

tamguo-oms/src/main/java/com/tamguo/modules/sys/model/SysCompanyEntity.java

@@ -18,6 +18,10 @@ import com.baomidou.mybatisplus.annotations.TableName;
 public class SysCompanyEntity implements Serializable {
 	private static final long serialVersionUID = 1L;
 	
+	public static final String ROOT_COMPANY_CODE = "0";
+	public static final String TREE_CODE_COMPANY_SEPARATE = ",";
+	public static final String TREE_NAME_COMPANY_SEPARATE = "/";
+
 	@TableId
 	private String companyCode;
 	private String areaCode;

tamguo-oms/src/main/java/com/tamguo/modules/sys/service/impl/SysCompanyServiceImpl.java

@@ -74,37 +74,21 @@ public class SysCompanyServiceImpl extends ServiceImpl<SysCompanyMapper, SysComp
 		company.setUpdateBy(ShiroUtils.getUserCode());
 		company.setUpdateDate(new Date());
 		company.setCompanyCode(company.getViewCode());
-		if(StringUtils.isEmpty(company.getParentCode())) {
-			company.setParentCode("0,");
-			company.setParentCodes("0,");
-			company.setTreeSorts(company.getTreeSort());
-			company.setTreeNames(company.getCompanyCode() + ",");
-			company.setTreeLeaf(true);
-			company.setTreeLevel(new BigDecimal(0));
-		}else {
-			SysCompanyEntity condition = new SysCompanyEntity();
-			condition.setCompanyCode(company.getParentCode());
-			SysCompanyEntity parentCompany = sysCompanyMapper.selectOne(condition);
 		
-			company.setTreeLeaf(true);
-			company.setTreeLevel(parentCompany.getTreeLevel().add(new BigDecimal(1)));
-			company.setParentCodes(parentCompany.getParentCodes() + parentCompany.getCompanyCode() + ",");
-			company.setTreeSorts(parentCompany.getTreeSorts() + parentCompany.getTreeSort() + ",");
-			company.setTreeNames(parentCompany.getTreeNames() + parentCompany.getCompanyName() + ",");
+		// 更新树形结构
+		this.handleTreeData(company);
 		
-			// 更新
-			parentCompany.setTreeLeaf(false);
-			parentCompany.setUpdateBy(ShiroUtils.getUserCode());
-			parentCompany.setUpdateDate(new Date());
-			sysCompanyMapper.updateById(parentCompany);
-		}
 		sysCompanyMapper.insert(company);
 	}
 
+	@SuppressWarnings("unchecked")
 	@Transactional(readOnly=false)
 	@Override
 	public void update(SysCompanyEntity company) {
 		SysCompanyEntity entity = sysCompanyMapper.selectByCode(company.getCompanyCode());
+		
+		String oldParentCode = entity.getParentCode();
+		
 		entity.setUpdateBy(ShiroUtils.getUserCode());
 		entity.setUpdateDate(new Date());
 		entity.setAreaCode(company.getAreaCode());
@@ -113,31 +97,55 @@ public class SysCompanyServiceImpl extends ServiceImpl<SysCompanyMapper, SysComp
 		entity.setCorpCode(company.getCorpCode());
 		entity.setCorpName(company.getCorpName());
 		entity.setFullName(company.getFullName());
+		entity.setParentCode(company.getParentCode());
+		
+		// 更新树形结构
+		this.handleTreeData(company);
+		
+		sysCompanyMapper.updateById(entity);
+		
+		// 更新旧的节点
+		Integer count = sysCompanyMapper.selectCount(Condition.create().eq("parent_code", oldParentCode).ne("company_code", oldParentCode));
+		if(count == 0) {
+		 	SysCompanyEntity oldParentCompany = sysCompanyMapper.selectById(oldParentCode);
+		 	oldParentCompany.setTreeLeaf(true);
+		 	sysCompanyMapper.updateById(oldParentCompany);
+		}
+	}
+	
+	// 处理树形结构
+	@SuppressWarnings("unchecked")
+	private SysCompanyEntity handleTreeData(SysCompanyEntity company) {
 		if(StringUtils.isEmpty(company.getParentCode())) {
-			entity.setParentCode("0");
-			entity.setParentCodes("0,");
-			entity.setTreeSorts(company.getTreeSort());
-			entity.setTreeNames(company.getCompanyCode() + ",");
-			entity.setTreeLeaf(true);
-			entity.setTreeLevel(new BigDecimal(0));
+			company.setParentCode(SysCompanyEntity.ROOT_COMPANY_CODE);
+			company.setParentCodes(SysCompanyEntity.ROOT_COMPANY_CODE + SysCompanyEntity.TREE_CODE_COMPANY_SEPARATE);
+			company.setTreeLeaf(true);
+			company.setTreeLevel(new BigDecimal(0));
+			company.setTreeNames(company.getCompanyName());
+			company.setTreeSorts("00000000" + company.getTreeSort() + SysCompanyEntity.ROOT_COMPANY_CODE);
 		} else {
-			SysCompanyEntity condition = new SysCompanyEntity();
-			condition.setCompanyCode(company.getParentCode());
-			SysCompanyEntity parentCompany = sysCompanyMapper.selectOne(condition);
+			SysCompanyEntity parentCompany = sysCompanyMapper.selectById(company.getParentCode());
 			
-			entity.setTreeLeaf(true);
-			entity.setTreeLevel(parentCompany.getTreeLevel().add(new BigDecimal(1)));
-			entity.setParentCodes(parentCompany.getParentCodes() + parentCompany.getCompanyCode() + ",");
-			entity.setTreeSorts(parentCompany.getTreeSorts() + parentCompany.getTreeSort() + ",");
-			entity.setTreeNames(parentCompany.getTreeNames() + parentCompany.getCompanyName() + ",");
+			company.setParentCodes(parentCompany.getParentCodes() + parentCompany.getCompanyCode() + SysCompanyEntity.TREE_CODE_COMPANY_SEPARATE);
+			company.setTreeLeaf(true);
 			
-			// 更新
+			company.setTreeLevel(parentCompany.getTreeLevel().add(new BigDecimal(1)));
+			company.setTreeNames(parentCompany.getTreeNames() + SysCompanyEntity.TREE_NAME_COMPANY_SEPARATE + parentCompany.getCompanyName());
+			company.setTreeSorts(parentCompany.getTreeSorts() + ("00000000" +company.getTreeSort()) + SysCompanyEntity.TREE_CODE_COMPANY_SEPARATE);
+			
+			if(parentCompany.getTreeLeaf()) {
 				parentCompany.setTreeLeaf(false);
-			parentCompany.setUpdateBy(ShiroUtils.getUserCode());
-			parentCompany.setUpdateDate(new Date());
 				sysCompanyMapper.updateById(parentCompany);
 			}
-		sysCompanyMapper.updateById(entity);
 		}
+		Integer count = sysCompanyMapper.selectCount(Condition.create().eq("parent_code", company.getCompanyCode()));
+		if(count > 0) {
+			company.setTreeLeaf(false);
+		}else {
+			company.setTreeLeaf(true);
+		}
+		return company;
+	}
+
 	
 }