多用戶即時通訊系統04

4.編碼實現03

4.7功能實現-伺服器推送消息功能實現

4.7.1思路分析

伺服器推送新聞，本質其實就是群發消息

在伺服器啟動一個獨立線程，專門負責推送新聞

該線程通過管理線程的集合，把所有的線程的socket拿到，併發送Messgae對象信息

客戶端通過接收，自然就拿到了新聞推送信息

4.7.2代碼實現

只需要改動服務端即可

1.創建Utility類

該類與客戶端的工具類一致，用於處理各種情況的輸入

package qqserver.utlis;

/**
 * 工具類的作用:
 * 處理各種情況的用戶輸入，並且能夠按照程式員的需求，得到用戶的控制台輸入。
 */

import java.util.Scanner;

/**


 */
public class Utility {
    //靜態屬性。。。
    private static Scanner scanner = new Scanner(System.in);


    /**
     * 功能：讀取鍵盤輸入的一個菜單選項，值：1——5的範圍
     * @return 1——5
     */
    public static char readMenuSelection() {
        char c;
        for (; ; ) {
            String str = readKeyBoard(1, false);//包含一個字元的字元串
            c = str.charAt(0);//將字元串轉換成字元char類型
            if (c != '1' && c != '2' &&
                    c != '3' && c != '4' && c != '5') {
                System.out.print("選擇錯誤，請重新輸入：");
            } else break;
        }
        return c;
    }

    /**
     * 功能：讀取鍵盤輸入的一個字元
     * @return 一個字元
     */
    public static char readChar() {
        String str = readKeyBoard(1, false);//就是一個字元
        return str.charAt(0);
    }

    /**
     * 功能：讀取鍵盤輸入的一個字元，如果直接按回車，則返回指定的預設值；否則返回輸入的那個字元
     * @param defaultValue 指定的預設值
     * @return 預設值或輸入的字元
     */

    public static char readChar(char defaultValue) {
        String str = readKeyBoard(1, true);//要麼是空字元串，要麼是一個字元
        return (str.length() == 0) ? defaultValue : str.charAt(0);
    }

    /**
     * 功能：讀取鍵盤輸入的整型，長度小於2位
     * @return 整數
     */
    public static int readInt() {
        int n;
        for (; ; ) {
            String str = readKeyBoard(10, false);//一個整數，長度<=10位
            try {
                n = Integer.parseInt(str);//將字元串轉換成整數
                break;
            } catch (NumberFormatException e) {
                System.out.print("數字輸入錯誤，請重新輸入：");
            }
        }
        return n;
    }

    /**
     * 功能：讀取鍵盤輸入的 整數或預設值，如果直接回車，則返回預設值，否則返回輸入的整數
     * @param defaultValue 指定的預設值
     * @return 整數或預設值
     */
    public static int readInt(int defaultValue) {
        int n;
        for (; ; ) {
            String str = readKeyBoard(10, true);
            if (str.equals("")) {
                return defaultValue;
            }

            //異常處理...
            try {
                n = Integer.parseInt(str);
                break;
            } catch (NumberFormatException e) {
                System.out.print("數字輸入錯誤，請重新輸入：");
            }
        }
        return n;
    }

    /**
     * 功能：讀取鍵盤輸入的指定長度的字元串
     * @param limit 限制的長度
     * @return 指定長度的字元串
     */

    public static String readString(int limit) {
        return readKeyBoard(limit, false);
    }

    /**
     * 功能：讀取鍵盤輸入的指定長度的字元串或預設值，如果直接回車，返回預設值，否則返回字元串
     * @param limit 限制的長度
     * @param defaultValue 指定的預設值
     * @return 指定長度的字元串
     */

    public static String readString(int limit, String defaultValue) {
        String str = readKeyBoard(limit, true);
        return str.equals("") ? defaultValue : str;
    }


    /**
     * 功能：讀取鍵盤輸入的確認選項，Y或N
     * 將小的功能，封裝到一個方法中.
     * @return Y或N
     */
    public static char readConfirmSelection() {
        System.out.println("請輸入你的選擇(Y/N): 請小心選擇");
        char c;
        for (; ; ) {//無限迴圈
            //在這裡，將接受到字元，轉成了大寫字母
            //y => Y n=>N
            String str = readKeyBoard(1, false).toUpperCase();
            c = str.charAt(0);
            if (c == 'Y' || c == 'N') {
                break;
            } else {
                System.out.print("選擇錯誤，請重新輸入：");
            }
        }
        return c;
    }

    /**
     * 功能： 讀取一個字元串
     * @param limit 讀取的長度
     * @param blankReturn 如果為true ,表示 可以讀空字元串。
     *                   如果為false表示 不能讀空字元串。
     *
     * 如果輸入為空，或者輸入大於limit的長度，就會提示重新輸入。
     * @return
     */
    private static String readKeyBoard(int limit, boolean blankReturn) {

        //定義了字元串
        String line = "";

        //scanner.hasNextLine() 判斷有沒有下一行
        while (scanner.hasNextLine()) {
            line = scanner.nextLine();//讀取這一行

            //如果line.length=0, 即用戶沒有輸入任何內容，直接回車
            if (line.length() == 0) {
                if (blankReturn) return line;//如果blankReturn=true,可以返回空串
                else continue; //如果blankReturn=false,不接受空串，必須輸入內容
            }

            //如果用戶輸入的內容大於了 limit，就提示重寫輸入
            //如果用戶如的內容 >0 <= limit ,我就接受
            if (line.length() < 1 || line.length() > limit) {
                System.out.print("輸入長度（不能大於" + limit + "）錯誤，請重新輸入：");
                continue;
            }
            break;
        }

        return line;
    }
}

2.創建SendNewsToAllService類

package qqserver.server;

import qqcommon.Message;
import qqcommon.MessageType;
import qqserver.utlis.Utility;

import java.io.IOException;
import java.io.ObjectOutputStream;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;


/**
 * @author 李
 * @version 1.0
 * 該類完成服務端新聞推送功能
 */
public class SendNewsToAllService implements Runnable {

    @Override
    public void run() {

        //為了可以推送多次新聞，使用while
        while (true) {
            System.out.println("請輸入伺服器要推送的新聞/消息[輸入exit表示退出推送服務]");
            String news = Utility.readString(100);

            if ("exit".equals(news)) {
                break;
            }

            //構建一個消息，群發消息
            Message message = new Message();
            message.setSender("伺服器");//發送者
            message.setMesType(MessageType.MESSAGE_TO_ALL_MES);//設置消息發送類型
            message.setContent(news);//新聞內容
            message.setSendTime(new Date().toString());//發送時間
            System.out.println("伺服器推送消息給所有人 說：" + news);

            //遍歷當前所有的通信線程，得到其socket，併發送 message

            HashMap<String, ServerConnectClientThread> hm = ManageClientThreads.getHm();
            Iterator<String> iterator = hm.keySet().iterator();
            while (iterator.hasNext()) {//遍歷
                String onlineUserId = iterator.next().toString();
                try {
                    ObjectOutputStream oos =
                            new ObjectOutputStream(hm.get(onlineUserId).getSocket().getOutputStream());
                    oos.writeObject(message);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}

3.修改QQServer類

在該類的構造方法中，啟動推送新聞的線程

//啟動推送新聞的線程
new Thread(new SendNewsToAllService()).start();

運行

1.啟動服務端

2.啟動三個客戶端，分別登錄三個用戶

3.在伺服器端推送消息

4.在三個用戶端都收到服務端消息

功能實現完畢