fixing TestFeatureResetIT

masseyke · Dec 6, 2023 · 3de223b · 3de223b
1 parent c1ce720
commit 3de223b
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/.../ml/src/main/java/org/elasticsearch/xpack/ml/utils/persistence/LimitAwareBulkIndexer.java b/.../ml/src/main/java/org/elasticsearch/xpack/ml/utils/persistence/LimitAwareBulkIndexer.java
@@ -62,18 +62,18 @@ private void execute() {
                 executor.accept(currentBulkRequest);
             } finally {
                 currentBulkRequest.close();
+                currentBulkRequest = new BulkRequest();
+                currentRamBytes = 0;
             }
-            currentBulkRequest = new BulkRequest();
-            currentRamBytes = 0;
         }
     }
 
     @Override
     public void close() {
-        try {
-            execute();
-        } finally {
-            currentBulkRequest.close();
+            try {
+                execute();
+            } finally {
+                currentBulkRequest.close();
+            }
         }
-    }
 }