单机信息总览场站玫瑰图

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

@@ -1,8 +1,22 @@
 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.model.auto.ProBasicWeatherStation;
+import com.gyee.runeconomy.service.TokenService;
+import io.swagger.annotations.ApiOperation;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletRequest;
+import java.util.List;
 
 /**
  * <p>
@@ -13,7 +27,23 @@ import org.springframework.web.bind.annotation.RestController;
  * @since 2023-04-04
  */
 @RestController
-@RequestMapping("//pro-basic-weather-station")
+@RequestMapping("//weatherstation")
 public class ProBasicWeatherStationController {
 
+    @Resource
+    private TokenService tokenService;
+    @GetMapping("/queryall")
+    @ResponseBody
+    @ApiOperation(value = "获得测风塔信息", notes = "获得测风塔信息")
+    public R queryAll() throws Exception {
+
+        HttpServletRequest request = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getRequest();
+        List<ProBasicWeatherStation> wslist = tokenService.getWswpls(request);
+
+        if (StringUtils.isNotNull(wslist)) {
+            return R.data(ResultMsg.ok(wslist));
+        } else {
+            return R.error(ResultMsg.error());
+        }
+    }
 }

web/runeconomy-xk/src/main/java/com/gyee/runeconomy/service/TokenService.java

@@ -24,6 +24,8 @@ public class TokenService {
     public static Map<String, List<ProBasicEquipment>> sessionWtMap = new ConcurrentHashMap<>();
     public static Map<String, List<ProBasicPowerstation>> sessionWpMap = new ConcurrentHashMap<>();
     public static Map<String, List<ProBasicSubStation>> sessionSubMap = new ConcurrentHashMap<>();
+
+    public static Map<String, List<ProBasicWeatherStation>> sessionWsMap = new ConcurrentHashMap<>();
     public static Map<String, List<ProBasicWeatherStation>> sessionWeMap = new ConcurrentHashMap<>();
 
     public static Map<String, TokenVo> sessionMap = new ConcurrentHashMap<>();
@@ -344,4 +346,79 @@ public class TokenService {
         }
         return wplist;
     }
+
+
+    public List<ProBasicWeatherStation> getWswpls(HttpServletRequest request) {
+        String token = request.getHeader("token");
+        String userId = request.getHeader("userId");
+        List<ProBasicWeatherStation> wplist = new ArrayList<>();
+
+        if (StringUtils.notEmp(token) && StringUtils.notEmp(userId)) {
+            if (!sessionWsMap.containsKey(token)) {
+
+                List<String> depls = sysUserService.getUserByuserId(Long.valueOf(userId));
+                if (!depls.isEmpty()) {
+                    String depId = depls.get(0);
+
+                    if (CacheContext.wpmap.containsKey(depId)) {
+                        for (ProBasicWeatherStation subStation : CacheContext.weawpls) {
+                            if (subStation.getWindpowerstationId().equals(depId)) {
+                                ProBasicWeatherStation newSub = new ProBasicWeatherStation();
+                                BeanUtil.copyProperties(subStation, newSub);
+                                wplist.add(newSub);
+                            }
+                        }
+                    } else if (CacheContext.cpwpmap.containsKey(depId)) {
+                        List<ProBasicPowerstation> wpls = CacheContext.cpwpmap.get(depId);
+
+                        for (ProBasicPowerstation wp : wpls) {
+                            for (ProBasicWeatherStation subStation : CacheContext.weawpls) {
+                                if (subStation.getWindpowerstationId().equals(wp.getId())) {
+                                    ProBasicWeatherStation newSub = new ProBasicWeatherStation();
+                                    BeanUtil.copyProperties(subStation, newSub);
+                                    wplist.add(newSub);
+                                    break;
+                                }
+                            }
+                        }
+                    } else if (CacheContext.rgmap.containsKey(depId)) {
+                        List<ProBasicCompany> rgcpls = CacheContext.rgcpmap.get(depId);
+                        for (ProBasicCompany cp : rgcpls) {
+                            List<ProBasicPowerstation> wpls = CacheContext.cpwpmap.get(cp.getId());
+
+                            for (ProBasicPowerstation wp : wpls) {
+                                for (ProBasicWeatherStation subStation : CacheContext.weawpls) {
+                                    if (subStation.getWindpowerstationId().equals(wp.getId())) {
+                                        ProBasicWeatherStation newSub = new ProBasicWeatherStation();
+                                        BeanUtil.copyProperties(subStation, newSub);
+                                        wplist.add(newSub);
+                                        break;
+                                    }
+                                }
+
+                            }
+                        }
+                    } else if (depId.equals("0")) {
+                        for (ProBasicWeatherStation subStation : CacheContext.weawpls) {
+                            ProBasicWeatherStation newSub = new ProBasicWeatherStation();
+                            BeanUtil.copyProperties(subStation, newSub);
+                            wplist.add(newSub);
+                        }
+
+                    }
+                }
+                sessionWsMap.put(token, wplist);
+            } else if (sessionMap.containsKey(token) && sessionWsMap.containsKey(token)) {
+                wplist = sessionWsMap.get(token);
+
+            }
+        } else {
+            for (ProBasicWeatherStation subStation : CacheContext.weawpls) {
+                ProBasicWeatherStation newSub = new ProBasicWeatherStation();
+                BeanUtil.copyProperties(subStation, newSub);
+                wplist.add(newSub);
+            }
+        }
+        return wplist;
+    }
 }