package com.ruoyi.common.websocket.service;

import com.ruoyi.common.utils.SecurityUtils;
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.*;


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

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

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

    @OnMessage
    public void onMessage( String message) {
        log.info("收到消息："+message);
        String userId = SecurityUtils.getUserId();
        Session session = ONLINE_USER_SESSIONS.get(userId);
        sendMessage(session,message);
//        sendMessageAll("用户[" + userid + "] : " + message);
    }

    @OnClose
    public void onClose( Session session) {
        String userId = SecurityUtils.getUserId();
        //当前的Session 移除
        ONLINE_USER_SESSIONS.remove(userId);
        //并且通知其他人当前用户已经离开聊天室了
        String message="用户[" + userId + "] 断开连接！";
        sendMessage(session,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());
    }
}