mithril_aggregator/database/query/open_message/
get_open_message.rs

1use chrono::{DateTime, Utc};
2use sqlite::Value;
3
4use mithril_common::{
5    entities::{Epoch, SignedEntityType},
6    StdResult,
7};
8use mithril_persistence::sqlite::{Query, SourceAlias, SqLiteEntity, WhereCondition};
9
10use crate::database::record::OpenMessageRecord;
11
12/// Simple queries to retrieve [OpenMessageRecord] from the sqlite database.
13pub struct GetOpenMessageQuery {
14    condition: WhereCondition,
15}
16
17impl GetOpenMessageQuery {
18    pub fn by_epoch_and_signed_entity_type(
19        epoch: Epoch,
20        signed_entity_type: &SignedEntityType,
21    ) -> StdResult<Self> {
22        let condition = Self::get_epoch_condition(epoch)
23            .and_where(Self::get_signed_entity_type_condition(signed_entity_type)?);
24
25        Ok(Self { condition })
26    }
27
28    pub fn by_expired_entity_type(
29        now: DateTime<Utc>,
30        signed_entity_type: &SignedEntityType,
31    ) -> StdResult<Self> {
32        let condition = Self::get_expired_condition(now)
33            .and_where(Self::get_signed_entity_type_condition(signed_entity_type)?);
34
35        Ok(Self { condition })
36    }
37
38    fn get_epoch_condition(epoch: Epoch) -> WhereCondition {
39        WhereCondition::new("epoch_setting_id = ?*", vec![Value::Integer(*epoch as i64)])
40    }
41
42    fn get_signed_entity_type_condition(
43        signed_entity_type: &SignedEntityType,
44    ) -> StdResult<WhereCondition> {
45        Ok(WhereCondition::new(
46            "signed_entity_type_id = ?* and beacon = ?*",
47            vec![
48                Value::Integer(signed_entity_type.index() as i64),
49                Value::String(signed_entity_type.get_json_beacon()?),
50            ],
51        ))
52    }
53
54    fn get_expired_condition(expires_at: DateTime<Utc>) -> WhereCondition {
55        WhereCondition::new(
56            "expires_at < ?*",
57            vec![Value::String(expires_at.to_rfc3339())],
58        )
59    }
60
61    #[cfg(test)]
62    pub fn by_id(open_message_id: &uuid::Uuid) -> Self {
63        Self {
64            condition: WhereCondition::new(
65                "open_message_id = ?*",
66                vec![Value::String(open_message_id.to_string())],
67            ),
68        }
69    }
70}
71
72impl Query for GetOpenMessageQuery {
73    type Entity = OpenMessageRecord;
74
75    fn filters(&self) -> WhereCondition {
76        self.condition.clone()
77    }
78
79    fn get_definition(&self, condition: &str) -> String {
80        let aliases = SourceAlias::new(&[
81            ("{:open_message:}", "open_message"),
82            ("{:single_signature:}", "single_signature"),
83        ]);
84        let projection = Self::Entity::get_projection().expand(aliases);
85
86        format!("select {projection} from open_message where {condition} order by created_at desc")
87    }
88}
mithril_aggregator/database/query/open_message/get_open_message.rs

mithril_aggregator/database/query/open_message/
get_open_message.rs
