基于Socket实现多人聊天室

当前支持：

1.仅文字

2.加入聊天室提醒

3.退出聊天室提醒

可能出现的BUG：

1.可能出现客户端发送信息后不能及时推送，需要下一个客户端发送信息后一起推送

服务端代码：

package com.example.demo.socket;

import org.springframework.util.ObjectUtils;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.*;

public class Server {
    public static void main(String[] args) {
        //信息队
        ConcurrentLinkedQueue str = new ConcurrentLinkedQueue<>();
        //连接对，暂未使用
        ConcurrentLinkedQueue<Map> socket = new ConcurrentLinkedQueue<>();
        //信息接收队
        ConcurrentLinkedQueue<Map> ins = new ConcurrentLinkedQueue<>();
        //信息发送队
        ConcurrentLinkedQueue<Map> outs = new ConcurrentLinkedQueue<>();
        //服务端端口号
        int port = 6666;
        try {
            ServerSocket serverSocket = new ServerSocket(port);
            //读取信息
            new Thread(new Runnable() {
                @Override
                public void run() {
                    // 参数 4 可以根据实际服务器cup设置 一般IO密集型设置为 CUP+1
                    ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(4);
                    while (true) {
                        if(!ObjectUtils.isEmpty(ins.peek())) {
                            Map in = ins.poll();
                            //延迟10毫秒执行，执行周期为  实际线程执行时间+10毫秒
                            ScheduledFuture<?> scheduledFuture = scheduledExecutorService.scheduleWithFixedDelay(new Runnable() {
                                @Override
                                public void run() {
                                    try {
                                        in.put("str", ((DataInputStream) in.get("in")).readUTF());
                                        str.add(in);
                                    } catch (IOException e) {
                                        ((ScheduledFuture)in.get("future")).cancel(true);
                                    }
                                }
                            }, 10, 10, TimeUnit.MICROSECONDS);
                            in.put("future",scheduledFuture);
                        }
                    }
                }
            }).start();
            //发送信息
            new Thread(new Runnable() {
                @Override
                public void run() {
                    while (true) {
                        if(!ObjectUtils.isEmpty(str.peek())) {
                            Map o = (Map) str.poll();
                            for(Map m : outs) {
                                if(!o.get("client").equals(m.get("client"))) {
                                    try {
                                        ((DataOutputStream)m.get("out")).writeUTF( o.get("client")+":"+o.get("str"));
                                    } catch (IOException e) {
                                        //异常下线大法，并发送群体下线通知
                                        System.out.printf("%s下线了！\n",m.get("client"));
                                        Map sv = new HashMap<>();
                                        sv.put("client","server");
                                        sv.put("str",m.get("client")+"退出聊天室！");
                                        str.add(sv);
                                        outs.remove(m);
                                    }
                                }
                            }
                        }
                    }
                }
            }).start();
            //循环获取客户端连接
            while (true) {
                Socket server = serverSocket.accept();
                Map m = new HashMap<>();
                m.put("socket",socket);
                m.put("in",new DataInputStream(server.getInputStream()));
                m.put("out",new DataOutputStream(server.getOutputStream()));

                //向客户端发送聊天室名，并获取客户端连接名
                ((DataOutputStream)m.get("out")).writeUTF("6666");
                m.put("client",((DataInputStream)m.get("in")).readUTF());
                System.out.printf("新人%s加入聊天室！\n",m.get("client"));

                ins.add(m);
                outs.add(m);
                socket.add(m);

                //发送群体上线通知
                Map sv = new HashMap<>();
                sv.put("client","server");
                sv.put("str",m.get("client")+"加入聊天室！");
                str.add(sv);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

客户端代码：

package com.example.demo.socket;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.Socket;
import java.util.Scanner;

public class Client {

    public static void main(String[] args) {
        //服务端 IP
        String serverName = "localhost";
        //服务端 端口号
        int port = 6666;
        try {
            Socket client = new Socket(serverName,port);
            DataInputStream in = new DataInputStream(client.getInputStream());
            DataOutputStream out = new DataOutputStream(client.getOutputStream());
            //向服务端发送连接名，并确认服务端收到连接名且返回聊天室名
            out.writeUTF("client");
            out.flush();
            System.out.printf("进入聊天室 %s \n",in.readUTF());
            try {
                new Thread(new Runnable() {
                    @Override
                    public void run() {
                        try {
                            while (true) {
                                System.out.println(in.readUTF());
                            }
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }).start();
                while (true) {
                    out.writeUTF(new Scanner(System.in).nextLine());
                    out.flush();
                }
            } finally {
                client.close();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }

    }
}