NIO-Buffer

目录

NIO-概览
NIO-Buffeer

什么是Buffer

Buffer是NIO用于存放特定基元类型数据的容器。缓冲区是特定基元类型的元素的线性有限序列。通过容量(capacity)、限制(limit)和位置(position)三个属性控制数据的写入大小和可读大小。

NIO-Buffer-LMLPHP

  1. 容量

    容量是它包含的元素数。 缓冲区在创建初始化容量之后容量就不会再更改。

  2. 偏移量

    偏移量是要读取或写入的下一个元素的索引。 偏移量不会大于其容量大小。

  3. 限制大小

    缓冲区的限制大小是最大可读或可写的索引位置,缓冲区限制大小不会大于其容量。

  4. 标志

    可以通过mark()方法打一个标志,通过reset()可以将偏移位置恢复到标志位置。

Buffer可以在写模式和读模式进行切换。在写模式写入数据后切换到读模式可以确保读取的数据不会超过写入数据的容量大小。

缓冲区类型

除了bool类型以外每个基元类型都会有缓冲区

缓冲区存储类型

缓冲区分为HeapBufferDirectBuffer

HeapBuffer是堆缓冲区,分配在堆上,有java虚拟机负责垃圾回收。
DirectBuffer是Java Native Interface(JNI,Java本地接口)在虚拟机外的内存中分配了一块缓冲区。这块缓冲区不直接有GC回收,在DirectBuffer包装类对象被回收时,会通过Java Reference机制来释放该内存块。即当引用了DirectBuffer对象被GC回收后,操作系统才会释放DirectBuffer空间。

当使用HeapBuffer时,如果我们要向硬盘读取数据时,硬盘的数据会先复制到操作系统内核空间,操作系统内核再复制到堆缓冲区中,最后我们在从堆缓冲区读取字节数据。
当使用DirectBuffer时,如果我们要向硬盘读取数据时,硬盘的数据会先复制到操作系统内核空间,我们直接从内核空间读取字节数据。

由于JVM堆中分配和释放内存比系统分配和释放内存更高效,因此DirectBuffer尽可能重用来提高性能。

字节存放顺序

大端模式(Big-Endian就是高位字节排放在内存的低地址端,低位字节排放在内存的高地址端)
小端模式:Little-Endian就是低位字节排放在内存的低地址端,高位字节排放在内存的高地址端。

在NIO中以BufferOrder来区分大端还是小端。

public final class ByteOrder {
    private String name;
    public static final ByteOrder BIG_ENDIAN = new ByteOrder("BIG_ENDIAN");
    public static final ByteOrder LITTLE_ENDIAN = new ByteOrder("LITTLE_ENDIAN");

    private ByteOrder(String var1) {
        this.name = var1;
    }

    public static ByteOrder nativeOrder() {
        return Bits.byteOrder();
    }

    public String toString() {
        return this.name;
    }
}

Buffer使用

接下来以ByteHeapBuffer为例,讲解如何使用Buffer。

Buffer

ByteBuffer

