From ce599e775f001342526ead7c0df0c00d2b93abfd Mon Sep 17 00:00:00 2001 From: Sam Rose Date: Thu, 7 Mar 2024 14:56:30 +0000 Subject: [PATCH] Add APM spans for request cleanup functions. --- packages/server/src/middleware/cleanup.ts | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/packages/server/src/middleware/cleanup.ts b/packages/server/src/middleware/cleanup.ts index a810879a65..43f945ab6b 100644 --- a/packages/server/src/middleware/cleanup.ts +++ b/packages/server/src/middleware/cleanup.ts @@ -1,5 +1,6 @@ import { Ctx } from "@budibase/types" import { context } from "@budibase/backend-core" +import { tracer } from "dd-trace" export default async (ctx: Ctx, next: any) => { const resp = await next() @@ -12,7 +13,9 @@ export default async (ctx: Ctx, next: any) => { let errors = [] for (let fn of current.cleanup) { try { - await fn() + await tracer.trace("cleanup", async span => { + await fn() + }) } catch (e) { // We catch errors here to ensure we at least attempt to run all cleanup // functions. We'll throw the first error we encounter after all cleanup