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

fix syntax

Browse Source
This commit is contained in:
Mostafa
2024-09-01 15:41:33 -05:00
parent 686b7ca7c5
commit 982d03c374
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/client.rs
View File

@@ -1730,7 +1730,7 @@ where
fn buffer_parse(&mut self, message: BytesMut, pool: &ConnectionPool) -> Result<(), Error> { fn buffer_parse(&mut self, message: BytesMut, pool: &ConnectionPool) -> Result<(), Error> {
// Avoid parsing if prepared statements not enabled // Avoid parsing if prepared statements not enabled
let client_given_name = Parse::get_name(&message)?; let client_given_name = Parse::get_name(&message)?;
if !self.prepared_statements_enabled || client_given_name.len() == 0 { if !self.prepared_statements_enabled || client_given_name.is_empty() {
debug!("Anonymous parse message"); debug!("Anonymous parse message");
self.extended_protocol_data_buffer self.extended_protocol_data_buffer
.push_back(ExtendedProtocolData::create_new_parse(message, None)); .push_back(ExtendedProtocolData::create_new_parse(message, None));
@@ -1774,7 +1774,7 @@ where
async fn buffer_bind(&mut self, message: BytesMut) -> Result<(), Error> { async fn buffer_bind(&mut self, message: BytesMut) -> Result<(), Error> {
// Avoid parsing if prepared statements not enabled // Avoid parsing if prepared statements not enabled
let client_given_name = Bind::get_name(&message)?; let client_given_name = Bind::get_name(&message)?;
if !self.prepared_statements_enabled || client_given_name.len() == 0 { if !self.prepared_statements_enabled || client_given_name.is_empty() {
debug!("Anonymous bind message"); debug!("Anonymous bind message");
self.extended_protocol_data_buffer self.extended_protocol_data_buffer
.push_back(ExtendedProtocolData::create_new_bind(message, None)); .push_back(ExtendedProtocolData::create_new_bind(message, None));
@@ -1833,7 +1833,7 @@ where
let describe: Describe = (&message).try_into()?; let describe: Describe = (&message).try_into()?;
let client_given_name = describe.statement_name.clone(); let client_given_name = describe.statement_name.clone();
if describe.target == 'P' || client_given_name.len() == 0 { if describe.target == 'P' || client_given_name.is_empty() {
debug!("Portal describe message"); debug!("Portal describe message");
self.extended_protocol_data_buffer self.extended_protocol_data_buffer
.push_back(ExtendedProtocolData::create_new_describe(message, None)); .push_back(ExtendedProtocolData::create_new_describe(message, None));

14
src/messages.rs
View File

@@ -952,7 +952,7 @@ pub struct Bind {
num_param_values: u16, num_param_values: u16,
param_values: Vec<(i32, BytesMut)>, param_values: Vec<(i32, BytesMut)>,
num_result_column_format_codes: u16, num_result_column_format_codes: u16,
result_columns_format_codes: Vec<u16>, result_columns_format_codes: Vec<i16>,
} }
impl TryFrom<&BytesMut> for Bind { impl TryFrom<&BytesMut> for Bind {
@@ -968,14 +968,14 @@ impl TryFrom<&BytesMut> for Bind {
let mut param_format_codes = Vec::new(); let mut param_format_codes = Vec::new();
for _ in 0..num_param_format_codes { for _ in 0..num_param_format_codes {
param_format_codes.push(cursor.get_u16()); param_format_codes.push(cursor.get_i16());
} }
let num_param_values = cursor.get_u16(); let num_param_values = cursor.get_u16();
let mut param_values = Vec::new(); let mut param_values = Vec::new();
for _ in 0..num_param_values { for _ in 0..num_param_values {
let param_len = cursor.get_u32(); let param_len = cursor.get_i32();
// There is special occasion when the parameter is NULL // There is special occasion when the parameter is NULL
// In that case, param length is defined as -1 // In that case, param length is defined as -1
// So if the passed parameter len is over 0 // So if the passed parameter len is over 0
@@ -997,7 +997,7 @@ impl TryFrom<&BytesMut> for Bind {
let mut result_columns_format_codes = Vec::new(); let mut result_columns_format_codes = Vec::new();
for _ in 0..num_result_column_format_codes { for _ in 0..num_result_column_format_codes {
result_columns_format_codes.push(cursor.get_u16()); result_columns_format_codes.push(cursor.get_i16());
} }
Ok(Bind { Ok(Bind {
@@ -1046,16 +1046,16 @@ impl TryFrom<Bind> for BytesMut {
bytes.put_slice(prepared_statement); bytes.put_slice(prepared_statement);
bytes.put_u16(bind.num_param_format_codes); bytes.put_u16(bind.num_param_format_codes);
for param_format_code in bind.param_format_codes { for param_format_code in bind.param_format_codes {
bytes.put_u16(param_format_code); bytes.put_i16(param_format_code);
} }
bytes.put_u16(bind.num_param_values); bytes.put_u16(bind.num_param_values);
for (param_len, param) in bind.param_values { for (param_len, param) in bind.param_values {
bytes.put_u32(param_len); bytes.put_i32(param_len);
bytes.put_slice(&param); bytes.put_slice(&param);
} }
bytes.put_u16(bind.num_result_column_format_codes); bytes.put_u16(bind.num_result_column_format_codes);
for result_column_format_code in bind.result_columns_format_codes { for result_column_format_code in bind.result_columns_format_codes {
bytes.put_u16(result_column_format_code); bytes.put_i16(result_column_format_code);
} }
Ok(bytes) Ok(bytes)