src/event/repo.rs


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

use sqlx::{SqliteConnection, Transaction, sqlite::Sqlite};

use crate::{
    clock::DateTime,
    event::{Instant, Sequence},
};

pub trait Provider {
    fn sequence(&mut self) -> Sequences;
}

impl Provider for Transaction<'_, Sqlite> {
    fn sequence(&mut self) -> Sequences {
        Sequences(self)
    }
}

pub struct Sequences<'t>(&'t mut SqliteConnection);

impl Sequences<'_> {
    pub async fn next(&mut self, at: &DateTime) -> Result<Instant, sqlx::Error> {
        let next = sqlx::query_scalar!(
            r#"
                update event_sequence
                set last_value = last_value + 1
                returning last_value as "next_value: Sequence"
            "#,
        )
        .fetch_one(&mut *self.0)
        .await?;

        Ok(Instant::new(*at, next))
    }

    pub async fn current(&mut self) -> Result<Sequence, sqlx::Error> {
        let next = sqlx::query_scalar!(
            r#"
                select last_value as "last_value: Sequence"
                from event_sequence
            "#,
        )
        .fetch_one(&mut *self.0)
        .await?;

        Ok(next)
    }
}