germanubis/sqlx: 🧰 The Rust SQL Toolkit. An async, pure Rust SQL crate featuring compile-time checked queries without a DSL. Supports PostgreSQL, MySQL, and SQLite.

sqlx-sqlite/src/transaction.rs

48 строк1 KB

Austin Bonander

Merge of #3427 (by @mpyw) and #3614 (by @bonsairobo) (#3765)

11 мар 2025, 00:29

11 мар 2025, 00:29393b731

use futures_core::future::BoxFuture;
use std::borrow::Cow;

use sqlx_core::error::Error;
use sqlx_core::transaction::TransactionManager;

use crate::{Sqlite, SqliteConnection};

/// Implementation of [`TransactionManager`] for SQLite.
pub struct SqliteTransactionManager;

impl TransactionManager for SqliteTransactionManager {
    type Database = Sqlite;

    fn begin<'conn>(
        conn: &'conn mut SqliteConnection,
        statement: Option<Cow<'static, str>>,
    ) -> BoxFuture<'conn, Result<(), Error>> {
        Box::pin(async {
            let is_custom_statement = statement.is_some();
            conn.worker.begin(statement).await?;
            if is_custom_statement {
                // Check that custom statement actually put the connection into a transaction.
                let mut handle = conn.lock_handle().await?;
                if !handle.in_transaction() {
                    return Err(Error::BeginFailed);
                }
            }
            Ok(())
        })
    }

    fn commit(conn: &mut SqliteConnection) -> BoxFuture<'_, Result<(), Error>> {
        Box::pin(conn.worker.commit())
    }

    fn rollback(conn: &mut SqliteConnection) -> BoxFuture<'_, Result<(), Error>> {
        Box::pin(conn.worker.rollback())
    }

    fn start_rollback(conn: &mut SqliteConnection) {
        conn.worker.start_rollback().ok();
    }

    fn get_transaction_depth(conn: &SqliteConnection) -> usize {
        conn.worker.shared.get_transaction_depth()
    }
}