From b7a8965a316208fda3241b7e2cf7997fcfe37f50 Mon Sep 17 00:00:00 2001
From: Joseph Page <josephpage@users.noreply.github.com>
Date: Mon, 17 Jun 2024 12:38:52 +0200
Subject: [PATCH] chore: fix APM ?

---
 src/oidc-provider/oidc.controller.ts | 15 ++++++++++++---
 src/oidc-provider/oidc.service.ts    |  3 +++
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/src/oidc-provider/oidc.controller.ts b/src/oidc-provider/oidc.controller.ts
index a0a75bd..b6d6c95 100644
--- a/src/oidc-provider/oidc.controller.ts
+++ b/src/oidc-provider/oidc.controller.ts
@@ -21,11 +21,20 @@ export class OidcController {
     'protocol/openid-connect/*',
     'clients-registrations/*'
   ])
-  public mountedOidc(@Req() req: Request, @Res() res: Response): Promise<void> {
+  public async mountedOidc(
+    @Req() req: Request,
+    @Res() res: Response
+  ): Promise<void> {
+    req.url = req.originalUrl.replace('/auth/realms/pass-emploi', '')
+
+    const span =
+      this.apmService.currentTransaction?.startSpan('node-oidc-provider')
     try {
-      req.url = req.originalUrl.replace('/auth/realms/pass-emploi', '')
-      return this.callback(req, res)
+      const result = await this.callback(req, res)
+      span?.end()
+      return result
     } catch (e) {
+      span?.end()
       this.logger.error(e)
       this.apmService.captureError(e)
       throw e
diff --git a/src/oidc-provider/oidc.service.ts b/src/oidc-provider/oidc.service.ts
index fa849fe..fefa417 100644
--- a/src/oidc-provider/oidc.service.ts
+++ b/src/oidc-provider/oidc.service.ts
@@ -408,6 +408,9 @@ export class OidcService {
     this.oidc.on('refresh_token.destroyed', refreshToken => {
       this.logger.debug('REFRESH DESTROYED %j', refreshToken)
     })
+    this.oidc.on('*', event =>
+      this.apmService.addLabels({ [event]: Date.now() })
+    )
     this.oidc.proxy = true
   }