Class ArrayNode

packages/effect/src/internal/hashMap.ts:623–802

@internal

/** @internal */
class ArrayNode<K, V> extends Node<K, V> {
  readonly _tag = "ArrayNode"

  edit: number
  private _size: number | undefined
  count: number
  children: Array<Node<K, V> | undefined>

  constructor(
    edit: number,
    count: number,
    children: Array<Node<K, V> | undefined>
  ) {
    super()
    this.edit = edit
    this.count = count
    this.children = children
  }

  get size(): number {
    if (this._size === undefined) {
      this._size = this.children.reduce<number>((acc, child) => acc + (child?.size ?? 0), 0)
    }
    return this._size
  }

  get(shift: number, hash: number, key: K): Option.Option<V> {
    const idx = mask(hash, shift)
    const child = this.children[idx]
    return child ? child.get(shift + SHIFT, hash, key) : Option.none()
  }

  has(shift: number, hash: number, key: K): boolean {
    const idx = mask(hash, shift)
    const child = this.children[idx]
    return child ? child.has(shift + SHIFT, hash, key) : false
  }

  set(
    edit: number,
    shift: number,
    hash: number,
    key: K,
    value: V,
    added: { value: boolean }
  ): Node<K, V> {
    const idx = mask(hash, shift)
    const child = this.children[idx]

    if (child) {
      const newChild = child.set(edit, shift + SHIFT, hash, key, value, added)
      if (child === newChild) {
        return this
      }

      if (this.canEdit(edit)) {
        this.children[idx] = newChild
        return this