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

Question

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

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

Array、Map 和 Set 使用 for..of 循环，但普通对象没有。为了一致性，我希望对象使用这个循环（并且应该迭代值而不是属性（因为 for...in 允许你这样做））。

这可以做到吗？

Answer 1

好吧，你可以做这样的事情（关于全局原型的通常警告适用）：

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))在您的代码中使用。