Plain to enc dump conversions

marlinprotocol · Jan 2, 2025 · 9077689 · 9077689
1 parent 6443805
commit 9077689
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/matching_engine/src/main.rs b/matching_engine/src/main.rs
@@ -2,6 +2,7 @@ use std::fs;
 use std::io;
 
 use dotenv::dotenv;
+use matching_engine::Dump;
 use matching_engine::EncryptedDump;
 use matching_engine::{MatchingEngine, MatchingEngineConfig};
 
@@ -31,13 +32,35 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         "../matching_engine_config/encrypted_dump.json",
         "./matching_engine_config/encrypted_dump.json",
     ];
+
     match read_file_from_paths(&encrypted_dump_paths) {
         Ok(dump_content) => {
             let enc_dump: EncryptedDump = serde_json::from_str(&dump_content)?;
             matching_engine
                 .run_from_encrypted_dump(enc_dump, encrypted_dump_paths[1].to_string())
                 .await?;
         }
+        Err(e) if e.kind() == io::ErrorKind::NotFound => {
+            // Handle the NotFound error case
+            let plain_dump = [
+                "../matching_engine_config/dump.json",
+                "./matching_engine_config/dump.json",
+            ];
+
+            match read_file_from_paths(&plain_dump) {
+                Ok(dump_content) => {
+                    let dump: Dump = serde_json::from_str(&dump_content)?;
+                    matching_engine
+                        .run_from_dump(dump, encrypted_dump_paths[1].to_string())
+                        .await?;
+                }
+                Err(_) => {
+                    matching_engine
+                        .run(encrypted_dump_paths[1].to_string())
+                        .await?;
+                }
+            }
+        }
         Err(_) => {
             matching_engine
                 .run(encrypted_dump_paths[1].to_string())