HPCC-32691 Add optional timestamp arg to createClientSpan() 2

Signed-off-by: M Kelly <[email protected]>
hpcc-systems · Sep 19, 2024 · e6f0432 · e6f0432
1 parent efa28e8
commit e6f0432
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/system/jlib/jtrace.cpp b/system/jlib/jtrace.cpp
@@ -558,7 +558,7 @@ class CSpan : public CInterfaceOf<ISpan>
     }
 
     ISpan * createClientSpan(const char * name, const SpanTimeStamp * spanStartTimeStamp = nullptr) override;
-    ISpan * createInternalSpan(const char * name) override;
+    ISpan * createInternalSpan(const char * name, const SpanTimeStamp * spanStartTimeStamp = nullptr) override;
 
     virtual void endSpan() final override
     {
@@ -919,7 +919,7 @@ class CNullSpan final : public CInterfaceOf<ISpan>
     virtual const char* queryLocalId() const override { return nullptr; }
 
     virtual ISpan * createClientSpan(const char * name, const SpanTimeStamp * spanStartTimeStamp = nullptr) override { return getNullSpan(); }
-    virtual ISpan * createInternalSpan(const char * name) override { return getNullSpan(); }
+    virtual ISpan * createInternalSpan(const char * name, const SpanTimeStamp * spanStartTimeStamp = nullptr) override { return getNullSpan(); }
 
 private:
     CNullSpan(const CNullSpan&) = delete;
@@ -930,9 +930,14 @@ class CNullSpan final : public CInterfaceOf<ISpan>
 class CChildSpan : public CSpan
 {
 protected:
-    CChildSpan(const char * spanName, CSpan * parent)
+    CChildSpan(const char * spanName, CSpan * parent, const SpanTimeStamp *spanStartTimeStamp = nullptr)
     : CSpan(spanName), localParentSpan(parent)
     {
+        if (spanStartTimeStamp && spanStartTimeStamp->isInitialized())
+        {
+            opts.start_system_time = opentelemetry::common::SystemTimestamp(spanStartTimeStamp->systemClockTime);
+            opts.start_steady_time = opentelemetry::common::SteadyTimestamp(spanStartTimeStamp->steadyClockTime);
+        }
         injectlocalParentSpan(localParentSpan);
     }
 
@@ -984,8 +989,8 @@ class CChildSpan : public CSpan
 class CInternalSpan : public CChildSpan
 {
 public:
-    CInternalSpan(const char * spanName, CSpan * parent)
-    : CChildSpan(spanName, parent)
+    CInternalSpan(const char * spanName, CSpan * parent, const SpanTimeStamp * spanStartTimeStamp = nullptr)
+    : CChildSpan(spanName, parent, spanStartTimeStamp)
     {
         opts.kind = opentelemetry::trace::SpanKind::kInternal;
         init(SpanFlags::None);
@@ -1002,13 +1007,8 @@ class CClientSpan : public CChildSpan
 {
 public:
     CClientSpan(const char * spanName, CSpan * parent, const SpanTimeStamp * spanStartTimeStamp = nullptr)
-    : CChildSpan(spanName, parent)
+    : CChildSpan(spanName, parent, spanStartTimeStamp)
     {
-        if (spanStartTimeStamp && spanStartTimeStamp->isInitialized())
-        {
-            opts.start_system_time = opentelemetry::common::SystemTimestamp(spanStartTimeStamp->systemClockTime);
-            opts.start_steady_time = opentelemetry::common::SteadyTimestamp(spanStartTimeStamp->steadyClockTime);
-        }
         opts.kind = opentelemetry::trace::SpanKind::kClient;
         init(SpanFlags::None);
     }
@@ -1025,9 +1025,9 @@ ISpan * CSpan::createClientSpan(const char * name, const SpanTimeStamp * spanSta
     return new CClientSpan(name, this, spanStartTimeStamp);
 }
 
-ISpan * CSpan::createInternalSpan(const char * name)
+ISpan * CSpan::createInternalSpan(const char * name, const SpanTimeStamp * spanStartTimeStamp)
 {
-    return new CInternalSpan(name, this);
+    return new CInternalSpan(name, this, spanStartTimeStamp);
 }
 
 class CServerSpan : public CSpan

diff --git a/system/jlib/jtrace.hpp b/system/jlib/jtrace.hpp
@@ -151,7 +151,7 @@ interface ISpan : extends IInterface
     virtual const char * querySpanId() const = 0;
 
     virtual ISpan * createClientSpan(const char * name, const SpanTimeStamp * spanStartTimeStamp = nullptr)  = 0;
-    virtual ISpan * createInternalSpan(const char * name) = 0;
+    virtual ISpan * createInternalSpan(const char * name, const SpanTimeStamp * spanStartTimeStamp = nullptr) = 0;
 
 //Old-style global/caller/local id interface functions
     virtual const char* queryGlobalId() const = 0;