如何进行异步迭代多播(&Q)?

是否可以以某种方式广播或多播async生成器,从而使其所有迭代器(&q;使用者&q;?订户?)是否接收所有值?

考虑此示例:

const fetchMock = () => "Example. Imagine real fetch";
async function* gen() {
  for (let i = 1; i <= 6; i++) {
    const res = await fetchMock();
    yield res.slice(0, 2) + i;
  }
}
const ait = gen();

(async() => {
  // first "consumer"
  for await (const e of ait) console.log('e', e);
})();
(async() => {
  // second...
  for await (const é of ait) console.log('é', é);
})();

迭代&消耗&q;一个值,因此只有一个或另一个获得该值。 如果这样的生成器能够以某种方式生成,我希望它们(以及以后的任何一个)都能获得每个yielded的值。(类似于Observable。)


解决方案

这不容易实现。您需要显式tee它。这与the situation for synchronous iterators类似,只是稍微复杂一点:

const AsyncIteratorProto = Object.getPrototypeOf(Object.getPrototypeOf(async function*(){}.prototype));
function teeAsync(iterable) {
    const iterator = iterable[Symbol.asyncIterator]();
    const buffers = [[], []];
    function makeIterator(buffer, i) {
        return Object.assign(Object.create(AsyncIteratorProto), {
            next() {
                if (!buffer) return Promise.resolve({done: true, value: undefined});
                if (buffer.length) return buffer.shift();
                const res = iterator.next();
                if (buffers[i^1]) buffers[i^1].push(res);
                return res;
            },
            async return() {
                if (buffer) {
                    buffer = buffers[i] = null;
                    if (!buffers[i^1]) await iterator.return();
                }
                return {done: true, value: undefined};
            },
        });
    }
    return buffers.map(makeIterator);
}

您应该确保两个迭代器的消耗速度大致相同,以便缓冲区不会变得太大。

相关文章