|
@@ -16,11 +16,13 @@ import com.huimv.env.common.service.IBasePigpenService;
|
|
import com.huimv.env.common.utils.Result;
|
|
import com.huimv.env.common.utils.Result;
|
|
import com.huimv.env.common.utils.ResultCode;
|
|
import com.huimv.env.common.utils.ResultCode;
|
|
import com.huimv.env.common.utils.ResultUtil;
|
|
import com.huimv.env.common.utils.ResultUtil;
|
|
|
|
+import com.huimv.env.common.utils.TreePigpenComparator;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.ArrayList;
|
|
|
|
+import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Map;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Collectors;
|
|
@@ -113,11 +115,14 @@ public class BasePigpenServiceImpl extends ServiceImpl<BasePigpenMapper, BasePig
|
|
queryWrapper.like(StringUtils.isNotBlank(buildName),"build_name", buildName);
|
|
queryWrapper.like(StringUtils.isNotBlank(buildName),"build_name", buildName);
|
|
queryWrapper.like(StringUtils.isNotBlank(stageCode),"stage_code", stageCode);
|
|
queryWrapper.like(StringUtils.isNotBlank(stageCode),"stage_code", stageCode);
|
|
queryWrapper.eq(StringUtils.isNotBlank(farmCode),"farm_id", farmCode);
|
|
queryWrapper.eq(StringUtils.isNotBlank(farmCode),"farm_id", farmCode);
|
|
- queryWrapper.orderByAsc("sort");
|
|
|
|
|
|
+ queryWrapper.orderByAsc("id");
|
|
//创建排序
|
|
//创建排序
|
|
List<BasePigpen> basePigpens = basePigpenMapper.selectList(queryWrapper);
|
|
List<BasePigpen> basePigpens = basePigpenMapper.selectList(queryWrapper);
|
|
//将结果List改为树
|
|
//将结果List改为树
|
|
List<TreeBasePigpen> treeBasePigpens = parseBizBaseArea(basePigpens);
|
|
List<TreeBasePigpen> treeBasePigpens = parseBizBaseArea(basePigpens);
|
|
|
|
+ for (TreeBasePigpen treeBasePigpen : treeBasePigpens) {
|
|
|
|
+ Collections.sort(treeBasePigpen.getChildNode(),new TreePigpenComparator());
|
|
|
|
+ }
|
|
return new Result(ResultCode.SUCCESS,treeBasePigpens);
|
|
return new Result(ResultCode.SUCCESS,treeBasePigpens);
|
|
}
|
|
}
|
|
|
|
|
|
@@ -142,6 +147,7 @@ public class BasePigpenServiceImpl extends ServiceImpl<BasePigpenMapper, BasePig
|
|
}).collect(Collectors.toMap(TreeBasePigpen::getId, b -> b,(k1, k2)->k1));
|
|
}).collect(Collectors.toMap(TreeBasePigpen::getId, b -> b,(k1, k2)->k1));
|
|
//创建数组存父亲节点
|
|
//创建数组存父亲节点
|
|
ArrayList<TreeBasePigpen> roots = new ArrayList<>();
|
|
ArrayList<TreeBasePigpen> roots = new ArrayList<>();
|
|
|
|
+
|
|
//构建树形
|
|
//构建树形
|
|
nodeMap.values().forEach(item->{
|
|
nodeMap.values().forEach(item->{
|
|
String parentCode = item.getParentId()+"";
|
|
String parentCode = item.getParentId()+"";
|