forward_server.cc

//
//
// Tencent is pleased to support the open source community by making tRPC available.
//
// Copyright (C) 2023 THL A29 Limited, a Tencent company.
// All rights reserved.
//
// If you have downloaded a copy of the tRPC source code from Tencent,
// please note that tRPC source code is licensed under the  Apache 2.0 License,
// A copy of the Apache 2.0 License is included in this file.
//
//

#include <memory>
#include <string>

#include "fmt/format.h"

#include "examples/features/prometheus/proxy/forward_service.h"
#include "trpc/common/trpc_app.h"

namespace examples::forward {

class ForwardServer : public ::trpc::TrpcApp {
 public:
  int Initialize() override {
    const auto& config = ::trpc::TrpcConfig::GetInstance()->GetServerConfig();
    // Set the service name, which must be the same as the value of the `server:service:name` configuration item
    // in the framework configuration file, otherwise the framework cannot receive requests normally
    std::string service_name = fmt::format("{}.{}.{}.{}", "trpc", config.app, config.server, "Forward");

    TRPC_FMT_INFO("service name:{}", service_name);

#ifdef TRPC_BUILD_INCLUDE_PROMETHEUS
    RegisterService(service_name, std::make_shared<ForwardServiceImpl>());
#endif

    return 0;
  }

  void Destroy() override {}
};

}  // namespace examples::forward

int main(int argc, char** argv) {
  examples::forward::ForwardServer forward_server;
  forward_server.Main(argc, argv);
  forward_server.Wait();

  return 0;
}