Question

我一直在努力解决这个问题。

如何编写一个异步生成器，给定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)

Answer 1

所以......从字面上看，几天后试图解决这个问题。这样做（只需将粘贴复制到最新的镀铬实验标记打开）

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

同时等待2个（或更多）异步生成器

1 个答案: