feat(db)!: add cone white flag order (#232)

iotaledger · May 31, 2022 · 6b936b5 · 6b936b5
1 parent 26a86ba
commit 6b936b5
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 5 deletions.
diff --git a/bin/inx-chronicle/src/stardust_inx/cone_stream.rs b/bin/inx-chronicle/src/stardust_inx/cone_stream.rs
@@ -25,11 +25,11 @@ impl ConeStream {
 
 #[async_trait]
 impl Actor for ConeStream {
-    type State = ();
+    type State = u32;
     type Error = InxError;
 
     async fn init(&mut self, _cx: &mut ActorContext<Self>) -> Result<Self::State, Self::Error> {
-        Ok(())
+        Ok(0)
     }
 
     fn name(&self) -> std::borrow::Cow<'static, str> {
@@ -58,16 +58,23 @@ impl HandleEvent<Result<inx::proto::BlockWithMetadata, Status>> for ConeStream {
         &mut self,
         _cx: &mut ActorContext<Self>,
         block_metadata_result: Result<inx::proto::BlockWithMetadata, Status>,
-        _state: &mut Self::State,
+        white_flag_index: &mut Self::State,
     ) -> Result<(), Self::Error> {
         log::trace!("Received Stardust Block Event");
 
         let inx_block_with_metadata: inx::BlockWithMetadata = block_metadata_result?.try_into()?;
         let BlockWithMetadata { metadata, block, raw } = inx_block_with_metadata;
 
         self.db
-            .insert_block_with_metadata(metadata.block_id.into(), block.into(), raw, metadata.into())
+            .insert_block_with_metadata(
+                metadata.block_id.into(),
+                block.into(),
+                raw,
+                metadata.into(),
+                *white_flag_index,
+            )
             .await?;
+        *white_flag_index += 1;
 
         log::trace!("Inserted block into database.");
 

diff --git a/src/db/collections/block.rs b/src/db/collections/block.rs
@@ -31,6 +31,8 @@ struct BlockDocument {
     raw: Vec<u8>,
     /// The block's metadata.
     metadata: BlockMetadata,
+    /// The index of this block in white flag order.
+    white_flag_index: u32,
 }
 
 impl BlockDocument {
@@ -120,12 +122,14 @@ impl MongoDb {
         block: Block,
         raw: Vec<u8>,
         metadata: BlockMetadata,
+        white_flag_index: u32,
     ) -> Result<(), Error> {
         let block_document = BlockDocument {
             block_id: block_id.clone(),
             block,
             raw,
             metadata,
+            white_flag_index,
         };
 
         self.0

diff --git a/tests/message_record.rs b/tests/message_record.rs
@@ -37,7 +37,7 @@ async fn test_test() -> Result<(), mongodb::error::Error> {
 
     db.clear().await?;
 
-    db.insert_block_with_metadata(block_id.clone(), block, raw, metadata)
+    db.insert_block_with_metadata(block_id.clone(), block, raw, metadata, 0)
         .await?;
 
     let result = db.get_block(&block_id).await?.unwrap();