JavaScript高级程序设计(第4版)
上QQ阅读APP看书,第一时间看更新

6.5 WeakMap

ECMAScript 6新增的“弱映射”(WeakMap)是一种新的集合类型,为这门语言带来了增强的键/值对存储机制。WeakMap是Map的“兄弟”类型,其API也是Map的子集。WeakMap中的“weak”(弱),描述的是JavaScript垃圾回收程序对待“弱映射”中键的方式。

6.5.1 基本API

可以使用new关键字实例化一个空的WeakMap:

    const wm = new WeakMap();

弱映射中的键只能是Object或者继承自Object的类型,尝试使用非对象设置键会抛出TypeError。值的类型没有限制。

如果想在初始化时填充弱映射,则构造函数可以接收一个可迭代对象,其中需要包含键/值对数组。可迭代对象中的每个键/值都会按照迭代顺序插入新实例中:

    const key1 = {id: 1},
          key2 = {id: 2},
          key3 = {id: 3};
    // 使用嵌套数组初始化弱映射
    const wm1 = new WeakMap([
      [key1, "val1"],
      [key2, "val2"],
      [key3, "val3"]
    ]);
    alert(wm1.get(key1)); // val1
    alert(wm1.get(key2)); // val2
    alert(wm1.get(key3)); // val3
    // 初始化是全有或全无的操作
    // 只要有一个键无效就会抛出错误,导致整个初始化失败
    const wm2 = new WeakMap([
      [key1, "val1"],
      ["BADKEY", "val2"],
      [key3, "val3"]
    ]);
    // TypeError: Invalid value used as WeakMap key
    typeof wm2;
    // ReferenceError: wm2 is not defined
    // 原始值可以先包装成对象再用作键
    const stringKey = new String("key1");
    const wm3 = new WeakMap([
      stringKey, "val1"
    ]);
    alert(wm3.get(stringKey)); // "val1"

初始化之后可以使用set()再添加键/值对,可以使用get()和has()查询,还可以使用delete()删除:

    const wm = new WeakMap();
    const key1 = {id: 1},
          key2 = {id: 2};
    alert(wm.has(key1)); // false
    alert(wm.get(key1)); // undefined
    wm.set(key1, "Matt")
      .set(key2, "Frisbie");
    alert(wm.has(key1)); // true
    alert(wm.get(key1)); // Matt
    wm.delete(key1);      // 只删除这一个键/值对
    alert(wm.has(key1)); // false
    alert(wm.has(key2)); // true

set()方法返回弱映射实例,因此可以把多个操作连缀起来,包括初始化声明:

    const key1 = {id: 1},
          key2 = {id: 2},
          key3 = {id: 3};
    const wm = new WeakMap().set(key1, "val1");
    wm.set(key2, "val2")
      .set(key3, "val3");
    alert(wm.get(key1)); // val1
    alert(wm.get(key2)); // val2
    alert(wm.get(key3)); // val3

6.5.2 弱键

WeakMap中“weak”表示弱映射的键是“弱弱地拿着”的。意思就是,这些键不属于正式的引用,不会阻止垃圾回收。但要注意的是,弱映射中值的引用可不是“弱弱地拿着”的。只要键存在,键/值对就会存在于映射中,并被当作对值的引用,因此就不会被当作垃圾回收。

来看下面的例子:

    const wm = new WeakMap();
    wm.set({}, "val");

set()方法初始化了一个新对象并将它用作一个字符串的键。因为没有指向这个对象的其他引用,所以当这行代码执行完成后,这个对象键就会被当作垃圾回收。然后,这个键/值对就从弱映射中消失了,使其成为一个空映射。在这个例子中,因为值也没有被引用,所以这对键/值被破坏以后,值本身也会成为垃圾回收的目标。

再看一个稍微不同的例子:

    const wm = new WeakMap();
    const container = {
      key: {}
    };
    wm.set(container.key, "val");
    function removeReference() {
      container.key = null;
    }

