Java NIO SocketChannel

MRZYD 發(fā)布于2019-08-16 13:26 / 2677人閱讀

摘要：非阻塞模式下方法在尚未讀取到任何數(shù)據(jù)時(shí)可能就返回了。非阻塞模式與選擇器非阻塞模式與選擇器搭配會(huì)工作的更好，通過(guò)將一或多個(gè)注冊(cè)到，可以詢問(wèn)選擇器哪個(gè)通道已經(jīng)準(zhǔn)備好了讀取，寫入等。

Java NIO中的SocketChannel是一個(gè)連接到TCP網(wǎng)絡(luò)套接字的通道。可以通過(guò)以下2種方式創(chuàng)建SocketChannel：

1.打開(kāi)一個(gè)SocketChannel并連接到互聯(lián)網(wǎng)上的某臺(tái)服務(wù)器。
2.一個(gè)新連接到達(dá)ServerSocketChannel時(shí)，會(huì)創(chuàng)建一個(gè)SocketChannel。

打開(kāi) SocketChannel
下面是SocketChannel的打開(kāi)方式：

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));// 綁定地址

關(guān)閉 SocketChannel

當(dāng)用完SocketChannel之后調(diào)用SocketChannel.close()關(guān)閉SocketChannel：

socketChannel.close();

從 SocketChannel 讀取數(shù)據(jù)

要從SocketChannel中讀取數(shù)據(jù)，調(diào)用一個(gè)read()的方法之一。以下是例子：

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

首先，分配一個(gè)Buffer。從SocketChannel讀取到的數(shù)據(jù)將會(huì)放到這個(gè)Buffer中。

然后，調(diào)用SocketChannel.read()。該方法將數(shù)據(jù)從SocketChannel 讀到Buffer中。

read()方法返回的int值表示讀了多少字節(jié)進(jìn)Buffer里。

如果返回的是-1，表示已經(jīng)讀到了流的末尾（連接關(guān)閉了）。

寫入 SocketChannel

寫數(shù)據(jù)到SocketChannel用的是SocketChannel.write()方法，該方法以一個(gè)Buffer作為參數(shù)。示例如下：

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()方法的調(diào)用是在一個(gè)while循環(huán)中的。Write()方法無(wú)法保證能寫多少字節(jié)到SocketChannel。所以，我們重復(fù)調(diào)用write()直到Buffer沒(méi)有要寫的字節(jié)為止。

非阻塞模式

可以設(shè)置 SocketChannel 為非阻塞模式（non-blocking mode）.設(shè)置之后，就可以在異步模式下調(diào)用connect(), read() 和write()了。

connect()

如果SocketChannel在非阻塞模式下，此時(shí)調(diào)用connect()，該方法可能在連接建立之前就返回了。為了確定連接是否建立，可以調(diào)用finishConnect()的方法。像這樣：

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

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

write()
非阻塞模式下，write()方法在尚未寫出任何內(nèi)容時(shí)可能就返回了。所以需要在循環(huán)中調(diào)用write()。

read()
非阻塞模式下,read()方法在尚未讀取到任何數(shù)據(jù)時(shí)可能就返回了。所以需要關(guān)注它的int返回值，它會(huì)告訴你讀取了多少字節(jié)。

非阻塞模式與選擇器
非阻塞模式與選擇器搭配會(huì)工作的更好，通過(guò)將一或多個(gè)SocketChannel注冊(cè)到Selector，可以詢問(wèn)選擇器哪個(gè)通道已經(jīng)準(zhǔn)備好了讀取，寫入等。