不会吧！做了这么久开发还有不会NIO的，看看BAT大佬是怎么用的吧

前言

• 在将NIO之前，我们必须要了解一下Java的IO部分知识。
• BIO（Blocking IO）
• 阻塞IO，在Java中主要就是通过ServerSocket.accept()实现的。
• NIO（Non-Blocking IO）
• 非阻塞IO，在Java主要是通过NIOSocketChannel + Seletor实现的。
• AIO（Asyc IO）
• 异步IO，目前不做学习。

BIO

简单实现服务器和客户端

package net.io;

import net.ByteUtil;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;

//NIO（NonBlocking IO）非阻塞IO
//通过一个事件监听器，吧这些客户端的连接保存起来，如果有时间发生再去处理，没时间发生不处理
public class Server {
    public Server(int port) {
        try {
            //创建服务器端，监听端口port
            ServerSocket serverSocket = new ServerSocket(port);
            //对客户端进行一个监听操作，如果有连接过来，就将连接返回（socket）-----阻塞方法
            while (true) {
                //监听，阻塞方法
                Socket socket = serverSocket.accept();
                //每个服务器和客户端的通信都是针对与socket进行操作
                System.out.println("客户端" + socket.getInetAddress());
                InputStream inputStream = socket.getInputStream();
                ObjectInputStream ois = new ObjectInputStream(inputStream);
                //获取客户端发送的message
                Object get = ois.readObject();
                System.out.println("接收到的消息为："  + get);

                //服务器需要给客户端进行一个回应
                OutputStream outputStream = socket.getOutputStream();
                ObjectOutputStream oos = new ObjectOutputStream(outputStream);
                String message = "客户端你好，我是服务器端";
                //我这里写了，不代表发送了，知识写到了输出流的缓冲区
                oos.writeObject(message);
                //发送并清空
                oos.flush();
            }

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        new Server(7000);
    }
}

package net.io;

import net.ByteUtil;

import java.io.*;
import java.net.Socket;

public class Client {
    public Client(int port){
        try {
            Socket socket = new Socket("localhost",port);

            //inputStream是输入流，从外面接收信息
            //outpurStream是输出流， 往外面输出信息
            OutputStream outputStream = socket.getOutputStream();
            ObjectOutputStream oos = new ObjectOutputStream(outputStream);
            //发送的信息
            String message = "服务器你好，我是客户端";
            //我这里写了，不代表发送了，知识写到了输出流的缓冲区
            oos.writeObject(message);
            //发送并清空
            oos.flush();

            //接收服务器的回应
            InputStream inputStream = socket.getInputStream();
            ObjectInputStream ois = new ObjectInputStream(inputStream);
            Object get = ois.readObject();
            System.out.println("接收的信息为：" + get);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        new Client(7000);
    }
}

针对于BIO，为什么是阻塞IO，是因为BIO是基于Socket完成数据的读写操作，Server调用accept()方法持续监听socket连接，所以就阻塞在accept()这里（后面的操作如果没有socket连接则无法执行），那这样就代表服务器只能同时处理一个socket的操作。

所以后序我们通过为socket连接创建一个线程执行，这样就可以增大服务器处理连接的数量了。

但是新的问题也就出来了，如果客户端的连接很多，那么就会导致服务器创建很多的线程

对socket进行处理，如果服务器端不断开连接的话，那么对应的线程也不会被销毁，这样大数量的线程的维护十分消耗资源。针对于这种情况设计出了Java的NIO。

NIO

首先我们需要介绍一下NIO。如果说BIO是面向socket进行读写操作的话，那么NIO则是面向channel进行读写操作（或者说面向buffer）。

这里我们在讲解NIO之前，需要先讲解一下这个buffer。众所周知这就是一个缓冲，并且我们知道socket具有输入流inputStream和输出流outputStream（读写分开的），但是我们的channel是同时具有read和write两个方法，而且两个方法都是基于buffer进行操作（这里就可以说明channel仅能比普通输入输出流好，相当于channel是一条双向，输入输出流是两条单向），所以我们可以知道buffer的重要性。

Buffer

诸如ByteBuffer，IntBuffer等都是Buffer的派生抽象类，需要调用抽象类的静态方法allocate(X capacity)方法进行一个初始化操作，该方法就是初始化buffer的大小,或者使用wrap(X x)方法，该方法相当于直接将信息存入缓冲中。至于存入buffer的put()方法和取出缓存的get()方法在下面代码中我就详细介绍（有底层知识，具有源码阅读能力的可以根据我的注释进行阅读），最关键的还有flip()方法，它是作为一个读写切换的作用，他使的缓存可又读又写，又使得读写相互隔离（需要注意的是使用buffer尽量是依次写完然后再一次读完，最后在调用clear()方法进行复位，不然会导致buffer容量越来越小，具体解释在下面代码）。

package net.nio.buffer;

import java.nio.IntBuffer;

public class TestBuffer {
    public static void main(String[] args) {

        /*
         *IntBuffer有四个重要参数
         * 1.mark 标记
         * 2.position 相当于当前下标、索引
         * 3.limit 代表缓冲的终点，读取不能超过该下标，当然也不能超过最大容量。（在调用flip时候会将当前下标position值赋值给limit，然后position置0）
         * 4.Capacity 最大容量，在初始化IntBuffer对象时候就定义好了，不能改变（IntBuffer.allocate(int capacity) ）
         *
         * ctrl+h 可以查看该类的子类
         */

        //intBuffer初始化
        IntBuffer intBuffer = IntBuffer.allocate(5);

        //放数据到缓冲区中
        intBuffer.put(10);
        intBuffer.put(11);
        intBuffer.put(12);
//        intBuffer.put(13);
//        intBuffer.put(14);

        /*
         *这里的读写反转的实现机制是：
         * 例如我们缓冲区容量为5，调用方法put()将数据写入缓冲区中，假如我们写入三个此时position为3，此时limit = capacity
         * 如果我们调用flip方法使得limit = 3 ,position = 0 ,mark我们现在先不管（下图源码已说明）
         *  public Buffer flip() {
         *      limit = position;
         *      position = 0;
         *      mark = -1;
         *      return this;
         * }
         *
         * 此时我们调用get()方法时候，取得下标是position的值，即从0下标读取。直到读取到position = limit = 3时候停止（不包括3）
         * 1.如果我们这个时候不调用flip()方法直接再次put()往缓冲区写入数据（即没从读状态切换到写状态），那么就会报错超过下标overflow
         * 2.如果我们调用一次flip()（即进入写状态）写入一个数据后，那么此时position = 0，limit = 3，此时我们最多存放3个数据（即下标0,1,2）
         * 如果我们不再次调用flip()切换状态那么就会导致，读取到错误数据，（即只存入了一个数据，但是却取出来了3个数据）
         *
         * 上述说明了一个问题，如果我们存取的数据越来越小，那么这个缓冲区逐渐缩小，导致并不能存取他的最大容量，可能会浪费内存，
         *（因为position是不能超过limit的，然而调用flip()方法后会使的limit = position（赋值操作），那么如果数据越来越少，
         * 就会导致缓冲区能使用的部分越来越小）
         *
         * 总结：缓冲区的大小设置应该根据实际使用进行设置（并且要及时调用clear() ），否则可能会导致缓冲区的内存浪费。
         */
        intBuffer.flip();
        //切换读写状态
        //判断缓存区是否还有剩余
        while (intBuffer.hasRemaining()) {
            System.out.println(intBuffer.get());
        }
    }
}

Channel

Channel是NIO实现的基础，对于NIO，Channel的地位相当于BIO的socket。

Channel具有非常多方法，其中使用最多的就是两个方法write(ByteBuffer buf)和read(ByteBuffer buf)方法。

（这里需要注意的是这个read和write是buffer作为主体的，即read()方法是channel往buffer里写数据，而write()方法是指buffer向channel写数据）

package net.nio.channel;

