diff --git a/.gitignore b/.gitignore index 3c654539..cf28bffe 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,7 @@ .idea /target *.deb -.vscode \ No newline at end of file +.vscode +.profraw +cov/ +lcov.info diff --git a/src/admin.rs b/src/admin.rs index 71d3e486..406b0fe0 100644 --- a/src/admin.rs +++ b/src/admin.rs @@ -100,6 +100,10 @@ where trace!("SHOW VERSION"); show_version(stream).await } + "USERS" => { + trace!("SHOW USERS"); + show_users(stream).await + } _ => error_response(stream, "Unsupported SHOW query against the admin database").await, }, _ => error_response(stream, "Unsupported query against the admin database").await, @@ -666,3 +670,32 @@ where } } } + +/// Show Users. +async fn show_users(stream: &mut T) -> Result<(), Error> +where + T: tokio::io::AsyncWrite + std::marker::Unpin, +{ + let mut res = BytesMut::new(); + + res.put(row_description(&vec![ + ("name", DataType::Text), + ("pool_mode", DataType::Text), + ])); + + for (user_pool, pool) in get_all_pools() { + let pool_config = &pool.settings; + res.put(data_row(&vec![ + user_pool.user.clone(), + pool_config.pool_mode.to_string(), + ])); + } + + res.put(command_complete("SHOW")); + + res.put_u8(b'Z'); + res.put_i32(5); + res.put_u8(b'I'); + + write_all_half(stream, &res).await +} diff --git a/tests/ruby/admin_spec.rb b/tests/ruby/admin_spec.rb index 2000e5d9..78364156 100644 --- a/tests/ruby/admin_spec.rb +++ b/tests/ruby/admin_spec.rb @@ -286,4 +286,14 @@ connections.map(&:close) end end + + describe "SHOW users" do + it "returns the right users" do + admin_conn = PG::connect(processes.pgcat.admin_connection_string) + results = admin_conn.async_exec("SHOW USERS")[0] + admin_conn.close + expect(results["name"]).to eq("sharding_user") + expect(results["pool_mode"]).to eq("transaction") + end + end end