修改一件调度获取在线用户

‘liusuyi’

2023-08-17 929c5384397b981d427c83e1e4c13ed62b0bda10

 ard-work/src/main/java/com/ruoyi/app/position/service/impl/AppPositionPush.java

@@ -3,6 +3,7 @@
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;
@@ -22,7 +23,7 @@
import java.util.stream.Collectors;

/**
 * @Description: app位置推送
 * @Description: app位置推送（弃用）
 * @ClassName: AppPositionPush
 * @Author: 刘苏义
 * @Date: 2023年07月25日16:07:04
@@ -33,42 +34,46 @@
    //app位置上传
    public static final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(20);
    public static final Map<Session, ScheduledFuture<?>> taskMap = new HashMap<>();
    public static final Map<String, List<ArdAppPosition>> defaultDataMap = new HashMap<>();//默认会话发送数据map
    public static final Map<String, List<ArdAppPosition>> initDataMap = new HashMap<>();//每个会话初始发送数据map

    // 初始定时任务，根据用户设置的频率定期推送实时位置信息
    public  static void initPushTask(String userId,Session session, int pushFrequency) {
        defaultDataMap.put(userId,getAppPositionList());
    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(() -> {
                List<ArdAppPosition> appPositionList = defaultDataMap.get(userId);
                Map newMap = new HashMap<>();
                newMap.put("50000", appPositionList);
                WebSocketUtils.sendMessage(session, newMap);
                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,String guideUserId, int pushFrequency) {
    public static void startLocationPushTask(String userId, Session session, String guideUserId, int pushFrequency) {
        // 取消之前设置的定时任务
        stopLocationPushTask(session);
        // 启动新的定时任务
        if (pushFrequency > 0) {
            ScheduledFuture<?> task = ScheduledFutureTask(session, userId,guideUserId, pushFrequency);
            taskMap.put(session, task); // 存储新的定时任务
        }
    }

    public static ScheduledFuture<?> ScheduledFutureTask(Session session, String userId,String guideUserId, Integer pushFrequency) {
        return scheduler.scheduleAtFixedRate(() -> {
            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);
            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); // 存储新的定时任务
        }
    }

    //取消定时任务
@@ -81,11 +86,15 @@

    //查询所有app用户的位置信息
    public static List<ArdAppPosition> getAppPositionList() {
        List<ArdAppPosition>ArdAppPositions = new ArrayList<>();
        List<ArdAppPosition> ArdAppPositions = new ArrayList<>();
        ISysUserService sysUserService = SpringUtils.getBean(ISysUserService.class);
        //获取所有app用户
        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());
            ArdAppPositions.add(ardAppPosition);