Log warning when threadpool for cache maintenance is null

Signed-off-by: owenhalpert <ohalpert@gmail.com>
owenhalpert · Jan 10, 2025 · 1a88678 · 1a88678
1 parent a7e5ff4
commit 1a88678
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/knn/index/memory/NativeMemoryCacheManager.java b/src/main/java/org/opensearch/knn/index/memory/NativeMemoryCacheManager.java
@@ -110,8 +110,11 @@ private void initialize(NativeMemoryCacheManagerDto nativeMemoryCacheDTO) {
         cacheCapacityReached = new AtomicBoolean(false);
         accessRecencyQueue = new ConcurrentLinkedDeque<>();
         cache = cacheBuilder.build();
+
         if (threadPool != null) {
             startMaintenance(cache);
+        } else {
+            logger.warn("ThreadPool is null during NativeMemoryCacheManager initialization. Maintenance will not be started.");
         }
     }
 

diff --git a/...java/org/opensearch/knn/quantization/models/quantizationState/QuantizationStateCache.java b/...java/org/opensearch/knn/quantization/models/quantizationState/QuantizationStateCache.java
@@ -78,8 +78,11 @@ private void buildCache() {
             )
             .removalListener(this::onRemoval)
             .build();
+
         if (threadPool != null) {
             startMaintenance(cache);
+        } else {
+            log.warn("ThreadPool is null during QuantizationStateCache initialization. Maintenance will not be started.");
         }
     }