Java UDP套接字编程（三）互相发送与接收消息

Java UDP套接字编程（三）互相发送与接收消息

发送消息线程：

package com.zzz.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 TalkSend implements Runnable {
    DatagramSocket socket = null;   //建立一个socket
    BufferedReader reader = null;   //准备从控制台读取 System.in

    private int port;
    private String toIP;
    private int toPort;

    public TalkSend(int fromPort, String toIP, int toPort) {
        this.port = fromPort;
        this.toIP = toIP;
        this.toPort = toPort;

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

    @Override
    public void run() {
        while (true) {
            try {
                String data = reader.readLine();
                byte[] sendData = data.getBytes();
                DatagramPacket packet = new DatagramPacket(sendData, 0, sendData.length, new InetSocketAddress(this.toIP, this.toPort));

                socket.send(packet);

                if(data.equals("bye")){
                    break;
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        socket.close();
    }
}

接收消息线程：

package com.zzz.chat;

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

public class TalkReceive implements Runnable{
    DatagramSocket socket = null;
    private int port;
    private String msgFrom;

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

    @Override
    public void run() {
        while (true) {
            try {
                byte[] container = new byte[1024];
                DatagramPacket packet = new DatagramPacket(container, 0, container.length);
                socket.receive(packet); //阻塞式接收包
                msgFrom = packet.getAddress().getHostName();

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

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

通信用户1：

package com.zzz.chat;

public class Person1 {
    public static void main(String[] args) {
        new Thread(new TalkSend(6666,"localhost",9999)).start();    //开启发送线程，指明自己的端口号和要发送到的端口号
        new Thread(new TalkReceive(7777)).start();      //开启接收线程，指明自己的接收端口号
    }
}

通信用户2：

package com.zzz.chat;

public class Person2 {
    public static void main(String[] args) {
        new Thread(new TalkSend(8888,"localhost",7777)).start();    //开启发送线程，指明自己的端口号和要发送到的端口号
        new Thread(new TalkReceive(9999)).start();      //开启接收线程，指明自己的接收端口号
    }
}