Как написать итератор для вывода нечетных чисел используя [Symbol.iterator]?

Question

[Vladislav]

Необходимо вывести нечетные числа не изменяя текущий код.

const arr = Object.freeze([0, 1, 2, 3, 4, 5, 6, 7, 8])

// code

for (item of arr) {
    console.log(item) // 1,3,5,7
}

Код, который работает некорректно, т.е. выводит все числа от 0 до 8:

const arr = Object.freeze([0, 1, 2, 3, 4, 5, 6, 7, 8])


arr[Symbol.iterator] = function () {
    let count = 0;
    const array = arr;
    const newArr = []

    for (let i = 0; i <= array.length; i++) {
        if (array[i] % 2 == 1) {
            newArr.push(array[i])
        }
    }
    return {
        next() {
            if (count <= newArr.length - 1) {
                return {
                    done: false,
                    value: newArr[count++]
                };
            } else {
                return {
                    done: true
                };
            }
        }
    }
};

for (item of arr) {
    console.log(item) // 1,3,5,7
}

Comments

[Fedor Vlasenko]

const arr = Object.freeze([0, 1, 2, 3, 4, 5, 6, 7, 8])
console.log(arr.filter(v=>v&1))

Незачем так извращаться, так как функционал бессмыслен
достаточно применить filter по последнему биту

[Vladislav]

Fedor Vlasenko, согласен, но такая вот задача )

Answer 1

[0xD34F]

Код, который работает некорректно...

Всё работает так, как и должно. После Object.freeze назначить новое свойство уже не получится. Можно записать итератор в прототип массива (ну и после того, как он отработает, вернуть оригинальный итератор на место):

const originalIterator = Array.prototype[Symbol.iterator];

Array.prototype[Symbol.iterator] = function() {
  let i = -1;

  return {
    next: () => (i += 2) < arr.length
      ? { done: false, value: this[i] }
      : { done: true },
  };
};

setTimeout(() => Array.prototype[Symbol.iterator] = originalIterator, 0);

Comments

[Alexandroppolus]

после того, как он отработает, верните оригинальный итератор на место

оригинальный итератор может понадобиться прямо во вложенном цикле. Так что лучше сделать "обёртку с итератором"

[0xD34F]

Alexandroppolus, не понадобится. Это учебная задача, суть которой - догадаться, как можно обойти freeze. Такое моё мнение.

[Alexandroppolus]

0xD34F, хех, точно. Я как-то завтыкал суть вопроса. Но тогда и возвращать старый итератор не надо, код всё равно "одноразовый"

[Vladislav]

0xD34F, Спасибо! Да, задача учебная и мой код работает действительно!
Я запускал код в VScode и в консоль ничего не падало, скопировал код в консоль браузера и все гуд!

Answer 2

[Alexandroppolus]

Зачем в итераторе filter? Это убивает всю идею. Надо помнить индекс последнего найденного числа, и искать следующее прямо в функции next(), начиная от сохраненного