UDP实现聊天

UDP实现聊天

虽然不完美。但基本实现通信

TalkSender.java

package com.example.network;

import lombok.SneakyThrows;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.SocketException;

public class TalkSender implements Runnable {
    DatagramSocket socket = null;
    BufferedReader bufferedReader;

    private int fromPort;
    private int toPort;

    public TalkSender(int fromPort, int toPort) {
        this.fromPort = fromPort;
        this.toPort = toPort;
        try {
            socket = new DatagramSocket(this.fromPort);
        } catch (SocketException e) {
            e.printStackTrace();
        }
        bufferedReader = new BufferedReader(new InputStreamReader(System.in));
    }

    @SneakyThrows
    @Override
    public void run() {
        while (true) {
            String msg = bufferedReader.readLine();
            byte[] msgByte = msg.getBytes();
            DatagramPacket packet = new DatagramPacket(msgByte, 0, msg.length(), InetAddress.getLocalHost(), toPort);
            socket.send(packet);
            if (msg.equals("bye")) {
                break;
            }
        }
        bufferedReader.close();
        socket.close();
    }
}

TalkReceiver.java

package com.example.network;

import lombok.SneakyThrows;

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

public class TalkReceiver implements Runnable {
    DatagramSocket socket = null;
    private int port;
    private String name;

    public TalkReceiver(int port, String name) {
        this.port = port;
        this.name = name;
        try {
            socket = new DatagramSocket(this.port);
        } catch (SocketException e) {
            e.printStackTrace();
        }
    }

    @SneakyThrows
    @Override
    public void run() {
        while (true) {
            byte[] msg = new byte[1024];
            DatagramPacket packet = new DatagramPacket(msg, 0, msg.length);
            socket.receive(packet);
            String msgStr = new String(packet.getData(), 0, packet.getLength());
            System.out.println(name + ":" + msgStr);
            if (msgStr.equals("bye")) {
                break;
            }
        }
        socket.close();
    }
}

TalkTeacher.java

package com.example.network;

public class TalkTeacher {
    public static void main(String[] args) {
        new Thread(new TalkReceiver(9090, "student")).start();
        new Thread(new TalkSender(8080, 7070), "teacher").start();
    }
}

TalkStudent.java

package com.example.network;

public class TalkStudent {
    public static void main(String[] args) {
        new Thread(new TalkReceiver(7070, "teacher")).start();
        new Thread(new TalkSender(7071, 9090), "student").start();
    }
}