default implementation for clean method (#80)

Signed-off-by: Thang PHAM <[email protected]>
powsybl · Oct 30, 2024 · 7eecbea · 7eecbea
1 parent be3fbde
commit 7eecbea
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/main/java/com/powsybl/ws/commons/computation/service/AbstractWorkerService.java b/src/main/java/com/powsybl/ws/commons/computation/service/AbstractWorkerService.java
@@ -153,12 +153,20 @@ public Consumer<Message<String>> consumeRun() {
                     this.handleNonCancellationException(resultContext, e, rootReporter);
                 }
             } finally {
-                futures.remove(resultContext.getResultUuid());
-                cancelComputationRequests.remove(resultContext.getResultUuid());
+                clean(resultContext);
             }
         };
     }
 
+    /**
+     * Perform cleaning
+     * @param resultContext The context of the computation
+     */
+    protected void clean(AbstractResultContext<C> resultContext) {
+        futures.remove(resultContext.getResultUuid());
+        cancelComputationRequests.remove(resultContext.getResultUuid());
+    }
+
     /**
      * Handle exception in consumeRun that is not a CancellationException
      * @param resultContext The context of the computation