1. 程式人生 > >四十三、NIO之Channel

四十三、NIO之Channel

Java NIO的通道類似流,但又有些不同:

  • 既可以從通道中讀取資料,又可以寫資料到通道。但流的讀寫通常是單向的。
  • 通道可以非同步地讀寫。
  • 通道中的資料總是要先讀到一個Buffer,或者總是要從一個Buffer中寫入。

正如上面所說,從通道讀取資料到緩衝區,從緩衝區寫入資料到通道。如下圖所示:

 

Channel的實現

這些是Java NIO中最重要的通道的實現:

  • FileChannel
  • DatagramChannel
  • SocketChannel
  • ServerSocketChannel

FileChannel 從檔案中讀寫資料。

DatagramChannel 能通過UDP讀寫網路中的資料。

SocketChannel 能通過TCP讀寫網路中的資料。

ServerSocketChannel可以監聽新進來的TCP連線,像Web伺服器那樣。對每一個新進來的連線都會建立一個SocketChannel。

基本的 Channel 示例

下面是一個使用FileChannel讀取資料到Buffer中的示例:

01 RandomAccessFile aFile = new RandomAccessFile("data/nio-data.txt""rw"
);
02 FileChannel inChannel = aFile.getChannel();
03  
04 ByteBuffer buf = ByteBuffer.allocate(48);
05  
06 int bytesRead = inChannel.read(buf);
07 while (bytesRead != -1) {
08  
09 System.out.println("Read " + bytesRead);
10 buf.flip();
11  
12 while(buf.hasRemaining()){
13 System.out.print((char) buf.get());
14 }
15  
16 buf.clear();
17 bytesRead = inChannel.read(buf);
18 }
19 aFile.close();

注意 buf.flip() 的呼叫,首先讀取資料到Buffer,然後反轉Buffer,接著再從Buffer中讀取資料。下一節會深入講解Buffer的更多細節。

二、SocketChannel

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

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

 

開啟 SocketChannel

下面是SocketChannel的開啟方式:

1 SocketChannel socketChannel = SocketChannel.open();
2 socketChannel.connect(new InetSocketAddress("http://jenkov.com"80));

關閉 SocketChannel

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

1 socketChannel.close();

從 SocketChannel 讀取資料

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

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

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

然後,呼叫SocketChannel.read()。該方法將資料從SocketChannel 讀到Buffer中。read()方法返回的int值表示讀了多少位元組進Buffer裡。如果返回的是-1,表示已經讀到了流的末尾(連線關閉了)。

寫入 SocketChannel

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

01 String newData = "New String to write to file..." + System.currentTimeMillis();
02  
03 ByteBuffer buf = ByteBuffer.allocate(48);
04 buf.clear();
05 buf.put(newData.getBytes());
06  
07 buf.flip();
08  
09 while(buf.hasRemaining()) {
10     channel.write(buf);
11 }

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

非阻塞模式

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

connect()

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

1 socketChannel.configureBlocking(false);
2 socketChannel.connect(new InetSocketAddress("http://jenkov.com"80));
3  
4 while(! socketChannel.finishConnect() ){
5     //wait, or do something else...
6 }

write()

非阻塞模式下,write()方法在尚未寫出任何內容時可能就返回了。所以需要在迴圈中呼叫write()。前面已經有例子了,這裡就不贅述了。

read()

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

非阻塞模式與選擇器

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