Check use_mutable_id_tracker_without_rocksdb feature flag when building new segment

ffuugoo · ffuugoo · commit 2cd54d279647 · 2025-03-20T17:16:25.000+01:00
diff --git a/lib/segment/src/segment_constructor/segment_builder.rs b/lib/segment/src/segment_constructor/segment_builder.rs
@@ -11,6 +11,7 @@ use atomic_refcell::AtomicRefCell;
 use bitvec::macros::internal::funty::Integral;
 use common::budget::ResourcePermit;
 use common::counter::hardware_counter::HardwareCounterCell;
+use common::flags::feature_flags;
 use common::small_uint::U24;
 use common::types::PointOffsetType;
 use io::storage_version::StorageVersion;
@@ -19,9 +20,10 @@ use tempfile::TempDir;
 use uuid::Uuid;
 
 use super::{
-    create_mutable_id_tracker, create_payload_storage, create_sparse_vector_index,
-    create_sparse_vector_storage, get_payload_index_path, get_vector_index_path,
-    get_vector_storage_path, new_segment_path, open_segment_db, open_vector_storage,
+    create_mutable_id_tracker, create_payload_storage, create_rocksdb_id_tracker,
+    create_sparse_vector_index, create_sparse_vector_storage, get_payload_index_path,
+    get_vector_index_path, get_vector_storage_path, new_segment_path, open_segment_db,
+    open_vector_storage,
 };
 use crate::common::error_logging::LogError;
 use crate::common::operation_error::{OperationError, OperationResult, check_process_stopped};
@@ -85,7 +87,11 @@ impl SegmentBuilder {
         let database = open_segment_db(temp_dir.path(), segment_config)?;
 
         let id_tracker = if segment_config.is_appendable() {
-            IdTrackerEnum::MutableIdTracker(create_mutable_id_tracker(temp_dir.path())?)
+            if feature_flags().use_mutable_id_tracker_without_rocksdb {
+                IdTrackerEnum::MutableIdTracker(create_mutable_id_tracker(temp_dir.path())?)
+            } else {
+                IdTrackerEnum::RocksDbIdTracker(create_rocksdb_id_tracker(database.clone())?)
+            }
         } else {
             IdTrackerEnum::InMemoryIdTracker(InMemoryIdTracker::new())
         };
