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_id.dart'; |
10 import 'asset_node.dart'; | 10 import 'asset_node.dart'; |
11 | 11 |
12 /// 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 |
13 /// the set as soon as they're marked as [AssetState.REMOVED]. | 13 /// the set as soon as they're marked as [AssetState.REMOVED]. |
14 /// | 14 /// |
15 /// Asset nodes may be accessed by their ids. This means that only one node with | 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. | 16 /// a given id may be stored in the set at a time. |
17 class AssetNodeSet extends DelegatingSet<AssetNode> { | 17 class AssetNodeSet extends DelegatingSet<AssetNode> { |
18 // TODO(nweiz): Use DelegatingMapSet when issue 18705 is fixed. | 18 // TODO(nweiz): Use DelegatingMapSet when issue 18705 is fixed. |
19 /// A map from asset ids to assets in the set. | 19 /// A map from asset ids to assets in the set. |
20 final _assetsById = new Map<AssetId, AssetNode>(); | 20 final _assetsById = new Map<AssetId, AssetNode>(); |
21 | 21 |
22 AssetNodeSet() | 22 AssetNodeSet() : super(new Set()); |
23 : super(new Set()); | |
24 | 23 |
25 /// Returns the asset node in the set with [id], or `null` if none exists. | 24 /// Returns the asset node in the set with [id], or `null` if none exists. |
26 AssetNode operator [](AssetId id) => _assetsById[id]; | 25 AssetNode operator [](AssetId id) => _assetsById[id]; |
27 | 26 |
28 bool add(AssetNode node) { | 27 bool add(AssetNode node) { |
29 if (node.state.isRemoved) return false; | 28 if (node.state.isRemoved) return false; |
30 node.whenRemoved(() { | 29 node.whenRemoved(() { |
31 super.remove(node); | 30 super.remove(node); |
32 _assetsById.remove(node.id); | 31 _assetsById.remove(node.id); |
33 }); | 32 }); |
34 _assetsById[node.id] = node; | 33 _assetsById[node.id] = node; |
35 return super.add(node); | 34 return super.add(node); |
36 } | 35 } |
37 | 36 |
38 /// Returns whether an asset node with the given [id] is in the set. | 37 /// Returns whether an asset node with the given [id] is in the set. |
39 bool containsId(AssetId id) => _assetsById.containsKey(id); | 38 bool containsId(AssetId id) => _assetsById.containsKey(id); |
40 | 39 |
41 void addAll(Iterable<AssetNode> nodes) => nodes.forEach(add); | 40 void addAll(Iterable<AssetNode> nodes) => nodes.forEach(add); |
42 } | 41 } |
OLD | NEW |