limit memory usage to 128 MiB

Signed-off-by: Bugen Zhao <[email protected]>
risingwavelabs · Jan 3, 2024 · c39b2eb · c39b2eb
1 parent 3987557
commit c39b2eb
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -118,7 +118,7 @@ hashbrown = { version = "0.14.0", features = [
 criterion = { version = "0.5", features = ["async_futures"] }
 tonic = { package = "madsim-tonic", version = "0.4.1" }
 tonic-build = { package = "madsim-tonic-build", version = "0.4.2" }
-otlp-embedded = "0.0.1"
+otlp-embedded = { git = "https://github.com/risingwavelabs/otlp-embedded", rev = "58c1f003484449d7c6dd693b348bf19dd44889cb" }
 prost = { version = "0.12" }
 icelake = { git = "https://github.com/icelake-io/icelake", rev = "3f7b53ba5b563524212c25810345d1314678e7fc", features = [
   "prometheus",

diff --git a/src/meta/node/src/server.rs b/src/meta/node/src/server.rs
@@ -491,7 +491,10 @@ pub async fn start_service_as_election_leader(
         MetadataManager::V2(_) => None,
     };
 
-    let trace_state = otlp_embedded::State::new(opts.cached_traces_num);
+    let trace_state = otlp_embedded::State::new(otlp_embedded::Config {
+        max_length: opts.cached_traces_num,
+        max_memory_usage: 1 << 27, // hard-code to limit memory usage to 128 MiB
+    });
     let trace_srv = otlp_embedded::TraceServiceImpl::new(trace_state.clone());
 
     #[cfg(not(madsim))]