feat: setRecorded for traceparent/tracecontext (#2687)

* feat: setRecorded for traceparent/tracecontext Part of #2302
elastic · May 12, 2022 · 8b6c11a · 8b6c11a
1 parent a289d44
commit 8b6c11a
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 2 deletions.
diff --git a/lib/tracecontext/index.js b/lib/tracecontext/index.js
@@ -82,6 +82,10 @@ class TraceContext {
       setter(carrier, 'tracestate', tracestate)
     }
   }
+
+  setRecorded () {
+    return this.traceparent.setRecorded()
+  }
 }
 
 TraceContext.FLAGS = TraceParent.FLAGS

diff --git a/lib/tracecontext/traceparent.js b/lib/tracecontext/traceparent.js
@@ -107,8 +107,10 @@ class TraceParent {
   constructor (buffer) {
     this[bufferSymbol] = buffer
     Object.defineProperty(this, 'recorded', {
-      value: !!(buffer[OFFSETS.flags] & FLAGS.recorded),
-      enumerable: true
+      enumerable: true,
+      get: function () {
+        return !!(this[bufferSymbol][OFFSETS.flags] & FLAGS.recorded)
+      }
     })
 
     defineLazyProp(this, 'version', hexSliceFn(buffer, OFFSETS.version, OFFSETS.traceId))
@@ -144,6 +146,14 @@ class TraceParent {
     return makeChild(Buffer.from(this[bufferSymbol]))
   }
 
+  setRecorded (value) {
+    if (value) {
+      this[bufferSymbol][OFFSETS.flags] |= FLAGS.recorded
+    } else {
+      this[bufferSymbol][OFFSETS.flags] &= ~FLAGS.recorded
+    }
+  }
+
   toString () {
     return `${this.version}-${this.traceId}-${this.id}-${this.flags}`
   }

diff --git a/test/tracecontext/traceparent.test.js b/test/tracecontext/traceparent.test.js
@@ -176,3 +176,17 @@ test('ensureParentId', t => {
 
   t.end()
 })
+
+test('setRecorded', t => {
+  const traceParent = TraceParent.fromString(header)
+
+  t.ok(traceParent.recorded)
+
+  traceParent.setRecorded(false)
+  t.ok(!traceParent.recorded)
+
+  traceParent.setRecorded(true)
+  t.ok(traceParent.recorded)
+
+  t.end()
+})