From d426f586b428a009c4260a35d3a37951ff1c0d62 Mon Sep 17 00:00:00 2001
From: Mostafa Abdelraouf <mostafa.abdelraouf@instacart.com>
Date: Mon, 8 Aug 2022 08:19:42 -0500
Subject: [PATCH] Report banned addresses as disabled

---
 src/admin.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/admin.rs b/src/admin.rs
index 170c851f..5c820ee7 100644
--- a/src/admin.rs
+++ b/src/admin.rs
@@ -252,10 +252,10 @@ where
         let pool_config = pool.settings.clone();
         for shard in 0..pool.shards() {
             let database_name = &pool_config.shards[&shard.to_string()].database;
-
             for server in 0..pool.servers(shard) {
                 let address = pool.address(shard, server);
                 let pool_state = pool.pool_state(shard, server);
+                let banned = pool.is_banned(address, shard, Some(address.role));
 
                 res.put(data_row(&vec![
                     address.name(),                         // name
@@ -270,7 +270,11 @@ where
                     pool_config.user.pool_size.to_string(), // max_connections
                     pool_state.connections.to_string(),     // current_connections
                     "0".to_string(),                        // paused
-                    "0".to_string(),                        // disabled
+                    match banned {
+                        // disabled
+                        true => "1".to_string(),
+                        false => "0".to_string(),
+                    },
                 ]));
             }
         }