com.ape9527.core.socket.IMWebSocketServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ape-core Show documentation
Ape low code platform core module
The newest version!
//package com.ape9527.core.socket;
//
//import com.ape9527.auth.service.LoginUserService;
//import com.ape9527.core.service.BaseDataService;
//import com.ape9527.utils.string.StringUtil;
//import lombok.extern.slf4j.Slf4j;
//import org.springframework.beans.factory.annotation.Autowired;
//import org.springframework.stereotype.Component;
//
//import javax.websocket.*;
//import javax.websocket.server.ServerEndpoint;
//import java.io.IOException;
//import java.util.concurrent.ConcurrentHashMap;
//
///**
// * @author YuanShuai[[email protected]]
// */
//@ServerEndpoint("/socket/im")
//@Component
//@Slf4j
//public class IMWebSocketServer {
//
//    /**
//     * 静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
//     */
//    private static int onlineCount = 0;
//
//    /**
//     * concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
//     */
//    private static ConcurrentHashMap webSocketMap = new ConcurrentHashMap<>();
//
//    /**
//     * 与某个客户端的连接会话，需要通过它来给客户端发送数据
//     */
//    private Session session;
//
//    /**
//     * 记录userId
//     */
//    private String userId;
//
//    private static LoginUserService loginUserService;
//
//    private static BaseDataService baseDataService;
//
//    @Autowired
//    public void setLoginUtil(LoginUserService loginUserService) {
//        IMWebSocketServer.loginUserService = loginUserService;
//    }
//
//    @Autowired
//    public void setBaseDataService(BaseDataService baseDataService) {
//        IMWebSocketServer.baseDataService = baseDataService;
//    }
//
//    @OnOpen
//    public void onOpen(Session session) {
//        // 从URL中获取token参数
//        String uri = session.getRequestURI().toString();
//        String token = uri.substring(uri.indexOf("=") + 1);
//        if(StringUtil.isEmpty(token) || "null".equals(token)){
//            return;
//        }
//        String userId = loginUserService.getUserId(token);
//        this.session = session;
//        this.userId = userId;
//        if (webSocketMap.containsKey(userId)) {
//            webSocketMap.remove(userId);
//            webSocketMap.put(userId, this);
//        } else {
//            webSocketMap.put(userId, this);
//            addOnlineCount();
//        }
//
//        log.info("用户连接:" + userId + ",当前在线人数为:" + getOnlineCount());
//
//        try {
//            sendMessage("服务器连接成功");
//        } catch (IOException e) {
//            log.error("用户:" + userId + ",网络异常!!!!!!");
//        }
//    }
//
//    /**
//     * 连接关闭调用的方法
//     */
//    @OnClose
//    public void onClose() {
//        if (StringUtil.isNotEmpty(userId) && webSocketMap.containsKey(userId)) {
//            webSocketMap.remove(userId);
//            //从set中删除
//            subOnlineCount();
//        }
//        log.info("用户退出:" + userId + ",当前在线人数为:" + getOnlineCount());
//    }
//
//    /**
//     * 收到客户端消息后调用的方法
//     *
//     * @param message 客户端发送过来的消息
//     */
//    @OnMessage
//    public void onMessage(String message, Session session) {
//        log.info("用户消息:" + userId + ",报文:" + message);
//        //可以群发消息
//        //消息保存到数据库、redis
//        if (StringUtil.isNotEmpty(message)) {
//            if ("ping".equals(message)) {
//                // 心跳保持机制，原样返回
//                try {
//                    sendMessage(message);
//                } catch (IOException e) {
//                    e.printStackTrace();
//                }
//            } else {
//                // 正常处理
//            }
//        }
//    }
//
//    /**
//     * @param session
//     * @param error
//     */
//    @OnError
//    public void onError(Session session, Throwable error) {
//        log.error("用户错误:" + this.userId + ",原因:" + error.getMessage());
//        error.printStackTrace();
//    }
//
//    /**
//     * 实现服务器主动推送
//     */
//    public void sendMessage(String message) throws IOException {
//        this.session.getBasicRemote().sendText(message);
//    }
//
//    /**
//     * 发送自定义消息
//     */
//    public static void sendInfo(String userId, String message) throws IOException {
//        log.info("发送消息到:" + userId + "，报文:" + message);
//        if (StringUtil.isNotEmpty(userId) && webSocketMap.containsKey(userId)) {
//            webSocketMap.get(userId).sendMessage(message);
//        } else {
//            log.error("用户" + userId + ",不在线！");
//        }
//    }
//
//    /**
//     * 消息群发至所有在线用户
//     */
//    public static void sendInfoAll(String userId, String message) {
//        log.info("发送消息到: 所有人，报文:" + message);
//        for (String toUserId : webSocketMap.keySet()) {
//            if(!userId.equals(toUserId)){
//                try {
//                    webSocketMap.get(toUserId).sendMessage(message);
//                } catch (IOException e) {
//                    log.error("向用户：{} 推送失败", toUserId);
//                }
//            }
//        }
//    }
//
//    public static synchronized int getOnlineCount() {
//        return onlineCount;
//    }
//
//    public static synchronized void addOnlineCount() {
//        IMWebSocketServer.onlineCount++;
//    }
//
//    public static synchronized void subOnlineCount() {
//        IMWebSocketServer.onlineCount--;
//    }
//
//}