mithril_aggregator/database/record/
signer.rs

1use chrono::{DateTime, Utc};
2
3use mithril_persistence::sqlite::{HydrationError, Projection, SqLiteEntity};
4
5/// Signer record is the representation of a stored signer.
6#[derive(Debug, PartialEq, Clone)]
7pub struct SignerRecord {
8    /// Signer id.
9    pub signer_id: String,
10
11    /// Pool ticker of the signer.
12    pub pool_ticker: Option<String>,
13
14    /// Date and time when the signer was created.
15    pub created_at: DateTime<Utc>,
16
17    /// Date and time when the signer was updated.
18    pub updated_at: DateTime<Utc>,
19
20    /// Date and time when the signer registered for the last time.
21    pub last_registered_at: Option<DateTime<Utc>>,
22}
23
24#[cfg(test)]
25impl SignerRecord {
26    pub(crate) fn fake_records(number_of_records: usize) -> Vec<SignerRecord> {
27        (0..number_of_records)
28            .map(|idx| SignerRecord {
29                signer_id: format!("signer-{idx}"),
30                pool_ticker: Some(format!("pool-ticker-{idx}")),
31                created_at: DateTime::parse_from_rfc3339("2023-01-19T13:43:05.618857482Z")
32                    .unwrap()
33                    .with_timezone(&Utc),
34                updated_at: DateTime::parse_from_rfc3339("2024-01-19T13:43:05.618857482Z")
35                    .unwrap()
36                    .with_timezone(&Utc),
37                last_registered_at: Some(
38                    DateTime::parse_from_rfc3339("2023-01-19T13:43:05.618857482Z")
39                        .unwrap()
40                        .with_timezone(&Utc),
41                ),
42            })
43            .collect()
44    }
45}
46
47impl SqLiteEntity for SignerRecord {
48    fn hydrate(row: sqlite::Row) -> Result<Self, HydrationError>
49    where
50        Self: Sized,
51    {
52        let signer_id = row.read::<&str, _>(0).to_string();
53        let pool_ticker = row.read::<Option<&str>, _>(1).map(|s| s.to_owned());
54        let created_at = row.read::<&str, _>(2);
55        let updated_at = row.read::<&str, _>(3);
56        let registered_at = row.read::<Option<&str>, _>(4);
57
58        let signer_record = Self {
59            signer_id,
60            pool_ticker,
61            created_at: DateTime::parse_from_rfc3339(created_at)
62                .map_err(|e| {
63                    HydrationError::InvalidData(format!(
64                        "Could not turn string '{created_at}' to rfc3339 Datetime. Error: {e}"
65                    ))
66                })?
67                .with_timezone(&Utc),
68            updated_at: DateTime::parse_from_rfc3339(updated_at)
69                .map_err(|e| {
70                    HydrationError::InvalidData(format!(
71                        "Could not turn string '{updated_at}' to rfc3339 Datetime. Error: {e}"
72                    ))
73                })?
74                .with_timezone(&Utc),
75            last_registered_at: registered_at
76                .map(|d| match DateTime::parse_from_rfc3339(d) {
77                    Ok(date) => Ok(date.with_timezone(&Utc)),
78                    Err(e) => Err(HydrationError::InvalidData(format!(
79                        "Could not turn string '{d}' to rfc3339 Datetime. Error: {e}"
80                    ))),
81                })
82                .transpose()?,
83        };
84
85        Ok(signer_record)
86    }
87
88    fn get_projection() -> Projection {
89        let mut projection = Projection::default();
90        projection.add_field("signer_id", "{:signer:}.signer_id", "text");
91        projection.add_field("pool_ticker", "{:signer:}.pool_ticker", "text");
92        projection.add_field("created_at", "{:signer:}.created_at", "text");
93        projection.add_field("updated_at", "{:signer:}.updated_at", "text");
94        projection.add_field(
95            "last_registered_at",
96            "{:signer:}.last_registered_at",
97            "text",
98        );
99
100        projection
101    }
102}
mithril_aggregator/database/record/signer.rs

mithril_aggregator/database/record/
signer.rs
