package cc.mrbird.febs.websocket;

import cn.hutool.core.util.StrUtil;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;

import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @date 2020-09-01
 **/
public class WsSessionManager {

    private static final ConcurrentHashMap<String, WebSocketSession> SESSIONS = new ConcurrentHashMap<>();

    public static void add(String key, WebSocketSession session) {
        SESSIONS.put(key, session);
    }


    public static WebSocketSession remove(String key) {
        return SESSIONS.remove(key);
    }

    public static void removeAndClose(String key) {
        WebSocketSession session = remove(key);
        if (session != null) {
            try {
                // 关闭连接
                session.close();
            } catch (IOException e) {
                // todo: 关闭出现异常处理
                e.printStackTrace();
            }
        }
    }

    public static WebSocketSession get(String key) {
        // 获得 session
        return SESSIONS.get(key);
    }

    /**
     * 发送消息
     *
     * @param key 用户手机号
     * @param msg 消息
     */
    public static void sendMsgToOne(String key, String msg) {
        TextMessage textMessage = new TextMessage(msg);
        try {
            if (SESSIONS.containsKey(key)) {
                SESSIONS.get(key).sendMessage(textMessage);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 批量发送
     *
     * @param keys 手机号集合, 逗号隔开
     * @param msg 消息
     */
    public static void sendMsgToMany(String keys, String msg) {
        TextMessage textMessage = new TextMessage(msg);

        List<String> keyList = StrUtil.splitTrim(keys, ",");
        for (Map.Entry<String, WebSocketSession> entry : SESSIONS.entrySet()) {
            if (keyList.contains(entry.getKey())) {
                try {
                    entry.getValue().sendMessage(textMessage);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

}