Java性能优化之使用NIO提高性能（Buffer和Channel）

在软件系统中，因为IO的速度要比内存慢，所以，I/O读写在不少场合都会成为系统的瓶颈。提高I/O速度，对提高系统总体性能有着很大的好处。

在Java的标准I/O中，提供了基于流的I/O实现，即InputStream和OutputStream。这种基于流的实现以字节为单位处理数据，而且很是容易创建各类过滤器。

NIO是New I/O的简称，具备如下特性：

1. 为全部的原始类型提供（Buffer）缓存支持；
2. 使用 java.nio.charset.Charset 做为字符集编码解码解决方案；
3. 增长通道（channel）对象，做为新的原始 I/O 抽象；
4. 支持锁和内存映射文件的文件访问接口；
5. 提供了基于 Selector 的异步网络 I/O。

与流式的 I/O 不一样，NIO是基于块（Block）的，它以块为基本单位处理数据。在NIO中，最为重要的两个组件是缓冲 Buffer 和通道 Channel 。缓冲是一块连续的内存块，是 NIO 读写数据的中转地。通道表示缓冲数据的源头或者目的地，它用于向缓冲读取或者写入数据，是访问缓冲的接口。

本文主要是介绍经过NIO中的Buffer和Channel，来提高系统性能。

1. NIO的Buffer类族和Channel

在NIO的实现中，Buffer是一个抽象类。JDK为每一种 Java 原生类型都建立了一个Buffer，如图

在NIO中和Buffer配合使用的还有 Channel 。Channel 是一个双向通道，便可读又可写。

下面列出Java NIO中最重要的集中Channel的实现：

• FileChannel
• DatagramChannel
• SocketChannel
• ServerSocketChannel

FileChannel用于文件的数据读写。 DatagramChannel用于UDP的数据读写。 SocketChannel用于TCP的数据读写。 ServerSocketChannel容许咱们监听TCP连接请求，每一个请求会建立会一个SocketChannel.

应用程序只能经过Buffer对Channel进行读写。好比，在读一个Channel的时候，须要先将数据读入到相应的Buffer，而后在Buffer中进行读取。

一个使用NIO进行文件复制的例子以下：

@Test
    public void test() throws IOException {
        //写文件通道
        FileOutputStream fileOutputStream = new FileOutputStream(new File(path_copy));
        FileChannel wChannel = fileOutputStream.getChannel();
        
        //读文件通道
        FileInputStream fileInputStream = new FileInputStream(new File(path));
        FileChannel rChannel = fileInputStream.getChannel();
        
        ByteBuffer byteBufferRead = ByteBuffer.allocate(1024);//从堆中分配缓冲区
        
        while(rChannel.read(byteBufferRead)!=-1){
            byteBufferRead.flip();//将Buffer从写状态切换到读状态
            while(byteBufferRead.hasRemaining()){
                wChannel.write(byteBufferRead);
            }
            byteBufferRead.clear();//为读入数据到Buffer作准备
        }
        wChannel.close();
        rChannel.close();
    }

2. Buffer的基本原理

buffer中有三个重要参数：位置（position）、容量（capacity）、上限（limit）。

• 位置（position）：当前缓冲区（Buffer）的位置，将从该位置日后读或写数据。
• 容量（capacity）：缓冲区的总容量上限。
• 上限（limit）：缓冲区的实际容量大小。

再回到上面的例子：

在建立ByteBuffer缓冲区实例后，位置（position）、容量（capacity）、上限（limit）均已初始化！position为0，capacity、limit均为最大长度值。rChannel 通道的 read() 方法会把文件数据写入ByteBuffer缓冲区，此时position的位置移动到下一个即将输入的位置，而limit，capacity不变。接着ByteBuffer缓冲区执行flip()操做，该操做会把会把limit移到position的位置，而且把position的位置重置为0。这样作是防止程序读到根本没有进行操做的区域。

接着 wChannel 通道的 write() 方法会读取ByteBuffer缓冲区的数据到文件，和 read() 操做同样，write() 操做也会设置position的位置到当前位置。为了便于下次读入数据到缓冲区，咱们调用clear()方法将position，capacity，limit初始化。

一、Buffer的建立

第一种从堆中建立

ByteBuffer byteBufferRead = ByteBuffer.allocate(1024);

