Skip to content
This repository was archived by the owner on Jan 14, 2025. It is now read-only.

Commit 99908ea

Browse files
authored
Close after sync on txt query (#7)
* close after sync on txt query * fix: Try not to flush after describe
1 parent 72545c3 commit 99908ea

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tokio-postgres/src/query.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,6 @@ where
8282
// prepare
8383
frontend::parse("", query, std::iter::empty(), buf).map_err(Error::encode)?;
8484
frontend::describe(b'S', "", buf).map_err(Error::encode)?;
85-
frontend::flush(buf);
8685

8786
// Bind, pass params as text, retrieve as binary
8887
match frontend::bind(
@@ -110,6 +109,9 @@ where
110109
// Sync
111110
frontend::sync(buf);
112111

112+
// Close
113+
frontend::close(b'S', "", buf).map_err(Error::encode)?;
114+
113115
Ok(buf.split().freeze())
114116
})?;
115117

@@ -210,7 +212,7 @@ async fn start(client: &InnerClient, buf: Bytes) -> Result<(Option<Statement>, R
210212

211213
loop {
212214
match responses.next().await? {
213-
Message::ParseComplete => {}
215+
Message::CloseComplete | Message::ParseComplete => {}
214216
Message::BindComplete => return Ok((statement, responses)),
215217
Message::ParameterDescription(body) => {
216218
parameter_description = Some(body); // tooo-o-ooo-o loooove

tokio-postgres/src/simple_query.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ pub async fn batch_execute(client: &InnerClient, query: &str) -> Result<(), Erro
5656
Message::ReadyForQuery(_) => return Ok(()),
5757
Message::CommandComplete(_)
5858
| Message::EmptyQueryResponse
59+
| Message::CloseComplete
5960
| Message::RowDescription(_)
6061
| Message::DataRow(_) => {}
6162
m => return Err(Error::unexpected_message(m)),

0 commit comments

Comments
 (0)