架构师讲解Java中websocket的应用(1)，详细学习指南

public static Collection getOnlineUser() {

List setUsers = new ArrayList();

Collection setUser = wsUserMap.values();

for (String u : setUser) {

setUsers.add(u);

}

return setUsers;

}

/**

• 移除连接池中的连接

• @param inbound

*/

public static boolean removeUser(WebSocket conn) {

if (wsUserMap.containsKey(conn)) {

wsUserMap.remove(conn); // 移除连接

return true;

} else {

return false;

}

}

/**

• 向特定的用户发送数据

• @param user

• @param message

*/

public static void sendMessageToUser(WebSocket conn, String message) {

if (null != conn && null != wsUserMap.get(conn)) {

conn.send(message);

}

}

/**

• 向所有的用户发送消息

• @param message

*/

public static void sendMessageToAll(String message) {

Set keySet = wsUserMap.keySet();

synchronized (keySet) {

for (WebSocket conn : keySet) {

String user = wsUserMap.get(conn);

if (user != null) {

conn.send(message);

}

}

}

}

}

3.接下来我们编写websocket的主程序类

该类用于管理[websocket]的生命周期。该类继承自WebSocketServer ，这是一个实现了runnable接口的类，他的构造函数需要传入一个端口，所以我们需要为websocket服务指定一个端口，该类有四个要重载的方法，[onOpen()]方法在连接创建成功以后调用，onClose在连接关闭以后调用，[onError方法]在连接发生错误的时候调用（一般连接出错以后触发了onError，也会紧接着触发onClose方法）。

onMessage方法在收到客户端发来消息的时候触发。我们可以在这个方法中处理客户端所传递过来的消息。

package com.xdx.websocket;

import java.net.InetSocketAddress;

import org.java_websocket.WebSocket;

import org.java_websocket.handshake.ClientHandshake;

import org.java_websocket.server.WebSocketServer;

public class WsServer extends WebSocketServer {

public WsServer(int port) {

super(new InetSocketAddress(port));

}

public WsServer(InetSocketAddress address) {

super(address);

}

@Override

public void onOpen(WebSocket conn, ClientHandshake handshake) {

// ws连接的时候触发的代码，onOpen中我们不做任何操作

}

@Override

public void onClose(WebSocket conn, int code, String reason, boolean remote) {

//断开连接时候触发代码

userLeave(conn); System.out.println(reason);