package com.ard.route.websocket;

import com.ard.work.component.websocket.utils.WebSocketUtil;
import org.springframework.web.socket.WebSocketSession;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

public class MiWebSocketSessionManager {
    private static final ConcurrentMap<Long, WebSocketSession> ONLINE_USER_SESSIONS = new ConcurrentHashMap<>();

    public static void addSession(Long userId, WebSocketSession session) {
        ONLINE_USER_SESSIONS.put(userId, session);
    }

    public static void removeSession(Long userId) {
        ONLINE_USER_SESSIONS.remove(userId);
    }

    public static WebSocketSession getSession(Long userId) {
        return ONLINE_USER_SESSIONS.get(userId);
    }

    public static void broadcast(String message) {
        ONLINE_USER_SESSIONS.forEach((userId, session) -> WebSocketUtil.sendMessage(session, message));
    }

    //返回当前在线用户数
    public static int getOnlineUserCount() {
        return ONLINE_USER_SESSIONS.size();
    }

    public static boolean containsUser(Long userId) {
        return ONLINE_USER_SESSIONS.containsKey(userId);
    }
}