mirror of
https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git
synced 2026-03-31 06:12:26 +00:00
20 lines
502 B
TypeScript
20 lines
502 B
TypeScript
export class AsyncQueue<T> implements AsyncIterable<T> {
|
|
private queue: T[] = []
|
|
private resolvers: ((value: T) => void)[] = []
|
|
|
|
push(item: T) {
|
|
const resolve = this.resolvers.shift()
|
|
if (resolve) resolve(item)
|
|
else this.queue.push(item)
|
|
}
|
|
|
|
async next(): Promise<T> {
|
|
if (this.queue.length > 0) return this.queue.shift()!
|
|
return new Promise((resolve) => this.resolvers.push(resolve))
|
|
}
|
|
|
|
async *[Symbol.asyncIterator]() {
|
|
while (true) yield await this.next()
|
|
}
|
|
}
|