PeerTube/server/core/middlewares/async.ts

49 lines
1.4 KiB
TypeScript
Raw Normal View History

2017-11-27 10:30:46 -06:00
import { NextFunction, Request, RequestHandler, Response } from 'express'
2019-07-24 08:48:47 -05:00
import { ValidationChain } from 'express-validator'
import { ExpressPromiseHandler } from '@server/types/express-handler.js'
import { retryTransactionWrapper } from '../helpers/database-utils.js'
2017-10-25 04:55:06 -05:00
2022-08-02 07:41:44 -05:00
// Syntactic sugar to avoid try/catch in express controllers/middlewares
2017-11-27 10:30:46 -06:00
2021-02-03 02:33:05 -06:00
export type RequestPromiseHandler = ValidationChain | ExpressPromiseHandler
2017-11-27 10:30:46 -06:00
function asyncMiddleware (fun: RequestPromiseHandler | RequestPromiseHandler[]) {
return async (req: Request, res: Response, next: NextFunction) => {
if (Array.isArray(fun) !== true) {
return Promise.resolve((fun as RequestHandler)(req, res, next))
.catch(err => next(err))
}
try {
2024-02-22 03:12:04 -06:00
for (const f of fun) {
await new Promise<void>((resolve, reject) => {
2022-08-02 07:41:44 -05:00
return asyncMiddleware(f)(req, res, err => {
if (err) return reject(err)
return resolve()
})
})
}
next()
} catch (err) {
next(err)
}
2017-10-25 04:55:06 -05:00
}
}
2019-07-24 08:48:47 -05:00
function asyncRetryTransactionMiddleware (fun: (req: Request, res: Response, next: NextFunction) => Promise<any>) {
2018-06-13 07:27:40 -05:00
return (req: Request, res: Response, next: NextFunction) => {
return Promise.resolve(
retryTransactionWrapper(fun, req, res, next)
).catch(err => next(err))
}
}
2017-10-25 04:55:06 -05:00
// ---------------------------------------------------------------------------
export {
2018-06-13 07:27:40 -05:00
asyncMiddleware,
asyncRetryTransactionMiddleware
2017-10-25 04:55:06 -05:00
}