目錄html

開門見山java

1、數據結構Map編程

2、保證線程安全數組

3、羣聊核心方法安全

4、聊天室具體設計服務器

0、用戶登陸服務器微信

一、查看當前上線用戶網絡

二、羣聊數據結構

三、私信多線程

開門見山

最近一個月記錄了學習Socket網絡編程的知識和實戰案例，相對來講，~~比較系統地~~學習了基於TCP協議實現網絡通訊，也是計算機網絡中重中之重，TCP/IP屬於網絡層，在java中，對該層的工做實現了封裝，在編程中，就更加容易地去實現通訊，而不用顧及底層的實現。固然，咱們須要熟悉五層協議，在實踐中體會其中的原理，理解更加深入。

因此，系列文章從入門開始，不斷完善C/S架構的Socket通訊，回憶一下，首先是實現了Server和Client的互相通訊，在這個過程發現問題，接着就使用多線程技術解決客戶端實時接收信息的問題，後來到了服務器端，發現多用戶鏈接服務器的「先到先得」問題，「後到者」沒法正常通訊，因此再使用線程池技術解決了多用戶服務器的問題。

到此，基本實現了一個簡單的客戶端-服務器應用，所以，本篇將基於前面所有內容，使用客戶端-服務器（C/S架構），結合多線程技術，模擬相似QQ、微信聊天功能，實現一個網絡實時聊天室，目前的功能包括：

(1)L(list):查看當前上線用戶;
(2)G(group):羣聊;
(3)O(one-one):私信;
(4)E(exit):退出當前聊天狀態;
(5)bye:離線;
(6)H(help):查看幫助

本篇將詳細記錄網絡實時聊天室的實現步驟，下面的系列文章爲前提基礎。

Java實現socket通訊網絡編程系列文章：

基於UDP協議網絡Socket編程(java實現C/S通訊案例)【https://blog.csdn.net/Charzous/article/details/109016215】

基於TCP協議網絡socket編程(java實現C/S通訊)【https://blog.csdn.net/Charzous/article/details/109016215】

Java多線程實現TCP網絡Socket編程(C/S通訊)【https://blog.csdn.net/Charzous/article/details/109283697】

Java多線程實現多用戶與服務端Socket通訊【https://blog.csdn.net/Charzous/article/details/109440277】

1、數據結構Map

前兩篇的TCPClientThreadFX和TCPThreadServer實現了多線程的通訊，但也只是客戶端和服務器的聊天，如何作到羣組的聊天？想法就是客戶A的聊天信息經過服務器轉發到同時在線的全部客戶。

具體作法是須要在服務器端新增記錄登錄客戶信息的功能，每一個用戶都有本身的標識。本篇將使用簡單的「在線方式」記錄客戶套接字，即採用集合來保存用戶登錄的套接字信息，來跟蹤用戶鏈接。

因此，咱們須要選擇一種合適的數據結構來保存用戶的Socket和用戶名信息，那在java中，提供了哪些數據結構呢？

Java經常使用的集合類型有：Map、List和Set。Map是保存Key-Value對，List相似數組，可保存可重複的值，而Set只保存不重複的值，至關因而只保存key，不保存value的Map。

若是是有用戶名、學號登陸的操做，就能夠採用Map類型的集合來存儲，例如可以使用key記錄用戶名+學號，value保存套接字。對於本篇的網絡聊天室的需求，須要採用Map，用來保存不一樣用戶的socket和登陸名。用戶套接字socket做爲key來標識一個在線用戶是比較方便的選擇，由於每個客戶端的IP地址+端口組合是不同的。

2、保證線程安全

很明顯，咱們須要使用到多線程技術，而在多線程環境中，對共享資源的讀寫存在線程併發安全的問題，例如HashMap、HaspSet等都不是線程安全的，能夠經過synchronized關鍵字進行加鎖，但還有更方便的方案：能夠直接使用Java標準庫的java.util.concurrent包提供的線程安全的集合。例如HashMap的線程安全是 ConcurrentHashMap，HashSet的線程安全Set是CopyOnWriteArraySet。

如圖，Map繼承體系：

在JDK1.8中，對HashMap進行了改進，當結點數量超過TREEIFY_THRESHOLD 則要轉換爲紅黑樹，這樣很大優化了查詢的效率，但仍然不是線程安全的。

這裏簡單瞭解一下，具體學習能夠查詢相關資料。有了以上的基本知識，下面開始進入網絡實時聊天室的具體實現。

3、羣聊核心方法

