從網路通訊的演進過程徹底搞懂Redis高效能通訊的原理(全網最詳細,建議收藏)

跟著Mic學架構發表於2021-10-18

我們一直說Redis的效能很快,那為什麼快?Redis為了達到效能最大化,做了哪些方面的優化呢?
深度解析Redis的資料結構
這篇文章中,其實從資料結構上分析了Redis效能高的一方面原因。

在目前的k-v資料庫的技術選型中,Redis幾乎是首選的用來實現高效能快取的方案,它的效能有多快呢?

根據官方的基準測試資料,一臺普通硬體配置的Linux機器上執行單個Redis例項,處理簡單命令(O(n)或者O(logn)),QPS可以達到8W,如果使用pipeline批處理功能,QPS最高可以達到10W。

Redis 為什麼那麼快

Redis的高效能主要依賴於幾個方面。

  • C語言實現,C語言在一定程度上還是比Java語言效能要高一些,因為C語言不需要經過JVM進行翻譯。
  • 純記憶體I/O,記憶體I/O比磁碟I/O效能更快
  • I/O多路複用,基於epoll的I/O多路複用技術,實現高吞吐網路I/O
  • 單執行緒模型,單執行緒無法利用到多核CPU,但是在Redis中,效能瓶頸並不是在計算上,而是在I/O能力,所以單執行緒能夠滿足高併發的要求。 從另一個層面來說,單執行緒可以避免多執行緒的頻繁上下文切換以及同步鎖機制帶來的效能開銷。

下面我們分別從上述幾個方面進行展開說明,先來看網路I/O的多路複用模型。

從請求處理開始分析

當我們在客戶端向Redis Server傳送一條指令,並且得到Redis回覆的整個過程中,Redis做了什麼呢?

image-20210707221959664

圖4-1

要處理命令,則redis必須完整地接收客戶端的請求,並將命令解析出來,再將結果讀出來,通過網路回寫到客戶端。整個工序分為以下幾個部分:

  • 接收,通過TCP接收到命令,可能會歷經多次TCP包、ack、IO操作
  • 解析,將命令取出來
  • 執行,到對應的地方將value讀出來
  • 返回,將value通過TCP返回給客戶端,如果value較大,則IO負荷會更重

其中解析執行是純cpu/記憶體操作,而接收和返回主要是IO操作,首先我們先來看通訊的過程。

網路IO的通訊原理

同樣,我也畫了一幅圖來描述網路資料的傳輸流程

首先,對於TCP通訊來說,每個TCP Socket的核心中都有一個傳送緩衝區和一個接收緩衝區

接收緩衝區把資料快取到核心,若應用程式一直沒有呼叫Socket的read方法進行讀取,那麼該資料會一直被快取在接收緩衝區內。不管程式是否讀取Socket,對端發來的資料都會經過核心接收並快取到Socket的核心接收緩衝區。

read所要做的工作,就是把核心接收緩衝區中的資料複製到應用層使用者的Buffer裡。

程式呼叫Socket的send傳送資料的時候,一般情況下是將資料從應用層使用者的Buffer裡複製到Socket的核心傳送緩衝區,然後send就會在上層返回。換句話說,send返回時,資料不一定會被髮送到對端。

1576066931883

網路卡中的緩衝區既不屬於核心空間,也不屬於使用者空間。它屬於硬體緩衝,允許網路卡與作業系統之間有個緩衝;
核心緩衝區在核心空間,在記憶體中,用於核心程式,做為讀自或寫往硬體的資料緩衝區;
使用者緩衝區在使用者空間,在記憶體中,用於使用者程式,做為讀自或寫往硬體的資料緩衝區

網路卡晶片收到網路資料會以中斷的方式通知CPU,我有資料了,存在我的硬體緩衝裡了,來讀我啊。
CPU收到這個中斷訊號後,會呼叫相應的驅動介面函式從網路卡的硬體緩衝裡把資料讀到核心緩衝區,正常情況下會向上傳遞給TCP/IP模組一層一層的處理。

NIO多路複用機制

Redis的通訊採用的是多路複用機制,什麼是多路複用機制呢?

由於Redis是C語言實現,為了簡化大家的理解,我們採用Java語言來描述這個過程。

在理解多路複用之前,我們先來了解一下BIO。

BIO模型

在Java中,如果要實現網路通訊,我們會採用Socket套接字來完成。

Socket這不是一個協議,而是一個通訊模型。其實它最初是BSD發明的,主要用來一臺電腦的兩個程式間通訊,然後把它用到了兩臺電腦的程式間通訊。所以,可以把它簡單理解為程式間通訊,不是什麼高階的東西。主要做的事情不就是:

  • A發包:發請求包給某個已經繫結的埠(所以我們經常會訪問這樣的地址182.13.15.16:1235,1235就是埠);收到B的允許;然後正式傳送;傳送完了,告訴B要斷開連結;收到斷開允許,馬上斷開,然後傳送已經斷開資訊給B。

  • B收包:繫結埠和IP;然後在這個埠監聽;接收到A的請求,發允許給A,並做好接收準備,主要就是清理快取等待接收新資料;然後正式接收;接受到斷開請求,允許斷開;確認斷開後,繼續監聽其它請求。

