在子迭代器上分配迭代器項

Question

我正在玩JavaScript中的迭代器。 我找到了鏈接它們的方法，因此不必為了可讀性而嵌套它們。

它的工作原理完全類似於數組映射鏈。

我想做的是將生成器的項目分布在幾個子迭代器上，捕獲結果並將它們提供給下一個迭代器。 這樣的結果將是這樣的：

4
6
8
7
9
11

給出這段代碼：

"use strict";

const _ = require('lodash');

let things = chain([
    gen,
    addOne,
    distribute([
        addOne,
        addTwo,
        addThree
    ]),
    addOne
]);

for(let thing of things) {
    console.log(thing);
}

//////////////// DEFENITIONS ////////////////////

function* gen() {
    yield* [1, 2, 3, 4, 5, 6];
}

function* addOne(iterator) {
    for(let item of iterator) {
        yield (item + 1)
    }
}

function* addTwo(iterator) {
    for(let item of iterator) {
        yield (item + 2)
    }
}

function* addThree(iterator) {
    for(let item of iterator) {
        yield (item + 3)
    }
}

const distribute = _.curry(function* (iterators, iterator) {
    // magic
});

function chain(iterators) {
    return iterators.reduce((prevIterator, thisIterator, index) => {
        if(index === 0) {
            return thisIterator();
        }
        return thisIterator(prevIterator);
    }, null);
}

最終，我想向分布迭代器添加一個分布函數，以便它可以確定將哪個項目傳遞給哪個子迭代器。 現在，它只是基於順序。

問題：如何編寫一個將幾個子迭代器作為參數並將其傳遞給下一個迭代器的分布迭代器。

Answer 1

似乎過於復雜，但是有效

const distribute = _.curry(function* (iterators, mainIterator) {
    let iteratorIndex = 0;
    let done = [];
    for(let iterator of iterators) {
        iterators[iteratorIndex] = iterator(mainIterator);
        done.push(false);
        iteratorIndex++;
    }
    while(true) {
        let iteratorIndex = 0;
        for(let iterator of iterators) {
            let next = iterator.next();
            done[iteratorIndex] = next.done;
            if(!next.done) {
                yield next.value;
            }
            iteratorIndex++;
        }
        if(done.every(done => done)) {
            return;
        }
    }
});

最后是分配函數：

const distribute = _.curry(function* (genIteratorIndex, iterators, mainIterator) {
    let iteratorIndex = 0;
    let done = [];

    // instantiate iterators
    for(let iterator of iterators) {
        iterators[iteratorIndex] = iterator(mainIterator);
        done.push(false);
        iteratorIndex++;
    }

    // Pass stuff through
    while(true) {
        let next = iterators[genIteratorIndex.next().value].next();
        done[iteratorIndex] = next.done;
        if(!next.done) {
            yield next.value;
        }
        if(done.every(done => done === true)) {
            return;
        }
    }
});

function* subSequent(len) {
    let curr = 0;
    while(true) {
        if(curr === len) {
            curr = 0;
        }
        yield curr;
        curr++;
    }
}


let things = chain([
    gen,
    addOne,
    distribute(subSequent(3), [
        addOne,
        addTwo,
        addThree
    ]),
    addOne
]);