From 2c97f9adf61f846edbce4ef458400416f0003ff8 Mon Sep 17 00:00:00 2001
From: liusuyi <13324259@qq.com>
Date: 星期五, 28 七月 2023 23:16:31 +0800
Subject: [PATCH] 修改exe程序调用方式 修改推送app实时位置为传参开启

---
 ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java |  117 ++++++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 88 insertions(+), 29 deletions(-)

diff --git a/ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java b/ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java
index f4186b2..6050250 100644
--- a/ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java
+++ b/ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java
@@ -4,6 +4,7 @@
 import com.ruoyi.app.position.domain.ArdAppPosition;
 import com.ruoyi.app.position.service.IArdAppPositionService;
 import com.ruoyi.app.position.service.impl.AppPositionPush;
+import com.ruoyi.app.position.service.impl.AppPositionPushService;
 import com.ruoyi.app.position.service.impl.ArdAppPositionServiceImpl;
 import com.ruoyi.common.core.domain.entity.SysUser;
 import com.ruoyi.common.utils.spring.SpringUtils;
@@ -18,10 +19,7 @@
 import javax.websocket.server.PathParam;
 import javax.websocket.server.ServerEndpoint;
 import java.io.IOException;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Optional;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -36,45 +34,93 @@
 @Slf4j(topic = "websocket")
 @ServerEndpoint("/websocket/{userId}")
 public class ChatServerEndpoint {
-
     @OnOpen
     public void openSession(@PathParam("userId") String userId, Session session) {
         WebSocketUtils.ONLINE_USER_SESSIONS.put(userId, session);
         String message = "鐢ㄦ埛[" + userId + "] 鎴愬姛杩炴帴锛�";
         log.info("鐢ㄦ埛鐧诲綍锛�" + message);
         WebSocketUtils.sendMessage(session, message);
-        //姣忎釜鐢ㄦ埛杩炴帴鏃讹紝鍚姩鍒濆鎵�鏈塧pp鐢ㄦ埛浣嶇疆鍙戦��
-        AppPositionPush.initDataMap.put(userId,AppPositionPush.getAppPositionList());
-        AppPositionPush.initPushTask(userId,session,3000);
+        //region 鏃ф柟娉�
+//        //姣忎釜鐢ㄦ埛杩炴帴鏃讹紝鍚姩鍒濆鎵�鏈塧pp鐢ㄦ埛浣嶇疆鍙戦��
+//        AppPositionPush.initDataMap.put(userId,AppPositionPush.getAppPositionList());
+//        AppPositionPush.initPushTask(userId,session,3000);
+        //endregion
     }
 
     @OnMessage
     public void onMessage(@PathParam("userId") String userId, String message) {
         log.info("鏀跺埌娑堟伅锛�" + message);
         Session session = WebSocketUtils.ONLINE_USER_SESSIONS.get(userId);
-       // WebSocketUtils.sendMessage(session, message);
+        WebSocketUtils.sendMessage(session, message);
 //        sendMessageAll("鐢ㄦ埛[" + userid + "] : " + message);
         // 鏍规嵁鐢ㄦ埛鏂扮殑棰戠巼閲嶆柊璋冩暣瀹氭椂浠诲姟
-        Map<String,Object> map = JSONObject.parseObject(message, Map.class);
-        if(map.size()>0) {
-            Integer frequency = (Integer)map.get("frequency");
+        Map<String, Object> map = JSONObject.parseObject(message, Map.class);
+        if (map.size() > 0) {
+            Boolean enabled = (Boolean) map.get("enabled");
+            if (enabled) {
+                //鍚姩鍏ㄥ眬鎺ㄩ��
+                List<ArdAppPosition> newList = new ArrayList<>(AppPositionPushService.initDataList);
+                AppPositionPushService.dataMap.put(userId, newList);
+                if (!AppPositionPushService.userTimerMap.containsKey(userId)) {
+                    Timer timer = AppPositionPushService.startTimerTask(userId, 3000);
+                    AppPositionPushService.userTimerMap.put(userId, timer);//灏嗗畾鏃跺櫒鍏宠仈鐢ㄦ埛
+                }
+            } else {
+                //鍋滄鍏ㄥ眬鎺ㄩ��
+                if (AppPositionPushService.userTimerMap.containsKey(userId)) {
+                    Timer timer = AppPositionPushService.userTimerMap.get(userId);
+                    AppPositionPushService.stopTimerTask(timer);
+                    AppPositionPushService.userTimerMap.remove(userId);
+                }
+                if (AppPositionPushService.tempUserTimerMap.containsKey(userId)) {
+                    Timer temptimer = AppPositionPushService.tempUserTimerMap.get(userId);
+                    AppPositionPushService.stopTimerTask(temptimer);
+                    AppPositionPushService.tempUserTimerMap.remove(userId);
+                }
+                AppPositionPushService.dataMap.remove(userId);
+                AppPositionPushService.tempDataMap.remove(userId);
+            }
+
+            Integer frequency = (Integer) map.get("frequency");
             String uId = (String) map.get("userId");
-            Map<String, List<ArdAppPosition>> dataMap = new HashMap<>();
-            List<ArdAppPosition> positionList = AppPositionPush.initDataMap.get(userId);//鑾峰彇褰撳墠鐢ㄦ埛鐨勯粯璁ゅ彂閫佺殑鏁版嵁map
-            if(frequency>0)
-            {
-                //鎺ユ敹鍒伴鐜囧ぇ浜�0锛屽睘浜庣敵璇峰紩瀵硷紝鑾峰彇闄ゅ紩瀵煎鐨勫叾浠朼pp鐢ㄦ埛鐨勪綅缃甃ist
-                List<ArdAppPosition> appPositionList = positionList.stream()
+            if (frequency > 0) {
+                List<ArdAppPosition> List = AppPositionPushService.initDataList.stream()
                         .filter(obj -> !obj.getUserId().equals(uId))
-                        .collect(Collectors.toList());
-                dataMap.put(userId,appPositionList);
+                        .collect(Collectors.toList());//鍘婚櫎寮曞鍚庣殑list
+                AppPositionPushService.dataMap.put(userId,List);
+
+                List<ArdAppPosition> tempList = AppPositionPushService.initDataList.stream()
+                        .filter(obj -> obj.getUserId().equals(uId))
+                        .collect(Collectors.toList());//鑾峰彇寮曞鐨刲ist
+                AppPositionPushService.tempDataMap.put(userId, tempList);
+                Timer timer = AppPositionPushService.startTempTimerTask(userId, frequency);
+                AppPositionPushService.tempUserTimerMap.put(userId, timer);
+            } else {
+                List<ArdAppPosition> newList = new ArrayList<>(AppPositionPushService.initDataList);
+                AppPositionPushService.dataMap.put(userId, newList);
+                Timer timer = AppPositionPushService.tempUserTimerMap.get(userId);
+                AppPositionPushService.stopTimerTask(timer);
+                AppPositionPushService.tempUserTimerMap.remove(userId);
+                AppPositionPushService.tempDataMap.remove(userId);
             }
-           else {
-                //鎺ユ敹鍒伴鐜囩瓑浜�0锛屽睘浜庨噴鏀惧紩瀵硷紝鑾峰彇鍒板叏閮╝pp鐢ㄦ埛鐨勪綅缃甃ist
-                dataMap.put(userId,AppPositionPush.getAppPositionList());
-            }
-            AppPositionPush.initDataMap.putAll(dataMap);//鏇存柊鍒濆浼氳瘽榛樿鏁版嵁map
-            AppPositionPush.startLocationPushTask(userId,session,uId, frequency);//寮�鍚柊浠诲姟
+            //region 鏃ф柟娉�
+//            Map<String, List<ArdAppPosition>> dataMap = new HashMap<>();
+//            List<ArdAppPosition> positionList = AppPositionPush.initDataMap.get(userId);//鑾峰彇褰撳墠鐢ㄦ埛鐨勯粯璁ゅ彂閫佺殑鏁版嵁map
+//            if(frequency>0)
+//            {
+//                //鎺ユ敹鍒伴鐜囧ぇ浜�0锛屽睘浜庣敵璇峰紩瀵硷紝鑾峰彇闄ゅ紩瀵煎鐨勫叾浠朼pp鐢ㄦ埛鐨勪綅缃甃ist
+//                List<ArdAppPosition> appPositionList = positionList.stream()
+//                        .filter(obj -> !obj.getUserId().equals(uId))
+//                        .collect(Collectors.toList());
+//                dataMap.put(userId,appPositionList);
+//            }
+//           else {
+//                //鎺ユ敹鍒伴鐜囩瓑浜�0锛屽睘浜庨噴鏀惧紩瀵硷紝鑾峰彇鍒板叏閮╝pp鐢ㄦ埛鐨勪綅缃甃ist
+//                dataMap.put(userId,AppPositionPush.getAppPositionList());
+//            }
+//            AppPositionPush.initDataMap.putAll(dataMap);//鏇存柊鍒濆浼氳瘽榛樿鏁版嵁map
+//            AppPositionPush.startLocationPushTask(userId,session,uId, frequency);//寮�鍚柊浠诲姟
+            //endregion
         }
     }
 
@@ -83,12 +129,25 @@
         //褰撳墠鐨凷ession 绉婚櫎
         WebSocketUtils.ONLINE_USER_SESSIONS.remove(userId);
 //        //骞朵笖閫氱煡鍏朵粬浜哄綋鍓嶇敤鎴峰凡缁忕寮�鑱婂ぉ瀹や簡
-       String message="鐢ㄦ埛[" + userId + "] 鏂紑杩炴帴锛�";
+        String message = "鐢ㄦ埛[" + userId + "] 鏂紑杩炴帴锛�";
 //        sendMessage(session,message);
-        log.info("娑堟伅锛�"+message);
+        log.info("娑堟伅锛�" + message);
         try {
             session.close();
-            AppPositionPush.stopLocationPushTask(session);
+            //鍋滄鍏ㄥ眬鎺ㄩ��
+            if (AppPositionPushService.userTimerMap.containsKey(userId)) {
+                Timer timer = AppPositionPushService.userTimerMap.get(userId);
+                AppPositionPushService.stopTimerTask(timer);
+                AppPositionPushService.userTimerMap.remove(userId);
+            }
+            if (AppPositionPushService.tempUserTimerMap.containsKey(userId)) {
+                Timer temptimer = AppPositionPushService.tempUserTimerMap.get(userId);
+                AppPositionPushService.stopTimerTask(temptimer);
+                AppPositionPushService.tempUserTimerMap.remove(userId);
+            }
+            AppPositionPushService.dataMap.remove(userId);
+            AppPositionPushService.tempDataMap.remove(userId);
+            // AppPositionPush.stopLocationPushTask(session);//鏃ф柟娉�
         } catch (IOException e) {
             log.error("onClose error", e);
         }

--
Gitblit v1.9.3