NIO(三、Channel)

目录

NIO(一、概述)
NIO(二、Buffer)
NIO(三、Channel)
NIO(四、Selector)

Channel

上文说了描述了Buffer的实现机制,那么这个章节就主要描述数据是如何进入缓冲区的,并且又是如何从缓冲区流出的。

类图纵览及核心类概述

Channel

  这张图只是简单概括了Channel的类图,当然,Channel的设计远比这个更复杂:例如SelectableChannel还有SocketChannel和ServerSocketChannel的实现,NetworkChannel继承Channel并抽象了更多的方法;例如FileChannel,除了继承AbstractInterruptibleChannel之外,还实现了GatheringByteChannel和ScatteringByteChannel接口。

  • Channel
      我们可以看到,Channel接口本身定义了 close() 和 isOpen() 方法,在继承Channel的接口中,又分别抽象了读通道(ReadableByteChannel)、写通道(WritableByteChannel)及可中断的异步通道(InterruptibleChannel)接口。读写通道自然不必说,下文也会有介绍。

  • InterruptibleChannel
      这里说下InterruptibleChannel,这是一个可以被中断的异步通道,继承了 close() 方法。当一个线程在I/O被阻塞时,另一个线程执行了close()方法,那么阻塞的线程会抛出 AsynchronousCloseException
    异常。当一个线程在I/O被阻塞时,另一个线程调用阻塞的线程中断(interrupt())方法,那么将会抛出ClosedByInterruptException异常。

  • AbstractInterruptibleChannel
      AbstractInterruptibleChannel抽象类,这是所有可中断通道实现的基类,我们可以看到,FileChannel正是直接继承自它,后文会介绍的SocketChannel和ServerSocketChannel继承自AbstractSelectableChannel抽象类,而AbstractSelectableChannel又继承自SelectableChannel抽象类,SelectableChannel上图就可以看出,同样继承AbstractInterruptibleChannel。我们不得不想一下,为什么FileChannel直接继承AbstractInterruptibleChannel抽象类,而另一个常用的比如SocketChannel却需要继承自SelectableChannel抽象类?此时我们需要只了解SelectableChannel是什么问题就迎刃而解了。

  • SelectableChannel
      第一个章节在概述中提过一个名词“多路复用”,当时也作了简单描述:一个线程通过选择器处理和管理多个通道,利用一个线程的资源去处理多个连接。SelectableChannel正是扮演着其中的一个重要角色。而类似SocketChannel一样实现NetworkChannel接口的通道,这种网络数据的传输尤其在高并发的压力下,让CPU利用率真正体现在处理数据上,而不是频繁上下文切换的开销,使用这种机制能明显提升处理性能,而FileChannel这种对文件操作是绝对不会使用到这种机制的。另外,SelectableChannel在阻塞模型中,每一个I/O操作都会阻塞到它完成为止,在非阻塞模型中,是不会出现阻塞的情况,同样返回的字节数可能少于要求或者干脆没有,是否阻塞我们能通过 isBlocking() 方法查看。
      SelectableChannel是通过选择器(Selector)复用的通道,为了配合与选择器一起使用,可使用 register() 方法,这个方法会返回一个SelectionKey对象,表示已经在选择器里注册了,这个通道会一直保持到注销为止,同时也包括已经分配在这个选择器上的这个通道的资源。一般情况下,通道自己不能直接在通道上注销,我们可以调用之前注册时返回的SelectionKey对象的 cancel() 方法,可以显式注销。另外,阅读代码我们发现,几乎所有的实现方法都有同步控制,所以,在多个并发线程下使用是安全的。

Scatter/Gather

  Channel,它既是缓冲区数据的入口,也是其数据的出口。
  在上面的类图我们可以看到,Channel分别抽象了ReadableByteChannel和WritableByteChannel接口,两个接口各自定义了 read() 和 write() 方法,读取和写入ByteBuffer对象。并在这个基础上又进一步定义了ScatteringByteChannel和GatheringByteChannel接口,同样,这两个接口同样存在继承自ReadableByteChannel和WritableByteChannel接口的read() 和 write() 方法,此外,还分别重载了 read() 和 write() 方法,增加了读取或写入ByteBuffer数组。
  

  • Scattering Reads
      读取数据从一个通道到多个缓冲区,这是简单示意图:

Scatter

  ByteBuffer   buffer1 = ByteBuffer.allocate(1024);
  ByteBuffer   buffer2 = ByteBuffer.allocate(1024);
  ByteBuffer[] buffers = {buffer1, buffer2};
  channel.read(buffers);
  • Gathering Reads
      写数据从多个缓冲区到一个通道,这是简单示意图:

Gather

  ByteBuffer   buffer1 = ByteBuffer.allocate(1024);
  ByteBuffer   buffer2 = ByteBuffer.allocate(1024);
  ByteBuffer[] buffers = {buffer1, buffer2};
  channel.write(buffers);

Channel之间数据传输

  FileChannel的 transferFrom() 和 transferTo() 两个方法实现了通道之间的数据传输,当有一方是FileChannel时,另一方实现了 ReadableByteChannel和WritableByteChannel接口的通道就能够与FileChannel传输数据。目前为止,只有文件通道(FileChannel)能够双向传输。

  • transferFrom() & transferTo()
FileChannel fromChannel = fromFile.getChannel();
FileChannel toChannel = toFile.getChannel();
// transferFrom
toChannel.transferFrom(fromChannel, position, count);
// transferTo
fromChannel.transferTo(position,count,toChannel);

  我们在讲Buffer的时候就已经说了position,意指读写位置,count指的是数据大小。

