mithril_aggregator/database/query/epoch_settings/
delete_epoch_settings.rs

1use sqlite::Value;
2
3use mithril_common::entities::Epoch;
4use mithril_persistence::sqlite::{Query, SourceAlias, SqLiteEntity, WhereCondition};
5
6use crate::database::record::EpochSettingsRecord;
7
8/// Query to delete old [EpochSettingsRecord] from the sqlite database
9pub struct DeleteEpochSettingsQuery {
10    condition: WhereCondition,
11}
12
13impl Query for DeleteEpochSettingsQuery {
14    type Entity = EpochSettingsRecord;
15
16    fn filters(&self) -> WhereCondition {
17        self.condition.clone()
18    }
19
20    fn get_definition(&self, condition: &str) -> String {
21        // it is important to alias the fields with the same name as the table
22        // since the table cannot be aliased in a RETURNING statement in SQLite.
23        let projection = Self::Entity::get_projection()
24            .expand(SourceAlias::new(&[("{:epoch_setting:}", "epoch_setting")]));
25
26        format!("delete from epoch_setting where {condition} returning {projection}")
27    }
28}
29
30impl DeleteEpochSettingsQuery {
31    #[cfg(test)]
32    /// Create the SQL condition to delete a record given the Epoch.
33    pub fn by_epoch(epoch: Epoch) -> Self {
34        let epoch_settings_id_value = Value::Integer(epoch.try_into().unwrap());
35
36        Self {
37            condition: WhereCondition::new("epoch_setting_id = ?*", vec![epoch_settings_id_value]),
38        }
39    }
40
41    /// Create the SQL condition to prune data older than the given Epoch.
42    pub fn below_epoch_threshold(epoch_threshold: Epoch) -> Self {
43        let epoch_settings_id_value = Value::Integer(epoch_threshold.try_into().unwrap());
44
45        Self {
46            condition: WhereCondition::new("epoch_setting_id < ?*", vec![epoch_settings_id_value]),
47        }
48    }
49}
50
51#[cfg(test)]
52mod tests {
53    use crate::database::query::GetEpochSettingsQuery;
54    use crate::database::test_helper::{insert_epoch_settings, main_db_connection};
55    use mithril_persistence::sqlite::ConnectionExtensions;
56
57    use super::*;
58
59    #[test]
60    fn test_delete_by_epoch() {
61        let connection = main_db_connection().unwrap();
62        insert_epoch_settings(&connection, &[1, 2]).unwrap();
63
64        let cursor = connection
65            .fetch(DeleteEpochSettingsQuery::by_epoch(Epoch(2)))
66            .unwrap();
67
68        assert_eq!(1, cursor.count());
69
70        let cursor = connection
71            .fetch(GetEpochSettingsQuery::by_epoch(Epoch(1)).unwrap())
72            .unwrap();
73
74        assert_eq!(1, cursor.count());
75
76        let cursor = connection
77            .fetch(GetEpochSettingsQuery::by_epoch(Epoch(2)).unwrap())
78            .unwrap();
79
80        assert_eq!(0, cursor.count());
81    }
82
83    #[test]
84    fn test_delete_below_threshold() {
85        let connection = main_db_connection().unwrap();
86        insert_epoch_settings(&connection, &[1, 2]).unwrap();
87
88        let cursor = connection
89            .fetch(DeleteEpochSettingsQuery::below_epoch_threshold(Epoch(2)))
90            .unwrap();
91
92        assert_eq!(1, cursor.count());
93
94        let cursor = connection
95            .fetch(GetEpochSettingsQuery::by_epoch(Epoch(1)).unwrap())
96            .unwrap();
97
98        assert_eq!(0, cursor.count());
99
100        let cursor = connection
101            .fetch(GetEpochSettingsQuery::by_epoch(Epoch(2)).unwrap())
102            .unwrap();
103
104        assert_eq!(1, cursor.count());
105    }
106}
mithril_aggregator/database/query/epoch_settings/delete_epoch_settings.rs

mithril_aggregator/database/query/epoch_settings/
delete_epoch_settings.rs
