@ms-cloudpack/telemetry 0.10.7 → 0.10.8

package/dist/{AppInsightsTelemetryClient-ZKKOJCSN.js → AppInsightsTelemetryClient-52HZATD6.js}

@@ -21,7 +21,7 @@ import {
   require_src2,
   require_src3,
   require_src4
-} from "./chunk-F3TAGUKV.js";
+} from "./chunk-AEUNAYUO.js";
 
 // ../../node_modules/.store/@azure-core-rest-pipeline-npm-1.17.0-aa1ea9ace7/package/dist/commonjs/pipeline.js
 var require_pipeline = __commonJS({

package/dist/{NoOpTelemetryClient-H75ZBB4T.js → NoOpTelemetryClient-DJPDMTNS.js}

@@ -6,8 +6,8 @@ const __filename = topLevelUrl.fileURLToPath(import.meta.url);
 const __dirname = topLevelPath.dirname(__filename);
 import {
   NoOpTelemetryClient
-} from "./chunk-KZSDB6J7.js";
-import "./chunk-F3TAGUKV.js";
+} from "./chunk-UPNW2UON.js";
+import "./chunk-AEUNAYUO.js";
 export {
   NoOpTelemetryClient
 };

package/dist/{chunk-F3TAGUKV.js → chunk-AEUNAYUO.js}

@@ -1733,9 +1733,9 @@ var init_esm = __esm({
   }
 });
 
-// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/AbstractAsyncHooksContextManager.js
+// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/AbstractAsyncHooksContextManager.js
 var require_AbstractAsyncHooksContextManager = __commonJS({
-  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/AbstractAsyncHooksContextManager.js"(exports) {
+  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/AbstractAsyncHooksContextManager.js"(exports) {
     "use strict";
     Object.defineProperty(exports, "__esModule", { value: true });
     exports.AbstractAsyncHooksContextManager = void 0;
@@ -1893,9 +1893,9 @@ var require_AbstractAsyncHooksContextManager = __commonJS({
   }
 });
 
-// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/AsyncHooksContextManager.js
+// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/AsyncHooksContextManager.js
 var require_AsyncHooksContextManager = __commonJS({
-  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/AsyncHooksContextManager.js"(exports) {
+  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/AsyncHooksContextManager.js"(exports) {
     "use strict";
     Object.defineProperty(exports, "__esModule", { value: true });
     exports.AsyncHooksContextManager = void 0;
@@ -1989,14 +1989,14 @@ var require_AsyncHooksContextManager = __commonJS({
       }
     };
     __name(_AsyncHooksContextManager, "AsyncHooksContextManager");
-    var AsyncHooksContextManager = _AsyncHooksContextManager;
-    exports.AsyncHooksContextManager = AsyncHooksContextManager;
+    var AsyncHooksContextManager2 = _AsyncHooksContextManager;
+    exports.AsyncHooksContextManager = AsyncHooksContextManager2;
   }
 });
 
-// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/AsyncLocalStorageContextManager.js
+// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/AsyncLocalStorageContextManager.js
 var require_AsyncLocalStorageContextManager = __commonJS({
-  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/AsyncLocalStorageContextManager.js"(exports) {
+  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/AsyncLocalStorageContextManager.js"(exports) {
     "use strict";
     Object.defineProperty(exports, "__esModule", { value: true });
     exports.AsyncLocalStorageContextManager = void 0;
@@ -2030,9 +2030,9 @@ var require_AsyncLocalStorageContextManager = __commonJS({
   }
 });
 
-// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/index.js
+// ../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/index.js
 var require_src = __commonJS({
-  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-67b6ccd860/package/build/src/index.js"(exports) {
+  "../../node_modules/.store/@opentelemetry-context-async-hooks-virtual-6b60acf600/package/build/src/index.js"(exports) {
     "use strict";
     Object.defineProperty(exports, "__esModule", { value: true });
     exports.AsyncLocalStorageContextManager = exports.AsyncHooksContextManager = void 0;
@@ -12471,13 +12471,77 @@ function createProxiedTracer(tracer) {
 }
 __name(createProxiedTracer, "createProxiedTracer");
 
+// src/CustomContextManager.ts
+init_esm();
+var import_context_async_hooks = __toESM(require_src(), 1);
+var _CustomContextManager = class _CustomContextManager {
+  constructor() {
+    this._contextManager = new import_context_async_hooks.AsyncHooksContextManager();
+    this._rootContext = void 0;
+  }
+  /**
+   * This is used to set the root context. If no root context is set, ROOT_CONTEXT is used.
+   * @param context - the root context for the telemetry client
+   */
+  setRootContext(context2) {
+    this._rootContext = context2;
+  }
+  /**
+   * @returns the active context
+   */
+  active() {
+    const activeContext = this._contextManager.active();
+    if (activeContext === ROOT_CONTEXT) {
+      return this._rootContext || ROOT_CONTEXT;
+    }
+    return activeContext;
+  }
+  /**
+   * Run the fn callback with object set as the current active context
+   * @param context Any object to set as the current active context
+   * @param fn A callback to be immediately run within a specific context
+   * @param thisArg optional receiver to be used for calling fn
+   * @param args optional arguments forwarded to fn
+   */
+  with(context2, fn, thisArg, ...args) {
+    return this._contextManager.with(context2, fn, thisArg, ...args);
+  }
+  /**
+   * Bind an object as the current context (or a specific one)
+   * @param [context] Optionally specify the context which you want to assign
+   * @param target Any object to which a context need to be set
+   */
+  bind(context2, target) {
+    return this._contextManager.bind(context2, target);
+  }
+  /**
+   * Enable context management
+   */
+  enable() {
+    this._contextManager.enable();
+    return this;
+  }
+  /**
+   * Disable context management
+   */
+  disable() {
+    this._contextManager.disable();
+    return this;
+  }
+};
+__name(_CustomContextManager, "CustomContextManager");
+var CustomContextManager = _CustomContextManager;
+
 // src/BaseTelemetryClient.ts
 var _BaseTelemetryClient = class _BaseTelemetryClient {
   constructor(options) {
     this._tracer = void 0;
+    this._contextManager = new CustomContextManager();
     diag2.debug("Initializing TelemetryClient with options:", options);
     metrics.disable();
     trace.disable();
+    this._contextManager.enable();
+    context.setGlobalContextManager(this._contextManager);
     const resource = createResourceFromOptions(options);
     this._tracerProvider = new import_sdk_trace_node.NodeTracerProvider({
       resource
@@ -12513,6 +12577,19 @@ var _BaseTelemetryClient = class _BaseTelemetryClient {
     }
     return this._tracer;
   }
+  /**
+   * Gets the OpenTelemetry context.
+   */
+  get context() {
+    return context;
+  }
+  /**
+   * Sets the root context.
+   * @param context - the root context for the telemetry client
+   */
+  setRootContext(context2) {
+    this._contextManager.setRootContext(context2);
+  }
 };
 __name(_BaseTelemetryClient, "BaseTelemetryClient");
 var BaseTelemetryClient = _BaseTelemetryClient;

package/dist/{chunk-KZSDB6J7.js → chunk-UPNW2UON.js}

@@ -7,7 +7,7 @@ const __dirname = topLevelPath.dirname(__filename);
 import {
   BaseTelemetryClient,
   __name
-} from "./chunk-F3TAGUKV.js";
+} from "./chunk-AEUNAYUO.js";
 
 // src/NoOpTelemetryClient.ts
 var _NoOpTelemetryClient = class _NoOpTelemetryClient extends BaseTelemetryClient {

package/dist/index.js

@@ -6,28 +6,28 @@ const __filename = topLevelUrl.fileURLToPath(import.meta.url);
 const __dirname = topLevelPath.dirname(__filename);
 import {
   NoOpTelemetryClient
-} from "./chunk-KZSDB6J7.js";
+} from "./chunk-UPNW2UON.js";
 import {
   DiagConsoleLogger,
   DiagLogLevel,
   __name,
   diag,
   init_esm
-} from "./chunk-F3TAGUKV.js";
+} from "./chunk-AEUNAYUO.js";
 
 // src/createTelemetryClient.ts
 init_esm();
 async function createTelemetryClient(options) {
   if (!options.connectionString) {
     console.debug("No connection string found. Telemetry will not be sent.");
-    const { NoOpTelemetryClient: NoOpTelemetryClient2 } = await import("./NoOpTelemetryClient-H75ZBB4T.js");
+    const { NoOpTelemetryClient: NoOpTelemetryClient2 } = await import("./NoOpTelemetryClient-DJPDMTNS.js");
     return new NoOpTelemetryClient2();
   }
   diag.setLogger(new DiagConsoleLogger(), {
     logLevel: (options.logLevel && DiagLogLevel[options.logLevel]) ?? DiagLogLevel.WARN,
     suppressOverrideMessage: true
   });
-  const { AppInsightsTelemetryClient } = await import("./AppInsightsTelemetryClient-ZKKOJCSN.js");
+  const { AppInsightsTelemetryClient } = await import("./AppInsightsTelemetryClient-52HZATD6.js");
   return new AppInsightsTelemetryClient(options);
 }
 __name(createTelemetryClient, "createTelemetryClient");

package/lib/BaseTelemetryClient.d.ts

@@ -1,3 +1,4 @@
+import { type ContextAPI, type Context } from '@opentelemetry/api';
 import type { TelemetryOptions } from './types/TelemetryOptions.js';
 import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
 import type { TelemetryClient } from './types/TelemetryClient.js';
@@ -5,6 +6,7 @@ import { type AsyncTracer } from './proxies/createProxiedTracer.js';
 export declare class BaseTelemetryClient implements TelemetryClient {
     protected readonly _tracerProvider: NodeTracerProvider;
     private _tracer;
+    private _contextManager;
     constructor(options: TelemetryOptions);
     /**
      * Flushes all telemetry and shuts down the telemetry client.
@@ -15,5 +17,14 @@ export declare class BaseTelemetryClient implements TelemetryClient {
      * @returns the OpenTelemetry tracer
      */
     get tracer(): AsyncTracer;
+    /**
+     * Gets the OpenTelemetry context.
+     */
+    get context(): ContextAPI;
+    /**
+     * Sets the root context.
+     * @param context - the root context for the telemetry client
+     */
+    setRootContext(context: Context | undefined): void;
 }
 //# sourceMappingURL=BaseTelemetryClient.d.ts.map

package/lib/CustomContextManager.d.ts

@@ -0,0 +1,41 @@
+import { type ContextManager, type Context } from '@opentelemetry/api';
+/**
+ * This is a wrapper around the AsyncHooksContextManager that allows for setting a root context
+ * which is being used if no context is available.
+ */
+export declare class CustomContextManager implements ContextManager {
+    private _contextManager;
+    private _rootContext;
+    /**
+     * This is used to set the root context. If no root context is set, ROOT_CONTEXT is used.
+     * @param context - the root context for the telemetry client
+     */
+    setRootContext(context: Context | undefined): void;
+    /**
+     * @returns the active context
+     */
+    active(): Context;
+    /**
+     * Run the fn callback with object set as the current active context
+     * @param context Any object to set as the current active context
+     * @param fn A callback to be immediately run within a specific context
+     * @param thisArg optional receiver to be used for calling fn
+     * @param args optional arguments forwarded to fn
+     */
+    with<A extends unknown[], F extends (...args: A) => ReturnType<F>>(context: Context, fn: F, thisArg?: ThisParameterType<F>, ...args: A): ReturnType<F>;
+    /**
+     * Bind an object as the current context (or a specific one)
+     * @param [context] Optionally specify the context which you want to assign
+     * @param target Any object to which a context need to be set
+     */
+    bind<T>(context: Context, target: T): T;
+    /**
+     * Enable context management
+     */
+    enable(): this;
+    /**
+     * Disable context management
+     */
+    disable(): this;
+}
+//# sourceMappingURL=CustomContextManager.d.ts.map

package/lib/types/TelemetryClient.d.ts

@@ -1,3 +1,4 @@
+import type { Context, ContextAPI } from '@opentelemetry/api';
 import type { AsyncTracer } from '../proxies/createProxiedTracer.js';
 export interface TelemetryClient {
     /**
@@ -9,5 +10,14 @@ export interface TelemetryClient {
      * @returns the OpenTelemetry tracer
      */
     get tracer(): AsyncTracer;
+    /**
+     * Gets the OpenTelemetry context.
+     */
+    get context(): ContextAPI;
+    /**
+     * Sets the root context.
+     * @param context - the root context for the telemetry client
+     */
+    setRootContext(context: Context | undefined): void;
 }
 //# sourceMappingURL=TelemetryClient.d.ts.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ms-cloudpack/telemetry",
-  "version": "0.10.7",
+  "version": "0.10.8",
   "description": "Helpers for reporting telemetry in Cloudpack.",
   "license": "MIT",
   "type": "module",
@@ -20,6 +20,7 @@
     "@ms-cloudpack/package-utilities": "^11.1.2",
     "@ms-cloudpack/scripts": "^0.0.1",
     "@opentelemetry/api": "~1.9.0",
+    "@opentelemetry/context-async-hooks": "~1.26.0",
     "@opentelemetry/core": "~1.26.0",
     "@opentelemetry/resources": "~1.26.0",
     "@opentelemetry/sdk-trace-base": "~1.26.0",