代码很少，却很优秀！RocketMQ的NameServer是如何做到的？-51CTO.COM

今天我们来一起深入分析 RocketMQ的注册中心 NameServer。

本文基于 RocketMQ release-5.2.0。

首先，我们回顾下 RocketMQ的内核原理鸟瞰图：

从上面的鸟瞰图，我们可以看出：Nameserver既和 Broker交互，也和 Producer和 Consumer交互，因此，在 RocketMQ中，Nameserver起到了一个纽带性的作用。

接着，我们再看看 NameServer的工程结构，如下图：

整个工程只有 11个类（老版本好像只有不到 10个类），为什么 RocketMQ可以用如此少的代码，设计出如此高性能且轻量的注册中心？

我觉得最核心的 3个点是：

AP设计思想
简单的数据结构
心跳机制

一、AP设计思想

像 ZooKeeper，采用了 Zab (Zookeeper Atomic Broadcast) 这种比较重的协议，必须大多数节点（过半数）可用，才能确保了数据的一致性和高可用，大大增加了网络开销和复杂度。

而 NameServer遵守了 CAP理论中 AP，在一个 NameServer集群中，NameServer节点之间是P2P(Peer to Peer)的对等关系，并且 NameServer之间并没有通信，减少很多不必要的网络开销，即便只剩一个 NameServer节点也能继续工作，足以保证高可用。

二、数据结构

NameServer维护了一套比较简单的数据结构，内部维护了一个路由表，该路由表包含以下几个核心元数据，对应的源码类RouteInfoManager如下：

public class RouteInfoManager {
    private final static long DEFAULT_BROKER_CHANNEL_EXPIRED_TIME = 1000 * 60 * 2; // broker失效时间 120s
    private final ReadWriteLock lock = new ReentrantReadWriteLock();
    private final Map<String/* topic */, Map<String, QueueData>> topicQueueTable;
    private final Map<String/* brokerName */, BrokerData> brokerAddrTable;
    private final Map<String/* clusterName */, Set<String/* brokerName */>> clusterAddrTable;
    private final Map<BrokerAddrInfo/* brokerAddr */, BrokerLiveInfo> brokerLiveTable;
    private final Map<BrokerAddrInfo/* brokerAddr */, List<String>/* Filter Server */> filterServerTable;
}

topicQueueTable：Topic消息队列路由信息，消息发送时根据路由表进行负载均衡
brokerAddrTable：Broker基础信息，包括brokerName、所属集群名称、主备Broker地址
clusterAddrTable：Broker集群信息，存储集群中所有Broker名称
brokerLiveTable：Broker状态信息，NameServer每次收到心跳包会替换该信息
filterServerTable：Broker上的FilterServer列表，用于过滤标签（Tag）或 SQL表达式，以减轻 Consumer的负担，提高消息消费的效率。

1.TopicRouteData

TopicRouteData是 NameServer中最重要的数据结构之一，它包括了 Topic对应的所有 Broker信息以及每个 Broker上的队列信息，filter服务器列表，其源码如下：

public class TopicRouteData {
    private List<QueueData> queueDatas;
    private List<BrokerData> brokerDatas;
    private HashMap<String, List<String>> filterServerTable;
    //It could be null or empty
    private Map<String/*brokerName*/, TopicQueueMappingInfo> topicQueueMappingByBroker;
}

2.BrokerData

BrokerData包含了 Broker的基本属性，状态，所在集群以及 Broker服务器的 IP地址，其源码如下：

public class BrokerData {
    private String cluster;//所在的集群
    private String brokerName;//所在的brokerName
    private HashMap<Long, String> brokerAddrs;//该broker对应的机器IP列表
    private String zoneName; // 区域名称
}

3.QueueData

QueueData包含了 BrokerName，readQueue的数量，writeQueue的数量等信息，对应的源码类是QueueData，其源码如下：

public class QueueData {
    private String brokerName;//所在的brokerName
    private int readQueueNums;// 读队列数量
    private int writeQueueNums;// 写队列数量
    private int perm; // 读写权限，参考PermName 类
    private int topicSysFlag; // topic同步标记，参考TopicSysFlag 类
}

