feat: implement 'with_closest_block_number_above_slot_number' for Car…

…dano transaction repository
input-output-hk · Sep 10, 2024 · 4309810 · 4309810
1 parent 0dc87db
commit 4309810
Showing 1 changed file with 106 additions and 37 deletions.
diff --git a/...nal/mithril-persistence/src/database/query/cardano_transaction/get_cardano_transaction.rs b/...nal/mithril-persistence/src/database/query/cardano_transaction/get_cardano_transaction.rs
@@ -72,11 +72,33 @@ impl GetCardanoTransactionQuery {
         Self { condition }
     }
 
-    pub fn by_slot_number(slot_number: SlotNumber) -> Self {
+    /// Get the closest block number above the given slot number with either:
+    ///
+    /// - a record with a slot number which is an exact match
+    /// - or the highest block number with a slot number strictly below the given slot number
+    pub fn with_closest_block_number_above_slot_number(slot_number: SlotNumber) -> Self {
         Self {
             condition: WhereCondition::new(
-                "slot_number = ?*",
-                vec![Value::Integer(*slot_number as i64)],
+                "block_number IN (
+                    (select 
+                    max(block_number) 
+                    from 
+                    cardano_tx 
+                    where 
+                    block_number < (
+                        select 
+                        min(block_number) 
+                        from 
+                        cardano_tx 
+                        where 
+                        slot_number > ?*
+                    )),
+                    (select block_number from cardano_tx where slot_number = ?*)
+                )",
+                vec![
+                    Value::Integer(*slot_number as i64),
+                    Value::Integer(*slot_number as i64),
+                ],
             ),
         }
     }
@@ -120,6 +142,18 @@ mod tests {
             .unwrap();
     }
 
+    fn transaction_record(
+        block_number: BlockNumber,
+        slot_number: SlotNumber,
+    ) -> CardanoTransactionRecord {
+        CardanoTransactionRecord::new(
+            format!("tx-hash-{}", slot_number),
+            block_number,
+            slot_number,
+            format!("block-hash-{}", block_number),
+        )
+    }
+
     #[test]
     fn with_highest_block_number() {
         let connection = cardano_tx_db_connection().unwrap();
@@ -132,30 +166,10 @@ mod tests {
         insert_transactions(
             &connection,
             vec![
-                CardanoTransactionRecord::new(
-                    "tx-hash-0",
-                    BlockNumber(10),
-                    SlotNumber(50),
-                    "block-hash-10",
-                ),
-                CardanoTransactionRecord::new(
-                    "tx-hash-1",
-                    BlockNumber(10),
-                    SlotNumber(51),
-                    "block-hash-10",
-                ),
-                CardanoTransactionRecord::new(
-                    "tx-hash-2",
-                    BlockNumber(11),
-                    SlotNumber(54),
-                    "block-hash-11",
-                ),
-                CardanoTransactionRecord::new(
-                    "tx-hash-3",
-                    BlockNumber(11),
-                    SlotNumber(55),
-                    "block-hash-11",
-                ),
+                transaction_record(BlockNumber(10), SlotNumber(50)),
+                transaction_record(BlockNumber(10), SlotNumber(51)),
+                transaction_record(BlockNumber(11), SlotNumber(54)),
+                transaction_record(BlockNumber(11), SlotNumber(55)),
             ],
         );
 
@@ -164,19 +178,74 @@ mod tests {
             .unwrap();
         assert_eq!(
             vec![
-                CardanoTransactionRecord::new(
-                    "tx-hash-2",
-                    BlockNumber(11),
-                    SlotNumber(54),
-                    "block-hash-11"
+                transaction_record(BlockNumber(11), SlotNumber(54)),
+                transaction_record(BlockNumber(11), SlotNumber(55)),
+            ],
+            records
+        );
+    }
+
+    #[test]
+    fn with_closest_block_number_above_slot_number() {
+        let connection = cardano_tx_db_connection().unwrap();
+
+        let cursor = connection
+            .fetch(
+                GetCardanoTransactionQuery::with_closest_block_number_above_slot_number(
+                    SlotNumber(51),
                 ),
-                CardanoTransactionRecord::new(
-                    "tx-hash-3",
-                    BlockNumber(11),
-                    SlotNumber(55),
-                    "block-hash-11"
+            )
+            .unwrap();
+        assert_eq!(0, cursor.count());
+
+        insert_transactions(
+            &connection,
+            vec![transaction_record(BlockNumber(2), SlotNumber(5))],
+        );
+
+        let records: Vec<CardanoTransactionRecord> = connection
+            .fetch_collect(
+                GetCardanoTransactionQuery::with_closest_block_number_above_slot_number(
+                    SlotNumber(5),
                 ),
+            )
+            .unwrap();
+        assert_eq!(
+            vec![transaction_record(BlockNumber(2), SlotNumber(5)),],
+            records
+        );
+
+        insert_transactions(
+            &connection,
+            vec![
+                transaction_record(BlockNumber(10), SlotNumber(50)),
+                transaction_record(BlockNumber(11), SlotNumber(51)),
+                transaction_record(BlockNumber(14), SlotNumber(54)),
+                transaction_record(BlockNumber(15), SlotNumber(55)),
             ],
+        );
+
+        let records: Vec<CardanoTransactionRecord> = connection
+            .fetch_collect(
+                GetCardanoTransactionQuery::with_closest_block_number_above_slot_number(
+                    SlotNumber(53),
+                ),
+            )
+            .unwrap();
+        assert_eq!(
+            vec![transaction_record(BlockNumber(11), SlotNumber(51)),],
+            records
+        );
+
+        let records: Vec<CardanoTransactionRecord> = connection
+            .fetch_collect(
+                GetCardanoTransactionQuery::with_closest_block_number_above_slot_number(
+                    SlotNumber(54),
+                ),
+            )
+            .unwrap();
+        assert_eq!(
+            vec![transaction_record(BlockNumber(14), SlotNumber(54)),],
             records
         );
     }