Class CacheMap

packages/shared/src/cache-map.ts:1–112

export class CacheMap<K, V> implements Map<K, V> {
  private cache: Map<K, V>
  private keysInUse: K[]
  private maxCacheSize: number

  constructor(maxCacheSize = 1000) {
    this.maxCacheSize = maxCacheSize
    this.cache = new Map<K, V>()
    this.keysInUse = []
  }

  get(key: K) {
    if (!this.cache.has(key)) {
      return undefined
    }
    this.updateKeyUsage(key)
    return this.cache.get(key)
  }

  set(key: K, value: V) {
    if (!this.cache.has(key) && this.cache.size === this.maxCacheSize) {
      this.evictLeastRecentlyUsed()
    }
    this.cache.set(key, value)
    this.updateKeyUsage(key)
    return this
  }

  delete(key: K) {
    const result = this.cache.delete(key)
    if (result) {
      const index = this.keysInUse.indexOf(key)
      if (index !== -1) {
        this.keysInUse.splice(index, 1)
      }
    }
    return result
  }

  private updateKeyUsage(key: K) {
    const index = this.keysInUse.indexOf(key)
    if (index !== -1) {
      this.keysInUse.splice(index, 1)
    }
    this.keysInUse.push(key)
  }

  private evictLeastRecentlyUsed() {
    const keyToEvict = this.keysInUse.shift()
    if (keyToEvict !== undefined) {
      this.cache.delete(keyToEvict)
    }
  }

  clear() {
    this.cache.clear()
    this.keysInUse = []
  }

  has(key: K): boolean {