class/RandomSeed.js

/**
 * @description 带缓存的随机数种子。
 * @example
 * const seed = new RandomSeed(5, 10, [6])
 * seed.do()
 * // 5 or 7 or 8 or 9 or 10
 */

export class RandomSeed {
  /**
   * @param {Number} lower - 下限（包含）。
   * @param {Number} upper - 上限（包含）。
   * @param {Array} initCache - 初始缓存数组。
   * @param {Boolean} loop - 如果是，当缓存满时，清空缓存。默认 true。
   */

  constructor (lower, upper, initCache, loop = true) {
    if (upper < lower) {
      throw new Error('params error')
    }
    this.lower = lower
    this.upper = upper
    this.loop = loop
    // 在随机范围内 数的个数
    this.length = this.upper - this.lower + 1
    this.setCache(initCache || [])
  }
  setCache (cache) {
    this.initCache = cache
    this.cache = cache.concat()
    // 缓存中界限外的数个数
    this.outBoundsNum = this.initCache.reduce((total, val) => {
      if (val < this.lower || val > this.upper) {
        return total + 1
      }
      return total
    }, 0)
    // cache 能放入的元素个数是
    // cache 的长度加 cache中 超出界限的数的个数
    this.cacheLimitLength = this.length + this.outBoundsNum
  }
  do () {
    if (!this.loop && this.cache.length === this.cacheLimitLength) {
      return
    }

    if (this.upper - this.lower === 0) {
      return this.lower
    }

    const random = Math.floor(Math.random() * this.length + this.lower)

    if (this.cache.length === this.cacheLimitLength) {
      const lastValue = this.cache[this.cache.length - 1]

      if (random === lastValue) {
        return this.do()
      }
      this.cache = this.initCache.concat() || []
    }

    if (this.cache.some(number => number === random)) {
      return this.do()
    }
    this.cache.push(random)
    return random
  }
}