package com.ruoyi.common.websocket.service;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;

import static com.ruoyi.common.websocket.util.WebSocketUtils.ONLINE_USER_SESSIONS;
import static com.ruoyi.common.websocket.util.WebSocketUtils.sendMessageAll;


/**
 * @ClassName ChatServerEndpoint
 * @Description:
 * @Author 刘苏义
 * @Date 2023/1/27 7:42
 * @Version 1.0
 */

@Component
@Slf4j(topic = "websocket")
@ServerEndpoint("/websocket/{userid}")
public class ChatServerEndpoint {

    @OnOpen
    public void openSession(@PathParam("userid") String userid, Session session) {
        ONLINE_USER_SESSIONS.put(userid, session);
        String message = "用户[" + userid + "] 成功连接！";
        log.info("用户登录："+message);
        sendMessageAll(message);
    }

    @OnMessage
    public void onMessage(@PathParam("userid") String userid, String message) {
        log.info("发送消息："+message);
        sendMessageAll("用户[" + userid + "] : " + message);
    }

    @OnClose
    public void onClose(@PathParam("userid") String userid, Session session) {
        //当前的Session 移除
        ONLINE_USER_SESSIONS.remove(userid);
        //并且通知其他人当前用户已经离开聊天室了
        String message="用户[" + userid + "] 断开连接！";
        sendMessageAll(message);
        log.info("发送消息："+message);
        try {
            session.close();
        } catch (IOException e) {
            log.error("onClose error",e);
        }
    }

    @OnError
    public void onError(Session session, Throwable throwable) {
        try {
            session.close();
        } catch (IOException e) {
            log.error("onError excepiton",e);
        }
        log.info("Throwable msg "+throwable.getMessage());
    }
}