UDP实现简单双向聊天

两个类

 * 消息发送
 * @author wcong
 * @version 1.0
 * @date 2020-09-11 14:55
 */
public class MsgSend implements Runnable {

    /**
     * 套接字
     */
    private DatagramSocket datagramSocket;

    /**
     * 发送到哪里
     */
    private String toHost;
    private int toPort;

    /**
     * 初始化工作
     * @param port 自己的端口
     * @param toHost 发送地点IP
     * @param toPort 发送地点端口
     * @throws SocketException
     */
    public MsgSend(int port,String toHost,int toPort) throws SocketException {
        datagramSocket = new DatagramSocket(port);
        this.toHost = toHost;
        this.toPort = toPort;
    }

    @Override
    public void run() {
        while (true){
            String str = new Scanner(System.in).next();
            if("bye".equals(str)){
                break;
            }
            DatagramPacket datagramPacket  = new DatagramPacket(str.getBytes(), str.getBytes().length,
                    new InetSocketAddress(toHost,toPort));
            try {
//        发送
                datagramSocket.send(datagramPacket);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        datagramSocket.close();
    }
}

/**
 * 消息接收
 * @author wcong
 * @version 1.0
 * @date 2020-09-11 15:03
 */
public class MsgReceive implements Runnable {

    private DatagramSocket datagramSocket;

    /**
     * 初始化
     * @param targetPort 接收消息的端口
     * @throws SocketException
     */
    public MsgReceive(int targetPort) throws SocketException {
        datagramSocket = new DatagramSocket(targetPort);
    }

    @Override
    public void run() {
        while (true){
            byte[] bytes = new byte[1024];
            DatagramPacket datagramPacket = new DatagramPacket(bytes,bytes.length);
            try {
                datagramSocket.receive(datagramPacket);
                System.out.println("对方说：" + new String(bytes));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

测试

public class MsgProviderTest {

    public static void main(String[] args) throws SocketException {
        new Thread(new MsgSend(10001,"127.0.0.1",8001)).start();
        new Thread(new MsgReceive(8002)).start();
    }

}

public class MsgConsumerTest {

    public static void main(String[] args) throws SocketException {
        new Thread(new MsgSend(10002,"127.0.0.1",8002)).start();
        new Thread(new MsgReceive(8001)).start();
    }

}