Skip to main content

Db

Struct Db 

Source
pub struct Db { /* private fields */ }
Expand description

The database pool. Take it in a handler with State(db): State<Db>.

Models, queries and sql() take &db (or &mut tx inside a Transaction). For something only sqlx can do, reach the underlying pool with Db::sqlite or Db::postgres (with the postgres feature).

Implementations§

Source§

impl Db

Source

pub fn dialect(&self) -> Dialect

The engine this pool talks to.

Source

pub fn sqlite(&self) -> Option<&SqlitePool>

The sqlx pool, when the database is SQLite.

Source

pub fn postgres(&self) -> Option<&PgPool>

The sqlx pool, when the database is PostgreSQL.

Source

pub async fn begin(&self) -> Result<Transaction, DbError>

Starts a transaction. Pass &mut tx wherever a &db goes, then tx.commit(); dropping it without committing rolls it back.

Source

pub async fn transaction<T, F>(&self, work: F) -> Result<T>
where F: for<'t> FnMut(&'t mut Transaction) -> BoxFuture<'t, Result<T>>,

Runs work in a transaction: committed when it returns Ok, rolled back when it returns Err.

let moved = db
    .transaction_retrying(3, |tx| {
        Box::pin(async move {
            renox::db::sql("UPDATE accounts SET balance = balance - 100 WHERE id = 1")
                .execute(&mut *tx)
                .await?;
            renox::db::sql("UPDATE accounts SET balance = balance + 100 WHERE id = 2")
                .execute(&mut *tx)
                .await?;
            Ok(100)
        })
    })
    .await?;
Source

pub async fn transaction_retrying<T, F>( &self, attempts: u32, work: F, ) -> Result<T>
where F: for<'t> FnMut(&'t mut Transaction) -> BoxFuture<'t, Result<T>>,

Like Db::transaction, trying up to attempts times when another transaction got in the way (SQLite busy, PostgreSQL serialization failure or deadlock), with a short, growing pause in between. work must be safe to run again.

Source

pub fn retrying<'a, T, F, Fut>( &'a self, attempts: u32, work: F, ) -> impl Future<Output = Result<T>> + Send + 'a
where T: Send + 'a, F: FnMut() -> Fut + Send + 'a, Fut: Future<Output = Result<T>> + Send + 'a,

Runs work again (up to attempts times, with a short, growing pause) while it fails because another transaction got in the way (SQLite busy, PostgreSQL serialization failure or deadlock). work opens and commits its own transaction, so unlike Db::transaction_retrying it can borrow from the caller, and it can roll back and still return a value (drop or rollback the transaction instead of committing).

enum Transfer { Done, Short }

let outcome = db
    .retrying(3, || async {
        let mut tx = db.begin().await?;
        let taken = renox::db::sql(
            "UPDATE accounts SET balance = balance - ? WHERE id = ? AND balance >= ?",
        )
        .bind(amount) // borrowed from the caller
        .bind(from)
        .bind(amount)
        .execute(&mut tx)
        .await?;
        if taken == 0 {
            tx.rollback().await?;
            return Ok(Transfer::Short); // rolled back, with an answer
        }
        renox::db::sql("UPDATE accounts SET balance = balance + ? WHERE id = ?")
            .bind(amount)
            .bind(to)
            .execute(&mut tx)
            .await?;
        tx.commit().await?;
        Ok(Transfer::Done)
    })
    .await?;
Source

pub async fn begin_immediate(&self) -> Result<Transaction, DbError>

A transaction that takes SQLite’s write lock at once (BEGIN IMMEDIATE), so what it reads can’t change before it writes: the SQLite counterpart of lock_for_update. Same as begin on PostgreSQL.

Source

pub async fn close(&self)

Closes every connection; later queries fail.

Trait Implementations§

Source§

impl Clone for Db

Source§

fn clone(&self) -> Self

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl Debug for Db

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
Source§

impl<'c> Executor<'c> for &'c Db

Source§

impl From<Pool<Postgres>> for Db

Available on crate feature postgres only.
Source§

fn from(pool: PgPool) -> Self

Converts to this type from the input type.
Source§

impl From<Pool<Sqlite>> for Db

Source§

fn from(pool: SqlitePool) -> Self

Converts to this type from the input type.
Source§

impl FromRef<AppState> for Db

Source§

fn from_ref(state: &AppState) -> Self

Converts to this type from a reference to the input type.

Auto Trait Implementations§

§

impl !RefUnwindSafe for Db

§

impl !UnwindSafe for Db

§

impl Freeze for Db

§

impl Send for Db

§

impl Sync for Db

§

impl Unpin for Db

§

impl UnsafeUnpin for Db

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> Fake for T

Source§

fn fake<U>(&self) -> U
where Self: FakeBase<U>,

Source§

fn fake_with_rng<U, R>(&self, rng: &mut R) -> U
where R: RngExt + ?Sized, Self: FakeBase<U>,

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> FromRef<T> for T
where T: Clone,

Source§

fn from_ref(input: &T) -> T

Converts to this type from a reference to the input type.
Source§

impl<A, B, T> HttpServerConnExec<A, B> for T
where B: Body,

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self> ⓘ

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self> ⓘ

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> PolicyExt for T
where T: ?Sized,

Source§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow only if self and other return Action::Follow. Read more
Source§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow if either self or other returns Action::Follow. Read more
Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = !

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, !>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self> ⓘ
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self> ⓘ

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more