|
@@ -10,11 +10,10 @@ import com.huimv.eartag2.common.utils.Result;
|
|
import com.huimv.eartag2.common.utils.ResultCode;
|
|
import com.huimv.eartag2.common.utils.ResultCode;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
+import org.springframework.util.CollectionUtils;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
import javax.annotation.Resource;
|
|
-import java.util.ArrayList;
|
|
|
|
-import java.util.List;
|
|
|
|
-import java.util.Map;
|
|
|
|
|
|
+import java.util.*;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
@Service
|
|
@Service
|
|
@@ -32,8 +31,8 @@ public class BizBaseAreaServiceImpl implements BizBaseAreaService {
|
|
@Override
|
|
@Override
|
|
public Result add(String name, String code, String parentCode) {
|
|
public Result add(String name, String code, String parentCode) {
|
|
BizBaseArea bizBaseArea = new BizBaseArea();
|
|
BizBaseArea bizBaseArea = new BizBaseArea();
|
|
- bizBaseArea.setName(name);
|
|
|
|
- bizBaseArea.setCode(code);
|
|
|
|
|
|
+ bizBaseArea.setAreaName(name);
|
|
|
|
+ bizBaseArea.setAreaCode(code);
|
|
bizBaseArea.setParentCode(parentCode);
|
|
bizBaseArea.setParentCode(parentCode);
|
|
return ResultUtil.result( bizBaseAreaMapper.insert(bizBaseArea));
|
|
return ResultUtil.result( bizBaseAreaMapper.insert(bizBaseArea));
|
|
}
|
|
}
|
|
@@ -47,8 +46,8 @@ public class BizBaseAreaServiceImpl implements BizBaseAreaService {
|
|
public Result update(Integer id, String name, String code, String parentCode) {
|
|
public Result update(Integer id, String name, String code, String parentCode) {
|
|
BizBaseArea bizBaseArea = bizBaseAreaMapper.selectById(id);
|
|
BizBaseArea bizBaseArea = bizBaseAreaMapper.selectById(id);
|
|
bizBaseArea.setParentCode(parentCode);
|
|
bizBaseArea.setParentCode(parentCode);
|
|
- bizBaseArea.setCode(code);
|
|
|
|
- bizBaseArea.setName(name);
|
|
|
|
|
|
+ bizBaseArea.setAreaCode(code);
|
|
|
|
+ bizBaseArea.setAreaName(name);
|
|
return ResultUtil.result(bizBaseAreaMapper.updateById(bizBaseArea));
|
|
return ResultUtil.result(bizBaseAreaMapper.updateById(bizBaseArea));
|
|
}
|
|
}
|
|
|
|
|
|
@@ -72,30 +71,32 @@ public class BizBaseAreaServiceImpl implements BizBaseAreaService {
|
|
* @return
|
|
* @return
|
|
*/
|
|
*/
|
|
private List<BaseAreaVo> parseBizBaseArea(List<BizBaseArea> bizBaseAreas){
|
|
private List<BaseAreaVo> parseBizBaseArea(List<BizBaseArea> bizBaseAreas){
|
|
|
|
+ System.out.println("----------------"+bizBaseAreas);
|
|
//构建需要展示的树形节点结构
|
|
//构建需要展示的树形节点结构
|
|
Map<String, BaseAreaVo> cityNodeMap = bizBaseAreas.stream().map(biZBaseAreasDo -> {
|
|
Map<String, BaseAreaVo> cityNodeMap = bizBaseAreas.stream().map(biZBaseAreasDo -> {
|
|
BaseAreaVo baseAreaVo = new BaseAreaVo();
|
|
BaseAreaVo baseAreaVo = new BaseAreaVo();
|
|
baseAreaVo.setChildBaseAreaVo(new ArrayList<>());
|
|
baseAreaVo.setChildBaseAreaVo(new ArrayList<>());
|
|
- baseAreaVo.setCode(biZBaseAreasDo.getCode());
|
|
|
|
- baseAreaVo.setName(biZBaseAreasDo.getName());
|
|
|
|
|
|
+ baseAreaVo.setCode(biZBaseAreasDo.getAreaCode());
|
|
|
|
+ baseAreaVo.setName(biZBaseAreasDo.getAreaName());
|
|
baseAreaVo.setParentCode(biZBaseAreasDo.getParentCode());
|
|
baseAreaVo.setParentCode(biZBaseAreasDo.getParentCode());
|
|
return baseAreaVo;
|
|
return baseAreaVo;
|
|
- }).collect(Collectors.toMap(BaseAreaVo::getCode, b -> b));
|
|
|
|
-
|
|
|
|
|
|
+ }).collect(Collectors.toMap(BaseAreaVo::getCode, b -> b,(k1,k2)->k1));
|
|
|
|
+ System.out.println("----------------"+cityNodeMap);
|
|
//创建数组存父亲节点
|
|
//创建数组存父亲节点
|
|
ArrayList<BaseAreaVo> roots = new ArrayList<>();
|
|
ArrayList<BaseAreaVo> roots = new ArrayList<>();
|
|
-
|
|
|
|
//构建树形
|
|
//构建树形
|
|
cityNodeMap.values().forEach(city->{
|
|
cityNodeMap.values().forEach(city->{
|
|
String parentCode = city.getParentCode();
|
|
String parentCode = city.getParentCode();
|
|
- if (StringUtils.isBlank(parentCode)){
|
|
|
|
|
|
+ if (cityNodeMap.get(parentCode) == null){
|
|
//父节点为空,说明当前节点就已经是父节点了 将该节点存起来
|
|
//父节点为空,说明当前节点就已经是父节点了 将该节点存起来
|
|
roots.add(city);
|
|
roots.add(city);
|
|
return;
|
|
return;
|
|
|
|
+ }else {
|
|
|
|
+ //父节点 不为空,说明有父节点,拿到该节点的父节点的孩子节点(就是我自己),将该节点存起来
|
|
|
|
+ cityNodeMap.get(parentCode).getChildBaseAreaVo().add(city);
|
|
}
|
|
}
|
|
- //父节点 不为空,说明有父节点,拿到该节点的父节点的孩子节点(就是我自己),将该节点存起来
|
|
|
|
- cityNodeMap.get(parentCode).getChildBaseAreaVo().add(city);
|
|
|
|
});
|
|
});
|
|
|
|
+ System.out.println(roots);
|
|
return roots;
|
|
return roots;
|
|
}
|
|
}
|
|
}
|
|
}
|