From 929c5384397b981d427c83e1e4c13ed62b0bda10 Mon Sep 17 00:00:00 2001 From: ‘liusuyi’ <1951119284@qq.com> Date: 星期四, 17 八月 2023 15:09:39 +0800 Subject: [PATCH] 修改一件调度获取在线用户 --- ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java | 69 ++++++++++++++++++++++------------ 1 files changed, 44 insertions(+), 25 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..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,11 +3,16 @@ 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; import java.util.List; import java.util.Map; @@ -15,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 @@ -28,37 +34,48 @@ //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>> 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 map = defaultDataMap.get(session); - WebSocketUtils.sendMessage(session, map); + if (appPositionList.size() > 0) { + 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); + 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); - WebSocketUtils.sendMessage(session, DataMap); - }, 0, pushFrequency, TimeUnit.MILLISECONDS); - } + //鍙栨秷瀹氭椂浠诲姟 public static void stopLocationPushTask(Session session) { ScheduledFuture<?> scheduledTask = taskMap.get(session); @@ -66,21 +83,23 @@ 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) { + 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