Java NIO SocketChannel

ShanXF發表於2019-01-19
Java

Java NIO中的SocketChannel是一個連線到TCP網路套接字的通道。可以通過以下2種方式建立SocketChannel：

1.開啟一個SocketChannel並連線到網際網路上的某臺伺服器。
2.一個新連線到達ServerSocketChannel時，會建立一個SocketChannel。

開啟 SocketChannel
下面是SocketChannel的開啟方式：

SocketChannel socketChannel = SocketChannel.open();
socketChannel.connect(new InetSocketAddress("http://jenkov.com", 80));
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
ServerSocket ss = serverSocketChannel.socket();
ss.bind(new InetSocketAddress("localhost", 9026));// 繫結地址

關閉 SocketChannel

當用完SocketChannel之後呼叫SocketChannel.close()關閉SocketChannel：

socketChannel.close();

從 SocketChannel 讀取資料

要從SocketChannel中讀取資料，呼叫一個read()的方法之一。以下是例子：

ByteBuffer buf = ByteBuffer.allocate(48);
int bytesRead = socketChannel.read(buf);

首先，分配一個Buffer。從SocketChannel讀取到的資料將會放到這個Buffer中。

然後，呼叫SocketChannel.read()。該方法將資料從SocketChannel 讀到Buffer中。

read()方法返回的int值表示讀了多少位元組進Buffer裡。

如果返回的是-1，表示已經讀到了流的末尾（連線關閉了）。

寫入 SocketChannel

寫資料到SocketChannel用的是SocketChannel.write()方法，該方法以一個Buffer作為引數。示例如下：

String newData = "New String to write to file..." + System.currentTimeMillis();

ByteBuffer buf = ByteBuffer.allocate(48);
buf.clear();
buf.put(newData.getBytes());

buf.flip();
while(buf.hasRemaining()) {
    channel.write(buf);
}

注意SocketChannel.write()方法的呼叫是在一個while迴圈中的。Write()方法無法保證能寫多少位元組到SocketChannel。所以，我們重複呼叫write()直到Buffer沒有要寫的位元組為止。

非阻塞模式

可以設定 SocketChannel 為非阻塞模式（non-blocking mode）.設定之後，就可以在非同步模式下呼叫connect(), read() 和write()了。

connect()

如果SocketChannel在非阻塞模式下，此時呼叫connect()，該方法可能在連線建立之前就返回了。為了確定連線是否建立，可以呼叫finishConnect()的方法。像這樣：

socketChannel.configureBlocking(false);//非阻塞
socketChannel.connect(new InetSocketAddress("http://jenkov.com", 80));

while(! socketChannel.finishConnect() ){
    //wait, or do something else...
}

write()
非阻塞模式下，write()方法在尚未寫出任何內容時可能就返回了。所以需要在迴圈中呼叫write()。

read()
非阻塞模式下,read()方法在尚未讀取到任何資料時可能就返回了。所以需要關注它的int返回值，它會告訴你讀取了多少位元組。

非阻塞模式與選擇器
非阻塞模式與選擇器搭配會工作的更好，通過將一或多個SocketChannel註冊到Selector，可以詢問選擇器哪個通道已經準備好了讀取，寫入等。

相關文章