RocketMq源码解读（四）Topic创建

上一章内容介绍了MQ Producer如何启动，并且向所有的broker注册Group的过程，本章内容主要集中在如何创建消息的topic。

 

一、topic的创建方法 

public class DefaultMQProducer extends ClientConfig implements MQProducer{
    // 所有的实际操作委托给这个实现
    protected final transient DefaultMQProducerImpl defaultMQProducerImpl;
    // 默认topic队列数
    private volatile int defaultTopicQueueNums = 4;
    private int sendMsgTimeout = 3000;
    private int compressMsgBodyOverHowmuch = 1024 * 4;
    private int retryTimesWhenSendFailed = 2;
    private int retryTimesWhenSendAsyncFailed = 2;
    private boolean retryAnotherBrokerWhenNotStoreOK = false;
    private int maxMessageSize = 1024 * 1024 * 4;
    ...
    @Override
    public void createTopic(String key, String newTopic, int queueNum) throws MQClientException {
        createTopic(key, newTopic, queueNum, 0);
    }
    @Override
    public void createTopic(String key, String newTopic, int queueNum, int topicSysFlag) throws MQClientException {
        this.defaultMQProducerImpl.createTopic(key, newTopic, queueNum, topicSysFlag);
    }
    ...
}

二、Topic创建的具体实现

1、委托给内部DefaultMQProducerImpl.create

public void createTopic(String key, String newTopic, int queueNum) throws MQClientException {
    createTopic(key, newTopic, queueNum, 0);
}
public void createTopic(String key, String newTopic, int queueNum, int topicSysFlag) throws MQClientException {
    this.makeSureStateOK();
    Validators.checkTopic(newTopic);
    this.mQClientFactory.getMQAdminImpl().createTopic(key, newTopic, queueNum, topicSysFlag);
}

2、makeSureStateOk 主要就是判断Producer是否是Running状态

checkTopic会校验topic是否合法

3、获取producer启动时生成的MQAdminImpl实例，里面提供了topic创建的进一步实现

4、MQAdminImpl创建topic

会尝试从NameSrv获取broker的地址信息，注意，这里因为同一个brokerName可能有多个broker，会选取同一个brokerName中的主broker，发送创建topic的信息

public void createTopic(String key, String newTopic, int queueNum, int topicSysFlag) throws MQClientException {
    try {
        // 从NameSrv获取topic对应的路由信息
        TopicRouteData topicRouteData = this.mQClientFactory.getMQClientAPIImpl().getTopicRouteInfoFromNameServer(key, timeoutMillis);
        List<BrokerData> brokerDataList = topicRouteData.getBrokerDatas();
        // 
        if (brokerDataList != null && !brokerDataList.isEmpty()) {
           Collections.sort(brokerDataList);
           boolean createOKAtLeastOnce = false;
           MQClientException exception = null;
           StringBuilder orderTopicString = new StringBuilder();
           // 从拉取到的broker信息，遍历
           for (BrokerData brokerData : brokerDataList) {
               // 获取master的addr
               String addr = brokerData.getBrokerAddrs().get(MixAll.MASTER_ID);
               if (addr != null) {
                   TopicConfig topicConfig = new TopicConfig(newTopic);
                   topicConfig.setReadQueueNums(queueNum);
                   topicConfig.setWriteQueueNums(queueNum);
                   topicConfig.setTopicSysFlag(topicSysFlag);
                   boolean createOK = false;
                   for (int i = 0; i < 5; i++) {
                       try {           
                            this.mQClientFactory.getMQClientAPIImpl().createTopic(addr, key, topicConfig, timeoutMillis);
                            createOK = true;
                            createOKAtLeastOnce = true;
                            break;
                         } catch (Exception e) {
                            if (4 == i) {
                               exception = new MQClientException("create topic to broker exception", e);
                            }
                         }
                   }
                   if (createOK) {
                            orderTopicString.append(brokerData.getBrokerName());
                            orderTopicString.append(":");
                            orderTopicString.append(queueNum);
                            orderTopicString.append(";");
                   }
               }
               if (exception != null && !createOKAtLeastOnce) {
                    throw exception;
               }
          } else {
              throw new MQClientException("Not found broker, maybe key is wrong", null);
          }         
    }catch (Exception e) {
        throw new MQClientException("create new topic failed", e);
    }
}

public void createTopic(final String addr, final String defaultTopic, final TopicConfig topicConfig, final long timeoutMillis)
        throws RemotingException, MQBrokerException, InterruptedException, MQClientException {
        CreateTopicRequestHeader requestHeader = new CreateTopicRequestHeader();
        requestHeader.setTopic(topicConfig.getTopicName());
        requestHeader.setDefaultTopic(defaultTopic);
        requestHeader.setReadQueueNums(topicConfig.getReadQueueNums());
        requestHeader.setWriteQueueNums(topicConfig.getWriteQueueNums());
        requestHeader.setPerm(topicConfig.getPerm());
        requestHeader.setTopicFilterType(topicConfig.getTopicFilterType().name());
        requestHeader.setTopicSysFlag(topicConfig.getTopicSysFlag());
        requestHeader.setOrder(topicConfig.isOrder());

        RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.UPDATE_AND_CREATE_TOPIC, requestHeader);

        RemotingCommand response = this.remotingClient.invokeSync(MixAll.brokerVIPChannel(this.clientConfig.isVipChannelEnabled(), addr),
            request, timeoutMillis);
        assert response != null;
        switch (response.getCode()) {
            case ResponseCode.SUCCESS: {
                return;
            }
            default:
                break;
        }

        throw new MQClientException(response.getCode(), response.getRemark());
    }

broker收到创建topic的消息后RequestCode.UPDATE_AND_CREATE_TOPIC，会建立broker和topic的关系，然后向NameSrv服务注册变化信息。