docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-27 10:46:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

comments

Browse Source
This commit is contained in:
Lev Kokotov
2022-02-04 16:08:18 -08:00
parent fa46e15318
commit 52c4cac175
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/client.rs
View File

@@ -149,7 +149,10 @@ impl Client {
let mut proxy = pool.get().await.unwrap(); let mut proxy = pool.get().await.unwrap();
let server = &mut *proxy; let server = &mut *proxy;
// TODO: maybe don't do this, I don't think it's useful.
server.set_name(&self.name).await?; server.set_name(&self.name).await?;
// Claim this server as mine for query cancellation.
server.claim(self.process_id, self.secret_key); server.claim(self.process_id, self.secret_key);
loop { loop {
@@ -280,6 +283,7 @@ impl Client {
} }
} }
/// Release the server from being mine. I can't cancel its queries anymore.
pub fn release(&mut self) { pub fn release(&mut self) {
let mut guard = self.client_server_map.lock().unwrap(); let mut guard = self.client_server_map.lock().unwrap();
guard.remove(&(self.process_id, self.secret_key)); guard.remove(&(self.process_id, self.secret_key));

1
src/server.rs
View File

@@ -305,6 +305,7 @@ impl Server {
self.bad = true; self.bad = true;
} }
/// Claim this server as mine for the purposes of query cancellation.
pub fn claim(&mut self, process_id: i32, secret_key: i32) { pub fn claim(&mut self, process_id: i32, secret_key: i32) {
let mut guard = self.client_server_map.lock().unwrap(); let mut guard = self.client_server_map.lock().unwrap();
guard.insert((process_id, secret_key), (self.backend_id, self.secret_key)); guard.insert((process_id, secret_key), (self.backend_id, self.secret_key));