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/app/position/service/impl/AppPositionPush.java | 47 +++++++++++++++++++++-------------------------- 1 files changed, 21 insertions(+), 26 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; } } -- Gitblit v1.9.3