单机信息总览场站玫瑰图

web/runeconomy-xk/src/main/java/com/gyee/runeconomy/controller/ProBasicSubStationController.java

@@ -4,6 +4,8 @@ package com.gyee.runeconomy.controller;
 import com.gyee.common.model.StringUtils;
 import com.gyee.runeconomy.dto.R;
 import com.gyee.runeconomy.dto.ResultMsg;
+import com.gyee.runeconomy.init.CacheContext;
+import com.gyee.runeconomy.model.auto.ProBasicEquipment;
 import com.gyee.runeconomy.model.auto.ProBasicSubStation;
 import com.gyee.runeconomy.service.TokenService;
 import io.swagger.annotations.ApiOperation;
@@ -16,6 +18,7 @@ import org.springframework.web.context.request.ServletRequestAttributes;
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -48,4 +51,34 @@ public class ProBasicSubStationController {
             return R.error(ResultMsg.error());
         }
     }
+
+    @GetMapping("/querywtbysubid")
+    @ResponseBody
+    @ApiOperation(value = "通过升压站编号火车风机列表", notes = "通过升压站编号火车风机列表")
+    public R queryWtBySubId(String subId) throws Exception {
+
+        HttpServletRequest request = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getRequest();
+        List<ProBasicSubStation> sublist = tokenService.getSubwpls(request);
+        List<ProBasicEquipment> subwtls=new ArrayList<>();
+        if(StringUtils.notEmp(subId))
+        {
+            for (ProBasicSubStation sub : sublist) {
+                if (sub.getId().equals(subId)) {
+
+                    if( CacheContext.subWtsMap.containsKey(subId))
+                    {
+                        subwtls=CacheContext.subWtsMap.get(subId);
+                    }
+                    break;
+                }
+            }
+        }
+
+
+        if (StringUtils.isNotNull(subwtls)) {
+            return R.data(ResultMsg.ok(subwtls));
+        } else {
+            return R.error(ResultMsg.error());
+        }
+    }
 }

web/runeconomy-xk/src/main/java/com/gyee/runeconomy/init/CacheContext.java

@@ -397,6 +397,16 @@ public class CacheContext implements CommandLineRunner {
                 map.put(wt.getId(), wt);
                 cmwtmap.put(wt.getCompanyId(), map);
             }
+
+
+            if (subWtsMap.containsKey(wt.getSubstationId())) {
+                subWtsMap.get(wt.getSubstationId()).add(wt);
+            } else {
+                List<ProBasicEquipment> subls = new ArrayList<>();
+                subls.add(wt);
+                subWtsMap.put(wt.getSubstationId(), subls);
+            }
+
         });
 
         //初始化标杆风机