deno.land / std@0.166.0 / async / tee.ts

نووسراو ببینە
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
// Copyright 2018-2022 the Deno authors. All rights reserved. MIT license.// This module is browser compatible.
// Utility for representing n-tupletype Tuple<T, N extends number> = N extends N ? number extends N ? T[] : TupleOf<T, N, []> : never;type TupleOf<T, N extends number, R extends unknown[]> = R["length"] extends N ? R : TupleOf<T, N, [T, ...R]>;
interface QueueNode<T> { value: T; next: QueueNode<T> | undefined;}
class Queue<T> { #source: AsyncIterator<T>; #queue: QueueNode<T>; head: QueueNode<T>;
done: boolean;
constructor(iterable: AsyncIterable<T>) { this.#source = iterable[Symbol.asyncIterator](); this.#queue = { value: undefined!, next: undefined, }; this.head = this.#queue; this.done = false; }
async next() { const result = await this.#source.next(); if (!result.done) { const nextNode: QueueNode<T> = { value: result.value, next: undefined, }; this.#queue.next = nextNode; this.#queue = nextNode; } else { this.done = true; } }}
/** * Branches the given async iterable into the n branches. * * Example: * * ```ts * import { tee } from "https://deno.land/std@$STD_VERSION/async/tee.ts"; * * const gen = async function* gen() { * yield 1; * yield 2; * yield 3; * } * * const [branch1, branch2] = tee(gen()); * * for await (const n of branch1) { * console.log(n); // => 1, 2, 3 * } * * for await (const n of branch2) { * console.log(n); // => 1, 2, 3 * } * ``` */export function tee<T, N extends number = 2>( iterable: AsyncIterable<T>, n: N = 2 as N,): Tuple<AsyncIterable<T>, N> { const queue = new Queue<T>(iterable);
async function* generator(): AsyncGenerator<T> { let buffer = queue.head; while (true) { if (buffer.next) { buffer = buffer.next; yield buffer.value; } else if (queue.done) { return; } else { await queue.next(); } } }
const branches = Array.from({ length: n }).map( () => generator(), ) as Tuple< AsyncIterable<T>, N >; return branches;}
std
Deno standard library
GitHub Repositorydenoland/deno_std
GitHub Stars
2458

Version Info

Tagged at
2 months ago