trpc_template_server.cc

// copyright skylanwei 2023-11-18

#include <string>

#include "trpc/common/trpc_app.h"
#include "trpc_template_service.h"
#include "trpc/util/log/default/default_log.h"
#include "trpc/util/log/default/sinks/local_file/local_file_sink.h"
#include "trpc/common/config/local_file_sink_conf.h"
#include "common/trace_id_formatter.h"
#include "common/include/opentelemetry_telemetry_api.h"


namespace trpc {
namespace sample {

class TrpcTemplateServer : 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 configuration file, otherwise the framework cannot receive requests normally.
    std::string service_name = fmt::format("{}.{}.{}.{}", "trpc", config.app, config.server, "TrpcTemplateService");
    TRPC_FMT_INFO("service name:{}", service_name);
    RegisterService(service_name, std::make_shared<TrpcTemplateServiceImpl>());

    return 0;
  }

  int RegisterPlugins() override {
    // Initializes the OpenTelemetry plugin and filters in RegisterPlugins
    //add spd custom flag
    Log* t = ::trpc::LogFactory::GetInstance()->Get().Get();
    DefaultLog* dpt = dynamic_cast<DefaultLog*>(t);
    dpt->SetCustomFlag<LocalFileSink, LocalFileSinkConfig, TraceIdFormatter>("default", 'q');

    ::trpc::opentelemetry::Init();
    return 0;
  }

  void Destroy() override {}

};

}  // sample 
}  // trpc 

int main(int argc, char** argv) {
  ::trpc::sample::TrpcTemplateServer server;
  server.Main(argc, argv);
  server.Wait();

  return 0;
}