File Explorer

1'use strict'2 3const { InvalidArgumentError, MaxOriginsReachedError } = require('../core/errors')4const { kClients, kRunning, kClose, kDestroy, kDispatch, kUrl } = require('../core/symbols')5const DispatcherBase = require('./dispatcher-base')6const Pool = require('./pool')7const Client = require('./client')8const util = require('../core/util')9 10const kOnConnect = Symbol('onConnect')11const kOnDisconnect = Symbol('onDisconnect')12const kOnConnectionError = Symbol('onConnectionError')13const kOnDrain = Symbol('onDrain')14const kFactory = Symbol('factory')15const kOptions = Symbol('options')16const kOrigins = Symbol('origins')17 18function defaultFactory (origin, opts) {19  return opts && opts.connections === 120    ? new Client(origin, opts)21    : new Pool(origin, opts)22}23 24class Agent extends DispatcherBase {25  constructor ({ factory = defaultFactory, maxOrigins = Infinity, connect, ...options } = {}) {26    if (typeof factory !== 'function') {27      throw new InvalidArgumentError('factory must be a function.')28    }29 30    if (connect != null && typeof connect !== 'function' && typeof connect !== 'object') {31      throw new InvalidArgumentError('connect must be a function or an object')32    }33 34    if (typeof maxOrigins !== 'number' || Number.isNaN(maxOrigins) || maxOrigins <= 0) {35      throw new InvalidArgumentError('maxOrigins must be a number greater than 0')36    }37 38    super(options)39 40    if (connect && typeof connect !== 'function') {41      connect = { ...connect }42    }43 44    this[kOptions] = { ...util.deepClone(options), maxOrigins, connect }45    this[kFactory] = factory46    this[kClients] = new Map()47    this[kOrigins] = new Set()48 49    this[kOnDrain] = (origin, targets) => {50      this.emit('drain', origin, [this, ...targets])51    }52 53    this[kOnConnect] = (origin, targets) => {54      this.emit('connect', origin, [this, ...targets])55    }56 57    this[kOnDisconnect] = (origin, targets, err) => {58      this.emit('disconnect', origin, [this, ...targets], err)59    }60 61    this[kOnConnectionError] = (origin, targets, err) => {62      this.emit('connectionError', origin, [this, ...targets], err)63    }64  }65 66  get [kRunning] () {67    let ret = 068    for (const { dispatcher } of this[kClients].values()) {69      ret += dispatcher[kRunning]70    }71    return ret72  }73 74  [kDispatch] (opts, handler) {75    let key76    if (opts.origin && (typeof opts.origin === 'string' || opts.origin instanceof URL)) {77      key = String(opts.origin)78    } else {79      throw new InvalidArgumentError('opts.origin must be a non-empty string or URL.')80    }81 82    if (this[kOrigins].size >= this[kOptions].maxOrigins && !this[kOrigins].has(key)) {83      throw new MaxOriginsReachedError()84    }85 86    const result = this[kClients].get(key)87    let dispatcher = result && result.dispatcher88    if (!dispatcher) {89      const closeClientIfUnused = (connected) => {90        const result = this[kClients].get(key)91        if (result) {92          if (connected) result.count -= 193          if (result.count <= 0) {94            this[kClients].delete(key)95            if (!result.dispatcher.destroyed) {96              result.dispatcher.close()97            }98          }99          this[kOrigins].delete(key)100        }101      }102      dispatcher = this[kFactory](opts.origin, this[kOptions])103        .on('drain', this[kOnDrain])104        .on('connect', (origin, targets) => {105          const result = this[kClients].get(key)106          if (result) {107            result.count += 1108          }109          this[kOnConnect](origin, targets)110        })111        .on('disconnect', (origin, targets, err) => {112          closeClientIfUnused(true)113          this[kOnDisconnect](origin, targets, err)114        })115        .on('connectionError', (origin, targets, err) => {116          closeClientIfUnused(false)117          this[kOnConnectionError](origin, targets, err)118        })119 120      this[kClients].set(key, { count: 0, dispatcher })121      this[kOrigins].add(key)122    }123 124    return dispatcher.dispatch(opts, handler)125  }126 127  [kClose] () {128    const closePromises = []129    for (const { dispatcher } of this[kClients].values()) {130      closePromises.push(dispatcher.close())131    }132    this[kClients].clear()133 134    return Promise.all(closePromises)135  }136 137  [kDestroy] (err) {138    const destroyPromises = []139    for (const { dispatcher } of this[kClients].values()) {140      destroyPromises.push(dispatcher.destroy(err))141    }142    this[kClients].clear()143 144    return Promise.all(destroyPromises)145  }146 147  get stats () {148    const allClientStats = {}149    for (const { dispatcher } of this[kClients].values()) {150      if (dispatcher.stats) {151        allClientStats[dispatcher[kUrl].origin] = dispatcher.stats152      }153    }154    return allClientStats155  }156}157 158module.exports = Agent159