docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-26 18:36:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

server version \d+ hack

Browse Source
This commit is contained in:
Lev Kokotov
2022-02-05 15:23:21 -08:00
parent 0e29d5e0ed
commit 7133d049c6
2 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/client.rs
View File

@@ -20,6 +20,7 @@ pub struct Client {
buffer: BytesMut,
name: String,
cancel_mode: bool,
transaction_mode: bool,
process_id: i32,
secret_key: i32,
client_server_map: ClientServerMap,
@@ -89,6 +90,7 @@ impl Client {
buffer: BytesMut::with_capacity(8196),
name: name,
cancel_mode: false,
transaction_mode: true,
process_id: process_id,
secret_key: secret_key,
client_server_map: client_server_map,
@@ -108,6 +110,7 @@ impl Client {
buffer: BytesMut::with_capacity(8196),
name: String::from("cancel_mode"),
cancel_mode: true,
transaction_mode: true,
process_id: process_id,
secret_key: secret_key,
client_server_map: client_server_map,
@@ -200,7 +203,7 @@ impl Client {
}
// Release server
if !server.in_transaction() {
if !server.in_transaction() && self.transaction_mode {
break;
}
}
@@ -253,7 +256,7 @@ impl Client {
}
// Release server
if !server.in_transaction() {
if !server.in_transaction() && self.transaction_mode {
break;
}
}
@@ -276,6 +279,12 @@ impl Client {
return Err(err);
}
};
// Release the server
if !server.in_transaction() && self.transaction_mode {
println!("Releasing after copy done");
break;
}
}
_ => {