move to stack based vector for websocket_depth_buffer in exchange stream

githubsands · githubsands · commit 7e69bb1c580c · 2025-08-19T12:44:38.000-07:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/exchange/Cargo.toml b/exchange/Cargo.toml
@@ -20,3 +20,5 @@ tracing.workspace = true
 itertools.workspace = true
 tokio-tungstenite.workspace = true
 pin-project-lite.workspace = true
+arrayvec = { git = "https://github.com/bluss/arrayvec" }
+
diff --git a/exchange/src/stream.rs b/exchange/src/stream.rs
@@ -1,3 +1,4 @@
+use arrayvec::ArrayVec;
 use std::error::Error;
 use std::{pin::Pin, task::Poll};
 
@@ -34,7 +35,8 @@ pin_project! {
         pub client_name: String,
         pub exchange_name: u8,
         pub snapshot_enabled: bool,
-        pub websocket_depth_buffer: Vec<DepthUpdate>,
+        // todo: update webssocket_depth_buffer take the array vecs size by pragma or something similar -- it should be configurable
+        pub websocket_depth_buffer: ArrayVec<DepthUpdate, 1000>,
         pub pull_retry_count: u8,
         pub http_snapshot_uri: String,
         pub buffer_websocket_depths: bool,
