同时等待 2 个(或更多)异步生成器

waiting on 2 (or more) async generators simultaneously

我为此苦恼了一段时间。

你会如何编写一个异步生成器,给定 n 个异步生成器,按照它们到达的顺序从 n 个生成器中生成值。

方法应采用以下格式: const merged = merge([gen1, gen2,...])

例如:

gen1 and gen2 are async generators.
gen1() yields 2 values: 1 in 100ms, 3 in 300ms  
gen2() yields 2 values: 2 in 200ms, 4 in 400ms

merge([gen1, gen2]) will yield 1, 2, 3, 4 (in that order, every 100ms)

所以...从字面上看,经过几天的努力解决这个问题。 这有效(只需将实验性标志打开,将粘贴复制到最新的 chrome)

function timer(time = 500) {
  return new Promise(resolve => setTimeout(() => resolve(), time));
}

async function* gen1() {
  await timer(100);
  yield 1;
  await timer(300);
  yield 4;
}

async function* gen2() {
  await timer(200);
  yield 2;
  await timer(100);
  yield 3;
}

function race(promises) {
  return new Promise(resolve =>
    promises.forEach((p, index) => {
      p.then(value => {
        resolve({index, value});
      });
    })
  );
}
async function* mergen(...gens) {
  let promises = gens.map((gen, index) =>
    gen.next().then(p => ({...p, gen}))
  );

  while (promises.length > 0) {
    yield race(promises)
    .then(({index, value: {value, done, gen}}) => {
      promises.splice(index, 1);
      if (!done)
        promises.push(
          gen.next().then(({value: newVal, done: newDone}) => ({
            value: newVal,
            done: newDone,
            gen
          }))
        );
      return value;
    });
  }
}

async function printGen(gen) {
  let max = 10;
  for await (x of gen) {
    if (x) console.log('Next up:', x);
    if (--max <= 0) break;
  }
}

printGen(mergen(gen1(), gen2())); // 1, 2, 3, 4

它是重构前的,所以请记住,它还不够漂亮和干净。 有趣的代码叫做 mergen()(明白了吗?merge-gen?) 它也使用此处包含的 Promise.race() 的修改版本。 使用修改后的 race 的原因是因为我需要先完成的承诺的索引。

更新: 现在它是一个 npm 模块 https://github.com/hesher/mergen