X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fmiddlewares%2Fcache.ts;h=091c82d92cdad17f6e23ff94013120aee33e1791;hb=8519cc92341545468104f0704fff64a05c84bec0;hp=1b44957d3004ac8106ea30df3620d5150e35eb53;hpb=328e607d32d7f1e960f8966a70097b030e386ab4;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/middlewares/cache.ts b/server/middlewares/cache.ts index 1b44957d3..091c82d92 100644 --- a/server/middlewares/cache.ts +++ b/server/middlewares/cache.ts @@ -1,14 +1,16 @@ 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.buildCachedRouteKey(req) + const redisKey = Redis.Instance.generateCachedRouteKey(req) try { await lock.acquire(redisKey, async (done) => { @@ -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() }