| 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 |