feat(tracing): context binding util (#32)

MapColonies · Jul 14, 2021 · 9725ed7 · 9725ed7
1 parent 47e9a9a
commit 9725ed7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/tracing/index.ts b/src/tracing/index.ts
@@ -1,3 +1,4 @@
 export { Tracing } from './tracing';
 export { getTraceContexHeaderMiddleware } from './middleware/traceOnHeaderMiddleware';
+export { contexBindingHelper } from './utils/tracing';
 export { logMethod } from './loggerHook';
diff --git a/src/tracing/utils/tracing.ts b/src/tracing/utils/tracing.ts
@@ -0,0 +1,6 @@
+import api, { Span } from '@opentelemetry/api';
+
+export const contexBindingHelper = <T>(parentSpan: Span, func: T): T => {
+  const ctx = api.trace.setSpan(api.context.active(), parentSpan);
+  return api.context.bind(ctx, func);
+};