4.元数据举例

为了更好地理解元数据，这里对每一种元数据都给出一个数据实例：

topicQueueTable:{
    "topicA":[
        {
            "brokeName":"broker-a",
            "readQueueNums":4,
            "writeQueueNums":4,
            "perm":6, 
            "topicSyncFlag":0 
        },
        {
            "brokeName":"broker-b",
            "readQueueNums":4,
            "writeQueueNums":4,
            "perm":6, 
            "topicSyncFlag":0
        }
    ],
    "topicB":[]
}

brokeAddrTable:{
    "broker-a":{
        "cluster":"cluster-1",
        "brokerName":"broker-a",
        "brokerAddrs":{
            0:"192.168.0.1:8000",
            1:"192.168.0.2:8000"
        }
    },
    "broker-b":{
        "cluster":"cluster-1",
        "brokerName":"broker-b",
        "brokerAddrs":{
            0:"192.168.0.3:8000",
            1:"192.168.0.4:8000"
        }
    }
}

三、心跳机制

心跳机制是 NameServer维护 Broker的路由信息最重要的一个抓手，主要分为接收心跳、处理心跳、心跳超时 3部分：

1.接收心跳

Broker每 30s会向所有的 NameServer发送心跳包，告诉它们自己还存活着，从而更新自己在 NameServer的状态，整体交互如下图：

2.处理心跳

NameServer收到心跳包时会更新 brokerLiveTable缓存中 BrokerLiveInfo的 lastUpdateTimeStamp信息，整体交互如下图：

处理逻辑可以参考源码：org.apache.rocketmq.namesrv.processor.DefaultRequestProcessor#processRequest#brokerHeartbeat:

public RemotingCommand brokerHeartbeat(ChannelHandlerContext ctx,
    RemotingCommand request) throws RemotingCommandException {
    final RemotingCommand response = RemotingCommand.createResponseCommand(null);
    final BrokerHeartbeatRequestHeader requestHeader =
        (BrokerHeartbeatRequestHeader) request.decodeCommandCustomHeader(BrokerHeartbeatRequestHeader.class);

    this.namesrvController.getRouteInfoManager().updateBrokerInfoUpdateTimestamp(requestHeader.getClusterName(), requestHeader.getBrokerAddr());

    response.setCode(ResponseCode.SUCCESS);
    response.setRemark(null);
    return response;
}

3.心跳超时

NameServer每隔 10s(每隔5s + 5s延迟)扫描 brokerLiveTable检查 Broker的状态，如果在 120s内未收到 Broker心跳，则认为 Broker异常，会从路由表将该 Broker摘除并关闭 Socket连接，同时还会更新路由表的其他信息，整体交互如下图：

private void startScheduleService() {
this.scanExecutorService.scheduleAtFixedRate(NamesrvController.this.routeInfoManager::scanNotActiveBroker,
        5, this.namesrvConfig.getScanNotActiveBrokerInterval(), TimeUnit.MILLISECONDS);
}

源码参考：org.apache.rocketmq.namesrv.routeinfo.RouteInfoManager#unRegisterBroker()，核心流程：

遍历brokerAddrTable
遍历broker地址
根据 broker地址移除 brokerAddr
如果当前 Topic只包含待移除的 Broker,则移除该 Topic

四、其他核心源码解读

NameServer启动

NameServer的启动类为：org.apache.rocketmq.namesrv.NamesrvStartup，整个流程如下图：

NameServer启动最核心的 3个事情是：

加载配置：NameServerConfig、NettyServerConfig主要是映射配置文件，并创建 NamesrvController。
启动 Netty通信服务：NettyRemotingServer是 NameServer和Broker，Producer，Consumer通信的底层通道 Netty服务器。
启动定时器和钩子程序：NameServerController实例一方面处理 Netty接收到消息后，一方面内部有多个定时器和钩子程序，它是 NameServer的核心控制器。

五、总结

NameServer并没有采用复杂的分布式协议来保持数据的一致性，而是采用 CAP理论中的 AP，各个节点之间是Peer to Peer的对等关系，数据的一致性通过心跳机制，定时器，延时感知来完成。