fix overflow bug

Signed-off-by: Little-Wallace <[email protected]>
risingwavelabs · Mar 15, 2024 · ab953a9 · ab953a9
1 parent 70d2248
commit ab953a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/compute/src/memory/config.rs b/src/compute/src/memory/config.rs
@@ -95,7 +95,7 @@ pub fn storage_memory_config(
 
     if meta_cache_capacity_mb != default_meta_cache_capacity_mb {
         default_block_cache_capacity_mb += default_meta_cache_capacity_mb;
-        default_block_cache_capacity_mb -= meta_cache_capacity_mb;
+        default_block_cache_capacity_mb = default_block_cache_capacity_mb.saturating_sub(meta_cache_capacity_mb);
     }
 
     let default_shared_buffer_capacity_mb = ((non_reserved_memory_bytes as f64
@@ -112,7 +112,7 @@ pub fn storage_memory_config(
             ));
     if shared_buffer_capacity_mb != default_shared_buffer_capacity_mb {
         default_block_cache_capacity_mb += default_shared_buffer_capacity_mb;
-        default_block_cache_capacity_mb -= shared_buffer_capacity_mb;
+        default_block_cache_capacity_mb = default_block_cache_capacity_mb.saturating_sub(shared_buffer_capacity_mb);
     }
     let block_cache_capacity_mb = storage_config
         .block_cache_capacity_mb