Sep 22, 2025 · Apr 2, 2025 · Sep 22, 2025
diff --git a/tokio-postgres/src/client.rs b/tokio-postgres/src/client.rs
 use crate::codec::BackendMessages;
 use crate::codec::{BackendMessages, FrontendMessage};
 use crate::config::{SslMode, SslNegotiation};
 use crate::connection::{Request, RequestMessages};
 use crate::copy_out::CopyOutStream;
 use futures_util::{future, pin_mut, ready, StreamExt, TryStreamExt};
 use parking_lot::Mutex;
 use postgres_protocol::message::backend::Message;
 use postgres_protocol::message::frontend;
 use postgres_types::BorrowToSql;
 use std::collections::HashMap;
 use std::fmt;
        self.inner.sender.is_closed()
    }

    #[doc(hidden)]
    pub fn __private_api_rollback(&self, name: Option<&str>) {
        let buf = self.inner().with_buf(|buf| {
            if let Some(name) = name {
                frontend::query(&format!("ROLLBACK TO {}", name), buf).unwrap();
            } else {
                frontend::query("ROLLBACK", buf).unwrap();
            }
            buf.split().freeze()
        });
        let _ = self
            .inner()
            .send(RequestMessages::Single(FrontendMessage::Raw(buf)));
    }

    #[doc(hidden)]
    pub fn __private_api_close(&mut self) {
        self.inner.sender.close_channel()
diff --git a/tokio-postgres/src/transaction.rs b/tokio-postgres/src/transaction.rs
 use crate::codec::FrontendMessage;
 use crate::connection::RequestMessages;
 use crate::copy_out::CopyOutStream;
 use crate::query::RowStream;
 #[cfg(feature = "runtime")]
 };
 use bytes::Buf;
 use futures_util::TryStreamExt;
 use postgres_protocol::message::frontend;
 use tokio::io::{AsyncRead, AsyncWrite};

 /// A representation of a PostgreSQL database transaction.
            return;
        }

        let query = if let Some(sp) = self.savepoint.as_ref() {
            format!("ROLLBACK TO {}", sp.name)
        } else {
            "ROLLBACK".to_string()
        };
        let buf = self.client.inner().with_buf(|buf| {
            frontend::query(&query, buf).unwrap();
            buf.split().freeze()
        });
        let _ = self
            .client
            .inner()
            .send(RequestMessages::Single(FrontendMessage::Raw(buf)));
        let name = self.savepoint.as_ref().map(|sp| sp.name.as_str());
        self.client.__private_api_rollback(name);
    }
 }

diff --git a/tokio-postgres/src/transaction_builder.rs b/tokio-postgres/src/transaction_builder.rs
 use postgres_protocol::message::frontend;

 use crate::{codec::FrontendMessage, connection::RequestMessages, Client, Error, Transaction};
 use crate::{Client, Error, Transaction};

 /// The isolation level of a database transaction.
 #[derive(Debug, Copy, Clone)]
                    return;
                }

                let buf = self.client.inner().with_buf(|buf| {
                    frontend::query("ROLLBACK", buf).unwrap();
                    buf.split().freeze()
                });
                let _ = self
                    .client
                    .inner()
                    .send(RequestMessages::Single(FrontendMessage::Raw(buf)));
                self.client.__private_api_rollback(None);
            }
        }
Original file line number	Diff line number	Diff line change
		@@ -1,4 +1,4 @@
		use crate::codec::BackendMessages;
		use crate::codec::{BackendMessages, FrontendMessage};
		use crate::config::{SslMode, SslNegotiation};
		use crate::connection::{Request, RequestMessages};
		use crate::copy_out::CopyOutStream;
Expand All		@@ -22,6 +22,7 @@ use futures_channel::mpsc;
		use futures_util::{future, pin_mut, ready, StreamExt, TryStreamExt};
		use parking_lot::Mutex;
		use postgres_protocol::message::backend::Message;
		use postgres_protocol::message::frontend;
		use postgres_types::BorrowToSql;
		use std::collections::HashMap;
		use std::fmt;
Expand DownExpand Up		@@ -601,6 +602,21 @@ impl Client {
		self.inner.sender.is_closed()
		}

		#[doc(hidden)]
		pub fn __private_api_rollback(&self, name: Option<&str>) {
		let buf = self.inner().with_buf(\|buf\| {
		if let Some(name) = name {
		frontend::query(&format!("ROLLBACK TO {}", name), buf).unwrap();
		} else {
		frontend::query("ROLLBACK", buf).unwrap();
		}
		buf.split().freeze()
		});
		let _ = self
		.inner()
		.send(RequestMessages::Single(FrontendMessage::Raw(buf)));
		}
fredr marked this conversation as resolved. Show resolvedHide resolved

		#[doc(hidden)]
		pub fn __private_api_close(&mut self) {
		self.inner.sender.close_channel()
Expand Down
Original file line number	Diff line number	Diff line change
		@@ -1,5 +1,3 @@
		use crate::codec::FrontendMessage;
		use crate::connection::RequestMessages;
		use crate::copy_out::CopyOutStream;
		use crate::query::RowStream;
		#[cfg(feature = "runtime")]
Expand All		@@ -14,7 +12,6 @@ use crate::{
		};
		use bytes::Buf;
		use futures_util::TryStreamExt;
		use postgres_protocol::message::frontend;
		use tokio::io::{AsyncRead, AsyncWrite};

		/// A representation of a PostgreSQL database transaction.
Expand All		@@ -39,19 +36,8 @@ impl Drop for Transaction<'_> {
		return;
		}

		let query = if let Some(sp) = self.savepoint.as_ref() {
		format!("ROLLBACK TO {}", sp.name)
		} else {
		"ROLLBACK".to_string()
		};
		let buf = self.client.inner().with_buf(\|buf\| {
		frontend::query(&query, buf).unwrap();
		buf.split().freeze()
		});
		let _ = self
		.client
		.inner()
		.send(RequestMessages::Single(FrontendMessage::Raw(buf)));
		let name = self.savepoint.as_ref().map(\|sp\| sp.name.as_str());
		self.client.__private_api_rollback(name);
		}
		}

Expand Down
Original file line number	Diff line number	Diff line change
		@@ -1,6 +1,4 @@
		use postgres_protocol::message::frontend;

		use crate::{codec::FrontendMessage, connection::RequestMessages, Client, Error, Transaction};
		use crate::{Client, Error, Transaction};

		/// The isolation level of a database transaction.
		#[derive(Debug, Copy, Clone)]
Expand DownExpand Up		@@ -119,14 +117,7 @@ impl<'a> TransactionBuilder<'a> {
		return;
		}

		let buf = self.client.inner().with_buf(\|buf\| {
		frontend::query("ROLLBACK", buf).unwrap();
		buf.split().freeze()
		});
		let _ = self
		.client
		.inner()
		.send(RequestMessages::Single(FrontendMessage::Raw(buf)));
		self.client.__private_api_rollback(None);
		}
		}

Expand Down