feat(koa): add layer type to request hook context

open-telemetry · Oct 11, 2022 · 2cda169 · 2cda169
1 parent b406b1d
commit 2cda169
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/plugins/node/opentelemetry-instrumentation-koa/src/instrumentation.ts b/plugins/node/opentelemetry-instrumentation-koa/src/instrumentation.ts
@@ -186,7 +186,11 @@ export class KoaInstrumentation extends InstrumentationBase<typeof koa> {
       if (this.getConfig().requestHook) {
         safeExecuteInTheMiddle(
           () =>
-            this.getConfig().requestHook!(span, { context, middlewareLayer }),
+            this.getConfig().requestHook!(span, {
+              context,
+              middlewareLayer,
+              layerType,
+            }),
           e => {
             if (e) {
               api.diag.error('koa instrumentation: request hook failed', e);

diff --git a/plugins/node/opentelemetry-instrumentation-koa/src/types.ts b/plugins/node/opentelemetry-instrumentation-koa/src/types.ts
@@ -35,6 +35,7 @@ export type KoaContext = ParameterizedContext<DefaultState, RouterParamContext>;
 export type KoaRequestInfo = {
   context: KoaContext;
   middlewareLayer: KoaMiddleware;
+  layerType: KoaLayerType;
 };
 
 /**

diff --git a/plugins/node/opentelemetry-instrumentation-koa/test/koa.test.ts b/plugins/node/opentelemetry-instrumentation-koa/test/koa.test.ts
@@ -531,6 +531,7 @@ describe('Koa Instrumentation', () => {
       const requestHook = sinon.spy((span: Span, info: KoaRequestInfo) => {
         span.setAttribute('http.method', info.context.request.method);
         span.setAttribute('app.env', info.context.app.env);
+        span.setAttribute('koa.layer', info.layerType);
       });
 
       plugin.setConfig({
@@ -564,6 +565,10 @@ describe('Koa Instrumentation', () => {
             requestHandlerSpan?.attributes['app.env'],
             'development'
           );
+          assert.strictEqual(
+            requestHandlerSpan?.attributes['koa.layer'],
+            'router'
+          );
         }
       );
     });