sqlx-repo 0.1.2

repository pattern on top of sqlx
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
34
35
36
37
38
39
40
41
42
43

use sqlx::{Database, Error};
use std::{borrow::Cow, future::Future};

pub trait AcquireExt<D: Database> {
    fn start_transaction(
        &self,
    ) -> impl Future<Output = Result<sqlx::Transaction<'_, D>, Error>> + Send + 'static;
}

impl AcquireExt<sqlx::Sqlite> for sqlx::Pool<sqlx::Sqlite> {
    fn start_transaction(
        &self,
    ) -> impl Future<Output = Result<sqlx::Transaction<'_, sqlx::Sqlite>, Error>> + Send + 'static
    {
        let conn = self.acquire();

        async move {
            sqlx::Transaction::begin(conn.await?, Some(Cow::Borrowed("BEGIN IMMEDIATE"))).await
        }
    }
}

impl AcquireExt<sqlx::Postgres> for sqlx::Pool<sqlx::Postgres> {
    fn start_transaction(
        &self,
    ) -> impl Future<Output = Result<sqlx::Transaction<'_, sqlx::Postgres>, Error>> + Send + 'static
    {
        let conn = self.acquire();

        async move { sqlx::Transaction::begin(conn.await?, None).await }
    }
}

impl AcquireExt<sqlx::MySql> for sqlx::Pool<sqlx::MySql> {
    fn start_transaction(
        &self,
    ) -> impl Future<Output = Result<sqlx::Transaction<'_, sqlx::MySql>, Error>> + Send + 'static
    {
        let conn = self.acquire();

        async move { sqlx::Transaction::begin(conn.await?, None).await }
    }
}