4.Java 网络编程TCP NIO通讯例子

在上篇例子会存在问题，在Java BIO模型中，会存在阻塞，即使使用线程池也会创建过多的线程，导致吞吐量不高，而Java NIO可以设置成非阻塞，解决这个问题。

TCP服务端

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.logging.Logger;

/**
 * tcp NIO 服务端 例子（双向发送数据）
 * @author terry
 * @version 1.0
 * @date 2021/12/18 15:28
 */
public class TcpServer {

    static Logger logger = Logger.getGlobal();

    public static void main(String[] args) throws IOException {
        ServerSocketChannel socket = ServerSocketChannel.open();
        socket.bind(new InetSocketAddress(5002));
        // 非阻塞
        socket.configureBlocking(false);
        logger.info("服务开启...");

        // 获取客户端连接
        SocketChannel channel = null;
        while (channel == null) {
            channel = socket.accept();
        }
        channel.configureBlocking(false);
        logger.info("新的客户端连接");

        // 读取客户端数据
        ByteBuffer buffer = ByteBuffer.allocate(5);
        while (buffer.hasRemaining()) {
            channel.read(buffer);
        }
        String str = new String(buffer.array());
        logger.info("接受客户端的数据：" + str);

        // 发送客户端数据
        ByteBuffer buffer2 = ByteBuffer.wrap("Hello".getBytes());
        while (buffer2.hasRemaining()) {
            channel.write(buffer2);
        }

        channel.close();
        socket.close();
    }
}

TCP客户端

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.charset.StandardCharsets;
import java.util.logging.Logger;

/**
 * tcp NIO 客户端 例子（双向发送数据）
 * @author terry
 * @version 1.0
 * @date 2021/12/18 15:28
 */
public class TcpClient {

    static Logger logger = Logger.getGlobal();

    public static void main(String[] args) throws IOException {
        // 连接服务端
        SocketChannel channel = SocketChannel.open();
        channel.configureBlocking(false);
        boolean isConnect = channel.connect(new InetSocketAddress("127.0.0.1", 5002));
        if (!isConnect) {
            // 如果连接不上一直重连
            while (!channel.finishConnect()){}
        }
        logger.info("连接成功...");

        // 发送数据给服务端
        ByteBuffer buffer = ByteBuffer.wrap("Hello Server !".getBytes());
        while (buffer.hasRemaining()) {
            channel.write(buffer);
        }

        // 读取服务端数据
        ByteBuffer buffer2 = ByteBuffer.allocate(5);
        while (buffer2.hasRemaining()) {
            channel.read(buffer2);
        }
        String str = new String(buffer2.array());
        logger.info("接受服务端数据：" + str);
    }
}

启动顺序 先启动TCP服务端 后启动TCP客户端

TCP 服务端打印：

十二月 18, 2021 3:55:50 下午 com.terry.demo1.TcpServer main
信息: 服务开启...
十二月 18, 2021 3:55:54 下午 com.terry.demo1.TcpServer main
信息: 新的客户端连接
十二月 18, 2021 3:55:54 下午 com.terry.demo1.TcpServer main
信息: 接受客户端的数据：Hello

TCP 客户端打印：

十二月 18, 2021 3:55:54 下午 com.terry.demo1.TcpClient main
信息: 连接成功...
十二月 18, 2021 3:55:54 下午 com.terry.demo1.TcpClient main
信息: 接受服务端数据：Hello