两个客户端通信的socket简单实现

两个客户端通信的socket简单实现

这段时间学习了socket，根据简单的服务器与客户端通信，做了一个两个客户端通过服务器来进行通信的小程序。

---

首先定义一个简单的消息类Message，该类只有两个属性，一个是接收消息的客户端的编号，一个是消息内容，代码如下：：

import java.io.Serializable;

public class Message implements Serializable {

    // 客户端编号
    int id;

    // 消息内容
    String msg;

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getMsg() {
        return msg;
    }

    public void setMsg(String msg) {
        this.msg = msg;
    }

}

接下来先创建服务器端，SocketNotifyServer类，该类创建了serverSocket对象，绑定本地的8888端口，声明静态变量sessionMap来保存所有已连接的客户端，并给每个客户端分配一个编号：

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;

public class SocketNotifyServer {
    // socket服务
    ServerSocket serverSocket = null;
    // 用来存放已连接的客户端的socket会话
    static Map<Integer, Socket> sessionMap = new HashMap<Integer, Socket>();

    public void socket() {
        try {
            // 创建serverSocket，绑定端口为8888
            serverSocket = new ServerSocket(8888);
            System.out.println("服务器开启。。。");
            // 客户端编号
            int i = 1;
            // 实现多个客户端连接
            while (true) {
                Socket socket = serverSocket.accept();
                System.out.println("客户端" + i + "连接成功。。。");
                if (socket != null) {
                    // 将socket放入map，key为客户端编号
                    sessionMap.put(i, socket);
                    // 开启线程处理本次会话
                    Thread thread = new Thread(new NotifyHandler(socket, sessionMap));
                    thread.setDaemon(true);
                    thread.start();
                    i++;
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws IOException {
        new SocketNotifyServer().socket();
    }
}

接下来是消息处理类，通过重写的构造方法，得到消息发送方以及所有的已连接客户端。
通过ObjectInputStream和ObjectOutputStream来对message对象进行读写：

import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.util.Map;

public class NotifyHandler extends Thread {
    Socket socket = null;
    InputStream in = null;
    Map<Integer, Socket> sessionMap = null;

    public NotifyHandler(Socket socket, Map<Integer, Socket> sessionMap) {
        this.socket = socket;
        this.sessionMap = sessionMap;
    }

    public void run() {
        try {
            in = socket.getInputStream();
            ObjectInputStream ois = new ObjectInputStream(in);
            // 实现一次连接多次通话
            while (true) {
                try {
                    Message msg = (Message) ois.readObject();
                    System.out.println("消息接受对象：客户端" + msg.getId() + "，消息内容：" + msg.getMsg());
                    // 发送数据
                    try {
                        Socket targetSocket = sessionMap.get(msg.getId());
                        OutputStream out = targetSocket.getOutputStream();
                        ObjectOutputStream oos = new ObjectOutputStream(out);
                        oos.writeObject(msg);
                        System.out.println("服务端已转发");
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                    if (socket.isClosed()) {
                        break;
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                    try {
                        socket.close();
                        break;
                    } catch (IOException e1) {
                        e1.printStackTrace();
                    }
                } catch (ClassNotFoundException e) {
                    e.printStackTrace();
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

最后是客户端的代码，由于没有界面以及其他原因，在客户端这里直接是默认[客户端1]和[客户端2]进行通信，所以代码中message对象的id属性是写死的，当[客户端1]启动的时候id属性的值为2，当[客户端2]启动时，id属性为1。此处把消息接收代码放在前面，是因为while (true){input = new InputStreamReader(System.in);这里在阻塞着等待输入，如果接收消息的代码放在这段代码下方，会导致在没有发过消息的时候，接收消息的线程不会启动，收不到别的客户端发来的消息：

import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.Socket;

public class MsgSocketClient {
    Socket socket = null;
    InputStreamReader input = null;
    InputStream in = null;
    OutputStream out = null;

    /** 
     * @param args 
     */
    public void socketStart() {
        try {
            socket = new Socket("127.0.0.1", 8888);
            System.out.println("客户端1启动.......");
            // 接受返回数据
            new Thread() {
                public void run() {
                    try {
                        while (true) {
                            in = socket.getInputStream();
                            ObjectInputStream ois = new ObjectInputStream(in);
                            Message msg = (Message) ois.readObject();
                            System.out.println("返回数据：" + msg.getMsg());
                        }
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            }.start();
            out = socket.getOutputStream();
            ObjectOutputStream oos = new ObjectOutputStream(out);
            while (true) {
                input = new InputStreamReader(System.in);
                String msg = new BufferedReader(input).readLine();
                Message message = new Message();
                message.setId(2);
                message.setMsg(msg);
                oos.writeObject(message);
                System.out.println("已发送");
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            // 关闭流和连接
            try {
                in.close();
                out.close();
                socket.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    public static void main(String[] args) {
        new MsgSocketClient().socketStart();
    }
}

运行结果如下图：

---

注意： in = socket.getInputStream(); ObjectInputStream ois = new ObjectInputStream(in);
这里不能直接写成 ObjectInputStream ois = (ObjectInputStream)socket.getInputStream();
直接强转会报异常。