15815213711
2024-08-26 67b8b6731811983447e053d4396b3708c14dfe3c
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
const LRUCache = require('lru-cache')
const dns = require('dns')
 
const defaultOptions = exports.defaultOptions = {
  family: undefined,
  hints: dns.ADDRCONFIG,
  all: false,
  verbatim: undefined,
}
 
const lookupCache = exports.lookupCache = new LRUCache({ max: 50 })
 
// this is a factory so that each request can have its own opts (i.e. ttl)
// while still sharing the cache across all requests
exports.getLookup = (dnsOptions) => {
  return (hostname, options, callback) => {
    if (typeof options === 'function') {
      callback = options
      options = null
    } else if (typeof options === 'number') {
      options = { family: options }
    }
 
    options = { ...defaultOptions, ...options }
 
    const key = JSON.stringify({
      hostname,
      family: options.family,
      hints: options.hints,
      all: options.all,
      verbatim: options.verbatim,
    })
 
    if (lookupCache.has(key)) {
      const [address, family] = lookupCache.get(key)
      process.nextTick(callback, null, address, family)
      return
    }
 
    dnsOptions.lookup(hostname, options, (err, address, family) => {
      if (err) {
        return callback(err)
      }
 
      lookupCache.set(key, [address, family], { ttl: dnsOptions.ttl })
      return callback(null, address, family)
    })
  }
}