feat(signer): store unique signer registration per epoch

Author: jpraynaud

mithril-signer/src/database/query/protocol_initializer/insert_protocol_initializer.rs

@@ -5,12 +5,12 @@ use mithril_persistence::sqlite::{Query, SourceAlias, SqLiteEntity, WhereConditi
 
 use crate::database::record::ProtocolInitializerRecord;
 
-/// Query to insert or replace [ProtocolInitializerRecord] in the sqlite database
-pub struct InsertOrReplaceProtocolInitializerQuery {
+/// Query to insert or ignore [ProtocolInitializerRecord] in the sqlite database
+pub struct InsertOrIgnoreProtocolInitializerQuery {
     condition: WhereCondition,
 }
 
-impl InsertOrReplaceProtocolInitializerQuery {
+impl InsertOrIgnoreProtocolInitializerQuery {
     pub fn one(record: ProtocolInitializerRecord) -> StdResult<Self> {
         let value = serde_json::to_string(&record.protocol_initializer).unwrap();
         let condition = WhereCondition::new(
@@ -26,7 +26,7 @@ impl InsertOrReplaceProtocolInitializerQuery {
     }
 }
 
-impl Query for InsertOrReplaceProtocolInitializerQuery {
+impl Query for InsertOrIgnoreProtocolInitializerQuery {
     type Entity = ProtocolInitializerRecord;
 
     fn filters(&self) -> WhereCondition {
@@ -41,6 +41,6 @@ impl Query for InsertOrReplaceProtocolInitializerQuery {
             "protocol_initializer",
         )]));
 
-        format!("insert or replace into protocol_initializer {condition} returning {projection}")
+        format!("insert or ignore into protocol_initializer {condition} returning {projection} ")
     }
 }

mithril-signer/src/database/repository/protocol_initializer_repository.rs

@@ -4,7 +4,7 @@ use anyhow::Ok;
 use async_trait::async_trait;
 
 use crate::database::query::{
-    DeleteProtocolInitializerQuery, InsertOrReplaceProtocolInitializerQuery,
+    DeleteProtocolInitializerQuery, InsertOrIgnoreProtocolInitializerQuery,
 };
 use crate::database::record::ProtocolInitializerRecord;
 use crate::{
@@ -62,7 +62,7 @@ impl ProtocolInitializerStorer for ProtocolInitializerRepository {
             created_at: chrono::Utc::now(),
         };
         self.connection
-            .apply(InsertOrReplaceProtocolInitializerQuery::one(record).unwrap())?;
+            .apply(InsertOrIgnoreProtocolInitializerQuery::one(record).unwrap())?;
 
         Ok(previous_protocol_initializer)
     }

mithril-signer/src/database/tests/protocol_initializer.rs

@@ -13,7 +13,7 @@ fn setup_protocol_initializers(nb_epoch: u64) -> Vec<(Epoch, ProtocolInitializer
     let mut values: Vec<(Epoch, ProtocolInitializer)> = Vec::new();
     for epoch in 1..=nb_epoch {
         let stake = (epoch + 1) * 100;
-        let protocol_initializer = fake_data::protocol_initializer("1", stake);
+        let protocol_initializer = fake_data::protocol_initializer(format!("{epoch:?}"), stake);
         values.push((Epoch(epoch), protocol_initializer));
     }
     values
@@ -84,6 +84,37 @@ mod request {
         );
     }
 
+    #[tokio::test]
+    async fn protocol_initializer_in_store_is_created_once_per_epoch_and_never_updated() {
+        let protocol_initializers = setup_protocol_initializers(2);
+        let epoch = protocol_initializers[0].0;
+        let first_protocol_initializer = protocol_initializers[0].1.clone();
+        let second_protocol_initializer = protocol_initializers[1].1.clone();
+
+        let store = init_store(&[], None).await;
+        store
+            .save_protocol_initializer(epoch, first_protocol_initializer.clone())
+            .await
+            .unwrap();
+
+        let res = store.get_protocol_initializer(epoch).await.unwrap().unwrap();
+        assert_eq!(
+            serde_json::to_string(&first_protocol_initializer).unwrap(),
+            serde_json::to_string(&res).unwrap()
+        );
+
+        store
+            .save_protocol_initializer(epoch, second_protocol_initializer.clone())
+            .await
+            .unwrap();
+
+        let res = store.get_protocol_initializer(epoch).await.unwrap().unwrap();
+        assert_eq!(
+            serde_json::to_string(&first_protocol_initializer).unwrap(),
+            serde_json::to_string(&res).unwrap()
+        );
+    }
+
     #[tokio::test]
     async fn get_protocol_initializer_for_empty_epoch() {
         let store = init_store(&setup_protocol_initializers(2), None).await;