基於前面這樣的想法：實現羣聊就是客戶A的聊天信息經過服務器轉發到同時在線的全部客戶，服務器端根據HashMap記錄登錄用戶的socket，向全部用戶轉發信息。

核心的羣組發送方法sendToAllMembers，用於給全部在線客服發送信息。

private void sendToMembers(String msg,String hostAddress,Socket mySocket) throws IOException{

    PrintWriter pw;
    OutputStream out;
    Iterator iterator=users.entrySet().iterator();
    while (iterator.hasNext()){
        Map.Entry entry=(Map.Entry) iterator.next();
        Socket tempSocket = (Socket) entry.getKey();
        String name = (String) entry.getValue();
        if (!tempSocket.equals(mySocket)){
            out=tempSocket.getOutputStream();
            pw=new PrintWriter(new OutputStreamWriter(out,"utf-8"),true);
            pw.println(hostAddress+"："+msg);
        }
    }

}

使用到了Map的遍歷，對其餘全部用戶發送信息。

相同的原理，咱們實現私聊的功能，轉化爲實現的思想，也就是當前用戶和指定用戶Socket之間的通訊，因此我寫了一個sendToOne的方法。

private void sendToOne(String msg,String hostAddress,Socket another) throws IOException{

    PrintWriter pw;
    OutputStream out;

    Iterator iterator=users.entrySet().iterator();
    while (iterator.hasNext()){

        Map.Entry entry=(Map.Entry) iterator.next();
        Socket tempSocket = (Socket) entry.getKey();
        String name = (String) entry.getValue();

        if (tempSocket.equals(another)){
            out=tempSocket.getOutputStream();
            pw=new PrintWriter(new OutputStreamWriter(out,"utf-8"),true);
            pw.println(hostAddress+"私信了你："+msg);
        }
    }
}

以上兩個方法是本網絡聊天室的關鍵，後面實現的功能將是對這兩個方法的靈活運用。

4、聊天室具體設計

目前聊天室的功能定位包括：1)查看當前上線用戶;2):羣聊;3)私信;4)退出當前聊天狀態;5)離線;6)查看幫助。

首先，初始化最關鍵的數據結構，做爲類成員變量，HashMap用來保存Socket和用戶名：

private ConcurrentHashMap<Socket,String> users=new ConcurrentHashMap();

每一個功能具體實現以下：

0、用戶登陸服務器

這裏是最開始的服務器端的信息處理，須要記錄每一個用戶的登陸信息，包括鏈接的套接字和自定義暱稱，方便後續使用。我採用的方法是當用戶鏈接服務器時候，提醒用戶輸入用戶名來進一步操做，也實現了不重名的判斷邏輯。代碼以下：

pw.println("From 服務器：歡迎使用服務！");
pw.println("請輸入用戶名：");
String localName = null;
while ((hostName=br.readLine())!=null){
    users.forEach((k,v)->{
        if (v.equals(hostName))
            flag=true;//線程修改了全局變量
    });

    if (!flag){
        localName=hostName;
        users.put(socket,hostName);
        flag=false;
        break;
    }
    else{
        flag=false;
        pw.println("該用戶名已存在，請修改！");
    }
}

登陸成功以後會向全部在線用戶告知上線信息。

一、查看當前上線用戶

其實就是將服務器端記錄在HashMap中的信息返回給請求用戶，經過約定的命令L來查看：

if (msg.trim().equalsIgnoreCase("L")){
       users.forEach((k,v)->{
       pw.println("用戶:"+v);
       });
       continue;
}

二、羣聊

else if (msg.trim().equals("G")){
    pw.println("您已進入羣聊。");
    while ((msg=br.readLine())!=null){
        if (!msg.equals("E")&&users.size()!=1)
            sendToMembers(msg,localName,socket);
        else if (users.size()==1){
            pw.println("當前羣聊無其餘用戶在線，已自動退出！");
            break;
        }
        else {
            pw.println("您已退出羣組聊天室！");
            break;
        }
    }

}

三、私信

同理，處理邏輯變爲一對一的通訊，與以前服務器-客戶端一對一相似，可是這裏須要更多的處理，保證邏輯正確，包括被私聊人的在線狀態，被私聊人用戶名是否正確等。

