Move ObjectPool from diskann to diskann-utils

diskann-disk/src/search/provider/disk_provider.rs

@@ -31,17 +31,15 @@ use diskann::{
         Accessor, BuildQueryComputer, DataProvider, DefaultContext, DelegateNeighbor, HasId,
         NeighborAccessor, NoopGuard,
     },
-    utils::{
-        object_pool::{ObjectPool, PoolOption, TryAsPooled},
-        IntoUsize, VectorRepr,
-    },
+    utils::{IntoUsize, VectorRepr},
     ANNError, ANNResult,
 };
 use diskann_providers::storage::StorageReadProvider;
 use diskann_providers::{
     model::{compute_pq_distance, compute_pq_distance_for_pq_coordinates},
     storage::{get_compressed_pq_file, get_disk_index_file, get_pq_pivot_file, LoadWith},
 };
+use diskann_utils::object_pool::{ObjectPool, PoolOption, TryAsPooled};
 
 use crate::search::pq::{quantizer_preprocess, PQData, PQScratch};
 use diskann_vector::{distance::Metric, DistanceFunction, PreprocessedDistanceFunction};

diskann-garnet/src/provider.rs

@@ -20,13 +20,11 @@ use diskann::{
         Accessor, BuildDistanceComputer, BuildQueryComputer, DataProvider, DelegateNeighbor,
         Delete, ElementStatus, HasId, NeighborAccessor, NeighborAccessorMut, NoopGuard, SetElement,
     },
-    utils::{
-        VectorRepr,
-        object_pool::{AsPooled, ObjectPool, PooledRef, Undef},
-    },
+    utils::VectorRepr,
 };
 use diskann_providers::model::graph::provider::async_::common::FullPrecision;
 use diskann_utils::Reborrow;
+use diskann_utils::object_pool::{AsPooled, ObjectPool, PooledRef, Undef};
 use diskann_vector::{PreprocessedDistanceFunction, contains::ContainsSimd, distance::Metric};
 use std::{
     future, mem,

diskann-providers/src/model/graph/provider/async_/bf_tree/quant_vector_provider.rs

@@ -9,12 +9,9 @@ use std::sync::Arc;
 
 use bf_tree::{BfTree, Config};
 use bytemuck::bytes_of;
-use diskann::{
-    ANNError, ANNErrorKind, ANNResult,
-    error::IntoANNResult,
-    utils::{VectorRepr, object_pool::ObjectPool},
-};
+use diskann::{ANNError, ANNErrorKind, ANNResult, error::IntoANNResult, utils::VectorRepr};
 use diskann_quantization::CompressInto;
+use diskann_utils::object_pool::ObjectPool;
 use diskann_vector::distance::Metric;
 use thiserror::Error;
 

diskann-providers/src/model/graph/provider/async_/fast_memory_quant_vector_provider.rs

@@ -14,12 +14,9 @@
 use std::sync::{Arc, Mutex};
 
 use crate::storage::{StorageReadProvider, StorageWriteProvider};
-use diskann::{
-    ANNError, ANNResult,
-    error::IntoANNResult,
-    utils::{VectorRepr, object_pool::ObjectPool},
-};
+use diskann::{ANNError, ANNResult, error::IntoANNResult, utils::VectorRepr};
 use diskann_quantization::CompressInto;
+use diskann_utils::object_pool::ObjectPool;
 use diskann_vector::distance::Metric;
 
 use super::common::{AlignedMemoryVectorStore, TestCallCount};

diskann-providers/src/model/graph/provider/async_/memory_quant_vector_provider.rs

@@ -14,9 +14,10 @@ use crate::storage::{StorageReadProvider, StorageWriteProvider};
 use arc_swap::{ArcSwap, Guard};
 #[cfg(test)]
 use diskann::utils::VectorRepr;
-use diskann::{ANNError, ANNResult, utils::object_pool::ObjectPool};
+use diskann::{ANNError, ANNResult};
 #[cfg(test)]
 use diskann_quantization::CompressInto;
+use diskann_utils::object_pool::ObjectPool;
 use diskann_vector::{DistanceFunction, PreprocessedDistanceFunction, distance::Metric};
 
 use super::{VectorGuard, common::TestCallCount};

diskann-providers/src/model/pq/distance/common.rs

@@ -3,7 +3,7 @@
  * Licensed under the MIT license.
  */
 
-use diskann::utils::object_pool::{self, ObjectPool};
+use diskann_utils::object_pool::{self, ObjectPool};
 
 use crate::model::pq::fixed_chunk_pq_table::FixedChunkPQTable;
 

diskann-providers/src/model/pq/distance/dynamic.rs

@@ -5,7 +5,8 @@
 
 use std::{ops::Deref, sync::Arc};
 
-use diskann::{ANNResult, utils::object_pool::ObjectPool};
+use diskann::ANNResult;
+use diskann_utils::object_pool::ObjectPool;
 use diskann_vector::{DistanceFunction, PreprocessedDistanceFunction, distance::Metric};
 
 // Concrete implementations

diskann-providers/src/model/pq/distance/innerproduct.rs

@@ -5,10 +5,8 @@
 
 use std::{ops::Deref, sync::Arc};
 
-use diskann::{
-    ANNResult,
-    utils::object_pool::{self, ObjectPool, PoolOption},
-};
+use diskann::ANNResult;
+use diskann_utils::object_pool::{self, ObjectPool, PoolOption};
 use diskann_vector::PreprocessedDistanceFunction;
 
 use super::common::get_lookup_table_size;

diskann-providers/src/model/pq/distance/l2.rs

@@ -5,10 +5,8 @@
 
 use std::{ops::Deref, sync::Arc};
 
-use diskann::{
-    ANNResult,
-    utils::object_pool::{self, ObjectPool, PoolOption},
-};
+use diskann::ANNResult;
+use diskann_utils::object_pool::{self, ObjectPool, PoolOption};
 use diskann_vector::PreprocessedDistanceFunction;
 
 use super::common::get_lookup_table_size;

diskann-utils/src/lib.rs

@@ -15,6 +15,7 @@ pub use lifetime::WithLifetime;
 pub mod future;
 
 pub mod io;
+pub mod object_pool;
 pub mod sampling;
 
 // Views

diskann/src/graph/index.rs

@@ -52,9 +52,9 @@ use crate::{
     utils::{
         IntoUsize, TryIntoVectorId, VectorId,
         async_tools::{self, DynamicBalancer},
-        object_pool::{ObjectPool, PooledRef},
     },
 };
+use diskann_utils::object_pool::{ObjectPool, PooledRef};
 
 #[derive(Debug)]
 pub struct DiskANNIndex<DP: DataProvider> {

diskann/src/graph/search/scratch.rs

@@ -11,8 +11,9 @@ use std::collections::VecDeque;
 
 use crate::{
     neighbor::{Neighbor, NeighborPriorityQueue},
-    utils::{VectorId, object_pool::AsPooled},
+    utils::VectorId,
 };
+use diskann_utils::object_pool::AsPooled;
 use hashbrown::HashSet;
 
 /// In-mem index related limits

diskann/src/utils/mod.rs

@@ -3,8 +3,6 @@
  * Licensed under the MIT license.
  */
 
-pub mod object_pool;
-
 pub mod async_tools;
 
 #[allow(clippy::module_inception)]