From 4a9c94514f49aa4e9880a8e0f5cf5a52d11ae993 Mon Sep 17 00:00:00 2001
From: Alex Auvolat <alex@adnab.me>
Date: Wed, 8 Nov 2023 16:41:00 +0100
Subject: avoid using layout_watch in System directly

---
 src/garage/admin/bucket.rs |  4 ++--
 src/garage/admin/mod.rs    | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

(limited to 'src/garage')

diff --git a/src/garage/admin/bucket.rs b/src/garage/admin/bucket.rs
index 34e48292..9e642f57 100644
--- a/src/garage/admin/bucket.rs
+++ b/src/garage/admin/bucket.rs
@@ -70,7 +70,7 @@ impl AdminRpcHandler {
 			.table
 			.get(&bucket_id, &EmptyKey)
 			.await?
-			.map(|x| x.filtered_values(&self.garage.system.layout_watch.borrow()))
+			.map(|x| x.filtered_values(&self.garage.system.cluster_layout()))
 			.unwrap_or_default();
 
 		let mpu_counters = self
@@ -79,7 +79,7 @@ impl AdminRpcHandler {
 			.table
 			.get(&bucket_id, &EmptyKey)
 			.await?
-			.map(|x| x.filtered_values(&self.garage.system.layout_watch.borrow()))
+			.map(|x| x.filtered_values(&self.garage.system.cluster_layout()))
 			.unwrap_or_default();
 
 		let mut relevant_keys = HashMap::new();
diff --git a/src/garage/admin/mod.rs b/src/garage/admin/mod.rs
index 006f71cd..c3fa801a 100644
--- a/src/garage/admin/mod.rs
+++ b/src/garage/admin/mod.rs
@@ -126,7 +126,7 @@ impl AdminRpcHandler {
 			opt_to_send.all_nodes = false;
 
 			let mut failures = vec![];
-			let layout = self.garage.system.layout_watch.borrow().clone();
+			let layout = self.garage.system.cluster_layout().clone();
 			for node in layout.node_ids().iter() {
 				let node = (*node).into();
 				let resp = self
@@ -163,7 +163,7 @@ impl AdminRpcHandler {
 	async fn handle_stats(&self, opt: StatsOpt) -> Result<AdminRpc, Error> {
 		if opt.all_nodes {
 			let mut ret = String::new();
-			let layout = self.garage.system.layout_watch.borrow().clone();
+			let layout = self.garage.system.cluster_layout().clone();
 
 			for node in layout.node_ids().iter() {
 				let mut opt = opt.clone();
@@ -275,7 +275,7 @@ impl AdminRpcHandler {
 		let mut ret = String::new();
 
 		// Gather storage node and free space statistics
-		let layout = &self.garage.system.layout_watch.borrow();
+		let layout = &self.garage.system.cluster_layout();
 		let mut node_partition_count = HashMap::<Uuid, u64>::new();
 		for short_id in layout.ring_assignment_data.iter() {
 			let id = layout.node_id_vec[*short_id as usize];
@@ -440,7 +440,7 @@ impl AdminRpcHandler {
 	) -> Result<AdminRpc, Error> {
 		if all_nodes {
 			let mut ret = vec![];
-			let layout = self.garage.system.layout_watch.borrow().clone();
+			let layout = self.garage.system.cluster_layout().clone();
 			for node in layout.node_ids().iter() {
 				let node = (*node).into();
 				match self
@@ -488,7 +488,7 @@ impl AdminRpcHandler {
 	) -> Result<AdminRpc, Error> {
 		if all_nodes {
 			let mut ret = vec![];
-			let layout = self.garage.system.layout_watch.borrow().clone();
+			let layout = self.garage.system.cluster_layout().clone();
 			for node in layout.node_ids().iter() {
 				let node = (*node).into();
 				match self
-- 
cgit v1.2.3