NIO 实现多人聊天室的案例
服务端
import java.io.IOException; import java.net.InetSocketAddress; import java.nio.ByteBuffer; import java.nio.channels.*; import java.nio.charset.Charset; /** * 聊天室服务端 */ public class NServer { private Selector selector = null; static final int PORT = 30000; private Charset charset = Charset.forName("UTF-8"); ServerSocketChannel server = null; public void init() throws IOException { selector = Selector.open(); server = ServerSocketChannel.open(); InetSocketAddress isa = new InetSocketAddress("127.0.0.1", PORT); // 将该 ServerSocketChannel 绑定到指定 IP 地址 server.bind(isa); //设置为以非阻塞方式工作 server.configureBlocking(false); // 将 server 注册到指定的 Selector server.register(selector, SelectionKey.OP_ACCEPT); while (selector.select() > 0) { for (SelectionKey sk : selector.selectedKeys()) { // 从 selector 上的已选择 Key 集中删除正在处理的 SelectionKey selector.selectedKeys().remove(sk); // 如果 sk 对应 channel 包含客户端的连接请求 if (sk.isAcceptable()) { // 接受请求 SocketChannel sc = server.accept(); // 采用非阻塞模式 sc.configureBlocking(false); // 将该 SocketChannel 也注册到 selector sc.register(selector, SelectionKey.OP_READ); // 将 sk 对应的 channel 设置成准备接受其他请求 sk.interestOps(SelectionKey.OP_ACCEPT); } // 如果 sk 对应的channel 有数据需要读取 if (sk.isReadable()) { SocketChannel sc = (SocketChannel) sk.channel(); ByteBuffer buffer = ByteBuffer.allocate(1024); String content = ""; try { // 读取数据操作 while (sc.read(buffer) > 0) { buffer.flip(); content += charset.decode(buffer); } System.out.println("读取的数据： " + content); sk.interestOps(SelectionKey.OP_READ); } catch (IOException e) { sk.cancel(); if (sk.channel() != null) { sk.channel().close(); } } // 如果 content 的长度大于 0，即聊天信息不为空 if (content.length() > 0) { // 遍历该 selector 里注册的所有 SelectionKey for (SelectionKey key : selector.keys()) { // 获取 channel Channel targetChannel = key.channel(); // 如果该 channel 是 SocketChannel if (targetChannel instanceof SocketChannel) { // 将读到的内容写到该 channel 中 SocketChannel dest = (SocketChannel) targetChannel; dest.write(charset.encode(content)); } } } } } } } public static void main(String[] args) throws IOException { new NServer().init(); } }
客户端