UDP网络编程实现多线程聊天

实现两个人都可以发送和接收消息
TalkSender.java

package com.zhang.chat;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetSocketAddress;
import java.net.SocketException;

public class TalkSender implements Runnable {

    private int fromPort;

    private String toIp;

    private int toPort;

    private DatagramSocket socket = null;

    private BufferedReader reader = null;

    public TalkSender(int fromPort, int toPort, String toIp) {
        this.fromPort = fromPort;
        this.toPort = toPort;
        this.toIp = toIp;

        try {
            socket = new DatagramSocket(fromPort);
            reader = new BufferedReader(new InputStreamReader(System.in));
        } catch (SocketException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void run() {
        while (true) {
            try {
                //发送端输入要发送的数据
                String data = reader.readLine();
                //数据转成字节数组
                byte[] dataBytes = data.getBytes();
                //创建一个数据报包
                DatagramPacket packet = new DatagramPacket(dataBytes, 0, dataBytes.length, new InetSocketAddress(this.toIp, this.toPort));
                //发送
                socket.send(packet);
                if (data.equals("bye")) break;
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        socket.close();
    }
}

TalkReveiver.java

package com.zhang.chat;

import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.SocketException;

public class TalkReceiver implements Runnable {

    //接收数据的端口
    private int port;

    //发送人的名字
    private String msgFrom;

    DatagramSocket socket = null;

    public TalkReceiver(int port, String msgFrom) {
        this.port = port;
        this.msgFrom = msgFrom;

        try {
            socket = new DatagramSocket(port);
        } catch (SocketException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void run() {
        while (true) {
            try {
                byte[] bytes = new byte[1024];
                //接收的数据报包
                DatagramPacket packet = new DatagramPacket(bytes, 0, bytes.length);
                socket.receive(packet);

                byte[] data = packet.getData();
                String receiveData = new String(data, 0, data.length);
                System.out.println(msgFrom + ": " + receiveData);

                if (receiveData.trim().equals("bye")) break;
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        socket.close();
    }
}

Student.java

package com.zhang.chat;

public class Student {
    public static void main(String[] args) {
        new Thread(new TalkSender(7777,9999, "localhost")).start();
        new Thread(new TalkReceiver(8888, "老师")).start();
    }
}

Teacher.java

package com.zhang.chat;

public class Teacher {
    public static void main(String[] args) {
        new Thread(new TalkSender(5555,8888, "localhost")).start();
        new Thread(new TalkReceiver(9999, "老师")).start();
    }
}

参考资料：狂神说java的视频