你說說RPC的一個請求的流程是怎麼樣的?

紀莫發表於2021-11-05

前言

面試的時候經常被問到RPC相關的問題,例如:你說說RPC實現原理、讓你實現一個RPC框架應該考慮哪些地方、RPC框架基礎上發起一個請求是怎樣一個流程等等。所以這次我就總結一波RPC的相關知識點,提前說明一下,本篇文章只是為了回答一些面試問題,所以只是解釋原理,並不會深入挖掘細節。

註冊中心

RPC(Remote Procedure Call)翻譯成中文就是$\color{red}{遠端過程呼叫}$。RPC框架起到的作用就是為了實現,呼叫遠端方法時,能夠做到和呼叫本地方法一樣,讓開發人員更專注於業務開發,不用去考慮網路程式設計等細節。

RPC框架怎麼就實現不讓開發人員關注網路程式設計等細節呢?

首先我們區分兩個角色一個服務提供方,一個是服務呼叫方。服務呼叫方其實是通過動態代理、負載均衡、網路呼叫等機制去服務提供方的機器上去執行對應的方法。服務提供方將方法執行完成後,將執行結果再通過網路傳輸返回到服務提供方。
大致過程如下:

遠端呼叫過程

但是現在的服務都是叢集部署,那麼服務呼叫方怎麼應該實時的知道服務提供方的叢集中的變化,例如服務提供方的IP地址變了,或者是服務重啟時怎麼能夠及時的切換流量呢?

這就需要$\color{red}{註冊中心}$起作用了,我們可以把註冊中心看作服務端,然後每個服務都看成客戶端,每個客戶端都需要將自己註冊到註冊中心,然後一個服務呼叫方要呼叫另一個服務時,需要從註冊中心獲取服務提供方的資訊,主要是獲取服務提供方的伺服器IP地址列表和埠資訊

服務呼叫方獲取到這些資訊後快取到自己本地,並且跟註冊中心保持一個長連線當服務提供方有任何變化時,註冊中心能夠實時的通知給服務呼叫方,呼叫方能夠及時更新自己本地快取的資訊(也可以採用定時輪詢的方式)。

服務呼叫方獲取到伺服器IP地址資訊後,根據自己的負載均衡策略選擇一個IP地址然後發起網路呼叫的請求。

註冊中心

那麼網路客戶端是通過什麼發起的網路呼叫呢?

可以自己使用JDK原生的BIO活NIO來實現一套網路通訊模組,但是這裡我們建議直接使用強大的網路通訊框架Netty。它是基於NIO的網路通訊框架,支援高併發,封裝完善,而且效能好傳輸快。

Netty不是我們本文的主要內容,這裡就不展開說了。

客戶端呼叫過程

因為我們知道資料在網路中傳輸的時候都是以二進位制的形式的,所以在呼叫方將呼叫的引數進行傳遞的時候是需要進行序列化的。服務提供方在接收到引數時也是需要進行反序列化的。

網路協議

呼叫方既然需要序列化,服務提供方又要進行反序列化,這樣雙方就要確定好一個協議,呼叫方傳輸什麼引數,服務提供方就按照這個協議去進行解析,而且在返回結果的時候也是按照這個協議進行結果解析。

那麼這個協議應該是怎麼樣的結構,都是什麼樣子的呢?
因為這個協議可以自定義,我們為了方便就以JSON的形式給舉個例子:

{
	"interfaces": "interface=com.jimoer.rpc.test.producer.TestService;method=printTest;parameter=com.jiomer.rpc.test.producer.TestArgs",
	"requestId": "3",
	"parameter": {
		"com.jiomer.rpc.test.producer.TestArgs": {
			"age": 20,
			"name": "Jimoer"
		}
	}
}

首先第一個引數interfaces是,我們要讓服務提供方知道呼叫方要呼叫哪個介面,以及介面中的哪個方法,並且方法的引數是什麼型別的。

第二個引數是當前一次請求的一個唯一標識,在多個執行緒同時請求一個方法時,用這個id來進行區分,以後無論是做鏈路追蹤還是日誌管理都可以以此id為依據。

第三個引數就是 實際的呼叫方法中的引數值。具體是什麼型別的,每個屬性值都是什麼。

呼叫

下面也是舉一個簡單的例子來說明一下呼叫的過程。我們一部分採用程式碼的形式一部分採用文字的形式來將整個呼叫過程串起來。

// 定義請求的URL
String tcpURL = "tcp://testProducer/TestServiceImpl";
// 定義介面請求
TestService testService = ProxyFactory.create(TestService.class, tcpURL);
// 組裝請求引數
TestArgs testArgs = new TestArgs(20,"Jimoer");
// 通過動態代理執行請求
String result = testService.printTest(testArgs);

通過檢視上面的程式碼我們可以看到整個呼叫過程最核心的地方在ProxyFactory.create()方法裡,這個方法裡面主要的過程是,動態代理生成介面的實際代理物件,然後使用Netty的介面發起網路請求。

