mithril_signer/database/record/
stake_pool.rs

1use chrono::{DateTime, Utc};
2
3use mithril_common::entities::{Epoch, PartyId};
4use mithril_persistence::sqlite::{HydrationError, Projection, SqLiteEntity};
5
6/// Stake pool as read from Chain.
7#[derive(Debug, PartialEq)]
8pub struct StakePool {
9    /// Pool Id
10    pub stake_pool_id: PartyId,
11
12    /// Total stake of this pool.
13    pub stake: u64,
14
15    /// Epoch at which this pool is valid.
16    pub epoch: Epoch,
17
18    /// DateTime of the record creation.
19    pub created_at: DateTime<Utc>,
20}
21
22impl SqLiteEntity for StakePool {
23    fn hydrate(row: sqlite::Row) -> Result<Self, HydrationError>
24    where
25        Self: Sized,
26    {
27        let epoch_int = row.read::<i64, _>(2);
28        let datetime = &row.read::<&str, _>(3);
29        let stake = row.read::<i64, _>(1);
30
31        let stake_pool = Self {
32            stake_pool_id: row.read::<&str, _>(0).to_string(),
33            stake: u64::try_from(stake).map_err(|e| {
34                HydrationError::InvalidData(format!(
35                    "Could not cast the StakePool.stake from internal db I64 → U64. Error: '{e}'.",
36                ))
37            })?,
38            epoch: Epoch(epoch_int.try_into().map_err(|e| {
39                HydrationError::InvalidData(format!(
40                    "Could not cast i64 ({epoch_int}) to u64. Error: '{e}'"
41                ))
42            })?),
43            created_at: DateTime::parse_from_rfc3339(datetime)
44                .map_err(|e| {
45                    HydrationError::InvalidData(format!(
46                        "Could not turn string '{datetime}' to rfc3339 Datetime. Error: {e}"
47                    ))
48                })?
49                .with_timezone(&Utc),
50        };
51
52        Ok(stake_pool)
53    }
54
55    fn get_projection() -> Projection {
56        let mut projection = Projection::default();
57        projection.add_field("stake_pool_id", "{:stake_pool:}.stake_pool_id", "text");
58        projection.add_field("stake", "{:stake_pool:}.stake", "integer");
59        projection.add_field("epoch", "{:stake_pool:}.epoch", "integer");
60        projection.add_field("created_at", "{:stake_pool:}.created_at", "text");
61
62        projection
63    }
64}
mithril_signer/database/record/stake_pool.rs

mithril_signer/database/record/
stake_pool.rs
