Class BlockHeightMap

packages/node-core/src/utils/blockHeightMap.ts:12–114

}

export class BlockHeightMap<T> {
  #map: Map<number, T>;

  constructor(initialValues: Map<number, T>) {
    // Ensure the values are sorted by key
    this.#map = new Map([...initialValues.entries()].sort((a, b) => a[0] - b[0]));
  }

  getAll(): Map<number, T> {
    return this.#map;
  }

  get(height: number): T {
    const details = this.getDetails(height);

    if (details === undefined) {
      throw new EntryNotFoundError(height);
    }

    return details.value;
  }

  // Same as get but wont throw when there is nothing in the block range
  getSafe(height: number): T | undefined {
    try {
      return this.get(height);
    } catch (e) {
      return undefined;
    }
  }

  getDetails(height: number): GetRange<T> | undefined {
    let result: GetRange<T> | undefined;

    const arr = [...this.#map.entries()];

    for (let i = 0; i < arr.length; i++) {
      const [currentHeight, value] = arr[i];
      const nextStart = arr[i + 1]?.[0];
      const r = {
        value,
        startHeight: currentHeight,
        endHeight: nextStart ? nextStart - 1 : undefined,
      };

      if (currentHeight === height) {
        result = r;
        break;
      }
      if (currentHeight <= height) {
        result = r;
      }

      if (currentHeight > height) {
        break;
      }
    }