src/tracelib.cpp

   1 /*
   2  * Copyright (c) 2019 AT&T Intellectual Property.
   3  * Copyright (c) 2018-2019 Nokia.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 #include "tracelibcpp.hpp"
  19 #include "config.hpp"
  20
  21 using namespace tracelibcpp;
  22
  23 std::string ConfMaker::getEnv(const char* envName, std::string defVal)
  24 {
  25     const char *ev = getenv(envName);
  26     if (!ev)
  27         return defVal;
  28     return std::string(ev);
  29 }
  30
  31 bool ConfMaker::isTracingEnabled()
  32 {
  33     std::string envValue = getEnv(TRACING_ENABLED_ENV, "false");
  34     if (envValue == "1" || boost::iequals(envValue, "true"))
  35         return true;
  36     else
  37         return false;
  38 }
  39
  40 jaegertracing::Config ConfMaker::makeNopTraceConfig()
  41 {
  42     return jaegertracing::Config(true,
  43             jaegertracing::samplers::Config("const", 0));
  44 }
  45
  46 jaegertracing::samplers::Config ConfMaker::getSamplerConfig()
  47 {
  48     std::string samplerType = getEnv(JAEGER_SAMPLER_TYPE_ENV, "const");
  49     // Use value 0.001 as default param, same way as jaeger does it
  50     double param = atof(getEnv(JAEGER_SAMPLER_PARAM_ENV, "0.001").c_str());
  51     return jaegertracing::samplers::Config(samplerType, param);
  52 }
  53
  54 jaegertracing::reporters::Config ConfMaker::getReporterConfig()
  55 {
  56     std::string agentHostPort = getEnv(JAEGER_AGENT_ADDR_ENV, jaegertracing::reporters::Config::kDefaultLocalAgentHostPort);
  57
  58     if (agentHostPort.find(':') == std::string::npos)
  59         agentHostPort += ":6831";
  60
  61     return jaegertracing::reporters::Config(
  62         0, std::chrono::seconds(0),     // use jaeger defaults
  63         getLoggingLevel() == LOG_ALL,   // log spans
  64         agentHostPort
  65     );
  66 }
  67
  68 LogLevel ConfMaker::getLoggingLevel()
  69 {
  70     std::string logLevel = getEnv(JAEGER_LOG_LEVEL_ENV, "error");
  71     if (boost::iequals(logLevel, "all"))
  72         return LOG_ALL;
  73     else if (boost::iequals(logLevel, "error"))
  74         return LOG_ERR;
  75     else
  76         return LOG_NONE;
  77 }
  78
  79 std::unique_ptr<jaegertracing::logging::Logger> ConfMaker::getLogger()
  80 {
  81     switch (getLoggingLevel())
  82     {
  83         case LOG_ALL:
  84         case LOG_ERR:
  85             return jaegertracing::logging::consoleLogger();
  86             break;
  87         default:
  88             return jaegertracing::logging::nullLogger();
  89     }
  90 }
  91
  92 jaegertracing::Config ConfMaker::getTraceConfig()
  93 {
  94     if (!isTracingEnabled())
  95         return makeNopTraceConfig();
  96     auto sampler = getSamplerConfig();
  97     auto reporter = getReporterConfig();
  98     return jaegertracing::Config(false, sampler, reporter);
  99 }
 100
 101 std::shared_ptr<opentracing::Tracer> tracelibcpp::createTracer(std::string serviceName)
 102 {
 103     auto cm = ConfMaker(serviceName);
 104     auto config = cm.getTraceConfig();
 105     try {
 106         return jaegertracing::Tracer::make(serviceName, config, cm.getLogger());
 107     } catch (std::exception& e)
 108     {
 109         if (cm.getLoggingLevel() != LOG_NONE)
 110             std::cerr << "Cannot create tracer: " << e.what() << std::endl;
 111         return jaegertracing::Tracer::make(serviceName, cm.makeNopTraceConfig());
 112     }
 113 }
 114
 115