Hide progress reporting behind Truffle boundary

enso-org · Feb 25, 2025 · a26bb2d · a26bb2d
1 parent f6156c5
commit a26bb2d
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/...time-instrument-common/src/main/java/org/enso/interpreter/service/ExecutionCallbacks.java b/...time-instrument-common/src/main/java/org/enso/interpreter/service/ExecutionCallbacks.java
@@ -100,23 +100,28 @@ public Object findCachedResult(IdExecutionService.Info info) {
       return result;
     } else {
       if (onProgressCallbackOrNull != null) {
-        if (cache.getPreferences().get(nodeId) == CachePreferences.Kind.BINDING_EXPRESSION) {
-          var newObserver =
-              ExecutionProgressObserver.startComputation(
-                  nodeId,
-                  (progress) -> {
-                    CompilerDirectives.transferToInterpreter();
-                    var expressionValue = ExpressionValue.progress(nodeId, progress, null);
-                    onProgressCallbackOrNull.accept(expressionValue);
-                  });
-          refreshObserver(newObserver);
-        }
+        reportEvaluationProgress(nodeId);
       }
     }
 
     return null;
   }
 
+  @CompilerDirectives.TruffleBoundary
+  private void reportEvaluationProgress(UUID nodeId) {
+    if (cache.getPreferences().get(nodeId) == CachePreferences.Kind.BINDING_EXPRESSION) {
+      var newObserver =
+          ExecutionProgressObserver.startComputation(
+              nodeId,
+              (progress) -> {
+                CompilerDirectives.transferToInterpreter();
+                var expressionValue = ExpressionValue.progress(nodeId, progress, null);
+                onProgressCallbackOrNull.accept(expressionValue);
+              });
+      refreshObserver(newObserver);
+    }
+  }
+
   private void refreshObserver(ExecutionProgressObserver newObserverOrNull) {
     var o = progressObserver;
     if (o != null) {