From 82f8f0065a27acf9f7e1558b04fa75ce68a7fa50 Mon Sep 17 00:00:00 2001
From: aijinhui <aijinhui>
Date: 星期二, 26 三月 2024 17:05:30 +0800
Subject: [PATCH] 31

---
 ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java |   72 +++++++++++++++++++-----------------
 1 files changed, 38 insertions(+), 34 deletions(-)

diff --git a/ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java b/ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java
index 3314c4d..fc5f294 100644
--- a/ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java
+++ b/ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java
@@ -3,10 +3,14 @@
 import com.ruoyi.app.position.domain.ArdAppPosition;
 import com.ruoyi.app.position.service.IArdAppPositionService;
 import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.spring.SpringUtils;
 import com.ruoyi.utils.websocket.util.WebSocketUtils;
 import com.ruoyi.system.service.ISysUserService;
+import org.springframework.stereotype.Component;
 
+import javax.annotation.PostConstruct;
+import javax.annotation.Resource;
 import javax.websocket.Session;
 import java.util.ArrayList;
 import java.util.HashMap;
@@ -16,9 +20,10 @@
 import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.ScheduledFuture;
 import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 /**
- * @Description: app浣嶇疆鎺ㄩ��
+ * @Description: app浣嶇疆鎺ㄩ�侊紙寮冪敤锛�
  * @ClassName: AppPositionPush
  * @Author: 鍒樿嫃涔�
  * @Date: 2023骞�07鏈�25鏃�16:07:04
@@ -29,50 +34,46 @@
     //app浣嶇疆涓婁紶
     public static final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(20);
     public static final Map<Session, ScheduledFuture<?>> taskMap = new HashMap<>();
-    public static final Map<Session, Map<String, ArdAppPosition>> defaultDataMap = new HashMap<>();//榛樿浼氳瘽鍙戦�佹暟鎹甿ap
+    public static final Map<String, List<ArdAppPosition>> initDataMap = new HashMap<>();//姣忎釜浼氳瘽鍒濆鍙戦�佹暟鎹甿ap
 
     // 鍒濆瀹氭椂浠诲姟锛屾牴鎹敤鎴疯缃殑棰戠巼瀹氭湡鎺ㄩ�佸疄鏃朵綅缃俊鎭�
-    public static void initPushTask(Session session, int pushFrequency) {
+    public static void initPushTask(String userId, Session session, int pushFrequency) {
+        initDataMap.put(userId, getAppPositionList());
+        List<ArdAppPosition> appPositionList = initDataMap.get(userId);
+        if (appPositionList.size() == 0) {
+            return;
+        }
         // 鍚姩鏂扮殑瀹氭椂浠诲姟
         if (pushFrequency > 0) {
             scheduler.scheduleAtFixedRate(() -> {
-                Map<String, ArdAppPosition> map = defaultDataMap.get(session);
-                Map newMap = new HashMap<>();
-                List<ArdAppPosition> positionList = new ArrayList<>();
-                for (String key : map.keySet()) {
-                    positionList.add(map.get(key));
+                if (appPositionList.size() > 0) {
+                    Map newMap = new HashMap<>();
+                    newMap.put("50000", appPositionList);
+                    WebSocketUtils.sendMessage(session, newMap);
                 }
-                newMap.put("50000", positionList);
-                WebSocketUtils.sendMessage(session, newMap);
             }, 0, pushFrequency, TimeUnit.MILLISECONDS);
         }
     }
 
     // 鏂扮殑瀹氭椂浠诲姟锛屾牴鎹敤鎴疯缃殑棰戠巼瀹氭湡鎺ㄩ�佸疄鏃朵綅缃俊鎭�
-    public static void startLocationPushTask(String userId, Session session, int pushFrequency) {
+    public static void startLocationPushTask(String userId, Session session, String guideUserId, int pushFrequency) {
         // 鍙栨秷涔嬪墠璁剧疆鐨勫畾鏃朵换鍔�
         stopLocationPushTask(session);
-
         // 鍚姩鏂扮殑瀹氭椂浠诲姟
         if (pushFrequency > 0) {
-            ScheduledFuture<?> task = ScheduledFutureTask(session, userId, pushFrequency);
+            List<ArdAppPosition> appPositionList = getAppPositionList().stream()
+                    .filter(obj -> obj.getUserId().equals(guideUserId))
+                    .collect(Collectors.toList());
+            if (appPositionList.size() == 0) {
+                return;
+            }
+            ScheduledFuture<?> task = scheduler.scheduleAtFixedRate(() -> {
+                Map newMap = new HashMap<>();
+                newMap.put("50000", appPositionList);
+                WebSocketUtils.sendMessage(session, newMap);
+            }, 0, pushFrequency, TimeUnit.MILLISECONDS);
             taskMap.put(session, task); // 瀛樺偍鏂扮殑瀹氭椂浠诲姟
         }
-    }
-
-    public static ScheduledFuture<?> ScheduledFutureTask(Session session, String userId, Integer pushFrequency) {
-        return scheduler.scheduleAtFixedRate(() -> {
-            ArdAppPosition ardAppPosition = getAppPositionList().get(userId);
-            Map<String, ArdAppPosition> DataMap = new HashMap<>();
-            DataMap.put(userId, ardAppPosition);
-            Map newMap = new HashMap<>();
-            List<ArdAppPosition> positionList = new ArrayList<>();
-            for (String key : DataMap.keySet()) {
-                positionList.add(DataMap.get(key));
-            }
-            newMap.put("50000", positionList);
-            WebSocketUtils.sendMessage(session, newMap);
-        }, 0, pushFrequency, TimeUnit.MILLISECONDS);
     }
 
     //鍙栨秷瀹氭椂浠诲姟
@@ -84,18 +85,21 @@
     }
 
     //鏌ヨ鎵�鏈塧pp鐢ㄦ埛鐨勪綅缃俊鎭�
-    public static Map<String, ArdAppPosition> getAppPositionList() {
-        Map<String, ArdAppPosition> userMap = new HashMap<>();
+    public static List<ArdAppPosition> getAppPositionList() {
+        List<ArdAppPosition> ArdAppPositions = new ArrayList<>();
         ISysUserService sysUserService = SpringUtils.getBean(ISysUserService.class);
         //鑾峰彇鎵�鏈塧pp鐢ㄦ埛
-        List<SysUser> list = sysUserService.selectAllocatedList(new SysUser());
+        List<SysUser> list = sysUserService.selectAllAppUserList(new SysUser());
         for (SysUser sysUser : list) {
+            if(StringUtils.isNull(sysUser.getAppOnlineState()))
+            {continue;}
+            if(sysUser.getAppOnlineState().equals("0"))
+            {continue; }
             IArdAppPositionService ardAppPositionService = SpringUtils.getBean(IArdAppPositionService.class);
             ArdAppPosition ardAppPosition = ardAppPositionService.selectLastArdAppPositionByUserId(sysUser.getUserId());
-            userMap.put(sysUser.getUserId(), ardAppPosition);
+            ArdAppPositions.add(ardAppPosition);
         }
-
-        return userMap;
+        return ArdAppPositions;
     }
 }
 

--
Gitblit v1.9.3