为了更清晰的说明缓冲区的功能,接下来以ByteBuffer举例。

  1. 申请缓冲区
    • allocate
    ByteBuffer byteBuffer = ByteBuffer.allocate(8);
    • allocateDirect
    ByteBuffer byteBuffer = ByteBuffer.allocateDirect(8);

    NIO-Buffer-LMLPHP

    DirectByteBuffer(int cap) {
        super(-1, 0, cap, cap);
        boolean pa = VM.isDirectMemoryPageAligned();  //是否页对齐
        int ps = Bits.pageSize();    //获取pageSize大小
        long size = Math.max(1L, (long) cap + (pa ? ps : 0));  //如果是页对齐的话,那么就加上一页的大小
        Bits.reserveMemory(size, cap);   //在系统中保存总分配内存(按页分配)的大小和实际内存的大小
    
        long base = 0;
        try {
            base = unsafe.allocateMemory(size);  //分配完堆外内存后就会返回分配的堆外内存基地址
        } catch (OutOfMemoryError x) {
            Bits.unreserveMemory(size, cap);
            throw x;
        }
        unsafe.setMemory(base, size, (byte) 0);   //初始化内存
        //计算地址
        if (pa && (base % ps != 0)) {
            address = base + ps - (base & (ps - 1));
        } else {
            address = base;
        }
        // 构建Cleaner对象用于跟踪DirectByteBuffer对象的垃圾回收,以实现当DirectByteBuffer被垃圾回收时,堆外内存也会被释放
        cleaner = Cleaner.create(this, new Deallocator(base, size, cap));
        att = null;
    }
  2. 写入数据

    byte[] data = new byte[] {'H','E','L','L','O'};
    byteBuffer.put(data);

    NIO-Buffer-LMLPHP

    • 堆缓冲区写入数据data

      public ByteBuffer put(byte[] src, int offset, int length) {
      //校验传入的参数是否合法
      checkBounds(offset, length, src.length);
      //在写入数据时首先会判断可写容量,大于容量则会抛出`BufferOverflowException`
      if (length > remaining())
          throw new BufferOverflowException();
      //将数据写入到指定的位置
      System.arraycopy(src, offset, hb, ix(position()), length);
      //更新偏移量
      position(position() + length);
      return this;
      }
    • 直接缓冲区写入数据

      public ByteBuffer put(byte[] src, int offset, int length) {
          //当写入长度大于JNI_COPY_FROM_ARRAY_THRESHOLD(6)时写入
          if ((length << $LG_BYTES_PER_VALUE$) > Bits.JNI_COPY_FROM_ARRAY_THRESHOLD) {
              checkBounds(offset, length, src.length);
              int pos = position();
              int lim = limit();
              assert (pos <= lim);
              int rem = (pos <= lim ? lim - pos : 0);
              if (length > rem)
                  throw new BufferOverflowException();
      
              Bits.copyFromArray(src, arrayBaseOffset, offset << $LG_BYTES_PER_VALUE$,
                                  ix(pos), length << $LG_BYTES_PER_VALUE$);
              position(pos + length);
          } else {
              //当长度小于6时,逐字节写入
              super.put(src, offset, length);
          }
      }
      //super.put(src, offset, length);
      public ByteBuffer put(byte[] var1, int var2, int var3) {
          checkBounds(var2, var3, var1.length);
          if (var3 > this.remaining()) {
              throw new BufferOverflowException();
          } else {
              int var4 = var2 + var3;
      
              for(int var5 = var2; var5 < var4; ++var5) {
                  this.put(var1[var5]);
              }
      
              return this;
          }
      }

      这里以6为界限的目的是什么?会有多少性能差异,哪位同学清楚的话麻烦告知一下。

  3. 转换为读模式

    byteBuffer.flip();

    NIO-Buffer-LMLPHP

    public final Buffer flip() {
        //当前可读位置指向,写入的位置
         this.limit = this.position;
         //读取开始位置置为0
         this.position = 0;
         this.mark = -1;
         return this;
     }
  4. 读取数据

     byte[] data1 = new byte[3];
     byteBuffer.get(data1);

    NIO-Buffer-LMLPHP

    • 堆缓冲区读取数据data
    public ByteBuffer get(byte[] dst, int offset, int length) {
        //检查传入参数
        checkBounds(offset, length, dst.length);
        //超过可读大小抛出BufferUnderflowException异常
        if (length > remaining())
            throw new BufferUnderflowException();
        //根据实际this.offset偏移后的位置读取数据
        System.arraycopy(hb, ix(position()), dst, offset, length);
        position(position() + length);
        return this;
    }
    • 直接缓冲区读取数据data
    public ByteBuffer get(byte[] dst, int offset, int length) {
        //当读取长度大于6时复制,小于6时逐字节复制
    
        if ((length << $LG_BYTES_PER_VALUE$) > Bits.JNI_COPY_TO_ARRAY_THRESHOLD) {
            checkBounds(offset, length, dst.length);
            int pos = position();
            int lim = limit();
            assert (pos <= lim);
            int rem = (pos <= lim ? lim - pos : 0);
            //超过可读大小抛出BufferUnderflowException异常
            if (length > rem)
                throw new BufferUnderflowException();
            Bits.copyToArray(ix(pos), dst, arrayBaseOffset,
                                offset << $LG_BYTES_PER_VALUE$,
                                length << $LG_BYTES_PER_VALUE$);
            position(pos + length);
        } else {
            super.get(dst, offset, length);
        }
        return this;
    }
  5. 缓冲区切片

     ByteBuffer sliceByteBuffer = byteBuffer.slice();

    NIO-Buffer-LMLPHP

  6. 初始化指针,清理已读取数据

     data.compact();

    此时将data初始化,会将未读取的2个字节复制到数组头部,同时转换为写模式。

    NIO-Buffer-LMLPHP

    public ByteBuffer compact() {
        //复制未读取的数据到初始位置
        System.arraycopy(this.hb, this.ix(this.position()), this.hb, this.ix(0), this.remaining());
        //设置当前偏移量为未读取的长度即5-3=2
        this.position(this.remaining());
        //设置限制大小为容量大小
        this.limit(this.capacity());
        //设置标记为-1
        this.discardMark();
        return this;
    }
  7. 初始化指针,清理所有数据

     data.clear();

    NIO-Buffer-LMLPHP

    完整代码

    public static void main(String[] args) {
    
         byte[] data = new byte[] {'H','E','L','L','O'};
         System.out.println(new String(data));
         ByteBuffer byteBuffer = ByteBuffer.allocate(8);
         byteBuffer.put(data);
         byteBuffer.flip();
         byte[] data1 = new byte[3];
         byteBuffer.get(data1);
         System.out.println(new String(data1));
         ByteBuffer sliceByteBuffer = byteBuffer.slice();
         byte[] data2 = new byte[2];
         sliceByteBuffer.get(data2);
         System.out.println(new String(data2));
         byteBuffer.compact();
         byteBuffer.clear();
     }

总结

NIO通过引入缓冲区的概念使得对字节操作比传统字节操作会方便一些,但是读写模式需要来回转换会让人有点头晕。

相关文献

  1. 解锁网络编程之NIO的前世今生
  2. 史上最强Java NIO入门:担心从入门到放弃的,请读这篇!
  3. Java NIO系列教程
  4. 深入理解DirectBuffer
  5. 《Java源码解析》NIO中的heap Buffer和direct Buffer区别
  6. Java Reference详解
  7. Direct Buffer vs. Heap Buffer
  8. JAVA之Buffer介绍
  9. 详解大端模式和小端模式
  10. 堆外内存 之 DirectByteBuffer 详解

12-07 11:26