get-it/src/util/speedometer.ts

Generic HTTP request library for node, browsers and workers

/**
 * Inlined variant of npm `speedometer` (https://github.com/mafintosh/speedometer),
 * MIT-licensed, Copyright (c) 2013 Mathias Buus.
 */

let tick = 1
const maxTick = 65535
const resolution = 4
let timer: ReturnType<typeof setInterval> | null = null

const inc = function () {
  tick = (tick + 1) & maxTick
}

export function speedometer(seconds: number) {
  if (!timer) {
    timer = setInterval(inc, (1000 / resolution) | 0)
    if (timer.unref) timer.unref()
  }

  const size = resolution * (seconds || 5)
  const buffer = [0]
  let pointer = 1
  let last = (tick - 1) & maxTick

  return {
    getSpeed: function (delta: number) {
      let dist = (tick - last) & maxTick
      if (dist > size) dist = size
      last = tick

      while (dist--) {
        if (pointer === size) pointer = 0
        buffer[pointer] = buffer[pointer === 0 ? size - 1 : pointer - 1]
        pointer++
      }

      if (delta) buffer[pointer - 1] += delta

      const top = buffer[pointer - 1]
      const btm = buffer.length < size ? 0 : buffer[pointer === size ? 0 : pointer]

      return buffer.length < resolution ? top : ((top - btm) * resolution) / buffer.length
    },
    clear: function () {
      if (timer) {
        clearInterval(timer)
        timer = null
      }
    },
  }
}