Channel重要实现

  • FileChannel:操作文件的读写
  • SocketChannel:通过TCP读写网络数据
  • ServerSocketChannel:监听TCP连接,你能利用它创建一个最简单的Web服务器
  • DatagramChannel:通过UDP读写网络数据

SocketChannel & ServerSocketChannel

  创建SocketChannel仅需要调用它的 open() 方法:

  SocketChannel channel = SocketChannel.open();

  事实上,可能在实际使用的时候,我们也会在的句柄对象的处理方法里,使用SelectionKey的 channel() 方法来获取,或者ServerSocketChannel对象的 accept() 方法来获取。

  SelectionKey key = ...
  //接受消息处理
  SocketChannel channel = ((ServerSocketChannel) key.channel()).accept();
  //或者
  SocketChannel channel = (SocketChannel) key.channel();

  在我们手动 open() 打开这个socket之后,其实还未连接,这时候我们对这个通道进行I/O操作会抛出NotYetConnectedException
。我们可以使用 connect() 方法来连接通道的socket,与此同时,也可以使用 isConnected()
 判断是否已连接。如果这个Socket在非阻塞模型中,就需要 finishConnect()
来确定连接已完成,也可以通过 isConnectionPending()
来测定该通道的连接是否正在进行中。

  // 设置非阻塞
  channel.configureBlocking(false);
  if(channel.isConnected()) {
	channel.connect(new InetSocketAddress(8080));
        while (channel.finishConnect()){
		    //connected -  do something
        }
  }

  当然,关于channel的读写操作,在讲Scatter/Gather时已经讲过,这里不再赘述。
  每次使用完成之后,都会调用 close() 方法关闭。需要说明的是,这里的 close() 方法支持异步关闭,当一个线程执行 close() 方法同时,另一个线程的在同一个通道上执行读操作时会被阻塞,然后这个读取操作不会返回任何数据,只会返回-1标识。当然,另一个线程如果是写操作的话也同样会被阻塞,不同的是会抛出 AsynchronousCloseException
异常。我们在上文中描述 InterruptibleChannel 这个可中断通道接口的时候也提到这个问题。

  channel.close();

  创建ServerSocketChannel与SocketChannel类似,不同的是ServerSocketChannel是监听套接字连接。所以在创建它的时候,需要将它绑定,如果没有绑定就执行 accept()
方法,那么会抛出 NotYetBoundException
  异常。

  ServerSocketChannel channel = ServerSocketChannel.open();
  channel.socket().bind(new InetSocketAddress(8080));

  监听到的连接可以使用 accept()
方法来返回该连接的 SocketChannel。执行 accept() 方法会一直阻塞直到有连接到达。

  SocketChannel channel = ((ServerSocketChannel) key.channel()).accept();

Simple Server

  下面是一段简单的ServerSocketChannel和SocketChannel应用,可以看到是如何使用这两个类的。当浏览器输入localhost:8080的时候,会在控制台打印出这个连接的请求信息。

   public static void main(String[] args) throws IOException {
		ServerSocketChannel channel = ServerSocketChannel.open();
		channel.socket().bind(new InetSocketAddress(8080));
		//设置非阻塞
		channel.configureBlocking(false);
		//注册
		Selector selector = Selector.open();
		channel.register(selector, SelectionKey.OP_ACCEPT);
		//处理器
		Handler handler = new Handler(1024);
		while (true) {
			//等待请求,每次等待阻塞5s,5s后线程继续向下执行,如果传入0或者不传参数将一直阻塞
			if (selector.select(5000) == 0) {
				continue;
			}
			//获取待处理的SelectionKey
			Iterator<SelectionKey> keyIterator = selector.selectedKeys().iterator();
			while (keyIterator.hasNext()) {
				SelectionKey key = keyIterator.next();
				//当接受到请求
				if (key.isAcceptable()) {
					handler.handleAccept(key);
				}
				try {
					//读数据
					if (key.isReadable()) {
						handler.handleRead(key);
					}
				} catch (IOException e) {
					keyIterator.remove();
					e.printStackTrace();
				}
				keyIterator.remove();
			}
		}
	}

	private static class Handler {
		private int    bufferSize   = 1024;
		private String localCharset = "UTF-8";

		public Handler() {
		}

		public Handler(int bufferSize) {
			this.bufferSize = bufferSize;
		}

		public Handler(String localCharset) {
			this.localCharset = localCharset;
		}

		public Handler(int bufferSize, String localCharset) {
			this.bufferSize = bufferSize;
			this.localCharset = localCharset;
		}

		public void handleAccept(SelectionKey key) {
			try {
				SocketChannel channel = ((ServerSocketChannel) key.channel()).accept();
				channel.configureBlocking(false);
				channel.register(key.selector(), SelectionKey.OP_READ, ByteBuffer.allocate(bufferSize));
			} catch (IOException e) {
				e.printStackTrace();
			}
		}

		public void handleRead(SelectionKey key) throws IOException {
			//获取channel
			SocketChannel channel = (SocketChannel) key.channel();
			//获取buffer 重置
			ByteBuffer buffer = (ByteBuffer) key.attachment();
			buffer.clear();
			if (channel.read(buffer) == -1) {
				channel.close();
			} else {
				//转为读状态
				buffer.flip();
				String receivedString = Charset.forName(localCharset)
						.newDecoder().decode(buffer).toString();

				System.out.printf("接受到客户端数据" + receivedString);

				//返回数据给客户端
				String sendString = "接受数据:" + receivedString;
				buffer = ByteBuffer.wrap(sendString.getBytes(localCharset));
				channel.write(buffer);

				channel.close();
			}
		}
	}
原文地址:https://www.cnblogs.com/sally-zhou/p/6583202.html