从既有数组中建立

byte[] bytes = new byte[1024];
ByteBuffer byteBufferRead = ByteBuffer.wrap(bytes);

二、重置和清空缓冲区

Buffer提供了一些用于重置和清空 Buffer 状态的函数，以下：

public final Buffer rewind()
public final Buffer clear()
public final Buffer flip()

rewind() 方法将position置零，并清除标志位（mark）。做用是为提取Buffer的有效数据作准备：

out.write(buf); //从buffer读取数据写入channel
buf.rewind();//回滚buffer
buf.get(array);//将buffer的有效数据复制到数组中

clear()方法将position置零，同时将limit设置为capacity的大小，并清除了mark。为从新写Buffer作准备：

buf.clear();//为读入数据到Buffer作准备
ch.read(buf);

flip()方法先将limit设置到position的位置，而且把position的位置重置为零，并清除mark。一般用于读写转换。

三、标志缓冲区

标志（mark）缓冲区是一项在数据处理时比较有用的功能，它就像书签同样，能够在数据处理过程当中。随时记录当前位置，而后在任意时刻，回到这个位置，从而加快或简化数据处理流程。主要函数以下：

public final Buffer mark()
public final Buffer reset()

mark()方法用于记录当前位置，reset()方法用于回到当前位置。

四、复制缓冲区

复制缓冲区是指以原缓冲区为基础，生成一个彻底同样的新缓冲区。方法以下：

public abstract ByteBuffer duplicate()

简单来讲，复制生成的新的缓冲区与原缓冲区共享相同内存数据，每一方的数据改动都是相互可见的。可是，二者又维护了各自的position、limit和mark。这就大大增长了程序的灵活性，为多方处理数据提供了可能。

五、缓冲区分片

缓存区分片使用slice()方法实现，它将在现有的缓冲区中，建立新的子缓冲区，子缓冲区和父缓冲区共享数据。

public abstract ByteBuffer slice()

新缓冲区的内容将今后缓冲区的当前位置开始。此缓冲区内容的更改在新缓冲区中是可见的，反之亦然；这两个缓冲区的position、limit和mark是相互独立的。 新缓冲区的position位置将为零，其容量和limit将为此缓冲区中所剩余的字节数量，其mark标记是不肯定的。当且仅当此缓冲区为只读时，新缓冲区才是只读的。

六、只读缓冲区

可使用缓冲区对象的asReadOnlyBuffer()方法获得一个与当前缓冲区一致的，而且共享内存数据的只读缓冲区。只读缓冲区对于数据安全很是有用。若是不但愿数据被随意修改，返回一个只读缓冲区是颇有帮助的。

public abstract ByteBuffer asReadOnlyBuffer()

七、文件映射到内存

NIO提供了一种将文件映射到内存的方法进行I/O操做，它能够比常规的基于流的方式快不少。这个操做主要由FileChannel.map()方法实现。以下

MappedByteBuffer map = channel.map(FileChannel.MapMode.READ_WRITE, 0, 1024);

以上代码将文件的前1024个字节映射到内存中。返回MappedByteBuffe，它是Buffer的子类，所以，能够像使用ByteBuffer那样使用它。

八、处理结构化数据（散射、汇集）

NIO提供处理结构化数据的方法，称之为散射（Scattering）和汇集（Gathering）。

散射是指将数据读入一组数据中，而不只仅是一个。汇集与之相反。

在JDK中，经过GatheringByteChannel, ScatteringByteChannel接口提供相关操做。

下面我用一个示例来讲明汇集写于散射读。

示例功能：写入两段话到文件，而后读取打印。

