前言

村里的老人说：“真正的强者，都是扮猪吃老虎。”

日常开发中经常需要用到弱引用，Dart 语言里也有提供弱引用的接口 ```WeakReference```，我们可以基于它开发更强大的复杂结构。

在前面的文章中，我们用到了一个以弱引用对象为元素的集合 ```WeakSet```，今天我们就来讲一下它是如何实现的。

Collections

常用的 collections 有 Set、List 和 Map，下面我将为大家介绍其对应的弱引用版本该如何快速实现。

WeakSet

弱集合也是集合，所以它应该实现一个集合接口 ```Set``` 的所有功能。

直接上代码：

class WeakSet<E extends Object> implements Set<E> {WeakSet() : _inner = {};final Set<WeakReference<E>> _inner;@overridevoid clear() => _inner.clear();@overrideSet<E> toSet() {Set<E> entries = {};E? item;for (WeakReference<E> wr in _inner) {item = wr.target;if (item != null) {entries.add(item);}}return entries;}@overrideList<E> toList({bool growable = true}) => toSet().toList(growable: growable);@overrideString toString() => toSet().toString();@overrideIterator<E> get iterator => toSet().iterator;@overrideint get length => toSet().length;@overridebool get isEmpty => toSet().isEmpty;@overridebool get isNotEmpty => toSet().isNotEmpty;@overridebool contains(Object? value) => toSet().contains(value);@overridebool containsAll(Iterable<Object?> other) => toSet().containsAll(other);@overrideE elementAt(int index) => toSet().elementAt(index);@overridebool add(E value) => !contains(value) && _inner.add(WeakReference(value));@overridebool remove(Object? value) {for (var wr in _inner) {if (wr.target == value) {return _inner.remove(wr);}}return false;}@overridevoid forEach(void Function(E element) action) => toSet().forEach(action);// 从略...}

首先，我们创建一个内置的集合 _inner，其元素是指向元素类型 E 的弱引用；
接下来实现 toSet() 函数，将 _inner 中所有实际指向的对象提取出来，构建一个新的集合返回；
然后其余的 Set 函数接口都可以基于这个 toSet() 来实现。

同样的思路，我们还可以继续实现列表与映射的弱引用版本。

WeakList

先实现最简单的那些接口函数：

class WeakList<E extends Object> implements List<E> {WeakList() : _inner = [];final List<WeakReference<E>> _inner;@overridevoid clear() => _inner.clear();@overrideList<E> toList({bool growable = true}) {List<E> entries = [];E? item;for (WeakReference<E> wr in _inner) {item = wr.target;if (item != null) {entries.add(item);}}return entries;}@overrideSet<E> toSet() => toList().toSet();@overrideString toString() => toList().toString();@overrideIterator<E> get iterator => toList().iterator;@overrideint get length => toList().length;@overridebool get isEmpty => toList().isEmpty;@overridebool get isNotEmpty => toList().isNotEmpty;@overrideE get first => toList().first;@overrideset first(E item) => _inner.first = WeakReference(item);@overrideE get last => toList().last;@overrideset last(E item) => _inner.last = WeakReference(item);@overrideE get single => toList().single;@overrideList<E> operator +(List<E> other) => toList() + other;@overrideE operator [](int index) => toList()[index];@overridevoid add(E value) => _inner.add(WeakReference(value));@overridevoid addAll(Iterable<E> iterable) {for (var item in iterable) {add(item);}}@overridebool contains(Object? element) => toList().contains(element);@overrideE elementAt(int index) => toList().elementAt(index);@overridevoid forEach(void Function(E element) action) => toList().forEach(action);@overrideint indexOf(E element, [int start = 0]) => toList().indexOf(element, start);@overrideint lastIndexOf(E element, [int? start]) => toList().lastIndexOf(element, start);@overrideString join([String separator = ""]) => toList().join(separator);@overridebool remove(Object? value) {/// Removes the first occurrence of [value] from this list.for (var wr in _inner) {if (wr.target == value) {return _inner.remove(wr);}}return false;}// 从略...}

跟上面的 WeakSet 类似，这里也是先创建一个内部的列表 _inner，其元素是指向实际对象的弱引用；接下来实现 toList() 函数，将内部列表中的实际指向对象提取出来构建新的列表；
然后其余关于 List 的接口函数基本都可以基于这个 toList() 函数来实现。

但是也有不同之处。由于 Set 是无序的，所以即使其中的弱引用所指向的对象已销毁，也不会影响集合运算。
而 List 是有序的，所以一些跟序号相关的操作则需要忽略那些对象已销毁的弱引用元素。
这里我采用的犯法是增加一个 purge() 函数，用于清除那些“无用”的元素，然后再基于它去实现那些和序号相关的接口函数。

class WeakList<E extends Object> implements List<E> {WeakList() : _inner = [];final List<WeakReference<E>> _inner;void purge() => _inner.removeWhere((wr) => wr.target == null);@overrideset length(int size) {purge();_inner.length = size;}@overridevoid operator []=(int index, E value) {purge();_inner[index] = WeakReference(value);}@overridevoid insert(int index, E element) {purge();_inner.insert(index, WeakReference(element));}@overridevoid insertAll(int index, Iterable<E> iterable) {purge();for (var item in iterable) {_inner.insert(index, WeakReference(item));index += 1;}}@overrideE removeAt(int index) {purge();var wr = _inner.removeAt(index);return wr.target;}@overrideE removeLast() {purge();var wr = _inner.removeLast();return wr.target;}// 从略...}

WeakValueMap

Talk is cheap, show you the codes!

class WeakValueMap<K, V> implements Map<K, V> {WeakValueMap() : _inner = {};final Map<K, WeakReference<dynamic>?> _inner;void purge() => _inner.removeWhere((key, wr) => wr?.target == null);@overridevoid clear() => _inner.clear();Map<K, V> toMap() {// remove empty entriespurge();// convert entriesreturn _inner.map((key, wr) => MapEntry(key, wr?.target));}@overrideString toString() => toMap().toString();@overridebool containsValue(Object? value) => toMap().containsValue(value);@overridebool containsKey(Object? key) => _inner[key]?.target != null;@overrideV? operator [](Object? key) => _inner[key]?.target;@overridevoid operator []=(K key, V value) =>_inner[key] = value == null ? null : WeakReference(value);@overrideIterable<MapEntry<K, V>> get entries => toMap().entries;@overridevoid addAll(Map<K, V> other) => other.forEach((key, value) {this[key] = value;});@overrideV? remove(Object? key) => _inner.remove(key)?.target;@overridevoid forEach(void Function(K key, V value) action) => _inner.forEach((key, wr) {V val = wr?.target;if (val != null) {action(key, val);}});@overrideIterable<K> get keys => toMap().keys;@overrideIterable<V> get values => toMap().values;@overrideint get length => toMap().length;@overridebool get isEmpty => toMap().isEmpty;@overridebool get isNotEmpty => toMap().isNotEmpty;// 从略...}

课余练习

• WeakKeyMap 如何实现？

提示：可以先实现一个用于包裹 key 的类，以 key 对应的对象为参数初始化（内部用一个弱引用指向它），同时将 key 对象的 hash 值等信息保存起来，以便后面 key 对象消失时仍然可以进行相应的运算。

应用示例

以上类的使用很简单，根普通的 Set、List、Map 几乎没有差别：

// WeakSet
Set<Observer> listenerSet = WeakSet();
listenerSet.add(obs1);
listenerSet.add(obs2);// WeakList
List<Observer> listenerList = WeakList();
listenerList.add(obs1);
listenerList.add(obs2);// WeakMap
Map<String, Observer> listenerMap = WeakValueMap();
listenerMap['Albert'] = obs1;
listenerMap['Ben'] = obs2;

代码引用

由于我已将这部分代码提交到了 pub.dev，所以在实际应用中，你只需要在项目工程文件 ```pubspec.yaml``` 中添加：

dependencies:
    object_key: ^0.1.1

然后在需要使用的 dart 文件头引入即可：

import 'package:object_key/object_key.dart';

全部源码

GitHub 地址：

https://github.com/moky/ObjectKey/tree/main/object_key/lib/src

结语

弱引用可以有效避免对象循环引用等原因导致的内存泄漏问题。
在某些实践中，还可以实现类似“自动退出”的效果，比如前面介绍的观察者模式的实战中，合理使用基于弱引用的集合，可以达到当观察者销毁时即自动注销的效果，十分方便！

如有其他问题，可以下载登录 Tarsier 与我交流（默认通讯录里找 Albert Moky / 章北海）