JAVA提供了一个API来使用NIO架构编写TCP服务器,如下所示。
import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.net.SocketAddress;
import java.nio.Buffer;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.text.ParseException;
import java.util.*;
public class NIOServer implements Runnable{
private InetAddress addr;
private int port;
private Selector selector;
public NIOServer(InetAddress addr, int port) throws IOException {
this.addr = addr;
this.port = port;
}
public void run(){
try {
startServer();
}catch(IOException ex){
System.out.println(ex.getMessage());
}
}
private void startServer() throws IOException {
this.selector = Selector.open();
ServerSocketChannel serverChannel = serverSocketChannel.open();
serverChannel.configureBlocking(false);
InetSocketAddress listenAddr = new InetSocketAddress(this.addr, this.port);
serverChannel.socket().bind(listenAddr);
serverChannel.register(this.selector, SelectionKey.OP_ACCEPT);
while (true) {
this.selector.select();
Iterator keys = this.selector.selectedKeys().iterator();
while (keys.hasNext()) {
SelectionKey key = (SelectionKey) keys.next();
keys.remove();
if (! key.isValid()) {
continue;
}
if (key.isAcceptable()) {
this.accept(key);
}
else if (key.isReadable()) {
this.read(key);
}
else if (key.isWritable()) {
this.write(key);
}
}
}
}
}这使用单个线程来处理诸如读、写和接受等事件。
与每个连接体系结构阻塞线程相比,这是非常可取的,因为它的非阻塞性质会导致最小的缓存丢失、线程开销、低cpu迁移。
然而,这个体系结构只使用一个线程。在多进程环境中(例如,4核心cpu),NIO体系结构浪费了其他核心。是否有一种设计方法可以用于使用所有具有NIO体系结构的核心?
NIO2 (基于前置器模式)就是这样的一种选择。但是底层的体系结构与原始的NIO非常不同。
发布于 2019-06-25 07:34:13
其基本思想是拆分任务
ExecuterService workers = Executors.newFixedThreadPool(50);
....
while (true) {
this.selector.select();
Iterator keys = this.selector.selectedKeys().iterator();
while (keys.hasNext()) {
SelectionKey key = (SelectionKey) keys.next();
keys.remove();
if (! key.isValid()) {
continue;
}
if (key.isAcceptable()) {
this.accept(key);
}
else if (key.isReadable()) {
workers.execute(new ReadTaskHandler(key));
}
else if (key.isWritable()) {
workers.execute(new WriteTaskHandler(key));
}
}
}class ReadTaskHandler implements Runnable {
SelectionKey key;
public ReadTaskHandler(SelectionKey key) {
this.key = key;
}
@Override
public void run() {
ByteBuffer buffer = ByteBuffer.allocate(1024);
SocketChannel channel = (SocketChannel) key.channel();
int size = 0;
try {
while ((size = channel.read(buffer)) > 0) {
System.out.println(new String(buffer.array()));
buffer.flip();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}事实上,关于NIO有很多模型。例如,我们还可以使用多个线程来处理接受任务(也称为多反应堆模型或多事件循环模型)。
顺便说一句,Netty是一个很好的事件驱动网络应用程序框架,打包了java NIO。
https://stackoverflow.com/questions/56747881
复制相似问题