基于java的websocket实现

基于tomcat的websocket实现中，如果会针对一个用户高频的发送消息的情况下，websocket自带的异步发送其实一点卵用都没有，所以，没办法，自己控制websocket的并发发送喽，不多说，直接贴代码

1.基础的websocket缓存对象，这个玩意儿每一个用户进来都会生成一个

public class WebSocketMessage {
    

//读写锁
public ReentrantReadWriteLock lock = new ReentrantReadWriteLock();

private Session session;

public ReentrantReadWriteLock getLock () { return lock ; } public Session getSession () { return session ; } public void setSession ( Session session ) { this . session = session ; } public Long getRoomId () { return roomId ; } public void setRoomId ( Long roomId ) { this . roomId = roomId ; } }

2：websocket的实现

@ServerEndpoint(value = "/client/socket/flop/{userId}")
public class WebSocket {

    private static UserMessageDao userMessageDao;

    /**
     * 在线人数
     */
    private static AtomicInteger atomicInteger = new AtomicInteger(0);

    /**
     * 房间信息缓存
     */
    private static ConcurrentHashMap<Long,FlopWebSocketMessage> user_flopWebSocketMessage = new ConcurrentHashMap<>();
    

/**
 * session对应用户id
 */
private static ConcurrentHashMap<Session,Long> session_userId = new ConcurrentHashMap<>();

/** * 获取在线人数 */ public static Integer getOnlineNumber (){ return atomicInteger . get ();} public static ConcurrentHashMap < Long , RoomMessage > getRoomId_roomMessage () { return roomId_roomMessage ; } public static ConcurrentHashMap < Long , EventMessage > getEvent_eventMessage () { return event_eventMessage ; } /** * 创建一个连接 * @param session 当前用户 session */ @OnOpen public void onOpen ( Session session , @PathParam ( "userId" ) Long userId ){ if (userId != null ){ FlopWebSocketMessage flopWebSocketMessage = new FlopWebSocketMessage (); flopWebSocketMessage . setSession (session ); user_flopWebSocketMessage . put (userId ,flopWebSocketMessage );
session_userId.put(session,userId); atomicInteger . addAndGet ( 1 ); FlopWebSocketResultMessage flopWebsocketResultMessage = new FlopWebSocketResultMessage (); flopWebsocketResultMessage . setCode ( ResponseCode . OK ); flopWebsocketResultMessage . setMessage ( " 链接成功 " ); sendMessage (userId , JSON .toJSONString (flopWebsocketResultMessage )); System . out . println (userId + " 连接成功 " ); } else { // 断开连接 try { session . close (); System . out . println ( " 连接失败 " ); } catch ( IOException e ) { e . printStackTrace (); } } } /** * 收到客户端消息后调用的方法 * @param message 客户端发送过来的消息 */ @OnMessage public void onMessage ( String message , Session session ) throws IOException { Long userId = session_userId . get (session ); // 发送信息接收成功的信息 FlopWebSocketResultMessage flopWebsocketResultMessage = new FlopWebSocketResultMessage (); flopWebsocketResultMessage . setCode ( ResponseCode . OK ); flopWebsocketResultMessage . setMessage ( " 信息接收成功 " ); sendMessage (userId , JSON .toJSONString (flopWebsocketResultMessage )); }

/**
 * 给某一个用户集合发送消息
 * @param userIds 用户集合
 * @param message 需要发送的消息
 */
public static void sendMessage(Set<Long> userIds, String message){
    for (Long userId : userIds){
        FlopWebSocketMessage flopWebSocketMessage = user_flopWebSocketMessage.get(userId);
        if (flopWebSocketMessage != null){
            ReentrantReadWriteLock reentrantLock = flopWebSocketMessage.getLock();
            //给这个用户的对象加上读写锁（针对到用户的读写锁，这样多个用户是并发发送消息的，但是单个用户是串行发送消息的）
            reentrantLock.writeLock().lock();
            try {
                flopWebSocketMessage.getSession().getBasicRemote().sendText(message);
            } catch (IOException e) {
                e.printStackTrace();
                flopWebSocketMessage.getLock().writeLock().unlock();
            }
            flopWebSocketMessage.getLock().writeLock().unlock();
        }
    }
}

/** * 发生错误时调用 * @param error 错误信息 */ @OnError public void onError ( Throwable error ){ System . out . println ( " 连接错误 :" +error . getMessage ()); } /** * 连接关闭调用的方法 */ @OnClose public void onClose ( Session session ){ Long userId = session_userId . get (session ); // 查找该用户是否有进入房间 if (userId != null && user_flopWebSocketMessage . get (userId ) != null ){ ReentrantReadWriteLock reentrantLock = user_flopWebSocketMessage . get (userId ). getLock (); user_flopWebSocketMessage . remove (userId ); reentrantLock . writeLock (). unlock (); } session_userId . remove (session ); atomicInteger . addAndGet (- 1 ); System . out . println ( " 连接关闭 :" +userId ); } }

注意：

千万不要在除了onOpen方法外使用@PathParam("userId")从连接的链接中获取携带的参数，绝对回给你意想不到的惊喜