liusuyi
2025-10-27 8a1be2c973eacb1fb661caf920321eb2c58ef87c
ard-work/src/main/java/com/ard/work/component/websocket/handler/BaseSubscriptionManager.java
@@ -24,8 +24,9 @@
     * 订阅指定设备的数据
     */
    public void subscribe(String boxSn, Long userId, Integer period) {
        subscriptions.computeIfAbsent(boxSn, k -> new HashSet<>()).add(userId);
        subscriptions.computeIfAbsent(boxSn, k -> ConcurrentHashMap.newKeySet()).add(userId);
        if (subscriptions.get(boxSn).size() == 1) {
            log.debug("首次订阅,启动推送任务");
            startPushTask(boxSn, period);
        }
    }
@@ -34,15 +35,19 @@
     * 取消订阅
     */
    public void unsubscribe(String boxSn, Long userId) {
        Set<Long> userIds = subscriptions.get(boxSn);
        if (userIds != null) {
        subscriptions.compute(boxSn, (key, userIds) -> {
            if (userIds == null) {
                return null;
            }
            userIds.remove(userId);
            if (userIds.isEmpty()) {
                stopPushTask(boxSn);
                log.debug("无人订阅设备 {},移除订阅记录", boxSn);
                return null; // 直接返回 null 等价于 remove(key)
            }
            return userIds;
        });
        }
    }
    /**
     * 启动推送任务
     */
@@ -52,7 +57,11 @@
            return;
        }
        ScheduledFuture<?> future = scheduler.scheduleAtFixedRate(() -> {
            try {
            sendDataToSubscribers(boxSn);
            } catch (Exception e) {
                log.error("Error in push task for boxSn={}", boxSn, e);
            }
        }, 0, period, TimeUnit.SECONDS);
        pushTasks.put(boxSn, future);