aboutsummaryrefslogtreecommitdiffstats
path: root/storage/src/vespa/storage/distributor/bucket_space_state_map.h
diff options
context:
space:
mode:
Diffstat (limited to 'storage/src/vespa/storage/distributor/bucket_space_state_map.h')
-rw-r--r--storage/src/vespa/storage/distributor/bucket_space_state_map.h3
1 files changed, 3 insertions, 0 deletions
diff --git a/storage/src/vespa/storage/distributor/bucket_space_state_map.h b/storage/src/vespa/storage/distributor/bucket_space_state_map.h
index 57eac9eac0d..6209f9f306c 100644
--- a/storage/src/vespa/storage/distributor/bucket_space_state_map.h
+++ b/storage/src/vespa/storage/distributor/bucket_space_state_map.h
@@ -64,6 +64,9 @@ public:
StateMap::const_iterator begin() const { return _map.begin(); }
StateMap::const_iterator end() const { return _map.end(); }
+ const BucketSpaceState& get(document::BucketSpace space) const;
+ BucketSpaceState& get(document::BucketSpace space);
+
void set_cluster_state(std::shared_ptr<const lib::ClusterState> cluster_state);
void set_distribution(std::shared_ptr<const lib::Distribution> distribution);