]>
Commit | Line | Data |
---|---|---|
4195cd2b | 1 | import * as express from 'express' |
b40f0575 | 2 | import * as AsyncLock from 'async-lock' |
8f0bc73d | 3 | import { parseDurationToMs } from '../helpers/core-utils' |
4195cd2b C |
4 | import { Redis } from '../lib/redis' |
5 | import { logger } from '../helpers/logger' | |
6 | ||
b40f0575 C |
7 | const lock = new AsyncLock({ timeout: 5000 }) |
8 | ||
98d3324d | 9 | function cacheRoute (lifetimeArg: string | number) { |
8519cc92 C |
10 | const lifetime = parseDurationToMs(lifetimeArg) |
11 | ||
fd4484f1 | 12 | return async function (req: express.Request, res: express.Response, next: express.NextFunction) { |
6e46de09 | 13 | const redisKey = Redis.Instance.generateCachedRouteKey(req) |
b40f0575 | 14 | |
98d3324d C |
15 | try { |
16 | await lock.acquire(redisKey, async (done) => { | |
17 | const cached = await Redis.Instance.getCachedRoute(req) | |
4195cd2b | 18 | |
98d3324d C |
19 | // Not cached |
20 | if (!cached) { | |
21 | logger.debug('No cached results for route %s.', req.originalUrl) | |
4195cd2b | 22 | |
98d3324d | 23 | const sendSave = res.send.bind(res) |
8d1fa36a | 24 | const redirectSave = res.redirect.bind(res) |
4195cd2b | 25 | |
98d3324d C |
26 | res.send = (body) => { |
27 | if (res.statusCode >= 200 && res.statusCode < 400) { | |
28 | const contentType = res.get('content-type') | |
98d3324d C |
29 | |
30 | Redis.Instance.setCachedRoute(req, body, lifetime, contentType, res.statusCode) | |
31 | .then(() => done()) | |
32 | .catch(err => { | |
33 | logger.error('Cannot cache route.', { err }) | |
34 | return done(err) | |
35 | }) | |
328e607d C |
36 | } else { |
37 | done() | |
98d3324d C |
38 | } |
39 | ||
40 | return sendSave(body) | |
b40f0575 C |
41 | } |
42 | ||
8d1fa36a C |
43 | res.redirect = url => { |
44 | done() | |
45 | ||
46 | return redirectSave(url) | |
47 | } | |
48 | ||
98d3324d | 49 | return next() |
fd4484f1 C |
50 | } |
51 | ||
98d3324d | 52 | if (cached.contentType) res.set('content-type', cached.contentType) |
4195cd2b | 53 | |
98d3324d C |
54 | if (cached.statusCode) { |
55 | const statusCode = parseInt(cached.statusCode, 10) | |
56 | if (!isNaN(statusCode)) res.status(statusCode) | |
57 | } | |
4195cd2b | 58 | |
98d3324d C |
59 | logger.debug('Use cached result for %s.', req.originalUrl) |
60 | res.send(cached.body).end() | |
4195cd2b | 61 | |
98d3324d C |
62 | return done() |
63 | }) | |
64 | } catch (err) { | |
328e607d | 65 | logger.error('Cannot serve cached route.', { err }) |
98d3324d C |
66 | return next() |
67 | } | |
4195cd2b | 68 | } |
4195cd2b C |
69 | } |
70 | ||
71 | // --------------------------------------------------------------------------- | |
72 | ||
73 | export { | |
98d3324d | 74 | cacheRoute |
4195cd2b | 75 | } |