change to reserved memory

opendistro-for-elasticsearch · Dec 21, 2020 · ce0654c · ce0654c
1 parent aaced0b
commit ce0654c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/com/amazon/opendistroforelasticsearch/ad/task/ADTaskCacheManager.java b/src/main/java/com/amazon/opendistroforelasticsearch/ad/task/ADTaskCacheManager.java
@@ -154,7 +154,7 @@ public void setThresholdModelTrained(String taskId, boolean trained) {
             long cacheSize = trainingDataMemorySize(size);
             taskCache.getThresholdModelTrainingData().clear();
             taskCache.getCacheMemorySize().getAndAdd(-cacheSize);
-            memoryTracker.releaseMemory(cacheSize, false, HISTORICAL_SINGLE_ENTITY_DETECTOR);
+            memoryTracker.releaseMemory(cacheSize, true, HISTORICAL_SINGLE_ENTITY_DETECTOR);
         }
     }
 
@@ -220,7 +220,7 @@ private long calculateADTaskCacheSize(ADTask adTask) {
      */
     public void remove(String taskId) {
         if (contains(taskId)) {
-            memoryTracker.releaseMemory(getBatchTaskCache(taskId).getCacheMemorySize().get(), false, HISTORICAL_SINGLE_ENTITY_DETECTOR);
+            memoryTracker.releaseMemory(getBatchTaskCache(taskId).getCacheMemorySize().get(), true, HISTORICAL_SINGLE_ENTITY_DETECTOR);
             taskCaches.remove(taskId);
         }
     }