多用戶即時通訊系統03

4.編碼實作02

4.2功能實作-拉取在線用戶

4.2.1思路分析

客戶端想要知道在線用戶串列，就要向服務器發送請求（Message），因為只有服務器端保持著所有與客戶端相連接的socket和uid資訊，

整個流程大致為：對Message的種類進行擴展，然后客戶端向服務器發送一個Message，服務端向客戶端回傳資料，客戶端接收資訊（Message），在接收的這個Message資訊里面包含了在線的用戶串列，

4.2.2代碼實作

1.客戶端：

1.修改：MessageType介面

拓展了一些Message訊息型別

package qqcommon;

/**
 * @author 李
 * @version 1.0
 * 表示訊息型別
 */
public interface MessageType {
    //在介面中定義類一些常量，不同的常量的表示不同的訊息型別
    String MESSAGE_LOGIN_SUCCEED = "1";//表示登錄成功
    String MESSAGE_LOGIN_FAIL = "2";//表示登錄失敗
    String MESSAGE_COMM_MES = "3";//表示普通資訊包
    String MESSAGE_GET_ONLINE_FRIEND = "4";//要求回傳在線用戶串列
    String MESSAGE_RET_ONLINE_FRIEND = "5";//回傳的在線用戶串列
    String MESSAGE_CLIENT_EXIT = "6";//客戶端請求退出

}

2.修改：UserClientService類

