Spaces:
Sleeping
Sleeping
File size: 998 Bytes
1070765 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | /**
* Split an async generator into multiple async generators, all drawing from the same source.
*/
export function splitAsyncGenerator<T>(source: AsyncGenerator<T>, n: number): Array<AsyncGenerator<T>> {
if (n <= 0) {
return [];
}
const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
let takenIndex: number | null = null;
const generators: AsyncGenerator<T>[] = [];
let remaining = n;
for (let i = 0; i < n; i++) {
generators.push({
next: async () => {
while (takenIndex !== null) {
await sleep(1);
}
takenIndex = i;
return source.next().then((r) => {
takenIndex = null;
return r;
});
},
return: async () => {
remaining--;
if (remaining === 0) {
return source.return(undefined);
}
return {
done: true,
value: undefined,
};
},
throw: async (error) => {
return source.throw(error);
},
[Symbol.asyncIterator]: () => generators[i],
});
}
return generators;
}
|