cpp/rocketmq-client-cpp-tests/cpp4.x/src/frame/BaseOperate.cpp

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "frame/BaseOperate.h"

extern std::shared_ptr<spdlog::logger> multi_logger;

std::string getTopic(MessageType messageType, const std::string &methodName, const std::string &brokerAddr, const std::string &namesrvAddr, const std::string &cluster)
{
  std::string messageTypeStr = MessageTypeToString[messageType];
  std::string topic = "topic_" + messageTypeStr + "_" + methodName + "_" + RandomUtils::getStringWithCharacter(6);
  multi_logger->info("[Topic] topic:{}, messageType:{}, methodName:{}", topic, messageTypeStr, methodName);
  if (messageType == MessageType::NORMAL)
  {
    MQAdminUtils::createTopic(topic, brokerAddr, "", namesrvAddr);
  }
  else if (messageType == MessageType::DELAY)
  {
    MQAdminUtils::createDelayTopic(topic, brokerAddr, "", namesrvAddr);
  }
  else if (messageType == MessageType::FIFO)
  {
    MQAdminUtils::createFIFOTopic(topic, brokerAddr, "", namesrvAddr);
  }
  else if (messageType == MessageType::TRANSACTION)
  {
    MQAdminUtils::createTransactionTopic(topic, brokerAddr, "", namesrvAddr);
  }

  return topic;
}

std::string getGroupId(const std::string &methodName)
{
  std::string randomStr = "";
  static const char CHARACTERS[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
  static const int STRING_LENGTH = 6;
  std::random_device rd;
  std::mt19937 gen(rd());
  std::uniform_int_distribution<> dis(0, sizeof(CHARACTERS) - 2);
  for (int i = 0; i < STRING_LENGTH; i++)
  {
    randomStr += CHARACTERS[dis(gen)];
  }
  std::string groupId = "GID_" + methodName + "_" + randomStr;
  std::cout << "[ConsumerGroupId] groupId:" << groupId << ", methodName:" << methodName << std::endl;
  return groupId;
}