rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-01-17 10:22:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(libs/proxy): make pending_repsonses an option instead of vecdeque, as it can only store 1 item

Browse Source
This commit is contained in:
Conrad Ludgate
2025-01-02 11:10:42 +00:00
committed by Conrad Ludgate
parent db1de41a43
commit 1d644b826d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libs/proxy/tokio-postgres2/src/connection.rs
View File

@@ -54,7 +54,7 @@ pub struct Connection<S, T> {
/// HACK: we need this in the Neon Proxy to forward params.
pub parameters: HashMap<String, String>,
receiver: mpsc::UnboundedReceiver<Request>,
pending_responses: VecDeque<BackendMessage>,
pending_responses: Option<BackendMessage>,
responses: VecDeque<Response>,
state: State,
}
@@ -73,7 +73,7 @@ where
stream,
parameters,
receiver,
pending_responses: VecDeque::new(),
pending_responses: None,
responses: VecDeque::new(),
state: State::Active,
}
@@ -83,7 +83,7 @@ where
&mut self,
cx: &mut Context<'_>,
) -> Poll<Option<Result<BackendMessage, Error>>> {
if let Some(message) = self.pending_responses.pop_front() {
if let Some(message) = self.pending_responses.take() {
trace!("retrying pending response");
return Poll::Ready(Some(Ok(message)));
}
@@ -157,7 +157,7 @@ where
}
Poll::Pending => {
self.responses.push_front(response);
self.pending_responses.push_back(BackendMessage::Normal {
self.pending_responses = Some(BackendMessage::Normal {
messages,
request_complete,
});