rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-01-05 20:42:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix query_string parsing for mgmt callback

Browse Source
This commit is contained in:
Stas Kelvich
2021-06-30 12:35:07 +03:00
parent aa404b60fe
commit 560f088f05
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
proxy/src/mgmt.rs
View File

@@ -78,15 +78,14 @@ impl postgres_backend::Handler for MgmtHandler {
pgb: &mut PostgresBackend,
query_string: Bytes,
) -> anyhow::Result<()> {
let (_, query_string) = query_string
.split_last()
.ok_or_else(|| anyhow::Error::msg("protocol violation"))?;
let mut query_string = query_string.to_vec();
if let Some(ch) = query_string.last() {
if *ch == 0 {
query_string.pop();
}
}
let (_, query_string) = query_string
.split_last()
.ok_or_else(|| anyhow::Error::msg("protocol violation"))?;
println!("Got mgmt query: {:?}", query_string);
println!("Got mgmt query: '{}'", std::str::from_utf8(&query_string)?);
let resp: PsqlSessionResponse = serde_json::from_slice(&query_string)?;
@@ -94,7 +93,7 @@ impl postgres_backend::Handler for MgmtHandler {
let sender = waiters
.get(&resp.session_id)
.ok_or_else(|| anyhow::Error::msg("psql session_id is not found"))?;
.ok_or_else(|| anyhow::Error::msg("psql_session_id is not found"))?;
match resp.result {
PsqlSessionResult::Success(db_info) => {