orca/cloud/apps/relay/src/assignment-identity-queue.ts

export type AssignmentIdentity = {
  userId: string
  relayHostId: string
}

export class AssignmentIdentityQueue {
  private readonly tails = new Map<string, Promise<void>>()

  async run<T>(identity: AssignmentIdentity, operation: () => Promise<T>): Promise<T> {
    const key = JSON.stringify([identity.userId, identity.relayHostId])
    const previous = this.tails.get(key) ?? Promise.resolve()
    const result = previous.catch(() => undefined).then(operation)
    const tail = result.then(
      () => undefined,
      () => undefined
    )
    this.tails.set(key, tail)
    try {
      return await result
    } finally {
      if (this.tails.get(key) === tail) this.tails.delete(key)
    }
  }
}