java nio socket 例子_Java NIO实现websocket聊天室实例

WebSocket协议

WebSocket是一种在单个TCP连接上进行全双工通信的协议。 WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。

WebSocket协议相比于Http协议来说，最大的特点就是可以实现服务端主动向客户端发送消息。在WebSocket出现之前，如果客户端想实时获取服务端的消息，就需要使用AJAX轮询，查询是否有消息，这样就很消耗服务器资源和带宽。但是用WebSocket就可以实现服务端主动向客户端发送数据，并且只需要占用一个TCP连接，节省了资源和带宽。

Java NIO 实现 websocket聊天室

接下来我们将利用Java NIO 来实现一个websocket实例，部分代码如下。

NIO的常规代码：

selector.select(1000);

Set selectionKeys = selector.selectedKeys();

Iterator it = selectionKeys.iterator();

while (it.hasNext()) {

SelectionKey key = it.next();

it.remove();

if (key.isAcceptable()) {

handleAccept(key);

}

if (key.isReadable()) {

handleRead(key);

}

}

接受连接：

public void handleAccept(SelectionKey key) {

ServerSocketChannel ssc = (ServerSocketChannel) key.channel();

SocketChannel sc;

try {

sc = ssc.accept();

sc.configureBlocking(false);

sc.register(selector, SelectionKey.OP_READ);

System.out.println(String.format("[server] -- client %s connected.", sc.getRemoteAddress().toString()));

} catch (IOException e) {

System.out.println(String.format("[server] -- error occur when accept: %s.", e.getMessage()));

key.cancel();<