Add unique constraint on 'open_message' table

Author: jpraynaud

mithril-aggregator/src/database/migration.rs

@@ -318,6 +318,14 @@ insert into signed_entity (signed_entity_id,
     from signed_entity_old 
     order by rowid asc;
 drop table signed_entity_old;
+"#,
+        ),
+        // Migration 12
+        // Alter `open_message` table
+        SqlMigration::new(
+            12,
+            r#"
+create unique index open_message_unique_index on open_message(signed_entity_type_id, beacon);
 "#,
         ),
     ]

mithril-aggregator/src/database/provider/open_message.rs

@@ -799,18 +799,25 @@ mod tests {
     async fn repository_clean_open_message() {
         let connection = get_connection().await;
         let repository = OpenMessageRepository::new(connection.clone());
+        let beacon = Beacon {
+            epoch: Epoch(1),
+            ..Beacon::default()
+        };
         let _ = repository
             .create_open_message(
-                Epoch(2),
-                &SignedEntityType::CardanoImmutableFilesFull(Beacon::default()),
+                beacon.epoch,
+                &SignedEntityType::CardanoImmutableFilesFull(beacon.clone()),
                 &ProtocolMessage::new(),
             )
             .await
             .unwrap();
         let _ = repository
             .create_open_message(
-                Epoch(1),
-                &SignedEntityType::CardanoImmutableFilesFull(Beacon::default()),
+                beacon.epoch,
+                &SignedEntityType::CardanoImmutableFilesFull(Beacon {
+                    epoch: Epoch(2),
+                    ..beacon
+                }),
                 &ProtocolMessage::new(),
             )
             .await