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()
|
||
|
|
}
|
||
|
|
}
|