【Flink集群RPC通讯机制（三）】AkkaRpcActor设计与实现：接收RPC消息以及处理逻辑

RPC请求发送后接收方的处理逻辑

在RpcEndpoint中创建的RemoteRpcInvocation消息，最终会通过Akka系统传递到被调用方。例如TaskExecutor向ResourceManager发送SlotReport请求的时候，会在TaskExecutor中将ResourceManagerGateway的方法名称和参数打包成RemoteRpcInvocation对象。然后经过网络发送到ResourceManager中的AkkaRpcActor，处理请求。

接下来深入了解AkkaRpcActor的设计与实现，了解在AkkaRpcActor中如何接收RemoteRpcInvocation消息并执行后续的操作。

 

1. 创建Receiver

如代码所示，首先在AkkaRpcActor中创建Receive对象，用于处理Akka系统接收的其他Actor发送过来的消息。

Receiver相关能力

在AkkaRpcActor中主要创建了RemoteHandshakeMessage、ControlMessages等消息对应的处理器，

• 其中RemoteHandshakeMessage主要用于进行正式RPC通信之前的网络连接检测，保障RPC通信正常。
• ControlMessages用于控制Akka系统，例如启动和停止Akka Actor等控制消息。这里我们重点关注第三种类型的消息，即在集群运行时中RPC组件通信使用的Message类型，此时会调用handleMessage()方法对这类消息进行处理。

public Receive createReceive() {
   
   
    return ReceiveBuilder.create()
        .match(RemoteHandshakeMessage.class, this::handleHandshakeMessage)
        .match(ControlMessages.class, this::handleControlMessage)
        .matchAny(this::handleMessage)
        .build();
}

 

2. 进行消息处理

在AkkaRpcActor.handleMessage()方法中，最终会调用handleRpcMessage()方法继续对RPC消息进行处理。

如下代码：

//根据RPC消息类型，进行不同方式处理
protected void handleRpcMessage(Object message) {