一对多聊天系统

一对多聊天系统

在上一篇博客中，我们写了一个基于TCP的客户端和服务器的一对一双向聊天系统，但只能开一个客户端与服务器通信，我们在之前的代码做一些改动，就实现了多个客户端与服务器通信的系统，且服务器能向所有客户端发送广播！

代码示例

创建服务器

import java.awt.BorderLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;

import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;

public class Server extends JFrame implements ActionListener {

    PrintWriter pw;
    BufferedReader br;
    JTextArea jta;
    JTextField jtf;
    JButton jb;
    JPanel jp;
    JScrollPane jsp;
    ArrayList<Socket> sockets;
    ServerSocket ss;

    public static void main(String[] args) {
        new Server();
    }

    public Server() {
        jta = new JTextArea();
        jsp = new JScrollPane(jta);
        jtf = new JTextField(15);
        jb = new JButton("发送");
        jb.addActionListener(this);
        jp = new JPanel();
        jp.add(jtf);
        jp.add(jb);
        this.add(jsp, BorderLayout.CENTER);
        this.add(jp, BorderLayout.SOUTH);
        this.setSize(300, 200);
        this.setTitle("服务器");
        this.setVisible(true);

        new Thread(new Runnable() {

            @Override
            public void run() {
                try {
                    ss = new ServerSocket(10000);
                    //存放客户端连接来的Socket
                    sockets = new ArrayList<>();
                    while (true) {
                        Socket socket = ss.accept();
                        sockets.add(socket);
                        jta.append("客户端已成功连接服务器\n");
                        Thread t = new Thread(new GetClientToServerMessage(Server.this, socket));
                        t.start();
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }).start();
    }

    @Override
    public void actionPerformed(ActionEvent e) {// 发送
        try {
            String str = jtf.getText();
            for (Socket socket : sockets) {
                pw = new PrintWriter(socket.getOutputStream(), true);
                pw.println(str);
                pw.flush();
            }
            jta.append("服务器广播：" + str + "\n");
            jtf.setText("");
        } catch (IOException e1) {
            e1.printStackTrace();
        }
    }

    public void read(Socket socket) {
        try {
            while (true) {
                br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
                jta.append(br.readLine() + "\n");
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

创建服务器接收客户端消息的线程类

import java.net.Socket;

public class GetClientToServerMessage implements Runnable {
    private Server server;
    private Socket socket;
    public GetClientToServerMessage(Server server,Socket socket) {
        this.server = server;
        this.socket = socket;
    }

    @Override
    public void run() { 
        server.read(socket);
    }
}

创建Client类

import java.awt.BorderLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.net.UnknownHostException;

import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;

public class Client extends JFrame implements ActionListener {
    Socket s;
    PrintWriter pw;
    BufferedReader br;
    JTextArea jta;
    JTextField jtf;
    JButton jb;
    JPanel jp;
    JScrollPane jsp;
    JOptionPane jop;
    String username;

    public static void main(String[] args) {
        new Client();
    }

    public Client() {
        jta = new JTextArea();
        jsp = new JScrollPane(jta);
        jtf = new JTextField(15);
        jb = new JButton("发送");
        jb.addActionListener(this);
        jp = new JPanel();
        jp.add(jtf);
        jp.add(jb);
        this.add(jsp, BorderLayout.CENTER);
        this.add(jp, BorderLayout.SOUTH);
        this.setSize(300, 200);
        username = JOptionPane.showInputDialog("请输入用户名"); 
        this.setTitle("客户端"+username);
        this.setVisible(true);


        try {
            s = new Socket("127.0.0.1", 10000);
            jta.append("您已成功连接服务器\n");
            Thread t = new Thread(new GetServerToClientMessage(Client.this));
            t.start();
        } catch (UnknownHostException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void actionPerformed(ActionEvent e) {
        try {
            pw = new PrintWriter(s.getOutputStream(), true);
            String str = jtf.getText();
            jta.append(username+"说：" + str + "\n");
            pw.println(username+"说："+str);
            pw.flush();
            jtf.setText("");
        } catch (IOException e1) {
            e1.printStackTrace();
        }
    }
    public void read(){
        try {
            while (true) {
                br = new BufferedReader(new InputStreamReader(s.getInputStream()));
                jta.append("服务器回复说：" + br.readLine()+ "\n");
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

创建接收服务器消息的线程类

public class GetServerToClientMessage implements Runnable {
    private Client client;

    public GetServerToClientMessage(Client client) {
        this.client = client;
    }

    @Override
    public void run() {
        client.read();
    }
}

运行结果