【java】【nio】【通信入门案例】

服务器端代码

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;

/**
 * nio 非阻塞式通信的入门案例
 */
public class Server {

    public static void main(String[] args) throws Exception {
        System.out.println("===================服务端启动================");
        //1.获取通道
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        //2.切换非阻塞模式
        serverSocketChannel.configureBlocking(false);
        //3.绑定连接端口
        serverSocketChannel.bind(new InetSocketAddress(9999));
        //4.获取选择器Selector
        Selector selector = Selector.open();
        //5.将通道都注册到选择器上，并且开始指定监听接收事件
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
        //6.使用Selector选择器轮询已经就绪好的事件
        while (selector.select() > 0){
            //7.获取选择器中所有注册的通道中已经就绪好的事件
            Iterator<SelectionKey> it = selector.selectedKeys().iterator();
            //8.开始遍历准备好的事件
            while (it.hasNext()){
                //9.提取当前事件
                SelectionKey key = it.next();
                //10.判断当前事件类型
                if(key.isAcceptable()){
                    //11.获取当前接入的客户端通道
                    SocketChannel socketChannel = serverSocketChannel.accept();
                    //12.将本客户端通道注册到选择器
                    socketChannel.configureBlocking(false);
                    //13.将本客户端通道注册到选择器
                    socketChannel.register(selector, SelectionKey.OP_READ);
                }else if(key.isReadable()){
                    //14.获取当前选择器上的读就绪事件
                    SocketChannel socketChannel = (SocketChannel) key.channel();
                    //15.读取数据
                    ByteBuffer buffer = ByteBuffer.allocate(1024);
                    int len = 0;
                    while ((len = socketChannel.read(buffer)) > 0 ){
                        buffer.flip();
                        System.out.println(new String(buffer.array(), 0, len));
                        buffer.clear();
                    }
                }
            }
            it.remove();
        }
    }
}

客户端代码

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

public class Client {

    public static void main(String[] args) throws IOException {
        //talk("李雷");
        talk("韩梅梅");
    }

    private static void talk( String userName ) throws IOException {
        //1.获取通道
        SocketChannel channel =
                SocketChannel.open(new InetSocketAddress("127.0.0.1",9999));
        //2.切换成非阻塞模式
        channel.configureBlocking(false);
        //3.分配指定缓冲区大小
        ByteBuffer buffer = ByteBuffer.allocate(1024);
        //4.发送数据给客户端
        Scanner scanner = new Scanner(System.in);
        while (true){
            System.out.println("请说");
            String msg = scanner.nextLine();
            buffer.put(( userName+ ": " + msg).getBytes());
            buffer.flip();
            channel.write(buffer);
            buffer.clear();
        }
    }
}