基于UDP协议实现多人聊天室

该聊天室分为三部分代码

第一部分:
服务端:聊天室,相当于一个转发容器,接收用户发送的过来的消息,然后将其转发给其他人。
服务端聊天室代码实现:

package com.company;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.SocketException;
import java.util.HashSet;

public class Chat_Room_UDP {
    public static void main(String[] args) throws IOException {
        // 基于UDP协议完成聊天室,能够完成接收消息,并且将消息转发给除了自己之外的人
        //第一步创建套接字对象
        DatagramSocket datagramSocket = new DatagramSocket(10000);
        // 创建一个集合HashSet集合用来存储端口号
        HashSet<Integer> set = new HashSet<>();
        // 创建两个byte字节数组用来发送和接受数据报包的使用
        byte[] rdata = new byte[1024 * 4];
        byte[] sdata = new byte[1024 * 4];
        // 创建两个数据报包来发送和接收消息
        DatagramPacket rPacket = new DatagramPacket(rdata, rdata.length);
        DatagramPacket sPacket = new DatagramPacket(sdata, sdata.length);
        System.out.println("*****聊天室*****");
        // 循环接收和转发消息
        while(true) {
            // 接收消息
            datagramSocket.receive(rPacket);
            // 判断该端口号是否存储在set集合中,如果没有就存储进去
            if(!set.contains(rPacket.getPort())) {
                set.add(rPacket.getPort());
            }
            // 使用String类的构造方法来将字符数组转换成字符串
            String msg = new String(rdata, 0, rPacket.getLength());
            System.out.println("聊天室: " + msg);
                // 循环转发消息
            for(int port: set) {
                // 转发给除了自己之外的其他人
                if(port != rPacket.getPort()) {
                    // 设置需要发送过去的IP地址
                    sPacket.setAddress(InetAddress.getLocalHost());
                    // 设置发送到的端口号
                    sPacket.setPort(port);
                    // 设置需要发送的消息,需要将其转换为字节数组再发送
                    sPacket.setData(msg.getBytes());
                    // 将数据报包发送出去
                    datagramSocket.send(sPacket);
                }
            }
        }
    }
}

第二部分:
客户端1:用户
客户端用户实现代码,该部分会发送消息和接受服务器转发过来其他人发送的消息

package com.company;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.*;
import java.util.Scanner;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class Client_UDP1 {
    public static void main(String[] args) throws SocketException {
        DatagramSocket datagramSocket = new DatagramSocket(10001);
        // 创建一个线程池来进行数据报包的接收和发送
        ThreadPoolExecutor tpe = new ThreadPoolExecutor(2, 4, 10,
                TimeUnit.SECONDS, new ArrayBlockingQueue<Runnable>(10));
        // 第一个线程用来发送消息
        tpe.execute(() ->{
            // 实例化数据报包对象
            byte[] sdata = new byte[1024 * 4];
            try {
                DatagramPacket sPacket = new DatagramPacket(sdata, sdata.length, InetAddress.getLocalHost(), 10000);
                Scanner scanner = new Scanner(System.in);
                while (true) {
                    System.out.println("请输入你要发送的内容 客户端1说: ");
                    String msg = scanner.nextLine();
                    // 设置发送的消息
                    sPacket.setData(msg.getBytes());
                    // 发送消息
                    datagramSocket.send(sPacket);
                }
            } catch (UnknownHostException  e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            }

        });
        // 这一个线程用来接收消息
        tpe.execute(() ->{
            // 实例化数据报包对象
            byte[] rdata = new byte[1024 * 4];
            DatagramPacket rpacket = new DatagramPacket(rdata, rdata.length);
            while(true){
                try {
                    // 接收数据报包
                    datagramSocket.receive(rpacket);
                    String msg = new String(rdata, 0, rpacket.getLength());
                    System.out.println("他说:" + msg);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        });
    }
}

第三部分:客户端2
这也是另一个用户与客户端1的功能相同,要想再多几个人可以再继续加客户端

package com.company;

import java.io.IOException;
import java.net.*;
import java.util.Scanner;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class Client_UDP2 {
    public static void main(String[] args) throws SocketException {
        // 创建实例化套接字对象
        DatagramSocket datagramSocket = new DatagramSocket(10002);
        // 创建一个线程池用来发送和接收数据报包
        ThreadPoolExecutor tpe = new ThreadPoolExecutor(2, 4, 10,
                TimeUnit.SECONDS, new ArrayBlockingQueue<Runnable>(10));
        // 第一个线程用来接收数据报包
        tpe.execute(() ->{
            // 实例化数据报包
            byte[] rdata = new byte[1024 * 4];
            DatagramPacket rPacket = new DatagramPacket(rdata, rdata.length);
            while (true){
                // 接收消息
                try {
                    datagramSocket.receive(rPacket);
                    // 获取接收到的消息,并将其转换为字符串
                    String msg = new String(rdata, 0 ,rPacket.getLength());
                    System.out.println("他说:" + msg);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        });
        tpe.execute(() ->{
            //实例化数据报包
            byte[] sdata = new byte[1024 * 4];
            try {
                DatagramPacket spacket = new DatagramPacket(sdata, sdata.length, InetAddress.getLocalHost(), 10000);
                Scanner scanner = new Scanner(System.in);
                while(true) {
                    System.out.println("请输入你要发送的内容 客户端2说: ");
                    String msg = scanner.nextLine();
                    // 设置需要发送的内容
                    spacket.setData(msg.getBytes());
                    // 发送数据报包
                    datagramSocket.send(spacket);
                }
            } catch (UnknownHostException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            }
        });
    }
}

最后实现的结果如下:
基于UDP协议实现多人聊天室_第1张图片
基于UDP协议实现多人聊天室_第2张图片
基于UDP协议实现多人聊天室_第3张图片

你可能感兴趣的:(基于UDP协议实现多人聊天室)