carryctx 0.0.3

Persistent project context for coding agents
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use rusqlite::{Connection, Transaction};

use crate::error::CarryCtxError;

/// Wraps a rusqlite immediate-mode transaction.
/// Provides access to the underlying connection for repository operations.
pub struct UnitOfWork<'conn> {
    tx: Option<Transaction<'conn>>,
}

impl<'conn> UnitOfWork<'conn> {
    pub fn new(tx: Transaction<'conn>) -> Self {
        Self { tx: Some(tx) }
    }

    /// Return a reference to the inner Connection.
    pub fn connection(&self) -> &Connection {
        self.tx.as_ref().expect("UnitOfWork already finalized")
    }

    /// Commit the transaction.
    pub fn commit(mut self) -> Result<(), CarryCtxError> {
        let tx = self.tx.take().expect("UnitOfWork already finalized");
        tx.commit().map_err(|e| {
            CarryCtxError::database_error(format!("Transaction commit failed: {e}")).with_source(e)
        })
    }

    /// Roll back the transaction.
    pub fn rollback(mut self) -> Result<(), CarryCtxError> {
        if let Some(tx) = self.tx.take() {
            tx.rollback().map_err(|e| {
                CarryCtxError::database_error(format!("Transaction rollback failed: {e}"))
                    .with_source(e)
            })
        } else {
            Ok(())
        }
    }
}

impl Drop for UnitOfWork<'_> {
    fn drop(&mut self) {
        if self.tx.is_some() {
            let _ = self.tx.take().unwrap().rollback();
        }
    }
}