Blame view

node_modules/cacache/lib/memoization.js 1.52 KB
aaac7fed   liuqimichale   add
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
  '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
    }
  }