优化上一篇文上的代码。
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.Iterator;
import java.util.Set;public class NonBlockingServer {public static void main(String[] args) throws IOException {// 创建 SelectorSelector selector = Selector.open();// 创建 ServerSocketChannel 并绑定端口ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();serverSocketChannel.bind(new InetSocketAddress(8080));serverSocketChannel.configureBlocking(false);// 将 ServerSocketChannel 注册到 Selector 上,并指定监听事件为接受连接事件serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);System.out.println("Server started, listening on port 8080...");while (true) {// 阻塞等待事件发生selector.select();// 获取发生事件的 SelectionKey 集合Set<SelectionKey> selectedKeys = selector.selectedKeys();Iterator<SelectionKey> iterator = selectedKeys.iterator();// 遍历处理事件while (iterator.hasNext()) {SelectionKey key = iterator.next();iterator.remove();// 创建新线程处理事件new Thread(new EventHandler(key)).start();}}}// 处理事件的线程类static class EventHandler implements Runnable {private SelectionKey key;public EventHandler(SelectionKey key) {this.key = key;}@Overridepublic void run() {try {// 处理接受连接事件if (key.isAcceptable()) {ServerSocketChannel serverChannel = (ServerSocketChannel) key.channel();SocketChannel clientChannel = serverChannel.accept();clientChannel.configureBlocking(false);clientChannel.register(key.selector(), SelectionKey.OP_READ);System.out.println("Client connected: " + clientChannel.getRemoteAddress());}// 处理读事件if (key.isReadable()) {SocketChannel clientChannel = (SocketChannel) key.channel();ByteBuffer buffer = ByteBuffer.allocate(1024);int bytesRead = clientChannel.read(buffer);if (bytesRead != -1) {buffer.flip();byte[] bytes = new byte[buffer.remaining()];buffer.get(bytes);String message = new String(bytes);System.out.println("Received message from client: " + message);// 原样返回给客户端clientChannel.write(ByteBuffer.wrap(bytes));} else {// 客户端关闭连接key.cancel();clientChannel.close();System.out.println("Client disconnected: " + clientChannel.getRemoteAddress());}}} catch (IOException e) {e.printStackTrace();}}}
}
在这个优化后的代码中,我们将处理事件的逻辑封装在 EventHandler
类中,并在遍历处理事件时创建新的线程来处理每个事件。这样可以实现并发处理不同连接的读写事件,提高系统的并发处理能力。