//一對一私聊
else if (msg.trim().equalsIgnoreCase("O")){
    pw.println("請輸入私信人的用戶名：");
    String name=br.readLine();

    //查找map中匹配的socket，與之創建通訊
    //有待改進，處理輸入的用戶名不存在的狀況
    users.forEach((k, v)->{
        if (v.equals(name)) {
            isExist=true;//全局變量與線程修改問題
        }

    });
    //已修復用戶不存在的處理邏輯
    Socket temp=null;
    for(Map.Entry<Socket,String> mapEntry : users.entrySet()){
        if(mapEntry.getValue().equals(name))
            temp = mapEntry.getKey();
//                            System.out.println(mapEntry.getKey()+":"+mapEntry.getValue()+'\n');
    }
    if (isExist){
        isExist=false;
        //私信後有一方用戶離開，另外一方未知，仍然發信息而未收到回覆，未處理這種狀況
        while ((msg=br.readLine())!=null){
            if (!msg.equals("E")&&!isLeaved(temp))
                sendToOne(msg,localName,temp);
            else if (isLeaved(temp)){
                pw.println("對方已經離開，已斷開鏈接！");
                break;
            }
            else{
                pw.println("您已退出私信模式！");
                break;
            }
        }
    }
    else
        pw.println("用戶不存在！");
}

四、退出當前聊天狀態

這個功能主要融入到羣聊和私聊中，可見上面兩個功能實現的內部調用，定義了一個方法isLeaved，判斷用戶是否已經下線。

//判斷用戶是否已經下線
private Boolean isLeaved(Socket temp){
    Boolean leave=true;
    for(Map.Entry<Socket,String> mapEntry : users.entrySet()) {
        if (mapEntry.getKey().equals(temp))
            leave = false;
    }
    return leave;
}

五、離線

這個功能比較簡單，經過約定的命令執行。

if (msg.trim().equalsIgnoreCase("bye")) {
     pw.println("From 服務器：服務器已斷開鏈接，結束服務！");

     users.remove(socket,localName);

     sendToMembers("我下線了",localName,socket);
     System.out.println("客戶端離開。");//加當前用戶名
     break;
}

六、查看幫助

經過命令H請求服務器的幫助，是指用戶查看哪些命令對應的功能，來進行選擇。

else if (msg.trim().equalsIgnoreCase("H")){
    pw.println("輸入命令功能：(1)L(list):查看當前上線用戶;(2)G(group):進入羣聊;(3)O(one-one):私信;(4)E(exit):退出當前聊天狀態;(5)bye:離線;(6)H(help):幫助");
    continue;//返回循環
}

5、聊天室服務完整代碼

聊天室實現主要工做在於服務端，聚焦於服務器線程處理的內部類Hanler，上面是各個功能具體介紹，下面完整給出代碼，只須要在前面文章的基礎上，見Java多線程實現多用戶與服務端Socket通訊。

修改服務器端線程處理代碼：

class Handler implements Runnable {
    private Socket socket;

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

