wip: wired up context manager, and segment synthesis

lib/context-manager/context.js

@@ -4,11 +4,13 @@
  */
 
 'use strict'
+const { otelSynthesis } = require('../symbols')
 
 module.exports = class Context {
-  constructor(transaction, segment) {
+  constructor(transaction, segment, parentContext) {
     this._transaction = transaction
     this._segment = segment
+    this._otelCtx = parentContext ? new Map(parentContext) : new Map()
   }
 
   get segment() {
@@ -19,11 +21,29 @@ module.exports = class Context {
     return this._transaction
   }
 
-  enterSegment({ segment, transaction = this.transaction }) {
+  enterSegment({ segment, transaction = this._transaction }) {
     return new this.constructor(transaction, segment)
   }
 
   enterTransaction(transaction) {
     return new this.constructor(transaction, transaction.trace.root)
   }
+
+  getValue(key) {
+    return this._otelCtx.get(key)
+  }
+
+  setValue(key, value) {
+    const { segment, transaction } = value[otelSynthesis]
+    segment.start()
+    const ctx = new this.constructor(transaction, segment, this._otelCtx)
+    ctx._otelCtx.set(key, value)
+    return ctx
+  }
+
+  deleteValue(key) {
+    const ctx = new this.constructor(this._transaction, this._segment, this._otelCtx)
+    ctx._otelCtx.delete(key)
+    return ctx
+  }
 }

lib/feature_flags.js

@@ -10,7 +10,7 @@ exports.prerelease = {
   // internal_test_only is used for testing our feature flag implementation.
   // It is not used to gate any features.
   internal_test_only: false,
-  otel_instrumentation: true,
+  otel_bridge: false,
   promise_segments: false,
   reverse_naming_rules: false,
   undici_async_tracking: true,

lib/instrumentations.js

@@ -35,7 +35,6 @@ module.exports = function instrumentations() {
     'mongodb': { type: InstrumentationDescriptor.TYPE_DATASTORE },
     'mysql': { module: './instrumentation/mysql' },
     'next': { module: './instrumentation/nextjs' },
-    'otel': { module: './instrumentation/otel' },
     'openai': { type: InstrumentationDescriptor.TYPE_GENERIC },
     'pg': { type: InstrumentationDescriptor.TYPE_DATASTORE },
     'pino': { module: './instrumentation/pino' },

lib/instrumentation/otel/context-manager.js → lib/otel/context-manager.js

@@ -5,25 +5,18 @@
 
 'use strict'
 
-const { ROOT_CONTEXT } = require('@opentelemetry/api')
-
 /**
  * @see https://open-telemetry.github.io/opentelemetry-js/interfaces/_opentelemetry_api.ContextManager.html
  */
 class ContextManager {
-  #agent
-
   #ctxMgr
 
   constructor(agent) {
-    this.#agent = agent
-    this.#ctxMgr = agent._contextManager
+    this.#ctxMgr = agent.tracer._contextManager
   }
 
   active() {
-    // TODO: get current span(segment?) from agent and prefer it
-    const storedContext = this.#ctxMgr.getContext()
-    return storedContext || ROOT_CONTEXT
+    return this.#ctxMgr.getContext()
   }
 
   bind(context, target) {
@@ -44,7 +37,7 @@ class ContextManager {
    * @param args
    */
   with(context, callback, thisRef, ...args) {
-    return this.#agent._contextManager.runInContext(context, callback, thisRef, args)
+    return this.#ctxMgr.runInContext(context, callback, thisRef, args)
   }
 
   enable() {

lib/otel/segments/http-external.js

@@ -6,13 +6,15 @@
 'use strict'
 const NAMES = require('../../metrics/names')
 const { SEMATTRS_HTTP_HOST } = require('@opentelemetry/semantic-conventions')
+const recordExternal = require('../../metrics/recorders/http_external')
 
 module.exports = function createHttpExternalSegment(agent, otelSpan) {
   const context = agent.tracer.getContext()
   const host = otelSpan.attributes[SEMATTRS_HTTP_HOST] || 'Unknown'
   const name = NAMES.EXTERNAL.PREFIX + host
   const segment = agent.tracer.createSegment({
     name,
+    recorder: recordExternal(host, 'http'),
     parent: context.segment,
     transaction: context.transaction
   })

lib/otel/segments/internal.js

@@ -4,13 +4,15 @@
  */
 
 'use strict'
+const customRecorder = require('../../metrics/recorders/custom')
 
 module.exports = function createInternalSegment(agent, otelSpan) {
   const context = agent.tracer.getContext()
-  const name = `Custom/${otelSpan.name}`
+  const name = otelSpan.name
   const segment = agent.tracer.createSegment({
     name,
     parent: context.segment,
+    recorder: customRecorder,
     transaction: context.transaction
   })
   return { segment, transaction: context.transaction }

lib/otel/setup.js

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2025 New Relic Corporation. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+'use strict'
+const { BasicTracerProvider } = require('@opentelemetry/sdk-trace-base')
+const NrSpanProcessor = require('./span-processor')
+const ContextManager = require('./context-manager')
+
+module.exports = function setupOtel(agent) {
+  if (agent.config.feature_flag.otel_bridge !== true) {
+    agent.logger.warn(
+      '`feature_flag.otel_bridge` is not enabled, not setting up opentelemetry_bridge.'
+    )
+    return
+  }
+
+  const provider = new BasicTracerProvider({
+    spanProcessors: [new NrSpanProcessor(agent)]
+  })
+  provider.register({
+    contextManager: new ContextManager(agent)
+    // propagator: // w3c trace propagator
+  })
+}