javascript - 修改对象的 for...of 循环行为

Question

鉴于 ES6 中新的“Symbol”原语，是否可以修改/设置对象中 for...of 循环的行为？

我正在创建一个小实用程序，用于从“可迭代”(我将其定义为对象、数组、映射或集合)中“深度提取”值。

Array、Map 和 Set 使用 for..of 循环，但普通对象则不然。为了保持一致性，我希望对象能够使用此循环(并且应该迭代值而不是属性(正如 for...in 允许您这样做))。

这可以做到吗？

Answer 1

好吧，你可以这样做(关于全局原型(prototype)的常见警告适用):

Object.prototype[Symbol.iterator] = function* () {
    yield* Object.values(this);
};

x = {a: 1, b: 2}

for (let y of x)
    console.log(y)

或

Object.defineProperty(
    Object.prototype,
    Symbol.iterator, {
        value: function*() {
            yield* Object.values(this);
        }
    }
);

myObj = {x: 1, y: 2}
console.log([...myObj])

但是，一个更简洁的解决方案是引入一个通用包装器，该包装器可以对本身不可迭代的对象启用迭代:

let iter = function* (a) {
    if (a === null || a === undefined)
        return;

    if (!a[Symbol.iterator]) {
        if (typeof a === 'object')
            a = Object.values(a);
        else
            a = [a];
    }

    for (let x of a)
        yield x;
};

console.log([...iter('abcd')])
console.log([...iter(123)])
console.log([...iter({a: 1, b: 2})])

然后只需在代码中使用 for (x of iter(whatever)) 即可。