|
@@ -11,7 +11,8 @@ import java.util.stream.Collectors;
|
|
|
|
|
|
@Service
|
|
@Service
|
|
public class SttreeService {
|
|
public class SttreeService {
|
|
- public List<ProBasicEnergyGroup> energytree() {
|
|
|
|
|
|
+ public List<ProBasicEnergyGroup> energytree(String tag) {
|
|
|
|
+
|
|
List<ProBasicEnergyGroup> energyls = CacheContext.energy;//集团
|
|
List<ProBasicEnergyGroup> energyls = CacheContext.energy;//集团
|
|
List<ProBasicRegion> regionsls = CacheContext.regions;//区域
|
|
List<ProBasicRegion> regionsls = CacheContext.regions;//区域
|
|
List<ProBasicCompany> cpls = CacheContext.cpls;//公司
|
|
List<ProBasicCompany> cpls = CacheContext.cpls;//公司
|
|
@@ -19,31 +20,59 @@ public class SttreeService {
|
|
List<ProBasicProject> prols = CacheContext.prols;//期次
|
|
List<ProBasicProject> prols = CacheContext.prols;//期次
|
|
List<ProBasicLine> list = CacheContext.lines;//期次
|
|
List<ProBasicLine> list = CacheContext.lines;//期次
|
|
|
|
|
|
- if (energyls.get(0).getChildren().size() > 0) return energyls;
|
|
|
|
|
|
+// if (energyls.get(0).getChildren().size() > 0) return energyls;
|
|
|
|
|
|
//集团-区域
|
|
//集团-区域
|
|
energyls.forEach(regls -> {
|
|
energyls.forEach(regls -> {
|
|
List<ProBasicRegion> collect = regionsls.stream().filter(r -> r.getGroupId().equals(regls.getId())).collect(Collectors.toList());
|
|
List<ProBasicRegion> collect = regionsls.stream().filter(r -> r.getGroupId().equals(regls.getId())).collect(Collectors.toList());
|
|
|
|
+ if (null != regls.getChildren()) {
|
|
|
|
+ regls.getChildren().clear();
|
|
|
|
+ }
|
|
|
|
+ collect.stream().forEach(c->{
|
|
|
|
+ c.getChildren().clear();
|
|
|
|
+ });
|
|
regls.getChildren().addAll(collect);
|
|
regls.getChildren().addAll(collect);
|
|
});
|
|
});
|
|
|
|
|
|
|
|
+ if("1".equals(tag)){
|
|
|
|
+ return energyls;
|
|
|
|
+ }
|
|
|
|
+
|
|
//区域-公司
|
|
//区域-公司
|
|
regionsls.forEach(cp -> {
|
|
regionsls.forEach(cp -> {
|
|
List<ProBasicCompany> region = cpls.stream().filter(c -> null != c.getRegionId() && c.getRegionId().equals(cp.getId())).collect(Collectors.toList());
|
|
List<ProBasicCompany> region = cpls.stream().filter(c -> null != c.getRegionId() && c.getRegionId().equals(cp.getId())).collect(Collectors.toList());
|
|
|
|
+ region.stream().forEach(r->{
|
|
|
|
+ r.getChildren().clear();
|
|
|
|
+ });
|
|
cp.getChildren().addAll(region);
|
|
cp.getChildren().addAll(region);
|
|
});
|
|
});
|
|
|
|
+ if("2".equals(tag)){
|
|
|
|
+ return energyls;
|
|
|
|
+ }
|
|
|
|
|
|
//公司-场站
|
|
//公司-场站
|
|
cpls.forEach(wp -> {
|
|
cpls.forEach(wp -> {
|
|
List<ProBasicWindpowerstation> co = wpls.stream().filter(w -> null != w.getCompanyId() && w.getCompanyId().equals(wp.getId())).collect(Collectors.toList());
|
|
List<ProBasicWindpowerstation> co = wpls.stream().filter(w -> null != w.getCompanyId() && w.getCompanyId().equals(wp.getId())).collect(Collectors.toList());
|
|
|
|
+ co.stream().forEach(c->{
|
|
|
|
+ c.getChildren().clear();
|
|
|
|
+ });
|
|
wp.getChildren().addAll(co);
|
|
wp.getChildren().addAll(co);
|
|
});
|
|
});
|
|
|
|
+ if("3".equals(tag)){
|
|
|
|
+ return energyls;
|
|
|
|
+ }
|
|
|
|
|
|
//场站-期次
|
|
//场站-期次
|
|
wpls.forEach(pl -> {
|
|
wpls.forEach(pl -> {
|
|
List<ProBasicProject> cz = prols.stream().filter(p -> null != p.getWindpowerstationId() && p.getWindpowerstationId().equals(pl.getId())).collect(Collectors.toList());
|
|
List<ProBasicProject> cz = prols.stream().filter(p -> null != p.getWindpowerstationId() && p.getWindpowerstationId().equals(pl.getId())).collect(Collectors.toList());
|
|
|
|
+ cz.stream().forEach(c->{
|
|
|
|
+ c.getChildren().clear();
|
|
|
|
+ });
|
|
pl.getChildren().addAll(cz);
|
|
pl.getChildren().addAll(cz);
|
|
});
|
|
});
|
|
|
|
+ if("4".equals(tag)){
|
|
|
|
+ return energyls;
|
|
|
|
+ }
|
|
|
|
|
|
//期次-线路
|
|
//期次-线路
|
|
prols.forEach(ll -> {
|
|
prols.forEach(ll -> {
|