X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fmiddlewares%2Fcache.ts;h=091c82d92cdad17f6e23ff94013120aee33e1791;hb=8519cc92341545468104f0704fff64a05c84bec0;hp=1e00fc7316aaa674c93b251302e3038de45090dd;hpb=54b3316099ed7d2dfcb6d708fdb686f1e125ce61;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/middlewares/cache.ts b/server/middlewares/cache.ts index 1e00fc731..091c82d92 100644 --- a/server/middlewares/cache.ts +++ b/server/middlewares/cache.ts @@ -1,12 +1,14 @@ import * as express from 'express' import * as AsyncLock from 'async-lock' -import { parseDuration } from '../helpers/core-utils' +import { parseDurationToMs } from '../helpers/core-utils' import { Redis } from '../lib/redis' import { logger } from '../helpers/logger' const lock = new AsyncLock({ timeout: 5000 }) function cacheRoute (lifetimeArg: string | number) { + const lifetime = parseDurationToMs(lifetimeArg) + return async function (req: express.Request, res: express.Response, next: express.NextFunction) { const redisKey = Redis.Instance.generateCachedRouteKey(req) @@ -19,11 +21,11 @@ function cacheRoute (lifetimeArg: string | number) { logger.debug('No cached results for route %s.', req.originalUrl) const sendSave = res.send.bind(res) + const redirectSave = res.redirect.bind(res) res.send = (body) => { if (res.statusCode >= 200 && res.statusCode < 400) { const contentType = res.get('content-type') - const lifetime = parseDuration(lifetimeArg) Redis.Instance.setCachedRoute(req, body, lifetime, contentType, res.statusCode) .then(() => done()) @@ -38,6 +40,12 @@ function cacheRoute (lifetimeArg: string | number) { return sendSave(body) } + res.redirect = url => { + done() + + return redirectSave(url) + } + return next() }