dnim/rust-postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use the client buffer for cleanup

Browse Source
This commit is contained in:
Steven Fackler 2019-10-12 18:07:09 -07:00
parent 20adbaf36b
commit 9e42c67b7b
3 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tokio-postgres/src/portal.rs
View File

@ -2,7 +2,6 @@ use crate::client::InnerClient;
use crate::codec::FrontendMessage;
use crate::connection::RequestMessages;
use crate::Statement;
use bytes::BytesMut;
use postgres_protocol::message::frontend;
use std::sync::{Arc, Weak};
@ -15,10 +14,12 @@ struct Inner {
impl Drop for Inner {
fn drop(&mut self) {
if let Some(client) = self.client.upgrade() {
let mut buf = BytesMut::new();
frontend::close(b'P', &self.name, &mut buf).expect("portal name not valid");
frontend::sync(&mut buf);
let _ = client.send(RequestMessages::Single(FrontendMessage::Raw(buf.freeze())));
let buf = client.with_buf(|buf| {
frontend::close(b'P', &self.name, buf).unwrap();
frontend::sync(buf);
buf.take().freeze()
});
let _ = client.send(RequestMessages::Single(FrontendMessage::Raw(buf)));
}
}
}

11
tokio-postgres/src/statement.rs
View File

@ -2,7 +2,6 @@ use crate::client::InnerClient;
use crate::codec::FrontendMessage;
use crate::connection::RequestMessages;
use crate::types::Type;
use bytes::BytesMut;
use postgres_protocol::message::frontend;
use std::sync::{Arc, Weak};
@ -16,10 +15,12 @@ struct StatementInner {
impl Drop for StatementInner {
fn drop(&mut self) {
if let Some(client) = self.client.upgrade() {
let mut buf = BytesMut::new();
frontend::close(b'S', &self.name, &mut buf).expect("statement name not valid");
frontend::sync(&mut buf);
let _ = client.send(RequestMessages::Single(FrontendMessage::Raw(buf.freeze())));
let buf = client.with_buf(|buf| {
frontend::close(b'S', &self.name, buf).unwrap();
frontend::sync(buf);
buf.take().freeze()
});
let _ = client.send(RequestMessages::Single(FrontendMessage::Raw(buf)));
}
}
}

9
tokio-postgres/src/transaction.rs
View File

@ -14,7 +14,6 @@ use crate::{
use bytes::IntoBuf;
use futures::{TryStream, TryStreamExt};
use postgres_protocol::message::frontend;
use postgres_types::private::BytesMut;
use std::error;
use tokio::io::{AsyncRead, AsyncWrite};
@ -34,17 +33,19 @@ impl<'a> Drop for Transaction<'a> {
return;
}
let mut buf = BytesMut::new();
let query = if self.depth == 0 {
"ROLLBACK".to_string()
} else {
format!("ROLLBACK TO sp{}", self.depth)
};
frontend::query(&query, &mut buf).unwrap();
let buf = self.client.inner().with_buf(|buf| {
frontend::query(&query, buf).unwrap();
buf.take().freeze()
});
let _ = self
.client
.inner()
.send(RequestMessages::Single(FrontendMessage::Raw(buf.freeze())));
.send(RequestMessages::Single(FrontendMessage::Raw(buf)));
}
}