mithril_signer/database/query/signed_beacon/
insert_signed_beacon.rs

1use sqlite::Value;
2
3use mithril_common::StdResult;
4use mithril_persistence::sqlite::{Query, SourceAlias, SqLiteEntity, WhereCondition};
5
6use crate::database::record::SignedBeaconRecord;
7
8/// Query to insert or replace [SignedBeaconRecord] in the sqlite database
9pub struct InsertSignedBeaconRecordQuery {
10    condition: WhereCondition,
11}
12
13impl InsertSignedBeaconRecordQuery {
14    pub fn one(record: SignedBeaconRecord) -> StdResult<Self> {
15        let condition =
16        WhereCondition::new(
17            "(epoch, beacon, signed_entity_type_id, initiated_at, signed_at) values (?*, ?*, ?*, ?*, ?*)",
18            vec![
19                Value::Integer(record.epoch.try_into()?),
20                Value::String(record.signed_entity_type.get_json_beacon()?),
21                Value::Integer(record.signed_entity_type.index() as i64),
22                Value::String(record.initiated_at.to_rfc3339()),
23                Value::String(record.signed_at.to_rfc3339()),
24            ],
25        );
26
27        Ok(Self { condition })
28    }
29}
30
31impl Query for InsertSignedBeaconRecordQuery {
32    type Entity = SignedBeaconRecord;
33
34    fn filters(&self) -> WhereCondition {
35        self.condition.clone()
36    }
37
38    fn get_definition(&self, condition: &str) -> String {
39        // it is important to alias the fields with the same name as the table
40        // since the table cannot be aliased in a RETURNING statement in SQLite.
41        let projection = Self::Entity::get_projection()
42            .expand(SourceAlias::new(&[("{:signed_beacon:}", "signed_beacon")]));
43
44        format!("insert into signed_beacon {condition} returning {projection}")
45    }
46}
47
48#[cfg(test)]
49mod tests {
50    use mithril_common::entities::{Epoch, SignedEntityType};
51    use mithril_persistence::sqlite::ConnectionExtensions;
52
53    use crate::database::test_helper::main_db_connection;
54
55    use super::*;
56
57    #[test]
58    fn insert_records_in_empty_db() {
59        let connection = main_db_connection().unwrap();
60
61        let record = SignedBeaconRecord::fake(
62            Epoch(5),
63            SignedEntityType::CardanoStakeDistribution(Epoch(6)),
64        );
65        let inserted_record = connection
66            .fetch_first(InsertSignedBeaconRecordQuery::one(record.clone()).unwrap())
67            .unwrap();
68
69        assert_eq!(Some(record), inserted_record);
70    }
71
72    #[test]
73    #[should_panic]
74    fn inserting_same_record_twice_should_fail() {
75        let connection = main_db_connection().unwrap();
76
77        let record = SignedBeaconRecord::fake(
78            Epoch(13),
79            SignedEntityType::CardanoStakeDistribution(Epoch(17)),
80        );
81
82        connection
83            .fetch_first(InsertSignedBeaconRecordQuery::one(record.clone()).unwrap())
84            .unwrap();
85        let _ = connection.fetch_first(InsertSignedBeaconRecordQuery::one(record.clone()).unwrap());
86    }
87}
mithril_signer/database/query/signed_beacon/insert_signed_beacon.rs

mithril_signer/database/query/signed_beacon/
insert_signed_beacon.rs
