1
0
Fork 0
peertube/server/middlewares/async.ts

40 lines
1.4 KiB
TypeScript
Raw Normal View History

import { eachSeries } from 'async'
2017-11-27 11:30:46 -05:00
import { NextFunction, Request, RequestHandler, Response } from 'express'
2019-07-24 09:48:47 -04:00
import { ValidationChain } from 'express-validator'
2021-02-03 03:33:05 -05:00
import { ExpressPromiseHandler } from '@server/types/express'
import { retryTransactionWrapper } from '../helpers/database-utils'
2017-10-25 05:55:06 -04:00
// Syntactic sugar to avoid try/catch in express controllers
// Thanks: https://medium.com/@Abazhenov/using-async-await-in-express-with-node-8-b8af872c0016
2017-11-27 11:30:46 -05:00
2021-02-03 03:33:05 -05:00
export type RequestPromiseHandler = ValidationChain | ExpressPromiseHandler
2017-11-27 11:30:46 -05:00
function asyncMiddleware (fun: RequestPromiseHandler | RequestPromiseHandler[]) {
2017-10-25 05:55:06 -04:00
return (req: Request, res: Response, next: NextFunction) => {
if (Array.isArray(fun) === true) {
return eachSeries(fun as RequestHandler[], (f, cb) => {
2021-04-12 10:35:04 -04:00
Promise.resolve(f(req, res, err => cb(err)))
2018-02-14 09:33:49 -05:00
.catch(err => next(err))
}, next)
}
return Promise.resolve((fun as RequestHandler)(req, res, next))
2018-02-14 09:33:49 -05:00
.catch(err => next(err))
2017-10-25 05:55:06 -04:00
}
}
2019-07-24 09:48:47 -04:00
function asyncRetryTransactionMiddleware (fun: (req: Request, res: Response, next: NextFunction) => Promise<any>) {
2018-06-13 08:27:40 -04:00
return (req: Request, res: Response, next: NextFunction) => {
return Promise.resolve(
retryTransactionWrapper(fun, req, res, next)
).catch(err => next(err))
}
}
2017-10-25 05:55:06 -04:00
// ---------------------------------------------------------------------------
export {
2018-06-13 08:27:40 -04:00
asyncMiddleware,
asyncRetryTransactionMiddleware
2017-10-25 05:55:06 -04:00
}