Proxy.newProxyInstance(getClass().getClassLoader(), interfaces.getClass().getInterfaces(), new InvocationHandler() {
            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

                // 第一步:獲取呼叫服務的地址列表
                ListregistryInfos = interfacesMethodRegistryList.get(clazz);

                if (registryInfos == null) {
                    throw new RuntimeException("無法找到服務提供者");
                }

                // 第二步: 通過自身的負載均衡策略選擇一個地址
                RegistryInfo registryInfo = loadBalancer.choose(registryInfos);

                // 第三步:Netty的網路請求處理
                ChannelHandlerContext ctx = channels.get(registryInfo);
                // 第四步:根據介面類的全路徑名和方法生成唯一標識
                String identify = InvokeUtils.buildInterfaceMethodIdentify(clazz, method);
                String requestId;
                // 第五步:通過加鎖的方式保證生成的requestId的唯一性
                synchronized (ApplicationContext.this) {
                    requestIdWorker.increment();
                    requestId = String.valueOf(requestIdWorker.longValue());
                }
                // 第六步: 組織引數
                JSONObject jsonObject = new JSONObject();
                jsonObject.put("interfaces", identify);
                jsonObject.put("parameter", param);
                jsonObject.put("requestId", requestId);
                System.out.println("傳送給服務端JSON為:" + jsonObject.toJSONString());
                // $$ 多條訊息之間的分隔符
                String msg = jsonObject.toJSONString() + "$$";
                ByteBuf byteBuf = Unpooled.buffer(msg.getBytes().length);
                byteBuf.writeBytes(msg.getBytes());
                // 第七步:這裡發起呼叫
                ctx.writeAndFlush(byteBuf);
                // 這裡會將執行緒進行阻塞,知道服務提供方將請求處理好之後返回結果,再喚醒。
                waitForResult();
                return result;

            }
        });

執行過程大致分為這幾步:

  1. 獲取呼叫服務的地址列表。
  2. 通過自身的負載均衡策略選擇一個地址。
  3. Netty的網路請求處理(選擇一個渠道Channel)。
  4. 根據介面類的全路徑名和方法生成唯一標識。
  5. 通過加鎖的方式保證生成的requestId的唯一性。
  6. 組織請求引數。
  7. 發起呼叫。
  8. 執行緒阻塞,直到服務提供方返回結果。
  9. 填充返回結果,返回到呼叫方。

服務端處理過程

上面也說了,服務呼叫方發起網路請求後,會阻塞住,直到服務提供方返回資料,所以服務提供方處理完呼叫方法的邏輯後,還是要喚醒阻塞的呼叫執行緒的。

服務提供方在處理請求時也是先通過Netty獲取到資料,然後再進行反序列化,然後再根據協議獲取到需要呼叫的方法,然後通過反射去進行呼叫。

Netty的返回入口在下面這部分邏輯裡

@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
    try {
        String message = (String) msg;
        if (messageCallback != null) {
        	// 將接收到的訊息放到回撥方法中
            messageCallback.onMessage(message);
        }
    } finally {
        ReferenceCountUtil.release(msg);
    }
}

Netty的client接收到響應的訊息後,先將結果返回到呼叫方,處理完成之後再去釋放之前的阻塞呼叫執行緒。

client.setMessageCallback(message -> {
    // 這裡收單服務端返回的訊息,先壓入佇列
    RpcResponse response = JSONObject.parseObject(message, RpcResponse.class);
	System.out.println("收到一個響應:" + response);
    String interfaceMethodIdentify = response.getInterfaceMethodIdentify();
    String requestId = response.getRequestId();
    // 設定唯一標識
    String key = interfaceMethodIdentify + "#" + requestId;
    Invoker invoker = inProgressInvoker.remove(key);
    // 將結果設定到代理物件中
    invoker.setResult(response.getResult());
	// 加鎖再釋放之前的阻塞執行緒。
    synchronized (ApplicationContext.this) {
        ApplicationContext.this.notifyAll();
    }
});

setResult()方法

@Override
public void setResult(String result) {
    synchronized (this) {
        this.result = JSONObject.parseObject(result, returnType);
        notifyAll();
    }
}

上面的步驟就是這樣,按照之前請求的唯一標識放入到返回的資訊中,然後將結果設定到代理物件中,再通過返回結果,然後喚醒之前的呼叫阻塞執行緒。

總結

其實整個RPC的請求過程就是如下(不含非同步呼叫):

呼叫過程
做一個總結,用大白話把一個RPC請求流程描述出來:
首先無論是呼叫方還是服務提供方都要註冊到註冊中心;

  1. 服務呼叫方把請求引數物件序列化成二進位制資料,通過動態代理生成代理物件,通過代理物件,使用Netty選擇一個從註冊中心拉取到的服務提供方的地址,然後發起網路請求。
  2. 服務提供方從TCP通道中接收到二進位制資料,根據定義的RPC網路協議,從二進位制資料中反序列化後,分割出介面地址和引數物件,再通過反射找到介面執行呼叫。
  3. 然後服務提供方再把呼叫執行結果序列化後,回傳到TCP通道中。
  4. 服務呼叫方獲取到應答二進位制資料後,再反序列化成結果物件。

這樣就完成了一次RPC網路呼叫,其實後面框架擴充套件後,還要考慮限流、熔斷、服務降級、序列化多樣性擴充套件,服務監控、鏈路追蹤等等功能。這些就要後面再擴充套件的講了,這次就先到這了。

參考:
如何設計一個短小精悍、可擴充的RPC框架?(含實現程式碼)
一篇文章瞭解RPC框架原理

相關文章