Basic Jaeger Tracing Added

Signed-off-by: Abhinav Singh <[email protected]>

src/common/jaegerTracer.h

@@ -0,0 +1,26 @@
+
+
+#ifndef JAEGER_TRACER_
+#define JAEGER_TRACER_
+
+#include <memory>
+#include<jaegertracing/Tracer.h>
+#include <opentracing/tracer.h>
+
+class jTracer
+{
+private:
+    static jaegertracing::Config configuration;
+public:
+    jTracer() {}
+    static void loadYamlConfigFile(const char *);
+    void initTracer(const char *);
+    std::unique_ptr<opentracing::Span> newSpan(const char *);
+    std::unique_ptr<opentracing::Span> childSpan(const char *, const std::unique_ptr<opentracing::Span> &);
+    std::unique_ptr<opentracing::Span> followUpSpan(const char *, const std::unique_ptr<opentracing::Span> &);
+    ~jTracer(){
+        opentracing::Tracer::Global()->Close();
+    }
+};
+
+#endif

src/common/jaegerTracing.cc

@@ -0,0 +1,34 @@
+
+#include <yaml-cpp/yaml.h>
+#include <iostream>
+#include "jaegerTracer.h"
+
+void jTracer::loadYamlConfigFile(const char *filePath)
+{
+    auto yaml = YAML::LoadFile(filePath);
+    jTracer::configuration = jaegertracing::Config::parse(yaml);
+}
+
+void jTracer::initTracer(const char *tracerName)
+{
+    auto tracer = jaegertracing::Tracer::make(
+        tracerName,
+        jTracer::configuration,
+        jaegertracing::logging::consoleLogger());
+    opentracing::Tracer::InitGlobal(
+        std::static_pointer_cast<opentracing::Tracer>(tracer));
+}
+std::unique_ptr<opentracing::Span> newSpan(const char * spanName){
+    auto Span=opentracing::Tracer::Global()->StartSpan(spanName);
+    return std::move(Span);
+}
+std::unique_ptr<opentracing::Span> jTracer::childSpan(const char *spanName, const std::unique_ptr<opentracing::Span> &parentSpan)
+{
+    auto Span = opentracing::Tracer::Global()->StartSpan(spanName, {opentracing::ChildOf(&parentSpan->context())});
+    return std::move(Span);
+}
+std::unique_ptr<opentracing::Span> jTracer::followUpSpan(const char *spanName, const std::unique_ptr<opentracing::Span> &parentSpan)
+{
+    auto Span = opentracing::Tracer::Global()->StartSpan(spanName, {opentracing::FollowsFrom(&parentSpan->context())});
+    return std::move(Span);
+}

src/rgw/rgw_aio.cc

@@ -49,9 +49,14 @@ Aio::OpFunc aio_abstract(Op&& op) {
   return [op = std::move(op)] (Aio* aio, AioResult& r) mutable {
       constexpr bool read = std::is_same_v<std::decay_t<Op>, librados::ObjectReadOperation>;
       auto s = new (&r.user_data) state(aio, r);
+      jTracer tracer;
       if constexpr (read) {
+        tracer.initTracer("Reading Started");
+        auto span=tracer.newSpan("Aio Read Operation started");
         r.result = r.obj.aio_operate(s->c, &op, &r.data);
       } else {
+        tracer.initTracer("Writing Started");
+        auto span=tracer.newSpan("Aio Write Operation started");
         r.result = r.obj.aio_operate(s->c, &op);
       }
       if (r.result < 0) {

src/rgw/rgw_aio.h

@@ -26,6 +26,7 @@
 #include "services/svc_rados.h" // cant forward declare RGWSI_RADOS::Obj
 
 #include "rgw_common.h"
+#include "common/jaegerTracer.h"
 
 #include "include/function2.hpp"
 

src/rgw/rgw_main.cc

@@ -38,6 +38,9 @@
 #include "rgw_frontend.h"
 #include "rgw_http_client_curl.h"
 #include "rgw_perf_counters.h"
+
+#include "common/jaegerTracer.h"
+
 #ifdef WITH_RADOSGW_AMQP_ENDPOINT
 #include "rgw_amqp.h"
 #endif
@@ -679,6 +682,9 @@ extern "C" {
 
 int radosgw_main(int argc, const char** argv)
 {
+  //loading the yaml configuration file for rgw
+  jTracer tracer;
+  tracer.loadYamlConfigFile("tracerConfiguration.yaml");
   return radosgw_Main(argc, argv);
 }