use futures_core::future::BoxFuture;
use crate::error::Error;
use crate::executor::Executor;
use crate::postgres::{PgConnection, Postgres};
use crate::transaction::{
begin_ansi_transaction_sql, commit_ansi_transaction_sql, rollback_ansi_transaction_sql,
TransactionManager,
};
pub struct PgTransactionManager;
impl TransactionManager for PgTransactionManager {
type Database = Postgres;
fn begin(conn: &mut PgConnection) -> BoxFuture<'_, Result<(), Error>> {
Box::pin(async move {
conn.execute(&*begin_ansi_transaction_sql(conn.transaction_depth))
.await?;
conn.transaction_depth += 1;
Ok(())
})
}
fn commit(conn: &mut PgConnection) -> BoxFuture<'_, Result<(), Error>> {
Box::pin(async move {
if conn.transaction_depth > 0 {
conn.execute(&*commit_ansi_transaction_sql(conn.transaction_depth))
.await?;
conn.transaction_depth -= 1;
}
Ok(())
})
}
fn rollback(conn: &mut PgConnection) -> BoxFuture<'_, Result<(), Error>> {
Box::pin(async move {
if conn.transaction_depth > 0 {
conn.execute(&*rollback_ansi_transaction_sql(conn.transaction_depth))
.await?;
conn.transaction_depth -= 1;
}
Ok(())
})
}
fn start_rollback(conn: &mut PgConnection) {
if conn.transaction_depth > 0 {
conn.queue_simple_query(&rollback_ansi_transaction_sql(conn.transaction_depth));
conn.transaction_depth -= 1;
}
}
}