如何使用生成器函数实现循环数组

Question

今天我想知道在 TypeScript 中提供循环数组最快捷的方法是什么，如下所示：

['one', 'two', 'three'] 

three 之后的下一个值是one，我认为它是生成器函数的一个很好的候选者。但是，它似乎对我不起作用。下面的代码有什么问题？

function* stepGen(){
  const steps = ['one', 'two', 'three'];

  let index = 0;

  if(index < steps.length - 1){
   index++;
  } else {
   index = 0;
  }
  yield steps[index];
}

let gen = stepGen();
console.log(gen.next().value); 
console.log(gen.next().value);
console.log(gen.next().value); // should be 'three'
console.log(gen.next().value); // should be 'one'
console.log(gen.next().value);

Answer 1

您的生成器代码中需要一个循环，否则只有一个 yield 发生：

function* stepGen(steps){
  let index = 0;
  while (true) {
    yield steps[index];
    index = (index+1)%steps.length;
  }
}

let gen = stepGen(['one', 'two', 'three']); // pass array to make it more reusable
console.log(gen.next().value); 
console.log(gen.next().value);
console.log(gen.next().value);
console.log(gen.next().value);
console.log(gen.next().value);

或者，您也可以使用yield*，它从一个可迭代对象中一一生成值：

function* stepGen(steps){
  while (true) yield* steps;
}

let gen = stepGen(['one', 'two', 'three']);
console.log(gen.next().value); 
console.log(gen.next().value);
console.log(gen.next().value);
console.log(gen.next().value);
console.log(gen.next().value);