From 6b44b988edd13970fb982284799eaa4593002d89 Mon Sep 17 00:00:00 2001
From: ‘liusuyi’ <1951119284@qq.com>
Date: 星期四, 27 七月 2023 17:24:32 +0800
Subject: [PATCH] 提交
---
ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java | 35 +++++++++++------
ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java | 2
ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java | 47 ++++++++++-------------
ard-work/src/main/java/com/ruoyi/app/position/service/impl/ArdAppPositionServiceImpl.java | 7 +++
4 files changed, 52 insertions(+), 39 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..8e3f633 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
@@ -6,7 +6,10 @@
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,6 +19,7 @@
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
/**
* @Description: app浣嶇疆鎺ㄩ��
@@ -29,48 +33,40 @@
//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>> defaultDataMap = new HashMap<>();//榛樿浼氳瘽鍙戦�佹暟鎹甿ap
// 鍒濆瀹氭椂浠诲姟锛屾牴鎹敤鎴疯缃殑棰戠巼瀹氭湡鎺ㄩ�佸疄鏃朵綅缃俊鎭�
- public static void initPushTask(Session session, int pushFrequency) {
+ public static void initPushTask(String userId,Session session, int pushFrequency) {
+ defaultDataMap.put(userId,getAppPositionList());
// 鍚姩鏂扮殑瀹氭椂浠诲姟
if (pushFrequency > 0) {
scheduler.scheduleAtFixedRate(() -> {
- Map<String, ArdAppPosition> map = defaultDataMap.get(session);
+ List<ArdAppPosition> appPositionList = defaultDataMap.get(userId);
Map newMap = new HashMap<>();
- List<ArdAppPosition> positionList = new ArrayList<>();
- for (String key : map.keySet()) {
- positionList.add(map.get(key));
- }
- newMap.put("50000", positionList);
+ newMap.put("50000", appPositionList);
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);
+ ScheduledFuture<?> task = ScheduledFutureTask(session, userId,guideUserId, pushFrequency);
taskMap.put(session, task); // 瀛樺偍鏂扮殑瀹氭椂浠诲姟
}
}
- public static ScheduledFuture<?> ScheduledFutureTask(Session session, String userId, Integer pushFrequency) {
+ public static ScheduledFuture<?> ScheduledFutureTask(Session session, String userId,String guideUserId, Integer pushFrequency) {
return scheduler.scheduleAtFixedRate(() -> {
- ArdAppPosition ardAppPosition = getAppPositionList().get(userId);
- Map<String, ArdAppPosition> DataMap = new HashMap<>();
- DataMap.put(userId, ardAppPosition);
+ List<ArdAppPosition> appPositionList = getAppPositionList().stream()
+ .filter(obj -> obj.getUserId().equals(guideUserId))
+ .collect(Collectors.toList());
Map newMap = new HashMap<>();
- List<ArdAppPosition> positionList = new ArrayList<>();
- for (String key : DataMap.keySet()) {
- positionList.add(DataMap.get(key));
- }
- newMap.put("50000", positionList);
+ newMap.put("50000", appPositionList);
WebSocketUtils.sendMessage(session, newMap);
}, 0, pushFrequency, TimeUnit.MILLISECONDS);
}
@@ -84,18 +80,17 @@
}
//鏌ヨ鎵�鏈塧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) {
IArdAppPositionService ardAppPositionService = SpringUtils.getBean(IArdAppPositionService.class);
ArdAppPosition ardAppPosition = ardAppPositionService.selectLastArdAppPositionByUserId(sysUser.getUserId());
- userMap.put(sysUser.getUserId(), ardAppPosition);
+ ArdAppPositions.add(ardAppPosition);
}
-
- return userMap;
+ return ArdAppPositions;
}
}
diff --git a/ard-work/src/main/java/com/ruoyi/app/position/service/impl/ArdAppPositionServiceImpl.java b/ard-work/src/main/java/com/ruoyi/app/position/service/impl/ArdAppPositionServiceImpl.java
index 18e068d..a33cb51 100644
--- a/ard-work/src/main/java/com/ruoyi/app/position/service/impl/ArdAppPositionServiceImpl.java
+++ b/ard-work/src/main/java/com/ruoyi/app/position/service/impl/ArdAppPositionServiceImpl.java
@@ -3,12 +3,18 @@
import com.ruoyi.app.position.domain.ArdAppPosition;
import com.ruoyi.app.position.mapper.ArdAppPositionMapper;
import com.ruoyi.app.position.service.IArdAppPositionService;
+import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.utils.DateUtils;
+import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.uuid.IdUtils;
+import com.ruoyi.system.service.ISysUserService;
import org.springframework.stereotype.Service;
+import javax.annotation.PostConstruct;
import javax.annotation.Resource;
+import java.util.HashMap;
import java.util.List;
+import java.util.Map;
/**
@@ -22,6 +28,7 @@
@Resource
private ArdAppPositionMapper ardAppPositionMapper;
+
/**
* 鏌ヨapp浣嶇疆
*
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 31aaffd..92c5fc3 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
@@ -2,17 +2,25 @@
import com.alibaba.fastjson2.JSONObject;
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.common.core.domain.entity.SysUser;
+import com.ruoyi.system.service.ISysUserService;
import com.ruoyi.utils.websocket.util.WebSocketUtils;
import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
+import javax.annotation.Resource;
import javax.websocket.*;
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.stream.Collectors;
/**
* @ClassName ChatServerEndpoint
@@ -34,9 +42,10 @@
log.info("鐢ㄦ埛鐧诲綍锛�" + message);
WebSocketUtils.sendMessage(session, message);
//姣忎釜鐢ㄦ埛杩炴帴鏃讹紝鍚姩鍒濆鎵�鏈塧pp鐢ㄦ埛浣嶇疆鍙戦��
- Map<String, ArdAppPosition> appPositionMap = AppPositionPush.getAppPositionList();
- AppPositionPush.defaultDataMap.put(session,appPositionMap);
- AppPositionPush.initPushTask(session,3000);
+
+ List<ArdAppPosition> positionList = AppPositionPush.defaultDataMap.get(userId);
+ AppPositionPush.defaultDataMap.put(userId,positionList);
+ AppPositionPush.initPushTask(userId,session,3000);
}
@OnMessage
@@ -50,20 +59,22 @@
if(map.size()>0) {
Integer frequency = (Integer)map.get("frequency");
String uId = (String) map.get("userId");
- Map allMap = AppPositionPush.defaultDataMap.get(session);//鑾峰彇褰撳墠session鐨勯粯璁ゅ彂閫佺殑鏁版嵁map
+ Map<String, List<ArdAppPosition>> dataMap = new HashMap<>();
+ List<ArdAppPosition> positionList = AppPositionPush.defaultDataMap.get(userId);//鑾峰彇褰撳墠鐢ㄦ埛鐨勯粯璁ゅ彂閫佺殑鏁版嵁map
if(frequency>0)
{
- //鎺ユ敹鍒伴鐜囧ぇ浜�0锛屽睘浜庣敵璇峰紩瀵硷紝灏哸pp浠庨粯璁ap涓Щ闄�
- allMap.remove(uId);
- AppPositionPush.defaultDataMap.putAll(allMap);
+ //鎺ユ敹鍒伴鐜囧ぇ浜�0锛屽睘浜庣敵璇峰紩瀵硷紝鑾峰彇闄ゅ紩瀵煎鐨勫叾浠朼pp鐢ㄦ埛鐨勪綅缃甃ist
+ List<ArdAppPosition> appPositionList = positionList.stream()
+ .filter(obj -> !obj.getUserId().equals(uId))
+ .collect(Collectors.toList());
+ dataMap.put(userId,appPositionList);
}
else {
- //鎺ユ敹鍒伴鐜囩瓑浜�0锛屽睘浜庨噴鏀惧紩瀵硷紝灏哸pp鎭㈠鍒板垵濮媘ap
- Map<String, ArdAppPosition> appPositionList = AppPositionPush.getAppPositionList();
- allMap.put(uId,appPositionList.get(uId));
- AppPositionPush.defaultDataMap.putAll(allMap);
+ //鎺ユ敹鍒伴鐜囩瓑浜�0锛屽睘浜庨噴鏀惧紩瀵硷紝鑾峰彇鍒板叏閮╝pp鐢ㄦ埛鐨勪綅缃甃ist
+ dataMap.put(userId,AppPositionPush.getAppPositionList());
}
- AppPositionPush.startLocationPushTask(uId, session, frequency);
+ AppPositionPush.defaultDataMap.putAll(dataMap);//鏇存柊鍒濆浼氳瘽榛樿鏁版嵁map
+ AppPositionPush.startLocationPushTask(userId,session,uId, frequency);//寮�鍚柊浠诲姟
}
}
diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java
index 67dbedb..826975b 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java
@@ -143,7 +143,7 @@
* @return 鐢ㄦ埛淇℃伅闆嗗悎淇℃伅
*/
@Override
- @DataScope(deptAlias = "d", userAlias = "u")
+ //@DataScope(deptAlias = "d", userAlias = "u")
public List<SysUser> selectAllAppUserList(SysUser user) {
return userMapper.selectAllAppUserList(user);
}
--
Gitblit v1.9.3