@Test
    public void test() throws IOException {
        String path = "D:\\test.txt";
        //汇集写
        //这是一组数据
        ByteBuffer byteBuffer1 = ByteBuffer.wrap("Java是最好的工具".getBytes(Charset.forName("UTF-8")));
        ByteBuffer byteBuffer2 = ByteBuffer.wrap("像风同样".getBytes(Charset.forName("UTF-8")));
        //记录数据长度
        int length1 = byteBuffer1.limit();
        int length2 = byteBuffer2.limit();
        //用 ByteBuffer 数组存放ByteBuffer实例的引用。
        ByteBuffer[] byteBuffers = new ByteBuffer[]{byteBuffer1, byteBuffer2};
        //获取文件写通道
        FileOutputStream fileOutputStream = new FileOutputStream(new File(path));
        FileChannel channel = fileOutputStream.getChannel();
        //开始写
        channel.write(byteBuffers);
        channel.close();
        
        //散射读
        byteBuffer1 = ByteBuffer.allocate(length1);
        byteBuffer2 = ByteBuffer.allocate(length2);
        byteBuffers = new ByteBuffer[]{byteBuffer1,byteBuffer2};
        //获取文件读通道
        FileInputStream fileInputStream = new FileInputStream(new File(path));
        channel = fileInputStream.getChannel();
        //开始读
        channel.read(byteBuffers);
        //读取
        System.out.println(new String(byteBuffers[0].array(),"utf-8"));
        System.out.println(new String(byteBuffers[1].array(),"utf-8"));
    }

执行完成后，咱们打开test.txt文件，看到：Java是最好的工具像风同样

并在控制台打印出：

Java是最好的工具
像风同样

九、直接内存访问

NIO的 Buffer 还提供了一个能够直接访问系统物理内存的类----DirectByteBuffer。

DirectByteBuffer继承自ByteBuffer，但和普通Buffer不一样。普通的ByteBuffer仍然在JVM堆上分配空间，其最大内存受到最大堆的限制。而DirectByteBuffer直接分配在物理内存中，并不占用堆空间。并且，DirectByteBuffer是一种更加接近系统底层的方法，因此，它的速度比普通的ByteBuffer更快。

使用很简单，只须要把 ByteBuffer.allocate(1024) 换成 ByteBuffer.allocateDirect(1024) 便可。该方法的源码为

public static ByteBuffer allocateDirect(int capacity) {
        return new DirectByteBuffer(capacity);
    }

有必要说明的是，使用参数-XX:MaxDirectMemorySize=10M 能够指定DirectByteBuffer的大小最可能是 10M。

DirectByteBuffer的读写比普通Buffer快，但建立和销毁却比普通Buffer慢。但若是能将DirectByteBuffer进行复用，那么，在读写频繁的状况下，它能够大幅改善系统性能。

3.与传统I/O的对比

I/O和NIO的最大区别就是 传统I/O是面向（缓冲）流，NIO是面向缓冲区。

使用 Java 进行 I/O操做有两种基本方法：

1. 使用基于InputStream 和 OutputStream 的方式；（字节流）
2. 使用 Writer 和 Reader。（字符流）

不管使用哪一种方式进行文件 I/O，若是能合理地使用缓冲，就能有效的提升I/O的性能。

用传统I/O实现刚开始的文件复制例子，代码以下：

@Test
    public void test6() throws IOException {
        //缓冲输出流
        BufferedOutputStream bufferedOutputStream = new BufferedOutputStream(new FileOutputStream(new File(path_copy)));
        //缓冲输入流
        BufferedInputStream bufferedInputStream = new BufferedInputStream(new FileInputStream(new File(path)));
        byte[] bytes = new byte[1024];
        while (bufferedInputStream.read(bytes) != -1) {
            bufferedOutputStream.write(bytes);
        }
        bufferedInputStream.close();
        bufferedOutputStream.close();
    }

须要注意的是，虽然使用ByteBuffer读写文件比Stream快不少，但不足以代表二者存在很如此之大的差距。这其中，因为ByteBuffer是将文件一次性读入内存再作后续处理，而Stream方式是则是边读文件边处理数据（虽然使用了缓冲组件 BufferedInputStream），这也是致使二者性能差别的缘由之一。虽如此，仍不能掩盖使用NIO的优点。使用NIO替代传统I/O操做，对系统总体性能的优化，应该是有立竿见影的效果的。

---

附录

位："位(bit)"是电子计算机中最小的数据单位。每一位的状态只能是0或1。

字节：8个二进制位构成1个"字节(Byte)"，它是存储空间的基本计量单位。1个字节能够储存1个英文字母或者半个汉字，换句话说，1个汉字占据2个字节的存储空间。

以1KB的文件举例：

1Byte = 8Bit
1KB = 1024Byte

当咱们进行byte[] bytes = new byte[1024]操做时，至关于开辟了1KB的内存空间。

参考

Java程序性能优化 葛一鸣著