Index: pkg/barback/lib/src/asset/asset_node_set.dart |
diff --git a/pkg/barback/lib/src/asset/asset_node_set.dart b/pkg/barback/lib/src/asset/asset_node_set.dart |
index 4474866553473ea96eb5754b27e51044a5527178..a1fc17bebcc9952dd07c2e10173244f2b4dd4afb 100644 |
--- a/pkg/barback/lib/src/asset/asset_node_set.dart |
+++ b/pkg/barback/lib/src/asset/asset_node_set.dart |
@@ -6,19 +6,37 @@ library barback.asset.asset_node_set; |
import 'package:collection/collection.dart'; |
+import 'asset_id.dart'; |
import 'asset_node.dart'; |
/// A set of [AssetNode]s that automatically ensures that nodes are removed from |
/// the set as soon as they're marked as [AssetState.REMOVED]. |
+/// |
+/// Asset nodes may be accessed by their ids. This means that only one node with |
+/// a given id may be stored in the set at a time. |
class AssetNodeSet extends DelegatingSet<AssetNode> { |
+ // TODO(nweiz): Use DelegatingMapSet when issue 18705 is fixed. |
+ /// A map from asset ids to assets in the set. |
+ final _assetsById = new Map<AssetId, AssetNode>(); |
+ |
AssetNodeSet() |
: super(new Set()); |
+ /// Returns the asset node in the set with [id], or `null` if none exists. |
+ AssetNode operator [](AssetId id) => _assetsById[id]; |
+ |
bool add(AssetNode node) { |
if (node.state.isRemoved) return false; |
- node.whenRemoved(() => super.remove(node)); |
+ node.whenRemoved(() { |
+ super.remove(node); |
+ _assetsById.remove(node.id); |
+ }); |
+ _assetsById[node.id] = node; |
return super.add(node); |
} |
+ /// Returns whether an asset node with the given [id] is in the set. |
+ bool containsId(AssetId id) => _assetsById.containsKey(id); |
+ |
void addAll(Iterable<AssetNode> nodes) => nodes.forEach(add); |
} |