Java网络编程四 非阻塞通信UDP

1. 服务端

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.DataInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.DatagramSocket;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.DatagramChannel;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.util.Set;

public class NioUdpServer {

    public static void main(String[] args) {
        try {
            Selector selector = Selector.open();
            DatagramChannel channel = DatagramChannel.open();
            channel.configureBlocking(false);
            DatagramSocket socket = channel.socket();
            socket.bind(new InetSocketAddress(1000));
            channel.register(selector, SelectionKey.OP_READ);
           
            ByteBuffer receiveBuffer = ByteBuffer.allocate(1024);
            final int PACKAGE_SIZE = 10;
            while(true){
                int n = selector.select();
                if(n == 0){
                    continue;
                }
                Set<SelectionKey> readyKeys = selector.selectedKeys();
                for(SelectionKey key : readyKeys){
                    readyKeys.remove(key);
                   
                    if(key.isReadable()){
                        DatagramChannel dc = (DatagramChannel)key.channel();
                        InetSocketAddress client = (InetSocketAddress)dc.receive(receiveBuffer); //接收来自任意一个Client的数据报
                        key.interestOps(SelectionKey.OP_READ);
                       
                        System.out.println("client ----> IP: " + client.getAddress().getHostAddress() + ", port: " + client.getPort());
                        System.out.println("receiveBuffer.position() = " + receiveBuffer.position());
                        if(receiveBuffer.position() >= PACKAGE_SIZE){
                            receiveBuffer.flip();
                            DataInputStream dis = new DataInputStream(new ByteArrayInputStream(receiveBuffer.array()));
                            System.out.println(dis.readInt());
                            BufferedReader d = new BufferedReader(new InputStreamReader(dis));
                            System.out.println(d.readLine());
                           
                            receiveBuffer.clear();
                        }else{
                            dc.register(selector, SelectionKey.OP_READ);
                        }
                    }//if
                }
            }//while
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

}

2. 客户端可以使用《Java网络编程二 UDP通信》中的客户端连接测试