    public void run() {
        //本地服務器控制檯顯示客戶端鏈接的用戶信息
        System.out.println("New connection accept:" + socket.getInetAddress().getHostAddress());
        try {
            BufferedReader br = getReader(socket);
            PrintWriter pw = getWriter(socket);

            pw.println("From 服務器：歡迎使用服務！");
            pw.println("請輸入用戶名：");
            String localName = null;
            while ((hostName=br.readLine())!=null){
                users.forEach((k,v)->{
                    if (v.equals(hostName))
                        flag=true;//線程修改了全局變量
                });

                if (!flag){
                    localName=hostName;
                    users.put(socket,hostName);
                    flag=false;//可能找出不一致問題
                    break;
                }
                else{
                    flag=false;
                    pw.println("該用戶名已存在，請修改！");
                }
            }

//                System.out.println(hostName+": "+socket);
            sendToMembers("我已上線",localName,socket);
            pw.println("輸入命令功能：(1)L(list):查看當前上線用戶;(2)G(group):進入羣聊;(3)O(one-one):私信;(4)E(exit):退出當前聊天狀態;(5)bye:離線;(6)H(help):幫助");

            String msg = null;
            //用戶鏈接服務器上線，進入聊天選擇狀態
            while ((msg = br.readLine()) != null) {
                if (msg.trim().equalsIgnoreCase("bye")) {
                    pw.println("From 服務器：服務器已斷開鏈接，結束服務！");

                    users.remove(socket,localName);

                    sendToMembers("我下線了",localName,socket);
                    System.out.println("客戶端離開。");//加當前用戶名
                    break;
                }
                else if (msg.trim().equalsIgnoreCase("H")){
                    pw.println("輸入命令功能：(1)L(list):查看當前上線用戶;(2)G(group):進入羣聊;(3)O(one-one):私信;(4)E(exit):退出當前聊天狀態;(5)bye:離線;(6)H(help):幫助");
                    continue;//返回循環
                }
                else if (msg.trim().equalsIgnoreCase("L")){
                    users.forEach((k,v)->{
                        pw.println("用戶:"+v);
                    });
                    continue;
                }
                //一對一私聊
                else if (msg.trim().equalsIgnoreCase("O")){
                    pw.println("請輸入私信人的用戶名：");
                    String name=br.readLine();

                    //查找map中匹配的socket，與之創建通訊
                    users.forEach((k, v)->{
                        if (v.equals(name)) {
                            isExist=true;//全局變量與線程修改問題
                        }

                    });
                    //已修復用戶不存在的處理邏輯
                    Socket temp=null;
                    for(Map.Entry<Socket,String> mapEntry : users.entrySet()){
                        if(mapEntry.getValue().equals(name))
                            temp = mapEntry.getKey();
                    }
                    if (isExist){
                        isExist=false;
                        //私信後有一方用戶離開，另外一方未知，仍然發信息而未收到回覆，未處理這種狀況
                        while ((msg=br.readLine())!=null){
                            if (!msg.equals("E")&&!isLeaved(temp))
                                sendToOne(msg,localName,temp);
                            else if (isLeaved(temp)){
                                pw.println("對方已經離開，已斷開鏈接！");
                                break;
                            }
                            else{
                                pw.println("您已退出私信模式！");
                                break;
                            }
                        }
                    }
                    else
                        pw.println("用戶不存在！");
                }
                //選擇羣聊
                else if (msg.trim().equals("G")){
                    pw.println("您已進入羣聊。");
                    while ((msg=br.readLine())!=null){
                        if (!msg.equals("E")&&users.size()!=1)
                            sendToMembers(msg,localName,socket);
                        else if (users.size()==1){
                            pw.println("當前羣聊無其餘用戶在線，已自動退出！");
                            break;
                        }
                        else {
                            pw.println("您已退出羣組聊天室！");
                            break;
                        }
                    }

                }
                else
                    pw.println("請選擇聊天狀態！");
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (socket != null)
                    socket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

6、效果演示：基於TCP的網絡實時聊天室

首先，開啓多個客戶端，鏈接服務器開始進入通訊狀態。

下面動圖演示了幾個基本功能，能夠看到三個用戶實現了實時通訊聊天，包括羣聊和私聊功能。其餘功能就留給你們去探索。

結語

系列文章從入門開始，不斷完善C/S架構的Socket通訊，回憶一下，首先是實現了Server和Client的互相通訊，在這個過程發現問題，接着就使用多線程技術解決客戶端實時接收信息的問題，後來到了服務器端，發現多用戶鏈接服務器的「先到先得」問題，「後到者」沒法正常通訊，因此再使用線程池技術解決了多用戶服務器的問題。

本篇基本實現了一個簡單的客戶端-服務器應用，使用客戶端-服務器（C/S架構），結合多線程技術，模擬相似QQ、微信聊天功能，實現一個網絡實時聊天室。

學習到的知識有：多線程、線程池、Socket通訊、TCP協議、HashMap、JavaFX等，全部知識的結合運用，並經過實戰練習，一步步理解知識！

若是以爲不錯歡迎「一鍵三連」哦，點贊收藏關注，有問題直接評論，交流學習！

Java實現socket通訊網絡編程系列文章：

基於UDP協議網絡Socket編程(java實現C/S通訊案例)【https://blog.csdn.net/Charzous/article/details/109016215】

基於TCP協議網絡socket編程(java實現C/S通訊)【https://blog.csdn.net/Charzous/article/details/109016215】

Java多線程實現TCP網絡Socket編程(C/S通訊)【https://blog.csdn.net/Charzous/article/details/109283697】

Java多線程實現多用戶與服務端Socket通訊【https://blog.csdn.net/Charzous/article/details/109440277】

個人CSDN博客：https://blog.csdn.net/Charzous/article/details/109540279

Java進階：基於TCP的網絡實時聊天室（socket通訊案例）

開門見山

1、數據結構Map

2、保證線程安全

3、羣聊核心方法

4、聊天室具體設計

0、用戶登陸服務器

一、查看當前上線用戶

二、羣聊

三、私信

四、退出當前聊天狀態

五、離線

六、查看幫助

5、聊天室服務完整代碼

6、效果演示：基於TCP的網絡實時聊天室

結語