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

Extended query protocol sharding (#339)

Browse Source 
* Prepared stmt sharding

s

tests

* len check

* remove python test

* latest rust

* move that to debug for sure

* Add the actual tests

* latest image

* Update tests/ruby/sharding_spec.rb
This commit is contained in:
Lev Kokotov
2023-03-10 07:55:22 -08:00
committed by GitHub
parent 76e195a8a4
commit b4baa86e8a
7 changed files with 320 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/client.rs
View File

@@ -675,14 +675,42 @@ where
// allocate a connection, we wouldn't be able to send back an error message
// to the client so we buffer them and defer the decision to error out or not
// to when we get the S message
'P' | 'B' | 'D' | 'E' => {
'D' | 'E' => {
self.buffer.put(&message[..]);
continue;
}
'Q' => {
if query_router.query_parser_enabled() {
query_router.infer(&message);
}
}
'P' => {
self.buffer.put(&message[..]);
if query_router.query_parser_enabled() {
query_router.infer(&message);
}
continue;
}
'B' => {
self.buffer.put(&message[..]);
if query_router.query_parser_enabled() {
query_router.infer_shard_from_bind(&message);
}
continue;
}
'X' => {
debug!("Client disconnecting");
return Ok(());
}
_ => (),
}
@@ -711,11 +739,7 @@ where
// Handle all custom protocol commands, if any.
match query_router.try_execute_command(&message) {
// Normal query, not a custom command.
None => {
if query_router.query_parser_enabled() {
query_router.infer(&message);
}
}
None => (),
// SET SHARD TO
Some((Command::SetShard, _)) => {
@@ -727,7 +751,7 @@ where
error_response(
&mut self.write,
&format!(
"shard {} is more than configured {}, staying on shard {}",
"shard {} is more than configured {}, staying on shard {} (shard numbers start at 0)",
query_router.shard(),
pool.shards(),
current_shard,