可見,Socket其實就是I/O操作,Socket並不僅限於網路通訊,在網路通訊中,它涵蓋了網路層、傳輸層、會話層、表示層、應用層——其實這都不需要記,因為Socket通訊時候用到了IP和埠,僅這兩個就表明了它用到了網路層和傳輸層;而且它無視多臺電腦通訊的系統差別,所以它涉及了表示層;一般Socket都是基於一個應用程式的,所以會涉及到會話層和應用層。

構建基礎的BIO通訊模型

BIOServerSocket

public class BIOServerSocket {
    //先定義一個埠號,這個埠的值是可以自己調整的。
    static final int DEFAULT_PORT=8080;
    public static void main(String[] args) throws IOException {
        //先定義一個埠號,這個埠的值是可以自己調整的。
        //在伺服器端,我們需要使用ServerSocket,所以我們先宣告一個ServerSocket變數
        ServerSocket serverSocket=null;
        //接下來,我們需要繫結監聽埠, 那我們怎麼做呢?只需要建立使用serverSocket例項
        //ServerSocket有很多構造過載,在這裡,我們把前邊定義的埠傳入,表示當前
        //ServerSocket監聽的埠是8080
        serverSocket=new ServerSocket(DEFAULT_PORT);
        System.out.println("啟動服務,監聽埠:"+DEFAULT_PORT);
        //回顧一下前面我們講的內容,接下來我們就需要開始等待客戶端的連線了。
        //所以我們要使用的是accept這個函式,並且當accept方法獲得一個客戶端請求時,會返回
        //一個socket物件, 這個socket物件讓伺服器可以用來和客戶端通訊的一個端點。

        //開始等待客戶端連線,如果沒有客戶端連線,就會一直阻塞在這個位置
        Socket socket=serverSocket.accept();
        //很可能有多個客戶端來發起連線,為了區分客戶端,我們們可以輸出客戶端的埠號
        System.out.println("客戶端:"+socket.getPort()+"已連線");
        //一旦有客戶端連線過來,我們就可以用到IO來獲得客戶端傳過來的資料。
        //使用InputStream來獲得客戶端的輸入資料
        //bufferedReader大家還記得吧,他維護了一個緩衝區可以減少資料來源讀取的頻率
        BufferedReader bufferedReader=new BufferedReader(new InputStreamReader(socket.getInputStream()));
        String clientStr=bufferedReader.readLine(); //讀取一行資訊
        System.out.println("客戶端發了一段訊息:"+clientStr);
        //服務端收到資料以後,可以給到客戶端一個回覆。這裡我們們用到BufferedWriter
        BufferedWriter bufferedWriter=new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
        bufferedWriter.write("我已經收到你的訊息了\n");
        bufferedWriter.flush(); //清空緩衝區觸發訊息傳送
    }
}

BIOClientSocket

public class BIOClientSocket {
    static final int DEFAULT_PORT=8080;
    public static void main(String[] args) throws IOException {

        //在客戶端這邊,我們們使用socket來連線到指定的ip和埠
        Socket socket=new Socket("localhost",8080);
        //使用BufferedWriter,像伺服器端寫入一個訊息
        BufferedWriter bufferedWriter=new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
        bufferedWriter.write("我是客戶端Client-01\n");
        bufferedWriter.flush();
        BufferedReader bufferedReader=new BufferedReader(new InputStreamReader(socket.getInputStream()));
        String serverStr=bufferedReader.readLine(); //通過bufferedReader讀取服務端返回的訊息
        System.out.println("服務端返回的訊息:"+serverStr);
    }
}

上述程式碼構建了一個簡單的BIO通訊模型,也就是服務端建立一個監聽,客戶端向服務端傳送一個訊息,實現簡單的網路通訊,那BIO有什麼弊端呢?

我們通過對BIOServerSocket進行改造,關注case1和case2部分。

  • case1: 增加了while迴圈,實現重複監聽
  • case2: 當服務端收到客戶端的請求後,不直接返回,而是等待20s。
public class BIOServerSocket {
    //先定義一個埠號,這個埠的值是可以自己調整的。
    static final int DEFAULT_PORT=8080;
    public static void main(String[] args) throws IOException, InterruptedException {
        ServerSocket serverSocket=null;
        serverSocket=new ServerSocket(DEFAULT_PORT);
        System.out.println("啟動服務,監聽埠:"+DEFAULT_PORT);

        while(true) { //case1: 增加迴圈,允許迴圈接收請求
            Socket socket = serverSocket.accept();
            System.out.println("客戶端:" + socket.getPort() + "已連線");
            BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
            String clientStr = bufferedReader.readLine(); //讀取一行資訊
            System.out.println("客戶端發了一段訊息:" + clientStr);
            Thread.sleep(20000); //case2: 修改:增加等待時間
            BufferedWriter bufferedWriter = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
            bufferedWriter.write("我已經收到你的訊息了\n");
            bufferedWriter.flush(); //清空緩衝區觸發訊息傳送
        }
    }
}

接著,把BIOClientSocket複製兩份(client1、client2),同時向BIOServerSocket發起請求。

執行後看到的現象應該是: client1先傳送請求到Server端,由於Server端等待20s才返回,導致client2的請求一直被阻塞。

