From 90d49d0597f52d6fc778d1433d9cf8f1674ad34e Mon Sep 17 00:00:00 2001
From: Ayaz Abbas <lfcayaz@gmail.com>
Date: Wed, 23 Oct 2024 00:07:09 +0100
Subject: [PATCH] increase max_bytes for nats stream and ensure stream config
 is updated

---
 src/utils.rs | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/src/utils.rs b/src/utils.rs
index 4f311fd..790f0ed 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -10,7 +10,7 @@ pub async fn setup_jetstream(nats_client: &async_nats::Client) -> Result<jetstre
         name: "PYTH_PRICE_UPDATES".to_string(),
         subjects: vec!["pyth.price.updates".to_string()],
         max_age: Duration::from_secs(300),
-        max_bytes: 1024 * 1024 * 100,
+        max_bytes: 1024 * 1024 * 4000,
         duplicate_window: Duration::from_secs(60),
         discard: stream::DiscardPolicy::New,
         max_messages_per_subject: 100,
@@ -19,7 +19,19 @@ pub async fn setup_jetstream(nats_client: &async_nats::Client) -> Result<jetstre
         ..Default::default()
     };
 
-    jetstream.get_or_create_stream(stream_config).await?;
-    info!("JetStream stream created or already exists: PYTH_PRICE_UPDATES");
+    // Try to get the existing stream
+    match jetstream.get_stream("PYTH_PRICE_UPDATES").await {
+        Ok(_) => {
+            // Stream exists, update its configuration
+            jetstream.update_stream(stream_config).await?;
+            info!("JetStream stream updated: PYTH_PRICE_UPDATES");
+        }
+        Err(_) => {
+            // Stream doesn't exist, create it
+            jetstream.create_stream(stream_config).await?;
+            info!("JetStream stream created: PYTH_PRICE_UPDATES");
+        }
+    }
+
     Ok(jetstream)
 }