rust_orm_gen 0.2.0

A comprehensive Rust ORM generator with schema visualization, real-time monitoring, and multiple output formats
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use tokio_postgres::{Client, Transaction};
use std::future::Future;
use std::pin::Pin;

pub struct TransactionManager<'a> {
    client: &'a mut Client,
}

impl<'a> TransactionManager<'a> {
    pub fn new(client: &'a mut Client) -> Self {
        TransactionManager { client }
    }

    pub async fn run<F, T, E>(&mut self, f: F) -> Result<T, Box<dyn std::error::Error>>
    where
        F: for<'b> FnOnce(&'b mut Transaction<'b>) -> Pin<Box<dyn Future<Output = Result<T, E>> + Send + 'b>>,
        E: std::error::Error + 'static,
    {
        let mut transaction = self.client.transaction().await?;
        let result = f(&mut transaction).await;

        match result {
            Ok(value) => {
                transaction.commit().await?;
                Ok(value)
            },
            Err(e) => {
                transaction.rollback().await?;
                Err(Box::new(e))
            }
        }
    }
}