這個情況會導致一個問題,如果服務端在同一個時刻只能處理一個客戶端的連線,而如果一個網站同時有1000個使用者訪問,那麼剩下的999個使用者都需要等待,而這個等待的耗時取決於前面的請求的處理時長,如圖4-2所示。

image-20210708152538953

圖4-2

基於多執行緒優化BIO

為了讓服務端能夠同時處理更多的客戶端連線,避免因為某個客戶端連線阻塞導致後續請求被阻塞,於是引入多執行緒技術,程式碼如下。

ServerSocket

public static void main(String[] args) throws IOException, InterruptedException {
    final int DEFAULT_PORT=8080;
    ServerSocket serverSocket=null;
    serverSocket=new ServerSocket(DEFAULT_PORT);
    System.out.println("啟動服務,監聽埠:"+DEFAULT_PORT);
    ExecutorService executorService= Executors.newFixedThreadPool(5);
    while(true) {
        Socket socket = serverSocket.accept();
        executorService.submit(new SocketThread(socket));
    }
}

SocketThread

public class SocketThread implements Runnable{
    Socket socket;

    public SocketThread(Socket socket) {
        this.socket = socket;
    }
    @Override
    public void run() {
        System.out.println("客戶端:" + socket.getPort() + "已連線");
        try {
            BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
            String clientStr = null; //讀取一行資訊
            clientStr = bufferedReader.readLine();
            System.out.println("客戶端發了一段訊息:" + clientStr);
            Thread.sleep(20000);
            BufferedWriter bufferedWriter = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
            bufferedWriter.write("我已經收到你的訊息了\n");
            bufferedWriter.flush(); //清空緩衝區觸發訊息傳送
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

如圖4-3所示,當引入了多執行緒之後,每個客戶端的連結(Socket),我們可以直接給到執行緒池去執行,而由於這個過程是非同步的,所以並不會同步阻塞影響後續連結的監聽,因此在一定程度上可以提升服務端連結的處理數量。

image-20210708160026412

圖4-3

NIO非阻塞IO

使用多執行緒的方式來解決這個問題,仍然有一個缺點,執行緒的數量取決於硬體配置,所以執行緒數量是有限的,如果請求量比較大的時候,執行緒本身會收到限制從而併發量也不會太高。那怎麼辦呢,我們可以採用非阻塞IO。

NIO 從JDK1.4 提出的,本意是New IO,它的出現為了彌補原本IO的不足,提供了更高效的方式,提出一個通道(channel)的概念,在IO中它始終以流的形式對資料的傳輸和接受,下面我們演示一下NIO的使用。

NioServerSocket

public class NioServerSocket {
    public static void main(String[] args) {
        try {
            ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
            serverSocketChannel.configureBlocking(false);
            serverSocketChannel.socket().bind(new InetSocketAddress(8080));
            while (true) {
                SocketChannel socketChannel = serverSocketChannel.accept();
                if (socketChannel != null) {
                    //讀取資料
                    ByteBuffer buffer = ByteBuffer.allocate(1024);
                    socketChannel.read(buffer);
                    System.out.println(new String(buffer.array()));
                    //寫出資料
                    Thread.sleep(10000); //阻塞一段時間
                    //當資料讀取到緩衝區之後,接下來就需要把緩衝區的資料寫出到通道,而在寫出之前必須要呼叫flip方法,實際上就是重置一個有效位元組範圍,然後把這個資料接觸到通道。
                    buffer.flip();
                    socketChannel.write(buffer);//寫出資料
                } else {
                    Thread.sleep(1000);
                    System.out.println("連線未就緒");
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

NioClientSocket

public class NioClientSocket {
    public static void main(String[] args) {
        try {
            SocketChannel socketChannel= SocketChannel.open();
            socketChannel.configureBlocking(false);
            socketChannel.connect(new InetSocketAddress("localhost",8080));
            if(socketChannel.isConnectionPending()){
                socketChannel.finishConnect();
            }
            ByteBuffer byteBuffer= ByteBuffer.allocate(1024);
            byteBuffer.put("Hello I'M SocketChannel Client".getBytes());
            byteBuffer.flip();
            socketChannel.write(byteBuffer);
            //讀取服務端資料
            byteBuffer.clear();
            while(true) {
                int i = socketChannel.read(byteBuffer);
                if (i > 0) {
                    System.out.println("收到服務端的資料:" + new String(byteBuffer.array()));
                } else {
                    System.out.println("服務端資料未準備好");
                    Thread.sleep(1000);
                }
            }
        } catch (IOException | InterruptedException e) {
            e.printStackTrace();
        }

    }
}

所謂的NIO(非阻塞IO),其實就是取消了IO阻塞和連線阻塞,當服務端不存在阻塞的時候,就可以不斷輪詢處理客戶端的請求,如圖4-4所示,表示NIO下的執行流程。

image-20210708165359843

圖4-4

上述這種NIO的使用方式,仍然存在一個問題,就是客戶端或者服務端需要通過一個執行緒不斷輪詢才能獲得結果,而這個輪詢過程中會浪費執行緒資源。

多路複用IO

大家站在全域性的角度再思考一下整個過程,有哪些地方可以優化呢?

我們回到NIOClientSocket中下面這段程式碼,當客戶端通過read方法去讀取服務端返回的資料時,如果此時服務端資料未準備好,對於客戶端來說就是一次無效的輪詢。

我們能不能夠設計成,當客戶端呼叫read方法之後,不僅僅不阻塞,同時也不需要輪詢。而是等到服務端的資料就緒之後, 告訴客戶端。然後客戶端再去讀取服務端返回的資料呢?

就像點外賣一樣,我們在網上下單之後,繼續做其他事情,等到外賣到了公司,外賣小哥主動打電話告訴你,你直接去前臺取餐即可。

while(true) {
    int i = socketChannel.read(byteBuffer);
    if (i > 0) {
        System.out.println("收到服務端的資料:" + new String(byteBuffer.array()));
    } else {
        System.out.println("服務端資料未準備好");
        Thread.sleep(1000);
    }
}

所以為了優化這個問題,引入了多路複用機制。

I/O多路複用的本質是通過一種機制(系統核心緩衝I/O資料),讓單個程式可以監視多個檔案描述符,一旦某個描述符就緒(一般是讀就緒或寫就緒),能夠通知程式進行相應的讀寫操作

什麼是fd:在linux中,核心把所有的外部裝置都當成是一個檔案來操作,對一個檔案的讀寫會呼叫核心提供的系統命令,返回一個fd(檔案描述符)。而對於一個socket的讀寫也會有相應的檔案描述符,成為socketfd。

常見的IO多路複用方式有【select、poll、epoll】,都是Linux API提供的IO複用方式,那麼接下來重點講一下select、和epoll這兩個模型

  • select:程式可以通過把一個或者多個fd傳遞給select系統呼叫,程式會阻塞在select操作上,這樣select可以幫我們檢測多個fd是否處於就緒狀態,這個模式有兩個缺點

    • 由於他能夠同時監聽多個檔案描述符,假如說有1000個,這個時候如果其中一個fd 處於就緒狀態了,那麼當前程式需要線性輪詢所有的fd,也就是監聽的fd越多,效能開銷越大。
    • 同時,select在單個程式中能開啟的fd是有限制的,預設是1024,對於那些需要支援單機上萬的TCP連線來說確實有點少
  • epoll:linux還提供了epoll的系統呼叫,epoll是基於事件驅動方式來代替順序掃描,因此效能相對來說更高,主要原理是,當被監聽的fd中,有fd就緒時,會告知當前程式具體哪一個fd就緒,那麼當前程式只需要去從指定的fd上讀取資料即可,另外,epoll所能支援的fd上線是作業系統的最大檔案控制程式碼,這個數字要遠遠大於1024

【由於epoll能夠通過事件告知應用程式哪個fd是可讀的,所以我們也稱這種IO為非同步非阻塞IO,當然它是偽非同步的,因為它還需要去把資料從核心同步複製到使用者空間中,真正的非同步非阻塞,應該是資料已經完全準備好了,我只需要從使用者空間讀就行】

I/O多路複用的好處是可以通過把多個I/O的阻塞複用到同一個select的阻塞上,從而使得系統在單執行緒的情況下可以同時處理多個客戶端請求。它的最大優勢是系統開銷小,並且不需要建立新的程式或者執行緒,降低了系統的資源開銷,它的整體實現思想如圖4-5所示。

客戶端請求到服務端後,此時客戶端在傳輸資料過程中,為了避免Server端在read客戶端資料過程中阻塞,服務端會把該請求註冊到Selector復路器上,服務端此時不需要等待,只需要啟動一個執行緒,通過selector.select()阻塞輪詢復路器上就緒的channel即可,也就是說,如果某個客戶端連線資料傳輸完成,那麼select()方法會返回就緒的channel,然後執行相關的處理即可。

image-20210708203509498

圖4-5

NIOServer的實現如下

測試訪問的時候,直接在cmd中通過telnet連線NIOServer,便可傳送資訊。

public class NIOServer implements Runnable{
    Selector selector;
    ServerSocketChannel serverSocketChannel;
    public NIOServer(int port) throws IOException {
        selector=Selector.open(); //多路複用器
        serverSocketChannel=ServerSocketChannel.open();
        //繫結監聽埠
        serverSocketChannel.socket().bind(new InetSocketAddress(port));
        serverSocketChannel.configureBlocking(false);//非阻塞配置
        //針對serverSocketChannel註冊一個ACCEPT連線監聽事件
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
    }
    @Override
    public void run() {
        while(!Thread.interrupted()){
            try {
                selector.select(); //阻塞等待事件就緒
                Set selected=selector.selectedKeys(); //得到事件列表
                Iterator it=selected.iterator();
                while(it.hasNext()){
                    dispatch((SelectionKey) it.next()); //分發事件
                    it.remove(); //移除當前時間
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    private void dispatch(SelectionKey key) throws IOException {
        if(key.isAcceptable()){ //如果是客戶端的連線事件,則需要針對該連線註冊讀寫事件
            register(key);
        }else if(key.isReadable()){
            read(key);
        }else if(key.isWritable()){
            write(key);
        }
    }
    private void register(SelectionKey key) throws IOException {
        //得到事件對應的連線
        ServerSocketChannel server=(ServerSocketChannel)key.channel();
        SocketChannel channel=server.accept(); //獲得客戶端的連結
        channel.configureBlocking(false);
        //把當前客戶端連線註冊到selector上,註冊事件為READ,
        // 也就是當前channel可讀時,就會觸發事件,然後讀取客戶端的資料
        channel.register(this.selector,SelectionKey.OP_READ);
    }

    private void read(SelectionKey key) throws IOException {
        SocketChannel channel=(SocketChannel)key.channel();
        ByteBuffer byteBuffer= ByteBuffer.allocate(1024);
        channel.read(byteBuffer); //把資料從channel讀取到緩衝區
        System.out.println("server receive msg:"+new String(byteBuffer.array()));
    }
    private void write(SelectionKey key) throws IOException {
        SocketChannel channel=(SocketChannel)key.channel();
        //寫一個資訊給到客戶端
        channel.write(ByteBuffer.wrap("hello Client,I'm NIO Server\r\n".getBytes()));
    }

    public static void main(String[] args) throws IOException {
        NIOServer server=new NIOServer(8888);
        new Thread(server).start();
    }
}

事實上NIO已經解決了上述BIO暴露的下面兩個問題:

  1. 同步阻塞IO,讀寫阻塞,執行緒等待時間過長。
  2. 在制定執行緒策略的時候,只能根據CPU的數目來限定可用執行緒資源,不能根據連線併發數目來制定,也就是連線有限制。否則很難保證對客戶端請求的高效和公平。

到這裡為止,通過NIO的多路複用機制,解決了IO阻塞導致客戶端連線處理受限的問題,服務端只需要一個執行緒就可以維護多個客戶端,並且客戶端的某個連線如果準備就緒時,會通過事件機制告訴應用程式某個channel可用,應用程式通過select方法選出就緒的channel進行處理。

單執行緒Reactor 模型(高效能I/O設計模式)

瞭解了NIO多路複用後,就有必要再和大家說一下Reactor多路複用高效能I/O設計模式,Reactor本質上就是基於NIO多路複用機制提出的一個高效能IO設計模式,它的核心思想是把響應IO事件和業務處理進行分離,通過一個或者多個執行緒來處理IO事件,然後將就緒得到事件分發到業務處理handlers執行緒去非同步非阻塞處理,如圖4-6所示。

Reactor模型有三個重要的元件:

  • Reactor :將I/O事件發派給對應的Handler
  • Acceptor :處理客戶端連線請求
  • Handlers :執行非阻塞讀/寫

image-20210708212057895

圖4-6

下面演示一個單執行緒的Reactor模型。

Reactor

Reactor 負責響應IO事件,一旦發生,廣播傳送給相應的Handler去處理。

public class Reactor implements Runnable{
    private final Selector selector;
    private final ServerSocketChannel serverSocketChannel;

    public Reactor(int port) throws IOException {
        //建立選擇器
        selector= Selector.open();
        //建立NIO-Server
        serverSocketChannel=ServerSocketChannel.open();
        serverSocketChannel.bind(new InetSocketAddress(port));
        serverSocketChannel.configureBlocking(false);
        SelectionKey key=serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
        // 繫結一個附加物件
        key.attach(new Acceptor(selector,serverSocketChannel));
    }

    @Override
    public void run() {
        while(!Thread.interrupted()){
            try {
                selector.select(); //阻塞等待就緒事件
                Set selectionKeys=selector.selectedKeys();
                Iterator it=selectionKeys.iterator();
                while(it.hasNext()){
                    dispatch((SelectionKey) it.next());
                    it.remove();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
    public void dispatch(SelectionKey key){
        //呼叫之前註冊時附加的物件,也就是attach附加的acceptor
        Runnable r=(Runnable)key.attachment();
        if(r!=null){
            r.run();
        }
    }

    public static void main(String[] args) throws IOException {
        new Thread(new Reactor(8888)).start();
    }
}

Acceptor

public class Acceptor implements Runnable{
    private Selector selector;
    private ServerSocketChannel serverSocketChannel;

    public Acceptor(Selector selector, ServerSocketChannel serverSocketChannel) {
        this.selector = selector;
        this.serverSocketChannel = serverSocketChannel;
    }

    @Override
    public void run() {
        SocketChannel channel;
        try {
            channel=serverSocketChannel.accept();
            System.out.println(channel.getRemoteAddress()+": 收到一個客戶端連線");
            channel.configureBlocking(false);
            //當channel連線中資料就緒時,呼叫DispatchHandler來處理channel
            //巧妙使用了SocketChannel的attach功能,將Hanlder和可能會發生事件的channel連結在一起,當發生事件時,可以立即觸發相應連結的Handler。
            channel.register(selector, SelectionKey.OP_READ,new DispatchHandler(channel));
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
}

Handler

public class DispatchHandler implements Runnable{
    private SocketChannel channel;

    public DispatchHandler(SocketChannel channel) {
        this.channel = channel;
    }

    @Override
    public void run() {
        System.out.println(Thread.currentThread().getName()+"---handler"); //case: 列印當前執行緒名稱,證明I/O是同一個執行緒來處理。
        ByteBuffer buffer=ByteBuffer.allocate(1024);
        int len=0,total=0;
        String msg="";
        try {
            do {
                len = channel.read(buffer);
                if (len > 0) {
                    total += len;
                    msg += new String(buffer.array());
                }
                buffer.clear();
            } while (len > buffer.capacity());
            System.out.println(channel.getRemoteAddress()+":Server Receive msg:"+msg);

        }catch (Exception e){
            e.printStackTrace();
            if(channel!=null){
                try {
                    channel.close();
                } catch (IOException ioException) {
                    ioException.printStackTrace();
                }
            }
        }

    }
}

演示方式,通過window的cmd視窗,使用telnet 192.168.1.102 8888 連線到Server端進行資料通訊;也可以通過下面這樣一個客戶端程式來訪問。

ReactorClient

public class ReactorClient {

    private static Selector selector;
    public static void main(String[] args) throws IOException {
        selector=Selector.open();
        //建立一個連線通道連線指定的server
        SocketChannel socketChannel= SocketChannel.open();
        socketChannel.configureBlocking(false);
        socketChannel.connect(new InetSocketAddress("192.168.1.102",8888));
        socketChannel.register(selector, SelectionKey.OP_CONNECT);
        while(true){
            selector.select();
            Set<SelectionKey> selectionKeys=selector.selectedKeys();
            Iterator<SelectionKey> iterator=selectionKeys.iterator();
            while(iterator.hasNext()){
                SelectionKey key=iterator.next();
                iterator.remove();
                if(key.isConnectable()){
                    handleConnection(key);
                }else if(key.isReadable()){
                    handleRead(key);
                }
            }
        }
    }
    private static void handleConnection(SelectionKey key) throws IOException {
        SocketChannel socketChannel=(SocketChannel)key.channel();
        if(socketChannel.isConnectionPending()){
            socketChannel.finishConnect();
        }
        socketChannel.configureBlocking(false);
        while(true) {
            Scanner in = new Scanner(System.in);
            String msg = in.nextLine();
            socketChannel.write(ByteBuffer.wrap(msg.getBytes()));
            socketChannel.register(selector,SelectionKey.OP_READ);
        }
    }
    private static void handleRead(SelectionKey key) throws IOException {
        SocketChannel channel=(SocketChannel)key.channel();
        ByteBuffer byteBuffer=ByteBuffer.allocate(1024);
        channel.read(byteBuffer);
        System.out.println("client receive msg:"+new String(byteBuffer.array()));
    }
}

這是最基本的單Reactor單執行緒模型(整體的I/O操作是由同一個執行緒完成的)

其中Reactor執行緒,負責多路分離套接字,有新連線到來觸發connect 事件之後,交由Acceptor進行處理,有IO讀寫事件之後交給hanlder 處理。

Acceptor主要任務就是構建handler ,在獲取到和client相關的SocketChannel之後 ,繫結到相應的hanlder上,對應的SocketChannel有讀寫事件之後,基於racotor 分發,hanlder就可以處理了(所有的IO事件都繫結到selector上,有Reactor分發)

Reactor 模式本質上指的是使用 I/O 多路複用(I/O multiplexing) + 非阻塞 I/O(non-blocking I/O)的模式。

多執行緒單Reactor模型

單執行緒Reactor這種實現方式有存在著缺點,從例項程式碼中可以看出,handler的執行是序列的,如果其中一個handler處理執行緒阻塞將導致其他的業務處理阻塞。由於handler和reactor在同一個執行緒中的執行,這也將導致新的無法接收新的請求,我們做一個小實驗:

  • 在上述Reactor程式碼的DispatchHandler的run方法中,增加一個Thread.sleep()。
  • 開啟多個客戶端視窗連線到Reactor Server端,其中一個視窗傳送一個資訊後被阻塞,另外一個視窗再發資訊時由於前面的請求阻塞導致後續請求無法被處理。

為了解決這種問題,有人提出使用多執行緒的方式來處理業務,也就是在業務處理的地方加入執行緒池非同步處理,將reactor和handler在不同的執行緒來執行,如圖4-7所示。

image-20210709154534593

圖4-7

多執行緒改造-MultiDispatchHandler

我們直接將4.2.5小節中的Reactor單執行緒模型改成多執行緒,其實我們就是把IO阻塞的問題通過非同步的方式做了優化,程式碼如下,

public class MultiDispatchHandler implements Runnable{
    private SocketChannel channel;

    public MultiDispatchHandler(SocketChannel channel) {
        this.channel = channel;
    }
    private static Executor executor = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors() << 1);

    @Override
    public void run() {
        processor();
    }
    private void processor(){
        executor.execute(new ReaderHandler(channel));
    }
    public static class ReaderHandler implements Runnable{
        private SocketChannel channel;

        public ReaderHandler(SocketChannel socketChannel) {
            this.channel = socketChannel;
        }
        @Override
        public void run() {
            System.out.println(Thread.currentThread().getName()+"---handler"); //case: 列印當前執行緒名稱,證明I/O是同一個執行緒來處理。
            ByteBuffer buffer= ByteBuffer.allocate(1024);
            int len=0;
            String msg="";
            try {
                do {
                    len = channel.read(buffer);
                    if (len > 0) {
                        msg += new String(buffer.array());
                    }
                    buffer.clear();
                } while (len > buffer.capacity());

                if(len>0) {
                    System.out.println(channel.getRemoteAddress() + ":Server Receive msg:" + msg);
                }
            }catch (Exception e){
                e.printStackTrace();
                if(channel!=null){
                    try {
                        channel.close();
                    } catch (IOException ioException) {
                        ioException.printStackTrace();
                    }
                }
            }
        }
    }
}

Acceptor

public class Acceptor implements Runnable{
    private Selector selector;
    private ServerSocketChannel serverSocketChannel;

    public Acceptor(Selector selector, ServerSocketChannel serverSocketChannel) {
        this.selector = selector;
        this.serverSocketChannel = serverSocketChannel;
    }

    @Override
    public void run() {
        SocketChannel channel;
        try {
            channel=serverSocketChannel.accept();
            System.out.println(channel.getRemoteAddress()+": 收到一個客戶端連線");
            channel.configureBlocking(false);
            //當channel連線中資料就緒時,呼叫DispatchHandler來處理channel
            //巧妙使用了SocketChannel的attach功能,將Hanlder和可能會發生事件的channel連結在一起,當發生事件時,可以立即觸發相應連結的Handler。
            channel.register(selector, SelectionKey.OP_READ,new MultiDispatchHandler(channel));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

多執行緒Reactor總結

在多執行緒Reactor模型中,新增了一個工作者執行緒池,並將非I/O操作從Reactor執行緒中移出轉交給工作者執行緒池來執行。這樣能夠提高Reactor執行緒的I/O響應,不至於因為一些耗時的業務邏輯而延遲對後面I/O請求的處理。

多Reactor多執行緒模式(主從多Reactor模型)

在多執行緒單Reactor模型中,我們發現所有的I/O操作是由一個Reactor來完成,而Reactor執行在單個執行緒中,它需要處理包括Accept()/read()/write/connect操作,對於小容量的場景,影響不大。但是對於高負載、大併發或大資料量的應用場景時,容易成為瓶頸,主要原因如下:

  • 一個NIO執行緒同時處理成百上千的鏈路,效能上無法支撐,即便NIO執行緒的CPU負荷達到100%,也無法滿足海量訊息的讀取和傳送;
  • 當NIO執行緒負載過重之後,處理速度將變慢,這會導致大量客戶端連線超時,超時之後往往會進行重發,這更加重了NIO執行緒的負載,最終會導致大量訊息積壓和處理超時,成為系統的效能瓶頸;

所以,我們還可以更進一步優化,引入多Reactor多執行緒模式,如圖4-8所示,Main Reactor負責接收客戶端的連線請求,然後把接收到的請求傳遞給SubReactor(其中subReactor可以有多個),具體的業務IO處理由SubReactor完成。

Multiple Reactors 模式通常也可以等同於 Master-Workers 模式,比如 Nginx 和 Memcached 等就是採用這種多執行緒模型,雖然不同的專案實現細節略有區別,但總體來說模式是一致的。

image-20210709162516832

圖4-8
  • Acceptor,請求接收者,在實踐時其職責類似伺服器,並不真正負責連線請求的建立,而只將其請求委託 Main Reactor 執行緒池來實現,起到一個轉發的作用。

  • Main Reactor,主 Reactor 執行緒組,主要負責連線事件,並將IO讀寫請求轉發到 SubReactor 執行緒池

  • Sub Reactor,Main Reactor 通常監聽客戶端連線後會將通道的讀寫轉發到 Sub Reactor 執行緒池中一個執行緒(負載均衡),負責資料的讀寫。在 NIO 中 通常註冊通道的讀(OP_READ)、寫事件(OP_WRITE)。

MultiplyReactor

public class MultiplyReactor {
    public static void main(String[] args) throws IOException {
        MultiplyReactor mr = new MultiplyReactor(8888);
        mr.start();
    }
    private static final int POOL_SIZE = Runtime.getRuntime().availableProcessors();
    // Reactor(Selector) 執行緒池,其中一個執行緒被 mainReactor 使用,剩餘執行緒都被 subReactor 使用
    static Executor mainReactorExecutor = Executors.newFixedThreadPool(POOL_SIZE);
    // 主 Reactor,接收連線,把 SocketChannel 註冊到從 Reactor 上
    private Reactor mainReactor;
    private int port;

    public MultiplyReactor(int port) {
        try {
            this.port = port;
            mainReactor = new Reactor();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    /**
     * 啟動主從 Reactor,初始化並註冊 Acceptor 到主 Reactor
     */
    public void start() throws IOException {
        new Acceptor(mainReactor.getSelector(), port); // 將 ServerSocketChannel 註冊到 mainReactor
        mainReactorExecutor.execute(mainReactor); //使用執行緒池來處理main Reactor的連線請求
    }
}

Reactor

public class Reactor implements Runnable{
    private ConcurrentLinkedQueue<AsyncHandler> events=new ConcurrentLinkedQueue<>();
    private final Selector selector;

    public Reactor() throws IOException {
        this.selector = Selector.open();
    }

    public Selector getSelector(){
        return selector;
    }
    @Override
    public void run() {
        try {
            while (!Thread.interrupted()) {
                AsyncHandler handler;
                while ((handler = events.poll()) != null) {
                    handler.getChannel().configureBlocking(false);
                    SelectionKey sk=handler.getChannel().register(selector, SelectionKey.OP_READ);
                    sk.attach(handler);
                    handler.setSk(sk);
                }
                selector.select(); //阻塞
                Set<SelectionKey> selectionKeys=selector.selectedKeys();
                Iterator<SelectionKey> it=selectionKeys.iterator();
                while(it.hasNext()){
                    SelectionKey key=it.next();
                    //獲取attach方法傳入的附加物件
                    Runnable runnable=(Runnable)key.attachment();
                    if(runnable!=null){
                        runnable.run();
                    }
                    it.remove();
                }
            }
        }catch (Exception e){
            e.printStackTrace();
        }
    }
    public void register(AsyncHandler asyncHandler){
        events.offer(asyncHandler);
        selector.wakeup();
    }
}

Acceptor

public class Acceptor implements Runnable{

    final Selector sel;
    final ServerSocketChannel serverSocket;
    int handleNext = 0;

    private final int POOL_SIZE=Runtime.getRuntime().availableProcessors();
    private Executor subReactorExecutor= Executors.newFixedThreadPool(POOL_SIZE);

    private Reactor[] subReactors=new Reactor[POOL_SIZE-1];

    public Acceptor(Selector sel, int port) throws IOException {
        this.sel = sel;
        serverSocket = ServerSocketChannel.open();
        serverSocket.socket().bind(new InetSocketAddress(port)); // 繫結埠
        // 設定成非阻塞模式
        serverSocket.configureBlocking(false);
        // 註冊到 選擇器 並設定處理 socket 連線事件
        serverSocket.register(sel, SelectionKey.OP_ACCEPT,this);
        init();
        System.out.println("mainReactor-" + "Acceptor: Listening on port: " + port);
    }
    public void init() throws IOException {
        for (int i = 0; i < subReactors.length; i++) {
            subReactors[i]=new Reactor();
            subReactorExecutor.execute(subReactors[i]);
        }
    }
    @Override
    public synchronized void run() {
        try {
            // 接收連線,非阻塞模式下,沒有連線直接返回 null
            SocketChannel sc = serverSocket.accept();
            if (sc != null) {
                // 把提示發到介面
                sc.write(ByteBuffer.wrap("Multiply Reactor Pattern Example\r\nreactor> ".getBytes()));
                System.out.println(Thread.currentThread().getName()+":Main-Reactor-Acceptor: " + sc.socket().getLocalSocketAddress() +" 註冊到 subReactor-" + handleNext);
                // 如何解決呢,直接呼叫 wakeup,有可能還沒有註冊成功又阻塞了。這是一個多執行緒同步的問題,可以藉助佇列進行處理
                Reactor subReactor = subReactors[handleNext];
                subReactor.register(new AsyncHandler(sc));
                if(++handleNext == subReactors.length) {
                    handleNext = 0;
                }
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }
}

AsyncHandler

public class AsyncHandler implements Runnable{
    private SocketChannel channel;
    private SelectionKey sk;

    ByteBuffer inputBuffer=ByteBuffer.allocate(1024);
    ByteBuffer outputBuffer=ByteBuffer.allocate(1024);
    StringBuilder builder=new StringBuilder(); //儲存客戶端的完整訊息

    public AsyncHandler(SocketChannel channel){
        this.channel=channel;
    }

    public SocketChannel getChannel() {
        return channel;
    }

    public void setSk(SelectionKey sk) {
        this.sk = sk;
    }

    @Override
    public void run() {
        try {
            if (sk.isReadable()) {
                read();
            } else if (sk.isWritable()) {
                write();
            }
        }catch (Exception e){
            try {
                this.sk.channel().close();
            } catch (IOException ioException) {
                ioException.printStackTrace();
            }
        }
    }

    protected void read() throws IOException {
        inputBuffer.clear();
        int n=channel.read(inputBuffer);
        if(inputBufferComplete(n)){
            System.out.println(Thread.currentThread().getName()+":Server端收到客戶端的請求訊息:"+builder.toString());
            outputBuffer.put(builder.toString().getBytes(StandardCharsets.UTF_8));
            this.sk.interestOps(SelectionKey.OP_WRITE); //更改服務的邏輯狀態以及處理的事件型別
        }
    }

    private boolean inputBufferComplete(int bytes) throws EOFException {
        if(bytes>0){
            inputBuffer.flip(); //轉化成讀取模式
            while(inputBuffer.hasRemaining()){ //判斷緩衝區中是否還有元素
                byte ch=inputBuffer.get(); //得到輸入的字元
                if(ch==3){ //表示Ctrl+c 關閉連線
                    throw new EOFException();
                }else if(ch=='\r'||ch=='\n'){ //表示換行符
                    return true;
                }else{
                    builder.append((char)ch); //拼接讀取到的資料
                }
            }
        }else if(bytes==-1){
            throw new EOFException(); //客戶端關閉了連線
        }
        return false;
    }

    private void write() throws IOException {
        int written=-1;
        outputBuffer.flip(); //轉化為讀模式,判斷是否有資料需要傳送
        if(outputBuffer.hasRemaining()){
            written=channel.write(outputBuffer); //把資料寫回客戶端
        }
        outputBuffer.clear();
        builder.delete(0,builder.length());
        if(written<=0){ //表示客戶端沒有輸資訊
            this.sk.channel().close();
        }else{
            channel.write(ByteBuffer.wrap("\r\nreactor>".getBytes()));
            this.sk.interestOps(SelectionKey.OP_READ);
        }
    }
}

關注[跟著Mic學架構]公眾號,獲取更多精品原創

相關文章