Merge pull request #172 from import-ai/refactor/trace

refactor(trace): add middleware

Author: ycdzj

src/app/app.module.ts

@@ -48,6 +48,7 @@ import { ScanResourceAttachments1755504936756 } from 'omniboxd/migrations/175550
 import { SharesAllResources1754471311959 } from 'omniboxd/migrations/1754471311959-shares-all-resources';
 import { ResourcesModule } from 'omniboxd/resources/resources.module';
 import { TraceModule } from 'omniboxd/trace/trace.module';
+import { UserInterceptor } from 'omniboxd/interceptor/user.interceptor';
 
 @Module({})
 export class AppModule implements NestModule {
@@ -72,6 +73,10 @@ export class AppModule implements NestModule {
           provide: APP_INTERCEPTOR,
           useClass: ClassSerializerInterceptor,
         },
+        {
+          provide: APP_INTERCEPTOR,
+          useClass: UserInterceptor,
+        },
       ],
       imports: [
         ConfigModule.forRoot({

src/interceptor/user.interceptor.ts

@@ -0,0 +1,26 @@
+import {
+  Injectable,
+  NestInterceptor,
+  ExecutionContext,
+  CallHandler,
+} from '@nestjs/common';
+import { Observable } from 'rxjs';
+import { trace, context } from '@opentelemetry/api';
+
+@Injectable()
+export class UserInterceptor implements NestInterceptor {
+  intercept(
+    executionContext: ExecutionContext,
+    next: CallHandler,
+  ): Observable<any> {
+    const req = executionContext.switchToHttp().getRequest();
+    const tracer = trace.getTracer('user-interceptor');
+    tracer.startActiveSpan('UserInterceptor', {}, context.active(), (span) => {
+      if (req.user?.id) {
+        span.setAttribute('user.id', req.user.id);
+      }
+      span.end();
+    });
+    return next.handle();
+  }
+}