33 lines
781 B
TypeScript
33 lines
781 B
TypeScript
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()
|
|
|
|
const current = context.getCurrentContext()
|
|
if (!current || !current.cleanup) {
|
|
return resp
|
|
}
|
|
|
|
let errors = []
|
|
for (let fn of current.cleanup) {
|
|
try {
|
|
await tracer.trace("cleanup", async () => {
|
|
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
|
|
// functions have been run.
|
|
errors.push(e)
|
|
}
|
|
}
|
|
delete current.cleanup
|
|
|
|
if (errors.length > 0) {
|
|
throw errors[0]
|
|
}
|
|
|
|
return resp
|
|
}
|