fx-durable-ga 0.3.0

Durable GA event driven optimization loop on PostgreSQL
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use super::Error;
use super::repository_tx::TxRepository;
use crate::models::{Request, RequestConclusion};
use crate::repositories::chainable::{Chain, FromOther, ToTx, TxType};
use futures::future::{BoxFuture, Future};
use sqlx::{PgPool, PgTransaction};
use tracing::instrument;
use uuid::Uuid;

/// Repository for managing genetic algorithm requests and their conclusions.
#[derive(Debug, Clone)]
pub(crate) struct Repository {
    pool: PgPool,
}

impl Repository {
    /// Creates a new Repository instance with the given database pool.
    pub(crate) fn new(pool: PgPool) -> Self {
        Self { pool }
    }

    /// Retrieves a request by its ID.
    #[instrument(level = "debug", skip(self), fields(request_id = %id))]
    pub(crate) async fn get_request(&self, id: Uuid) -> Result<Request, Error> {
        super::queries::get_request(&self.pool, &id).await
    }

    /// Retrieves a request conclusion by request ID. Returns None if no conclusion exists.
    #[instrument(level = "debug", skip(self), fields(request_id = %id))]
    pub(crate) fn get_request_conclusion(
        &self,
        id: &Uuid,
    ) -> impl Future<Output = Result<Option<RequestConclusion>, Error>> {
        super::queries::get_request_conclusion(&self.pool, id)
    }

    /// Creates a new request conclusion record in the database.
    #[instrument(level = "debug", skip(self), fields(request_id = %request_conclusion.request_id, concluded_at = %request_conclusion.concluded_at, concluded_with = ?request_conclusion.concluded_with))]
    pub(crate) fn new_request_conclusion(
        &self,
        request_conclusion: &RequestConclusion,
    ) -> impl Future<Output = Result<RequestConclusion, Error>> {
        super::queries::new_request_conclusion(&self.pool, request_conclusion)
    }
}

impl<'tx> TxType<'tx> for Repository {
    type TxType = TxRepository<'tx>;
    type TxError = Error;
}

impl<'tx> FromOther<'tx> for Repository {
    fn from_other(&self, other: impl ToTx<'tx>) -> Self::TxType {
        TxRepository::new(other.tx())
    }
}

impl<'tx> Chain<'tx> for Repository {
    #[instrument(level = "debug", skip(self, f))]
    fn chain<F, R, T>(&'tx self, f: F) -> BoxFuture<'tx, Result<T, Self::TxError>>
    where
        R: ToTx<'tx>,
        F: FnOnce(Self::TxType) -> BoxFuture<'tx, Result<(R, T), anyhow::Error>>
            + Send
            + Sync
            + 'tx,
        T: Send + Sync + 'tx,
    {
        Box::pin(async move {
            let pool = self.pool.clone();
            let tx = pool.begin().await?;

            let (tx, ret) = f(TxRepository::new(tx))
                .await
                .map_err(|err| Error::Tx(err))?;

            let tx: PgTransaction<'_> = tx.tx();
            tx.commit()
                .await
                .map_err(|err| Error::Tx(anyhow::Error::new(err)))?;

            Ok(ret)
        })
    }
}