refactor(trace): add span hook

Author: ycdzj

src/tracing.ts

@@ -8,6 +8,8 @@ import { ExpressInstrumentation } from '@opentelemetry/instrumentation-express';
 import { NestInstrumentation } from '@opentelemetry/instrumentation-nestjs-core';
 import { TypeormInstrumentation } from '@opentelemetry/instrumentation-typeorm';
 import { isEmpty } from 'omniboxd/utils/is-empty';
+import { Span } from '@opentelemetry/api';
+import { ServerResponse } from 'http';
 
 const env = process.env.ENV || 'unknown';
 const serviceName = `omnibox-backend`;
@@ -43,6 +45,12 @@ if (isTracingEnabled()) {
         ignoreIncomingRequestHook: (req) => {
           return excludedUrls.some((url) => req.url?.includes(url)) || false;
         },
+        responseHook: (span: Span, response: ServerResponse) => {
+          const req = response.req as any;
+          if (req.user?.id) {
+            span.setAttribute('user.id', req.user.id);
+          }
+        },
       }),
       new ExpressInstrumentation(),
       new NestInstrumentation(),