        import java.io.FileOutputStream;
        import java.nio.ByteBuffer;
        import java.nio.channels.FileChannel;

public class TestChannel {
    public static void main(String[] args) throws Exception{
        String abc = "我写入文件了";

        //写入的文件地址与文件名
        FileOutputStream fileOutputStream = new FileOutputStream("C:\\xxx\\xxx\\xxx\\test.txt");

        //从输出流中获取channel实例
        FileChannel channel = fileOutputStream.getChannel();

        //创建字节缓冲区
        ByteBuffer byteBuffer = ByteBuffer.allocate(1024);

        //将字符串转化成字节数组并放入缓冲区中，然后缓冲区转换读写状态，由写入状态变为读取状态
        byteBuffer.put(abc.getBytes());
        byteBuffer.flip();

        //将缓冲区数据写入到channel中（这里write代表从缓冲区写入，read代表从channel读取到缓冲区）
        channel.write(byteBuffer);
        //关闭通道和输出流
        channel.close();
        fileOutputStream.close();
    }
}

简单NIO实现

上面在介绍NIO时候讲过，NIO是需要一个Selector线程去监听那些客户端有实现发生，从而在进行处理，而不是BIO的一个线程维护一个socket。

下面针对于NIO我们先不引入Selector，就用BIO的方式实现一个客户端和服务器端。（相当于作为一个练手）

Server

package net.nio.socket;

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Arrays;

public class Server {
    public static void main(String[] args) throws Exception {
        //开启nio的服务器端，并且绑定8000端口进行监听
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        InetSocketAddress inetSocketAddress = new InetSocketAddress(8000);
        serverSocketChannel.bind(inetSocketAddress);

        //创建缓冲区数组
        ByteBuffer byteBuffer = ByteBuffer.allocate(40);

        //服务器端接收来自客户端的请求，创建客户端的socket的实例
        SocketChannel socketChannel = serverSocketChannel.accept();
        //将客户端发送的数据读取到buffer数组中
        socketChannel.read(byteBuffer);
        byte[] array = byteBuffer.array();
        String msg = new String(array);
        System.out.println("服务器收到信息 ： " + msg);

        //对buffer数组进行读写反转，由读状态到写状态
        byteBuffer.flip();

        //将数据回显到客户端去
        byteBuffer.put("ok".getBytes());

        //做完一套读写操作后，需要进行clear
        byteBuffer.clear();
        }
}

Client

package net.nio.serverclient;

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;

public class NIOClient {
    public static void main(String[] args) throws Exception {
        SocketChannel socketChannel = SocketChannel.open();
        InetSocketAddress inetSocketAddress = new InetSocketAddress("localhost",8000);
        socketChannel.configureBlocking(false);

        //如果客户端未连接上服务器
        if (!socketChannel.connect(inetSocketAddress)) {
            System.out.println("客户端连接不上服务器。。。。");

            //如果客户端没有完成连接
            while (!socketChannel.finishConnect()) {
                System.out.println("连接中。。。。");
            }
        }

        //进入到这里说明连接成功
        String message = "hello , Server!";
        ByteBuffer byteBuffer = ByteBuffer.wrap(message.getBytes());

        //将buffer中的数据写入socketChannel
        socketChannel.write(byteBuffer);
        System.out.println("发送完毕");
    }
}

NIO实现（基于Selector）

首先我们需要知道Selector是什么?

Selector是一个选择器，既然是一个选择器，那么肯定是先有选项再有选择，理解这个后就知道channel肯定有的就是rigister()方法（因为需要将自己注册到Selector中）。

既然选项有了，那么如何选择呢？

Selector是针对已注册的channel中对有事件（例如：服务器：接受，读写，客户端：读写，服务器是在服务器开始就将自己注册，客户端是连接成功后由服务器将其注册）发生的channel进行处理。

Selector注册的不是简单的channel，而是将channel和其监听事件封装成一个SelectionKey保存在Selector底层的Set集合中。

Selector的keys()和selectedKeys()两个方法需要注意：

keys()方法是返回已注册的所有selectionKey。

selectedKeys()方法是返回有事件发生的selectionKey。

上面就是Selector的简单工作流程，下面我将附上代码，因为有较详细的注释，所以除了重要知识点我不再多介绍。

Server

package net.nio.serverclient;

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Iterator;
import java.util.Set;

public class NIOServer {
    public static void main(String[] args) throws Exception {
        //开启ServerSocketChannel的监听
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();

        //绑定端口8000
        serverSocketChannel.bind(new InetSocketAddress(8000));

        //创建Selector对象
        Selector selector = Selector.open();

        //设置监听为非阻塞
        serverSocketChannel.configureBlocking(false);

        //将ServerSocketChannel注册到Selector中（注册事件为ACCEPT）
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

        //Selector监听ACCEPT时间
        while (true) {
            //未有事件发生（下面是等待），返回值为int,代表事件发生个数
            if (selector.select(1000) == 0) {
                System.out.println("服务器等待了1S，无事件发生。。。。");
                continue;
            }

            //有客户端请求过来，就获取到相关的selectionKeys集合
            Set<SelectionKey> selectionKeys = selector.selectedKeys();
            Iterator<SelectionKey> iterator = selectionKeys.iterator();
            while (iterator.hasNext()) {
                //获取到事件
                SelectionKey selectionKey = iterator.next();
                //移出读取过的事件
                iterator.remove();

                //根据对应事件对应处理
                if (selectionKey.isAcceptable()) {
                    //有新的客户端连接服务器
                    SocketChannel socketChannel = serverSocketChannel.accept();
                    //给客户端设置非阻塞
                    socketChannel.configureBlocking(false);
                    //设置该SocketChannel为读事件，并为它绑定一个Buffer
                    socketChannel.register(selector,SelectionKey.OP_READ,ByteBuffer.allocate(1024));
                }
                if (selectionKey.isReadable()) {
                    //通过Key反向获取到事件的channel
                    SocketChannel socketChannel = (SocketChannel) selectionKey.channel();
                    //获取到事件绑定的buffer
                    ByteBuffer byteBuffer = (ByteBuffer) selectionKey.attachment();
                    socketChannel.read(byteBuffer);
                    //重置缓冲
                    byteBuffer.clear();
                    String message = new String(byteBuffer.array());
                    System.out.println("接收到客户端信息为: "+ message);
                }
            }
        }
    }
}

Client

package net.nio.serverclient;

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;
import java.util.Scanner;

public class NIOClient {
    public static void main(String[] args) throws Exception {
        SocketChannel socketChannel = SocketChannel.open();
        InetSocketAddress inetSocketAddress = new InetSocketAddress("localhost",8000);
        socketChannel.configureBlocking(false);

        //如果客户端未连接上服务器
        if (!socketChannel.connect(inetSocketAddress)) {
            System.out.println("客户端连接不上服务器。。。。");

            //如果客户端没有完成连接
            while (!socketChannel.finishConnect()) {
                System.out.println("连接中。。。。");
            }
        }

        //进入到这里说明连接成功
        while(true) {
            Scanner scanner = new Scanner(System.in);
            String message = scanner.nextLine();
            ByteBuffer byteBuffer = ByteBuffer.wrap(message.getBytes());

            //将buffer中的数据写入socketChannel
            socketChannel.write(byteBuffer);
            System.out.println("发送完毕");
        }

    }
}

我们这里可以发现，只需要主函数中进行一个死循环，死循环中对selector注册的channel进行监听（select()方法），有事件发生则根据channel注册的监听事件对应进行处理。

这里需要注意的是需要将ServerSocketChannel和SocketChannel编程非阻塞（调用configureBlocking(false)），不然是无法注册到Selector中。

还有一件事需要注意：我们每次是通过iterator（迭代器）遍历发生时间的Set ，为了避免重复处理时间，我们在获取发生时间的selctionKey以后，就将其remove()。

最后

感谢你看到这里，看完有什么的不懂的可以在评论区问我，觉得文章对你有帮助的话记得给我点个赞，每天都会分享java相关技术文章或行业资讯，欢迎大家关注和转发文章！