Clean connection state up after protocol named prepared statement (#163)

* Clean connection state up after protocol named prepared statement

* Avoid cloning + add test

* fmt
This commit is contained in:
Mostafa Abdelraouf
2022-09-07 22:37:17 -05:00
committed by GitHub
parent 6d41640ea9
commit 9514b3b2d1
3 changed files with 29 additions and 0 deletions

View File

@@ -879,6 +879,23 @@ where
self.buffer.put(&original[..]);
// Clone after freeze does not allocate
let first_message_code = (*self.buffer.get(0).unwrap_or(&0)) as char;
// Almost certainly true
if first_message_code == 'P' {
// Message layout
// P followed by 32 int followed by null-terminated statement name
// So message code should be in offset 0 of the buffer, first character
// in prepared statement name would be index 5
let first_char_in_name = *self.buffer.get(5).unwrap_or(&0);
if first_char_in_name != 0 {
// This is a named prepared statement
// Server connection state will need to be cleared at checkin
server.mark_dirty();
}
}
self.send_and_receive_loop(
code,
self.buffer.clone(),