sqlx_sqlite/
transaction.rs

1use futures_core::future::BoxFuture;
2use std::borrow::Cow;
3
4use sqlx_core::error::Error;
5use sqlx_core::transaction::TransactionManager;
6
7use crate::{Sqlite, SqliteConnection};
8
9/// Implementation of [`TransactionManager`] for SQLite.
10pub struct SqliteTransactionManager;
11
12impl TransactionManager for SqliteTransactionManager {
13    type Database = Sqlite;
14
15    fn begin<'conn>(
16        conn: &'conn mut SqliteConnection,
17        statement: Option<Cow<'static, str>>,
18    ) -> BoxFuture<'conn, Result<(), Error>> {
19        Box::pin(async {
20            let is_custom_statement = statement.is_some();
21            conn.worker.begin(statement).await?;
22            if is_custom_statement {
23                // Check that custom statement actually put the connection into a transaction.
24                let mut handle = conn.lock_handle().await?;
25                if !handle.in_transaction() {
26                    return Err(Error::BeginFailed);
27                }
28            }
29            Ok(())
30        })
31    }
32
33    fn commit(conn: &mut SqliteConnection) -> BoxFuture<'_, Result<(), Error>> {
34        Box::pin(conn.worker.commit())
35    }
36
37    fn rollback(conn: &mut SqliteConnection) -> BoxFuture<'_, Result<(), Error>> {
38        Box::pin(conn.worker.rollback())
39    }
40
41    fn start_rollback(conn: &mut SqliteConnection) {
42        conn.worker.start_rollback().ok();
43    }
44
45    fn get_transaction_depth(conn: &SqliteConnection) -> usize {
46        conn.worker.shared.get_transaction_depth()
47    }
48}
sqlx_sqlite/transaction.rs

sqlx_sqlite/
transaction.rs
