mithril_persistence/database/query/block_range_root/
insert_block_range.rs

1use std::iter::repeat_n;
2
3use sqlite::Value;
4
5use mithril_common::StdResult;
6
7use crate::database::record::BlockRangeRootRecord;
8use crate::sqlite::{Query, SourceAlias, SqLiteEntity, WhereCondition};
9
10/// Query to insert [BlockRangeRootRecord] in the sqlite database
11pub struct InsertBlockRangeRootQuery {
12    condition: WhereCondition,
13}
14
15impl InsertBlockRangeRootQuery {
16    /// Query that insert multiples records.
17    pub fn insert_many(block_range_records: Vec<BlockRangeRootRecord>) -> StdResult<Self> {
18        let columns = "(start, end, merkle_root)";
19        let values_columns: Vec<&str> =
20            repeat_n("(?*, ?*, ?*)", block_range_records.len()).collect();
21
22        let values: StdResult<Vec<Value>> =
23            block_range_records.into_iter().try_fold(vec![], |mut vec, record| {
24                vec.append(&mut vec![
25                    Value::Integer(record.range.start.try_into()?),
26                    Value::Integer(record.range.end.try_into()?),
27                    Value::String(record.merkle_root.to_hex()),
28                ]);
29                Ok(vec)
30            });
31        let condition = WhereCondition::new(
32            format!("{columns} values {}", values_columns.join(", ")).as_str(),
33            values?,
34        );
35
36        Ok(Self { condition })
37    }
38}
39
40impl Query for InsertBlockRangeRootQuery {
41    type Entity = BlockRangeRootRecord;
42
43    fn filters(&self) -> WhereCondition {
44        self.condition.clone()
45    }
46
47    fn get_definition(&self, condition: &str) -> String {
48        let aliases = SourceAlias::new(&[("{:block_range_root:}", "block_range_root")]);
49        let projection = Self::Entity::get_projection().expand(aliases);
50
51        format!("insert or ignore into block_range_root {condition} returning {projection}")
52    }
53}