这一次,container对象维护着一个对弱映射键的引用,因此这个对象键不会成为垃圾回收的目标。不过,如果调用了removeReference(),就会摧毁键对象的最后一个引用,垃圾回收程序就可以把这个键/值对清理掉。

6.5.3 不可迭代键

因为WeakMap中的键/值对任何时候都可能被销毁,所以没必要提供迭代其键/值对的能力。当然,也用不着像clear()这样一次性销毁所有键/值的方法。WeakMap确实没有这个方法。因为不可能迭代,所以也不可能在不知道对象引用的情况下从弱映射中取得值。即便代码可以访问WeakMap实例,也没办法看到其中的内容。

WeakMap实例之所以限制只能用对象作为键,是为了保证只有通过键对象的引用才能取得值。如果允许原始值,那就没办法区分初始化时使用的字符串字面量和初始化之后使用的一个相等的字符串了。

6.5.4 使用弱映射

WeakMap实例与现有JavaScript对象有着很大不同,可能一时不容易说清楚应该怎么使用它。这个问题没有唯一的答案,但已经出现了很多相关策略。

1.私有变量

弱映射造就了在JavaScript中实现真正私有变量的一种新方式。前提很明确:私有变量会存储在弱映射中,以对象实例为键,以私有成员的字典为值。

下面是一个示例实现:

    const wm = new WeakMap();
    class User {
      constructor(id) {
        this.idProperty = Symbol('id');
        this.setId(id);
      }
      setPrivate(property, value) {
        const privateMembers = wm.get(this) || {};
        privateMembers[property] = value;
        wm.set(this, privateMembers);
      }
      getPrivate(property) {
        return wm.get(this)[property];
      }
      setId(id) {
        this.setPrivate(this.idProperty, id);
      }
      getId() {
        return this.getPrivate(this.idProperty);
      }
    }
    const user = new User(123);
    alert(user.getId()); // 123
    user.setId(456);
    alert(user.getId()); // 456
    // 并不是真正私有的
    alert(wm.get(user)[user.idProperty]); // 456

慧眼独具的读者会发现,对于上面的实现,外部代码只需要拿到对象实例的引用和弱映射,就可以取得“私有”变量了。为了避免这种访问,可以用一个闭包把WeakMap包装起来,这样就可以把弱映射与外界完全隔离开了:

    constUser=(()=>{
      const wm = new WeakMap();
      class User {
        constructor(id) {
          this.idProperty = Symbol('id');
          this.setId(id);
        }
        setPrivate(property, value) {
          const privateMembers = wm.get(this) || {};
          privateMembers[property] = value;
          wm.set(this, privateMembers);
        }
        getPrivate(property) {
          return wm.get(this)[property];
        }
        setId(id) {
          this.setPrivate(this.idProperty, id);
        }
        getId(id) {
          return this.getPrivate(this.idProperty);
        }
      }
      returnUser;
    })();
    const user = new User(123);
    alert(user.getId()); // 123
    user.setId(456);
    alert(user.getId()); // 456

这样,拿不到弱映射中的健,也就无法取得弱映射中对应的值。虽然这防止了前面提到的访问,但整个代码也完全陷入了ES6之前的闭包私有变量模式。

2.DOM节点元数据

因为WeakMap实例不会妨碍垃圾回收,所以非常适合保存关联元数据。来看下面这个例子,其中使用了常规的Map:

    const m = new Map();
    const loginButton = document.querySelector('#login');
    // 给这个节点关联一些元数据
    m.set(loginButton, {disabled: true});

假设在上面的代码执行后,页面被JavaScript改变了,原来的登录按钮从DOM树中被删掉了。但由于映射中还保存着按钮的引用,所以对应的DOM节点仍然会逗留在内存中,除非明确将其从映射中删除或者等到映射本身被销毁。

如果这里使用的是弱映射,如以下代码所示,那么当节点从DOM树中被删除后,垃圾回收程序就可以立即释放其内存(假设没有其他地方引用这个对象):

    const wm = new WeakMap();
    const loginButton = document.querySelector('#login');
    // 给这个节点关联一些元数据
    wm.set(loginButton, {disabled: true});