package websocket;

import io.jboot.utils.ArrayUtils;

import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArraySet;

/**
 * @author Michael Yang 杨福海 （fuhai999@gmail.com）
 * @version V1.0
 * @Title: (请输入文件名称)
 * @Description: (用一句话描述该文件做什么)
 * @Package websocket
 */
@ServerEndpoint("/myWebsocketServer")
public class WebSocketServer {


    static Map<String, CopyOnWriteArraySet<Session>> rooms = new ConcurrentHashMap<>();

    public static void sendMessage(String roomId, String message) {
        CopyOnWriteArraySet<Session> sessions = rooms.get(roomId);

        if (ArrayUtils.isNullOrEmpty(sessions)) {
            return;
        }
        for (Session session : sessions) {
            if (session.isOpen()) {
                session.getAsyncRemote().sendObject(message);
            } else {
                sessions.remove(session);
                quietClose(session);
            }
        }
    }

    public static void close(Session session) {

        String roomId = session.getRequestParameterMap().get("roomId").toString();
        CopyOnWriteArraySet<Session> sessions = rooms.get(roomId);

        if (ArrayUtils.isNotEmpty(sessions)) {
            sessions.remove(session);
        }

        quietClose(session);


    }

    private static void quietClose(Session session) {
        try {
            session.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }


    /**
     * 有新的连接的时候
     */
    @OnOpen
    public void onOpen(Session session) {
        String roomId = session.getRequestParameterMap().get("roomId").toString();
        CopyOnWriteArraySet<Session> sessions = rooms.get(roomId);
        if (sessions == null) {
            synchronized (roomId) {
                if (sessions == null) {
                    sessions = new CopyOnWriteArraySet<>();
                }
            }
        }
        sessions.add(session);

        rooms.put(roomId, sessions);
        System.out.println("有新的连接进来");
    }

    /**
     * 收到客户端消息时触发
     *
     * @param message
     * @return
     */
    @OnMessage
    public void onMessage(Session session, String message) {
        System.out.println("接受到新的消息:" + message);
        String roomId = session.getRequestParameterMap().get("roomId").toString();
        sendMessage(roomId, message);
    }

    /**
     * 异常时触发
     *
     * @param session
     */
    @OnError
    public void onError(Throwable throwable, Session session) {
        System.out.print("onError : " + session.getId() + throwable.toString());
        close(session);
    }

    /**
     * 关闭连接时触发
     *
     * @param session
     */
    @OnClose
    public void onClose(Session session) {
        System.out.print("onClose :  " + session.getId());
        close(session);
    }
}