欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-06-30 09:17:42
...

该聊天室分为三部分代码

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

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协议实现多人聊天室
基于UDP协议实现多人聊天室
基于UDP协议实现多人聊天室