重构app用户由角色转为用户配置

‘liusuyi’

2023-07-27 4155135bb0247685ce3652fe0257c39ff3e21ced

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package com.ruoyi.app.position.service.impl;
 
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.spring.SpringUtils;
import com.ruoyi.utils.websocket.util.WebSocketUtils;
import com.ruoyi.system.service.ISysUserService;
 
import javax.websocket.Session;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
 
/**
 * @Description: app位置推送
 * @ClassName: AppPositionPush
 * @Author: 刘苏义
 * @Date: 2023年07月25日16:07:04
 * @Version: 1.0
 **/
public class AppPositionPush {
 
    //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<>();//默认会话发送数据map
 
    // 初始定时任务，根据用户设置的频率定期推送实时位置信息
    public static void initPushTask(Session session, int pushFrequency) {
        // 启动新的定时任务
        if (pushFrequency > 0) {
            scheduler.scheduleAtFixedRate(() -> {
                Map<String, ArdAppPosition> map = defaultDataMap.get(session);
                Map newMap = new HashMap<>();
                List<ArdAppPosition> positionList = new ArrayList<>();
                for (String key : map.keySet()) {
                    positionList.add(map.get(key));
                }
                newMap.put("50000", positionList);
                WebSocketUtils.sendMessage(session, newMap);
            }, 0, pushFrequency, TimeUnit.MILLISECONDS);
        }
    }
 
    // 新的定时任务，根据用户设置的频率定期推送实时位置信息
    public static void startLocationPushTask(String userId, Session session, int pushFrequency) {
        // 取消之前设置的定时任务
        stopLocationPushTask(session);
 
        // 启动新的定时任务
        if (pushFrequency > 0) {
            ScheduledFuture<?> task = ScheduledFutureTask(session, userId, pushFrequency);
            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);
            Map newMap = new HashMap<>();
            List<ArdAppPosition> positionList = new ArrayList<>();
            for (String key : DataMap.keySet()) {
                positionList.add(DataMap.get(key));
            }
            newMap.put("50000", positionList);
            WebSocketUtils.sendMessage(session, newMap);
        }, 0, pushFrequency, TimeUnit.MILLISECONDS);
    }
 
    //取消定时任务
    public static void stopLocationPushTask(Session session) {
        ScheduledFuture<?> scheduledTask = taskMap.get(session);
        if (scheduledTask != null) {
            scheduledTask.cancel(false);
        }
    }
 
    //查询所有app用户的位置信息
    public static Map<String, ArdAppPosition> getAppPositionList() {
        Map<String, ArdAppPosition> userMap = new HashMap<>();
        ISysUserService sysUserService = SpringUtils.getBean(ISysUserService.class);
        //获取所有app用户
        List<SysUser> list = sysUserService.selectAllocatedList(new SysUser());
        for (SysUser sysUser : list) {
            IArdAppPositionService ardAppPositionService = SpringUtils.getBean(IArdAppPositionService.class);
            ArdAppPosition ardAppPosition = ardAppPositionService.selectLastArdAppPositionByUserId(sysUser.getUserId());
            userMap.put(sysUser.getUserId(), ardAppPosition);
        }
 
        return userMap;
    }
}