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 | 52 +++++++++++++++++++++++++++++++--------------------- 1 files changed, 31 insertions(+), 21 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 a3c2f72..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,8 +6,12 @@ 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; import java.util.List; import java.util.Map; @@ -15,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浣嶇疆鎺ㄩ�� @@ -28,37 +33,44 @@ //app浣嶇疆涓婁紶 public static final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(20); public static final Map<Session, ScheduledFuture<?>> taskMap = new HashMap<>(); - public static final Map<Session, Map> 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 map = defaultDataMap.get(session); - WebSocketUtils.sendMessage(session, map); + List<ArdAppPosition> appPositionList = defaultDataMap.get(userId); + Map newMap = new HashMap<>(); + 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); - WebSocketUtils.sendMessage(session, DataMap); + List<ArdAppPosition> appPositionList = getAppPositionList().stream() + .filter(obj -> obj.getUserId().equals(guideUserId)) + .collect(Collectors.toList()); + Map newMap = new HashMap<>(); + newMap.put("50000", appPositionList); + WebSocketUtils.sendMessage(session, newMap); }, 0, pushFrequency, TimeUnit.MILLISECONDS); } + //鍙栨秷瀹氭椂浠诲姟 public static void stopLocationPushTask(Session session) { ScheduledFuture<?> scheduledTask = taskMap.get(session); @@ -66,21 +78,19 @@ scheduledTask.cancel(false); } } + //鏌ヨ鎵�鏈塧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.selectAppUserListNoDataScope(new SysUser()); - for(SysUser sysUser:list) - { + 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