File Explorer

1'use strict'2 3const { InvalidArgumentError } = require('../core/errors')4const { kClients, kRunning, kClose, kDestroy, kDispatch, kInterceptors } = require('../core/symbols')5const DispatcherBase = require('./dispatcher-base')6const Pool = require('./pool')7const Client = require('./client')8const util = require('../core/util')9const createRedirectInterceptor = require('../interceptor/redirect-interceptor')10 11const kOnConnect = Symbol('onConnect')12const kOnDisconnect = Symbol('onDisconnect')13const kOnConnectionError = Symbol('onConnectionError')14const kMaxRedirections = Symbol('maxRedirections')15const kOnDrain = Symbol('onDrain')16const kFactory = Symbol('factory')17const kOptions = Symbol('options')18 19function defaultFactory (origin, opts) {20  return opts && opts.connections === 121    ? new Client(origin, opts)22    : new Pool(origin, opts)23}24 25class Agent extends DispatcherBase {26  constructor ({ factory = defaultFactory, maxRedirections = 0, connect, ...options } = {}) {27    if (typeof factory !== 'function') {28      throw new InvalidArgumentError('factory must be a function.')29    }30 31    if (connect != null && typeof connect !== 'function' && typeof connect !== 'object') {32      throw new InvalidArgumentError('connect must be a function or an object')33    }34 35    if (!Number.isInteger(maxRedirections) || maxRedirections < 0) {36      throw new InvalidArgumentError('maxRedirections must be a positive number')37    }38 39    super(options)40 41    if (connect && typeof connect !== 'function') {42      connect = { ...connect }43    }44 45    this[kInterceptors] = options.interceptors?.Agent && Array.isArray(options.interceptors.Agent)46      ? options.interceptors.Agent47      : [createRedirectInterceptor({ maxRedirections })]48 49    this[kOptions] = { ...util.deepClone(options), connect }50    this[kOptions].interceptors = options.interceptors51      ? { ...options.interceptors }52      : undefined53    this[kMaxRedirections] = maxRedirections54    this[kFactory] = factory55    this[kClients] = new Map()56 57    this[kOnDrain] = (origin, targets) => {58      this.emit('drain', origin, [this, ...targets])59    }60 61    this[kOnConnect] = (origin, targets) => {62      this.emit('connect', origin, [this, ...targets])63    }64 65    this[kOnDisconnect] = (origin, targets, err) => {66      this.emit('disconnect', origin, [this, ...targets], err)67    }68 69    this[kOnConnectionError] = (origin, targets, err) => {70      this.emit('connectionError', origin, [this, ...targets], err)71    }72  }73 74  get [kRunning] () {75    let ret = 076    for (const client of this[kClients].values()) {77      ret += client[kRunning]78    }79    return ret80  }81 82  [kDispatch] (opts, handler) {83    let key84    if (opts.origin && (typeof opts.origin === 'string' || opts.origin instanceof URL)) {85      key = String(opts.origin)86    } else {87      throw new InvalidArgumentError('opts.origin must be a non-empty string or URL.')88    }89 90    let dispatcher = this[kClients].get(key)91 92    if (!dispatcher) {93      dispatcher = this[kFactory](opts.origin, this[kOptions])94        .on('drain', this[kOnDrain])95        .on('connect', this[kOnConnect])96        .on('disconnect', this[kOnDisconnect])97        .on('connectionError', this[kOnConnectionError])98 99      // This introduces a tiny memory leak, as dispatchers are never removed from the map.100      // TODO(mcollina): remove te timer when the client/pool do not have any more101      // active connections.102      this[kClients].set(key, dispatcher)103    }104 105    return dispatcher.dispatch(opts, handler)106  }107 108  async [kClose] () {109    const closePromises = []110    for (const client of this[kClients].values()) {111      closePromises.push(client.close())112    }113    this[kClients].clear()114 115    await Promise.all(closePromises)116  }117 118  async [kDestroy] (err) {119    const destroyPromises = []120    for (const client of this[kClients].values()) {121      destroyPromises.push(client.destroy(err))122    }123    this[kClients].clear()124 125    await Promise.all(destroyPromises)126  }127}128 129module.exports = Agent130