OLD | NEW |
1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file |
2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
4 | 4 |
5 library barback.asset.asset_node_set; | 5 library barback.asset.asset_node_set; |
6 | 6 |
7 import 'package:collection/collection.dart'; | 7 import 'package:collection/collection.dart'; |
8 | 8 |
| 9 import 'asset_id.dart'; |
9 import 'asset_node.dart'; | 10 import 'asset_node.dart'; |
10 | 11 |
11 /// A set of [AssetNode]s that automatically ensures that nodes are removed from | 12 /// A set of [AssetNode]s that automatically ensures that nodes are removed from |
12 /// the set as soon as they're marked as [AssetState.REMOVED]. | 13 /// the set as soon as they're marked as [AssetState.REMOVED]. |
| 14 /// |
| 15 /// Asset nodes may be accessed by their ids. This means that only one node with |
| 16 /// a given id may be stored in the set at a time. |
13 class AssetNodeSet extends DelegatingSet<AssetNode> { | 17 class AssetNodeSet extends DelegatingSet<AssetNode> { |
| 18 // TODO(nweiz): Use DelegatingMapSet when issue 18705 is fixed. |
| 19 /// A map from asset ids to assets in the set. |
| 20 final _assetsById = new Map<AssetId, AssetNode>(); |
| 21 |
14 AssetNodeSet() | 22 AssetNodeSet() |
15 : super(new Set()); | 23 : super(new Set()); |
16 | 24 |
| 25 /// Returns the asset node in the set with [id], or `null` if none exists. |
| 26 AssetNode operator [](AssetId id) => _assetsById[id]; |
| 27 |
17 bool add(AssetNode node) { | 28 bool add(AssetNode node) { |
18 if (node.state.isRemoved) return false; | 29 if (node.state.isRemoved) return false; |
19 node.whenRemoved(() => super.remove(node)); | 30 node.whenRemoved(() { |
| 31 super.remove(node); |
| 32 _assetsById.remove(node.id); |
| 33 }); |
| 34 _assetsById[node.id] = node; |
20 return super.add(node); | 35 return super.add(node); |
21 } | 36 } |
22 | 37 |
| 38 /// Returns whether an asset node with the given [id] is in the set. |
| 39 bool containsId(AssetId id) => _assetsById.containsKey(id); |
| 40 |
23 void addAll(Iterable<AssetNode> nodes) => nodes.forEach(add); | 41 void addAll(Iterable<AssetNode> nodes) => nodes.forEach(add); |
24 } | 42 } |
OLD | NEW |