File Explorer

1'use strict'2 3const {4  PoolBase,5  kClients,6  kNeedDrain,7  kAddClient,8  kGetDispatcher,9  kRemoveClient10} = require('./pool-base')11const Client = require('./client')12const {13  InvalidArgumentError14} = require('../core/errors')15const util = require('../core/util')16const { kUrl } = require('../core/symbols')17const buildConnector = require('../core/connect')18 19const kOptions = Symbol('options')20const kConnections = Symbol('connections')21const kFactory = Symbol('factory')22 23function defaultFactory (origin, opts) {24  return new Client(origin, opts)25}26 27class Pool extends PoolBase {28  constructor (origin, {29    connections,30    factory = defaultFactory,31    connect,32    connectTimeout,33    tls,34    maxCachedSessions,35    socketPath,36    autoSelectFamily,37    autoSelectFamilyAttemptTimeout,38    allowH2,39    clientTtl,40    ...options41  } = {}) {42    if (connections != null && (!Number.isFinite(connections) || connections < 0)) {43      throw new InvalidArgumentError('invalid connections')44    }45 46    if (typeof factory !== 'function') {47      throw new InvalidArgumentError('factory must be a function.')48    }49 50    if (connect != null && typeof connect !== 'function' && typeof connect !== 'object') {51      throw new InvalidArgumentError('connect must be a function or an object')52    }53 54    if (typeof connect !== 'function') {55      connect = buildConnector({56        ...tls,57        maxCachedSessions,58        allowH2,59        socketPath,60        timeout: connectTimeout,61        ...(typeof autoSelectFamily === 'boolean' ? { autoSelectFamily, autoSelectFamilyAttemptTimeout } : undefined),62        ...connect63      })64    }65 66    super(options)67 68    this[kConnections] = connections || null69    this[kUrl] = util.parseOrigin(origin)70    this[kOptions] = { ...util.deepClone(options), connect, allowH2, clientTtl, socketPath }71    this[kOptions].interceptors = options.interceptors72      ? { ...options.interceptors }73      : undefined74    this[kFactory] = factory75 76    this.on('connect', (origin, targets) => {77      if (clientTtl != null && clientTtl > 0) {78        for (const target of targets) {79          Object.assign(target, { ttl: Date.now() })80        }81      }82    })83 84    this.on('connectionError', (origin, targets, error) => {85      // If a connection error occurs, we remove the client from the pool,86      // and emit a connectionError event. They will not be re-used.87      // Fixes https://github.com/nodejs/undici/issues/389588      for (const target of targets) {89        // Do not use kRemoveClient here, as it will close the client,90        // but the client cannot be closed in this state.91        const idx = this[kClients].indexOf(target)92        if (idx !== -1) {93          this[kClients].splice(idx, 1)94        }95      }96    })97  }98 99  [kGetDispatcher] () {100    const clientTtlOption = this[kOptions].clientTtl101    for (const client of this[kClients]) {102      // check ttl of client and if it's stale, remove it from the pool103      if (clientTtlOption != null && clientTtlOption > 0 && client.ttl && ((Date.now() - client.ttl) > clientTtlOption)) {104        this[kRemoveClient](client)105      } else if (!client[kNeedDrain]) {106        return client107      }108    }109 110    if (!this[kConnections] || this[kClients].length < this[kConnections]) {111      const dispatcher = this[kFactory](this[kUrl], this[kOptions])112      this[kAddClient](dispatcher)113      return dispatcher114    }115  }116}117 118module.exports = Pool119