Java语言Scoket编程进阶（一）（多线程、双向通信）

在《Java语言Scoket编程及实现原理浅析》一文中介绍了Java中Socket、ServerSocket的基本使用方法，本文将介绍一些Socket的进阶用法，实现一个多人的聊天室，技术要点主要包含以下两个方面：

一、Socket结合线程使用，实现服务端对客户端的一对多链接（TCP协议）

我们知道对于单个服务端的IP和端口是固定的，即我们在创建ServerSocket对象时传入的构造参数，那么为什么同一个Server启动时可以接受多个来自客户端的链接呢？

这里我们一定要正确认识Socket和ServerSocket的本质区别，前者是Java用来建立Socket链接的对象，而ServerSocket是通过阻塞、监听来产生Socket的像，如果加以抽象可以理解为实质上是两个Socket对象间的通信，而不是Socket和ServerSocket间的通信。这里我们还需要了解一点TCP协议的知识：TCP的链接由本机IP、本机端口、远程IP、远程端口四个元素确定，其中任一元素不同即可视为一条新的链接，在这样的条件下我们便可以将ServerSocket.accept()方法放入while的死循环中，不停的监听、创建多个Socket对象，然后我们将每个Socket对象放入独立的线程中进行通信。在通信过程中我们使用ConcurrentLinkedQueue做为消息队列，对多个客户端进行消息广播，实现消息的一对多分发。

 

二、Socket的双向通信（TCP协议）

这部分内容我并没有进行高并发下的测试，可作为您编程时的参考，我不能保证在实际生产中的可靠性。

Tcp协议本身是一种全双工协议，即同一个链接可以同时进行数据的读、写，基于这个前提我们将使用一个线程对ConcurrentLinkedQueue队列中的消息进行遍历，然后通过不同的Socket对象逐一发送。

 

综上所述我们实现聊天室功能的大体步骤如下：

1、启动线程遍历消息队列，通过不同的Socket向多个客户端发送信息;

2、将ServerSocket.accept()放入死循环中;

3、当有客户端接入时，将产生的Socket放入一个新的线程中进行通信;

4、将产生的Socket的对象放入集合中，以便后续通过遍历集合对多个客户端发送信息。

5、在每个独含有Socket的线程中读取客户端的传递过来的信息，并放入消息队列中。

 

服务端：

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ExecutorService;

import static java.util.concurrent.Executors.newFixedThreadPool;

/**
 * @author haye
 * @date 10/7/19
 * @description:
 */
public class Server {
    private final static int port=9999;
    private final static ConcurrentLinkedQueue<String> msgQueue=new ConcurrentLinkedQueue();
    private final static Collection<Socket> sockets= Collections.synchronizedCollection(new ArrayList<>());

    public static void main(String[] args) {
        try {
            ServerSocket socketServer=new ServerSocket(port);
            Socket socket;
            //用于便利消息队列中的消息，并向所有已链接的客户端进行分发
            new Thread(()->{
                while (true){
                    if(!msgQueue.isEmpty()){
                        String msg=msgQueue.poll();//取出队首的消息
                        sockets.forEach((client)->{ //遍历客户端，发送消息
                            if(null!=client&&client.isConnected()){
                                try {
                                    BufferedWriter bufferedWriter = new BufferedWriter(new OutputStreamWriter(client.getOutputStream(),"UTF-8"));
                                    bufferedWriter.write(msg+"\n");//向客户端发送消息
                                    System.out.println(msg);
                                    bufferedWriter.flush();
                                } catch (IOException e) {
                                    e.printStackTrace();
                                }
                            }else{
                                //如果客户端已经断开链接将其从集合中移除
                                sockets.remove(client);
                            }
                        });
                    }
                }
            }).start();

            while (true){
                socket= socketServer.accept();
                //将产生的Socket对象丢入线程池中
                ExecutorService fixedThreadPool = newFixedThreadPool(100);
                fixedThreadPool.execute(new ServerThread(socket));
                //将Socket对象丢入集合中，便于之后的消息分发
                sockets.add(socket);
            }

        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public final static class ServerThread implements Runnable{

        private Socket socket;

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


        @Override
        public void run() {
            //读取消息并放入消息队列中
            String msg=null;
                if(null!=socket&&socket.isConnected()){
                    try {
                        BufferedReader bufferedReader=new BufferedReader(new InputStreamReader(socket.getInputStream(),"UTF-8"));
                        while((msg=bufferedReader.readLine())!=null){
                            msgQueue.offer(msg);//将消息放入队列尾部
                        }
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
        }
    }

}

客户端：

import java.io.*;
import java.net.Socket;

/**
 * @author haye
 * @date 10/7/19
 * @description:
 */
public class Client {
    public static void main(String[] args) {


        try {
            System.out.println("输入昵称后开始聊天：");
            String loginName=new BufferedReader(new InputStreamReader(System.in,"UTF-8")).readLine();//登录名用来区分客户端

//初始化一个socket

            final Socket socket =new Socket("127.0.0.1",9999);

            //通过socket获取字符流

            BufferedWriter bufferedWriter =new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));

            //通过标准输入流获取字符流

            BufferedReader inputBufferedReader =new BufferedReader(new InputStreamReader(System.in,"UTF-8"));

            new Thread(()->{
                try {
                    BufferedReader socketBufferReader=new BufferedReader(new InputStreamReader(socket.getInputStream(),"UTF-8"));
                    String msg=null;
                    while ((msg=socketBufferReader.readLine())!=null){
                        System.out.println(msg);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }).start();

            while (true){

                String str = inputBufferedReader.readLine();

                bufferedWriter.write(loginName+":"+str);

                bufferedWriter.write("\n");

                bufferedWriter.flush();

            }

        }catch (IOException e) {

            e.printStackTrace();

        }

    }
}

以上的观点、代码如有不当之处还请指正！