Carabao Dev - Start transaction on r2d2 connection checkout

r2d2_test_connect.rs

use diesel::{
    pg::PgConnection,
    r2d2::{ConnectionManager, Error as DieselPoolError},
    result::ConnectionError,
};
use r2d2::{CustomizeConnection, Pool};

#[derive(Debug)]
struct TestConnection;

impl CustomizeConnection<PgConnection, DieselPoolError> for TestConnection {
    fn on_acquire(&self, conn: &mut PgConnection) -> Result<(), DieselPoolError> {
        conn.begin_test_transaction().map_err(|err| {
            DieselPoolError::ConnectionError(ConnectionError::CouldntSetupConfiguration(err))
        })?;

        Ok(())
    }

    fn on_release(&self, _conn: PgConnection) {}
}


fn main() {
    let manager = ConnectionManager::<PgConnection>::new("MYPOSTGRESURL");

    Pool::builder()
        .max_size(*pool_size as u32)
        .build(manager)
        .unwrap();
}