Class DisposableMap

packages/agent-core/src/di/lifecycle.ts:468–553

}

export class DisposableMap<K, V extends IDisposable = IDisposable>
  implements IDisposable
{
  private readonly _store: Map<K, V>;
  private _isDisposed = false;

  constructor(store: Map<K, V> = new Map<K, V>()) {
    this._store = store;
    trackDisposable(this);
  }

  dispose(): void {
    if (this._isDisposed) return;
    this._isDisposed = true;
    markAsDisposed(this);
    this.clearAndDisposeAll();
  }

  clearAndDisposeAll(): void {
    if (this._store.size === 0) return;
    try {
      dispose(this._store.values());
    } finally {
      this._store.clear();
    }
  }

  has(key: K): boolean {
    return this._store.has(key);
  }

  get size(): number {
    return this._store.size;
  }

  get(key: K): V | undefined {
    return this._store.get(key);
  }

  set(key: K, value: V, skipDisposeOnOverwrite = false): void {
    if (this._isDisposed) {
      // eslint-disable-next-line no-console
      console.warn(
        new Error(
          'Trying to add a disposable to a DisposableMap that has already been disposed of. The added object will be leaked!',
        ).stack,
      );
      return;
    }
    if (!skipDisposeOnOverwrite) {
      const prev = this._store.get(key);
      if (prev !== undefined && prev !== value) {
        prev.dispose();
      }
    }
    this._store.set(key, value);
    setParentOfDisposable(value, this);
  }