在該類中添加onlineFriendList()方法，該方法向服務器發送要求，請求在線用戶串列

    //先服務器端請求在線用戶串列
    public void onlineFriendList() {
        //向服務器發送一個Message，型別MESSAGE_GET_ONLINE_FRIEND,要求回傳在線用戶串列
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_GET_ONLINE_FRIEND);
         message.setSender(u.getUserId());

        //發送給服務器
        try {
            //從管理執行緒的集合里面，通過userId，得到這個執行緒物件
            ClientConnectServerThread clientConnectServerThread =
                    ManageClientConnectServerThread.getClientConnectServerThread(u.getUserId());
            //通過這個執行緒中獲取關聯的socket
            Socket socket = clientConnectServerThread.getSocket();
            //得到當前執行緒的Socket對應的ObjectOutputStream物件
            ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
            oos.writeObject(message);//發送一個Message物件向服務器，要求在線用戶串列
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

3.修改：ClientConnectServerThread類

在該類中的run方法中，增加判斷Message的型別，然后做相應的業務處理

@Override
public void run() {
    //因為Thread需要在后臺和服務器通信，因此我們使用while回圈
    while (true) {
        try {
            System.out.println("客戶端執行緒，等待讀取從服務端發送的訊息");
            ObjectInputStream ois = new ObjectInputStream(socket.getInputStream());

            //如果服務器沒有發送Message物件，執行緒會阻塞在這里
            Message message = (Message) ois.readObject();

            //判斷Message的型別，然后做相應的業務處理
            //如果讀取到的是 服務端回傳的在線用戶串列(MESSAGE_RET_ONLINE_FRIEND)
            if (message.getMesType().equals(MessageType.MESSAGE_RET_ONLINE_FRIEND)) {
                //取出在線用戶串列資訊，并展示
                //這里假定回傳的用戶串列是用空格隔開的id名（如：100 200 紫霞仙子 至尊寶 唐僧）
                String[] onlineUsers = message.getContent().split(" ");
                System.out.println("\n=======當前在線用戶串列=======");
                for (int i = 0; i < onlineUsers.length; i++) {
                    System.out.println("用戶：" + onlineUsers[i]);
                }

            } else {
                System.out.println("讀取到的是其他型別的message，暫時不處理");
            }

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

4.修改QQView類

修改該類中的第54行

使用上述撰寫的方法，完成拉取在線用戶串列的操作

//這里寫一個拉取用戶在線串列的方法
userClientService.onlineFriendList();

2.服務端

大致思路為：前面我們在服務端的執行緒的run方法中，使用for回圈在不停地請求讀取資料，這里我們可以在run方法里面拓展功能，讀取客戶端發送過來的拉取用戶的Message物件，回傳在線用戶串列

1.修改：MessageType介面

和客戶端一樣，拓展Message訊息型別

package qqcommon;

/**
 * @author 李
 * @version 1.0
 * 表示訊息型別
 */
public interface MessageType {
    //在介面中定義類一些常量
    //不同的常量的表示不同的訊息型別
    String MESSAGE_LOGIN_SUCCEED = "1";//表示登錄成功
    String MESSAGE_LOGIN_FAIL = "2";//表示登錄失敗
    String MESSAGE_COMM_MES = "3";//表示普通資訊包
    String MESSAGE_GET_ONLINE_FRIEND = "4";//要求回傳在線用戶串列
    String MESSAGE_RET_ONLINE_FRIEND = "5";//回傳的在線用戶串列
    String MESSAGE_CLIENT_EXIT = "6";//客戶端請求退出
}

2.修改：ServerConnectClientThread類

在該類中的run方法添加了客戶請求拉取在線用戶串列的業務操作

package qqserver.server;

import qqcommon.Message;
import qqcommon.MessageType;

import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.Socket;

/**
 * @author 李
 * @version 1.0
 * 該類的一個物件和某個客戶端保持通信
 */
public class ServerConnectClientThread extends Thread {
    private Socket socket;
    private String userId;//連接到服務端的用戶id


    public ServerConnectClientThread(Socket socket, String userId) {
        this.socket = socket;
        this.userId = userId;
    }

    @Override
    public void run() {//這里執行緒處于run的狀態，可以發送/接收訊息

        while (true) {
            try {
                System.out.println("服務端和客戶端" + userId + "保持通信，讀取資料...");
                ObjectInputStream ois = new ObjectInputStream(socket.getInputStream());
                Message message = (Message) ois.readObject();

                //后面會使用message,根據message的型別，做相應的業務處理

                //業務-：客戶請求拉取在線用戶串列
                if (message.getMesType().equals(MessageType.MESSAGE_GET_ONLINE_FRIEND)) {
                    //客戶請求拉取在線用戶串列
                    //假定回傳的用戶串列是用空格隔開的id名（如：100 200 紫霞仙子 至尊寶 唐僧）
                    System.out.println(message.getSender()+" 要在線用戶串列");
                    String onlineUser = ManageClientThreads.getOnlineUser();
                    
                    //回傳message
                    //構建一個Message物件（這個Message物件包含了在線用戶串列資訊），回傳給客戶端
                    Message message2 = new Message();
                    //設定訊息型別--回傳的在線用戶串列型別-客戶端會根據回傳的訊息型別來進行相應的業務處理
                    message2.setMesType(MessageType.MESSAGE_RET_ONLINE_FRIEND);
                    message2.setContent(onlineUser);//回傳用戶訊息串列
                    //服務器發送的訊息的接收者Getter 就是服務器接收的資訊 的發送者Sender
                    message2.setGetter(message.getSender());
                    
                    //回傳給客戶端
                    ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
                    oos.writeObject(message2);

                }else{
                    System.out.println("其他型別的message，暫時不處理");
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}

3.修改：ManageClientThreads類

在該類中增加了getOnlineUser()方法，用來遍歷取出userId，并在上面的ServerConnectClientThread的run方法中使用，

//這里撰寫方法，可以回傳在線用戶串列
public static String getOnlineUser() {
    //遍歷集合，遍歷 hashmap 的 key
    Iterator<String> iterator = hm.keySet().iterator();//獲取hm集合的ketSet集合的迭代器（這里的ketSet就是userId）
    String onlineUserList = "";
    while (iterator.hasNext()) {//遍歷
        onlineUserList += iterator.next().toString() + " ";//遍歷所有的userId，用空格拼接起來
    }
    return onlineUserList;
}

運行：