Java NIO Buffer 分析

引言

JDK 1.4 之后引入了NIO(New IO或 Non Blocking IO),我觉的可以称其为New IO,因为NIO基本重写所有标准IO的API,完全可以替代标准的Java IO API。并且NIO支持面向缓冲区(Buffer)的、基于通道(Channel)的IO操作,可以以更加高效的方式进行文件的读写操作。

NIO的核心主要包括3部分

  • Buffer 缓冲区
  • Channel 通道
    • 文件通道
    • Socket通道
  • Selector 选择器

本篇文章会对NIOBuffer缓冲区的使用及实现原理做分析

体系

Java NIO 提供了以下几种Buffer类型可供使用

  1. ByteBuffer
  2. MappedByteBuffer
  3. CharBuffer
  4. DoubleBuffer
  5. FloatBuffer
  6. IntBuffer
  7. LongBuffer
  8. ShortBuffer

这些Buffer的名字已经非常清晰的指明了这些Buffer所各自承载的不同数据类型

下面的分析我都将以ByteBuffer为例子

基本使用

ByteBuffer的使用比较简单,但若不理解其内部实现原理,也非常容易搞混。我们先来介绍其简单的使用

创建缓冲区

创建缓冲区主要有两种常用方式

  • allocate 开辟指定大小的缓冲区
  • wrap 使用字节数组开辟新的缓冲区,对缓冲区的修改将导致数组被修改。新缓冲区的容量(position)和限制(limit)将为该数组的长度
// 通过 allocate 创建
ByteBuffer byteBuffer = ByteBuffer.allocate(1024);

// 通过 wrap 创建
byte [] buf = new byte[1024];
ByteBuffer.wrap(buf);

读写数据

使用ByteBuffer读写数据比较简单

写数据

往Buffer中写数据只需要调用put方法,该方法有多个重载,可以以不同的方式写数据
这里只列举3个

  • put(byte b) 在当前位置将给定字节写入此缓冲区
  • put(int i, byte b) 在指定位置将给定字节写入此缓冲区
  • put(byte[] src, int offset, int length) 将字节从给定源数组传输到此缓冲区。如果从数组复制的字节比数大于剩余的缓冲区容量,会抛出BufferOverflowException异常

读数据

从Buffer中读取数据只需要调用get方法,该方法同样有多个重载,可以以不同的方式读数据

  • get() 获取当前位置的单个字节
  • get(int i) 获取指定位置的单个字节
  • get(byte[] dst, int offset, int length) 从此缓冲区中获取字节并传输到给定的目标数组中,如果缓冲区中的剩余字节少于满足请求所需的字节数,会抛出BufferUnderflowException异常

下面是一个抛出BufferUnderflowException的例子

ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
String s =  "hello";
byteBuffer.put(s.getBytes());
byteBuffer.flip(); // 翻转 下文会提到
byte [] bufs = new byte[9]; // 缓冲区实际只有5个字节大小,大于此大小就会抛出异常
byteBuffer.get(bufs); // 抛出异常

flip 方法

上文抛出BufferUnderflowException的例子,不知道大家有没发现,我们的代码在读写转换时用了flip方法来进行读写操作的转换,缓冲区在进行读写转换的时候必须调用flip方法,那么为何要这么做呢?下面我将会对flip方法的内部实现原理进行分析,在分析之前,我们有必要先认识一下Buffercapacity position limit mark四个变量属性

  • capacity 表示缓冲区的容量,大小不可变且不可为负
  • position 下一个要读取或写入的元素的位置索引
  • limit 下一个不应该被读取或写入的元素的位置索引
  • mark 标记位,可以通过这个标记位返回此位置

四者的大小关系 mark <= position <= limit <= capacity

知道了这四个变量的含义之后,我们再来根据例子来看看一步一步往下分析吧

一. 首先,我们通过ByteBuffer.allocate(6)开辟一个大小为6的缓冲区

// 通过allocate 开辟缓冲区
public static ByteBuffer allocate(int capacity) {
    if (capacity < 0)
        throw new IllegalArgumentException();
    return new HeapByteBuffer(capacity, capacity);
}

// HeapByteBuffer 初始化
HeapByteBuffer(int cap, int lim) {            // package-private
      // 这里前四个参数依次是 mark position limit capacity
    super(-1, 0, lim, cap, new byte[cap], 0);
}

通过查看allocate方法的源码,可知其内部初始结构如下图(Buffer内部其实是数组实现,下标从0开始)

  • mark = -1
  • position = 0
  • limit = capacity = 6
Java NIO Buffer 分析_第1张图片
image.png

二. 然后我们现在通过put方法加入1个字节数据
此时的内部结构如下图

  • mark = -1
  • position = 1
  • limit = capacity = 6
Java NIO Buffer 分析_第2张图片
image.png

三. 加一个太少了,我们再添加3个字节数据至缓冲区中
此时的内部结构如下图

  • mark = -1
  • position = 4
  • limit = capacity = 6
Java NIO Buffer 分析_第3张图片
image.png

四. 现在数据已经全部加完了,我们准备读数据了,必须要调用flip方法
先来看看flip方法的内部实现

public final Buffer flip() {
    limit = position;
    position = 0;
    mark = -1;
    return this;
}

flip内部其实是将 position的值赋给了limit,并将position位置归0,如下图显示

Java NIO Buffer 分析_第4张图片
image.png

五. 成功翻转之后,就可以往外读数据了,假设往外读4个数据
此时内部结构如下图,并且 position = limit了,达到了最大可读字节数

  • mark = -1
  • position = 4
  • limit = 4
  • capacity = 6


    Java NIO Buffer 分析_第5张图片
    image.png

通过上面这个例子,对flip方法的翻转逻辑已经进行了非常详细的分析,并且可以得出结论capacity的是绝对不变的,不会随着读写操作而改变。mark变量的值会在下文reset mark 方法中用到

reset mark 方法

顾名思义,mark就是打标记的意思,reset表示重置的意思。这两个操做是紧密联系的,由mark()方法在当前position位置做标记,然后在需要重置到标记位置的时候,调用reset方法重置

看一下这两个方法的内部实现

public final Buffer mark() {
    mark = position;
    return this;
}
public final Buffer reset() {
    int m = mark;
    if (m < 0)
        throw new InvalidMarkException();
    position = m;
    return this;
}

方法的内部实现非常简单,通过对变量mark的编辑来标记当前位置,在需要重置的时候,把变量mark的值赋给当前下标索引position来达到重置的目的。

clear

clear方法比较简单,将position = 0limit = capacitymark = -1。有点类似初始化时的操作

compact

最后再来介绍一下compact方法,首先我们先来看一个场景

  while (in.read(buf) >= 0 || buf.position != 0) {
     buf.flip();
     out.write(buf);
     buf.clear();
 }

上面这个例子,在非阻塞模式下是会有问题的,因为write方法我们并不知道一次性可以写多少个字节,所以有可能还有未写入的字节数据,这时候我们却又重新读取了新的数据,就会导致覆盖了原有还未写入的数据。

只需要改成以下的方式即可,该方法的作用是将positionlimit之间的数据复制到buffer的开始位置,复制后position = limit -position,limit = capacity

  while (in.read(buf) >= 0 || buf.position != 0) {
     buf.flip();
     out.write(buf);
     buf.compact();
 }

总结

本编文章比较详细的分析了Java NIOBuffer的基本使用及内部实现原理。Buffer的内部实现其实比较简单,需要着重理解的其实就是capacity position limit mark这四个变量属性。

博客原文地址戳这里

你可能感兴趣的:(Java NIO Buffer 分析)