30秒学会 JavaScript 片段 – Flat iterator

Creates a generator that iterates over an iterable, flattening nested iterables.

• Use recursion.
• Use a for...of loop to iterate over the values of the given iterable.
• Use Symbol.iterator to check if each value is an iterable. If it is, use the yield* expression to recursively delegate to the same generator function. Otherwise, yield the current value.

代码实现

const flatIterator = function* (itr) {
  for (let item of itr) {
    if (item[Symbol.iterator]) yield* flatIterator(item);
    else yield item;
  }
};

const arr = [1, 2, [3, 4], [5, [6, [7], 8]], 9, new Set([10, 11])];
[...flatIterator(arr)]; // 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11

翻译自：https://www.30secondsofcode.org/js/s/flat-iterator