【Java NIO】之 Buffer

下面以一个实例介绍 NIO 的 Buffer

---

/**
 * 一、缓冲区（Buffer）：在 Java NIO 中负责数据的存取。缓冲区就是数组。用于存储不同数据类型的数据
 *
 * 根据数据类型不同（boolean 除外），提供了相应类型的缓冲区：
 * ByteBuffer
 * CharBuffer
 * ShortBuffer
 * IntBuffer
 * LongBuffer
 * FloatBuffer
 * DoubleBuffer
 *
 * 上述缓冲区的管理方式几乎一致，通过 allocate() 获取缓冲区
 *
 * 二、缓冲区存取数据的两个核心方法：
 * put() : 存入数据到缓冲区中
 * get() : 获取缓冲区中的数据
 *
 * 三、缓冲区中的四个核心属性：
 * capacity : 容量，表示缓冲区中最大存储数据的容量。一旦声明不能改变。
 * limit : 界限，表示缓冲区中可以操作数据的大小。（limit 后数据不能进行读写）
 * position : 位置，表示缓冲区中正在操作数据的位置。
 *
 * mark : 标记，表示记录当前 position 的位置。可以通过 reset() 恢复到 mark 的位置
 *
 * 0 <= mark <= position <= limit <= capacity
 *
 * 四、直接缓冲区与非直接缓冲区：
 * 非直接缓冲区：通过 allocate() 方法分配缓冲区，将缓冲区建立在 JVM 的内存中
 * 直接缓冲区：通过 allocateDirect() 方法分配直接缓冲区，将缓冲区建立在物理内存中。可以提高效率
 */
public class TestBuffer {

    @Test
    public void testBuffer() {
        // 分配指定大小的缓冲区
        ByteBuffer buffer = ByteBuffer.allocate(1024);
        System.out.println("----------------- allocate ---------------");
        System.out.println("position" + " : " + buffer.position());
        System.out.println("limit" + " : " + buffer.limit());
        System.out.println("capacity" + " : " + buffer.capacity());

        // 使用 put() 方法把数据存入缓冲区
        buffer.put("abcdefgh".getBytes());
        System.out.println("----------------- put ---------------");
        System.out.println("position" + " : " + buffer.position());
        System.out.println("limit" + " : " + buffer.limit());
        System.out.println("capacity" + " : " + buffer.capacity());

        // 切换到读模式
        buffer.flip();
        System.out.println("----------------- flip ---------------");
        System.out.println("position" + " : " + buffer.position());
        System.out.println("limit" + " : " + buffer.limit());
        System.out.println("capacity" + " : " + buffer.capacity());

        // 使用 get() 方法读取缓冲区数据
        byte[] bytes = new byte[buffer.limit()];
        buffer.get(bytes);
        System.out.println("----------------- get ---------------");
        System.out.println("data : " + new String(bytes));
        System.out.println("position" + " : " + buffer.position());
        System.out.println("limit" + " : " + buffer.limit());
        System.out.println("capacity" + " : " + buffer.capacity());

        // 调用 rewind() 方法实现重复读
        buffer.rewind();
        System.out.println("----------------- rewind ---------------");
        System.out.println("position" + " : " + buffer.position());
        System.out.println("limit" + " : " + buffer.limit());
        System.out.println("capacity" + " : " + buffer.capacity());

        // clear() : 清空缓冲区. 但是缓冲区中的数据依然存在，但是处于“被遗忘”状态
        buffer.clear();
        System.out.println("----------------- clear ---------------");
        System.out.println("position" + " : " + buffer.position());
        System.out.println("limit" + " : " + buffer.limit());
        System.out.println("capacity" + " : " + buffer.capacity());
        System.out.println((char)buffer.get());
    }

    @Test
    public void testDirectBuffer() {
        // 分配直接缓冲区
        ByteBuffer buffer = ByteBuffer.allocateDirect(1024);

        System.out.println(buffer.isDirect());
    }

    @Test
    public void testMark() {
        // 分配指定大小的缓冲区
        ByteBuffer buffer = ByteBuffer.allocate(1024);

        // 使用 put() 方法把数据存入缓冲区
        buffer.put("abcdefgh".getBytes());

        // 读取数据
        buffer.flip();
        byte[] bytes = new byte[buffer.limit()];
        buffer.get(bytes, 0, 2);
        System.out.println("data : " + new String(bytes, 0, 2));
        System.out.println("position" + " : " + buffer.position());

        // mark() : 标记一下当前 positon 的位置
        buffer.mark();

        // 继续读取数据
        buffer.get(bytes, 2, 2);
        System.out.println("data : " + new String(bytes, 2, 2));
        System.out.println("position" + " : " + buffer.position());

        // reset() : 恢复到 mark 位置
        buffer.reset();
        System.out.println("--------- after reset --------");
        System.out.println("position" + " : " + buffer.position());

        // 判断缓冲区是否有剩余可操作数据，即：position < limit
        if (buffer.hasRemaining()) {
            System.out.println(buffer.remaining());
        }
    }
}