|
|
- 'use strict'
-
- const LRU = require('lru-cache')
-
- const MAX_SIZE = 50 * 1024 * 1024 // 50MB
- const MAX_AGE = 3 * 60 * 1000
-
- let MEMOIZED = new LRU({
- max: MAX_SIZE,
- maxAge: MAX_AGE,
- length: (entry, key) => {
- if (key.startsWith('key:')) {
- return entry.data.length
- } else if (key.startsWith('digest:')) {
- return entry.length
- }
- }
- })
-
- module.exports.clearMemoized = clearMemoized
- function clearMemoized () {
- const old = {}
- MEMOIZED.forEach((v, k) => {
- old[k] = v
- })
- MEMOIZED.reset()
- return old
- }
-
- module.exports.put = put
- function put (cache, entry, data, opts) {
- pickMem(opts).set(`key:${cache}:${entry.key}`, { entry, data })
- putDigest(cache, entry.integrity, data, opts)
- }
-
- module.exports.put.byDigest = putDigest
- function putDigest (cache, integrity, data, opts) {
- pickMem(opts).set(`digest:${cache}:${integrity}`, data)
- }
-
- module.exports.get = get
- function get (cache, key, opts) {
- return pickMem(opts).get(`key:${cache}:${key}`)
- }
-
- module.exports.get.byDigest = getDigest
- function getDigest (cache, integrity, opts) {
- return pickMem(opts).get(`digest:${cache}:${integrity}`)
- }
-
- class ObjProxy {
- constructor (obj) {
- this.obj = obj
- }
- get (key) { return this.obj[key] }
- set (key, val) { this.obj[key] = val }
- }
-
- function pickMem (opts) {
- if (!opts || !opts.memoize) {
- return MEMOIZED
- } else if (opts.memoize.get && opts.memoize.set) {
- return opts.memoize
- } else if (typeof opts.memoize === 'object') {
- return new ObjProxy(opts.memoize)
- } else {
- return MEMOIZED
- }
- }
|