From 499612dd76551bc1c79bd5d2522b677e6e26ebc7 Mon Sep 17 00:00:00 2001 From: Mostafa Abdelraouf Date: Wed, 3 Aug 2022 20:16:53 -0500 Subject: [PATCH] Add user to SHOW STATS query (#108) * Add user to SHOW STATS query * user_name => username --- src/admin.rs | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/admin.rs b/src/admin.rs index 831ca0b..170c851 100644 --- a/src/admin.rs +++ b/src/admin.rs @@ -370,6 +370,7 @@ where { let columns = vec![ ("database", DataType::Text), + ("user", DataType::Text), ("total_xact_count", DataType::Numeric), ("total_query_count", DataType::Numeric), ("total_received", DataType::Numeric), @@ -390,7 +391,7 @@ where let mut res = BytesMut::new(); res.put(row_description(&columns)); - for (_, pool) in get_all_pools() { + for ((_db_name, username), pool) in get_all_pools() { for shard in 0..pool.shards() { for server in 0..pool.servers(shard) { let address = pool.address(shard, server); @@ -400,8 +401,9 @@ where }; let mut row = vec![address.name()]; + row.push(username.clone()); - for column in &columns[1..] { + for column in &columns[2..] { row.push(stats.get(column.0).unwrap_or(&0).to_string()); }