综合简单聊天系统

在上一篇博客中，我们写了一个简单的群聊系统，但是我们要想单独只对某一个客户端聊天，却不能实现。下面，我们综合之前写的，写一个综合的既能群聊，又能单聊的简单聊天系统。

代码示例

创建客户端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.OutputStream;import java.io.PrintWriter;import java.net.Socket;import java.net.UnknownHostException;import java.util.Calendar;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;import net.sf.json.JSONObject;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);            JOptionPane.showMessageDialog(this, "连接成功");            write(username);            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) {        String str = creatMessage();        write(str);        jtf.setText("");    }    /**     *      * @return 返回一个发送给单人的信息     */    public String creatMessage() {        JSONObject obj = new JSONObject();        String msg = jtf.getText();        jta.append(username + "说：" + msg + "\n");        obj.put("message", msg);        obj.put("type", "toSingle");        obj.put("from", username);        obj.put("to", "2");        Calendar cal = Calendar.getInstance();        long time = cal.getTimeInMillis();        obj.put("time", "" + time);        return obj.toString();    }    /**     * 将此消息发送给服务器     *      * @param msg 发送的消息     *            */    public void write(String msg) {        try {            pw = new PrintWriter(s.getOutputStream());            pw.println(msg);            pw.flush();        } catch (IOException e) {            e.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 String getUsername(){        return username;    }}

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

public class GetServerToClientMessage implements Runnable {    private Client client;    public GetServerToClientMessage(Client client) {        this.client = client;    }    @Override    public void run() {        client.read();    }}

创建服务器端Server

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 java.util.HashMap;import java.util.Iterator;import java.util.Set;import javax.swing.JButton;import javax.swing.JFrame;import javax.swing.JPanel;import javax.swing.JScrollPane;import javax.swing.JTextArea;import javax.swing.JTextField;import net.sf.json.JSONObject;public class Server extends JFrame implements ActionListener {    PrintWriter pw;    BufferedReader br;    JTextArea jta;    JTextField jtf;    JButton jb;    JPanel jp;    JScrollPane jsp;    HashMap<String, 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);                    sockets = new HashMap<>();                    while (true) {                        Socket socket = ss.accept();                        BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream()));                        String username = br.readLine();                        jta.append("客户端" + username + "已成功连接服务器\n");                        sockets.put(username, socket);                        Thread t = new Thread(new GetClientToServerMessage(Server.this, socket));                        t.start();                    }                } catch (IOException e) {                    e.printStackTrace();                }            }        }).start();    }    @Override    public void actionPerformed(ActionEvent e) {// 发送        try {            Set<String> set = sockets.keySet();            Iterator<String> it = set.iterator();            while (it.hasNext()) {                Socket socket = sockets.get(it.next());                pw = new PrintWriter(socket.getOutputStream(), true);                String str = jtf.getText();                jta.append("服务器说：" + str + "\n");                pw.println("服务器公告："+str);                pw.flush();            }            jtf.setText("");        } catch (IOException e1) {            e1.printStackTrace();        }    }    public void readMsg(String msg) {        JSONObject obj = JSONObject.fromObject(msg);        String type = obj.getString("type");        if (type.equals("toSingle")) {            String message = obj.getString("message");            String to = obj.getString("to");            String from = obj.getString("from");            String time = obj.getString("time");            Socket toSocket = sockets.get(to);            try {                PrintWriter pw = new PrintWriter(toSocket.getOutputStream());                pw.println(from + "说：" + message);                pw.flush();            } catch (IOException e) {                e.printStackTrace();            }        } else if (type.equals("toAll")) {            String message = obj.getString("message");            String from = obj.getString("from");            try {                Set<String> set = sockets.keySet();                Iterator<String> it = set.iterator();                while (it.hasNext()) {                    String otherName = it.next().toString();                    if (!otherName.equals(from)) {                        Socket socket = sockets.get(otherName);                        pw = new PrintWriter(socket.getOutputStream(), true);                        pw.println(from + "说：" + message);                        pw.flush();                    }                }            } catch (IOException e1) {                e1.printStackTrace();            }        }        jta.append(msg + "\n");    }}

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

import java.io.BufferedReader;import java.io.IOException;import java.io.InputStreamReader;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() {        try {                BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream()));                while (true) {                String msg = br.readLine();                server.readMsg(msg);            }        } catch (IOException e) {            e.printStackTrace();        }    }}

运行结果：

下面我们进行测试，登录客户端，用户名为1，将createMessage()方法中“type”改为“toAll”，让1实现群发。登录客户端，用户名为2，将createMessage()方法中“type”的值改为“toSingle”，“to”得值改为“3”，让2单独给3聊天。登录客户端，用户名为3，将createMessage()方法中“type”的值改为“toSingle”，“to”得值改为“2”，让3单独给2聊天。我们也可以将此实现做到客户端界面上去，这样就不用修改代码了！这里我就不做了！给大家做一个测试！