【Java学习】基于Socket的多用户聊天Demo

继 《Java网络通信基石Socket》 ，Socket理论的实战Demo


使用Java编写，基于Socket的多用户聊天Demo

---

一、设计思想

多人聊天，，需要有一个服务器和多个客户端

【服务器】时刻监听客户发送过来的消息，，并将消息发送到各个客户端。

服务器需要有一下几个模块（端口，主机名就不用说）
【1】记录客户端的集合（需要从服务器端群发消息）
【2】循环，阻塞监听，，serverSocket.accept()，有连接则，创建一个子线程，处理socket请求
【3】子线程内容，接收客户端Socket发送的内容，并群发到各个Socket

【客户端】
【1】一个主线程，，中启动两个子线程
【子线程1】发消息线程，用于监听控制台输入信息，发送信息到服务器，时刻监听。
【子线程2】时刻等待服务器回应，，并输出到控制台

【注意】

【服务器端】

• 不关闭Socket，保持长链接
• socket和输入输出流都不关闭，否则与客户端的连接会断开，无法进行群发。
• 输入输出流也无需关闭，，否则会关闭Socket，，时刻等待客户端请求即可。（虽然有点耗资源，不过博主还没有更好的解决方法，，读者有好方法请评论，交流学习）

【客户端】

• 线程中的输入流当客户端单向断开时会产生异常【SocketException】，所以需要捕获一下，同时该线程应该停止。
• 当服务器停止时，客户端的输入流也会抛出【SocketException】，需要捕获并关闭Socket（因为服务器都没了，不关闭也没用了）
• 客户端的主线程不能挂了，，挂了会连子线程一块销毁，，所以再main方法中需要加一个while，保证主线程不挂掉。
• 想到了再加…

---

二、参考代码如下

【服务器】SocketServer.java

package xatu.zsl.SocketDemo;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketException;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * Created by zsl on 2017/8/30.
 */
public class SocketServer {
    //用于保存客户端Socket
    public static List<Socket> clientSocketList = new ArrayList<Socket>();
    //服务器端口
    public static final int port = 12345;
    //服务器主机名
    public static final String address = "localhost";
    //服务器接收消息，以日志形式记录，动态
    private static Logger log = Logger.getLogger("server");

    public static void main(String[] args) throws IOException {
        //创建ServerSocket监听
        ServerSocket serverSocket = new ServerSocket(port);
        log.log(Level.INFO, "服务器开启监听，端口：" + port);
        while (true) {
            Socket client = serverSocket.accept();//阻塞监听
            new SocketDoWith(client).start();//创建线程对其进行操作
            clientSocketList.add(client);//将链接添加到，集合中，用以群发
        }
    }

    //服务器处理客户端Socket消息线程
    static class SocketDoWith extends Thread {
        Logger log = Logger.getLogger("server");
        private Socket socket = null;

        public SocketDoWith(Socket socket) {
            this.socket = socket;
        }

        public void run() {
            if (socket == null) return;
            try {
                String s = null;
                while (!socket.isClosed()) {
                    //将输入封装成对象流（处理起来更方便）
                    ObjectInputStream oi = new ObjectInputStream(socket.getInputStream());
                    s = (String) oi.readObject();
                    //将输入作为日志打印
                    log.log(Level.INFO, "服务器接收内容：" + s);

                    //把信息输出到，当前连接的所有客户端。
                    for (Socket client : clientSocketList) {
                        if (!client.isClosed()) {//防止发现送消息给，，断连客户端。
                            ObjectOutputStream oos = new ObjectOutputStream(client.getOutputStream());
                            oos.writeObject(s);
                            oos.flush();
                            log.log(Level.INFO, "服务器发送内容：" + s);//+ client.toString() + "  "
                        } else {//断开的Socket就移除
                            clientSocketList.remove(client);//移除
                        }
                    }
                }
            } catch (SocketException e) {
                log.log(Level.INFO, "客户端断开连接！！！");
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

---

【客户端】SocketClient.java（名叫小红）

package xatu.zsl.SocketDemo;

import java.io.*;
import java.net.Socket;
import java.net.SocketException;
import java.util.Scanner;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * Created by zsl on 2017/8/30.
 */
public class SocketClient {
    public static final int port = SocketServer.port;
    public static final String address = SocketServer.address;
    private static Logger log = Logger.getLogger("client");
    //为了方便使用直接定义为静态属性，，调用很方便。
    private static Socket client = null;
    //给客户端起个名字
    private static String ClientName = "小红";

    public static void main(String[] args) throws IOException, InterruptedException {
        client = new Socket(address, port);
        new SendThread().start();
        new ReceiveThread().start();
        while (!client.isClosed()) {
            Thread.sleep(1000);
        }
    }

    /**
     * 发送Socket的socket
     *
     * @param outStr 待发送信息
     * @param client 客户端Socket
     */
    private static void sendSocketServer(String outStr, Socket client) {
        try {
            ObjectOutputStream oos = new ObjectOutputStream(client.getOutputStream());
            oos.writeObject(outStr);
            oos.flush();//刷新，将流发出去
//            log.log(Level.INFO, "客户端，" + SocketClient.class.getName() + "发送了：" + outStr);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 发送信息线程
     */
    static class SendThread extends Thread {
        @Override
        public void run() {
            super.run();
            while (!client.isClosed()) {//插入停止条件
                Scanner in = new Scanner(System.in);//接收输入
                String inputStr = in.next();
                if (inputStr != null) {
                    sendSocketServer(ClientName + "：" + inputStr, client);//发送消息
                }
            }
        }
    }

    /**
     * 接受信息线程，，运行在后台，，等待输入信息
     */
    static class ReceiveThread extends Thread {
        @Override
        public void run() {
            super.run();
            try {
                InputStream serverInputStream = client.getInputStream();
                while (!client.isClosed() && serverInputStream != null) {//插入停止条件
                    ObjectInputStream ois = new ObjectInputStream(serverInputStream);
                    String inputStr = (String) ois.readObject();
                    //获取输出流，经过测试发现，输入流貌似是阻塞，，也就是没有输入时，
                    // 他就停在这里了，，一直等着输入，，所以无需加入Thread.sleep().
                    System.out.println(inputStr);
                }
            } catch (SocketException e) {
                log.log(Level.INFO, "服务器断开连接！！！");
                try {
                    client.close();//服务器断开连接此时需要关闭客户端连接
                } catch (IOException e1) {
                    e1.printStackTrace();
                }
            } catch (IOException e) {
                e.printStackTrace();
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            }
        }
    }
}

---

测试结果如下图：
【客户端：小红】



【客户端：小白】



【服务器】

---

**就到这里，，，，花了一个多小时搞出来的Demo，可能考虑的不全**