Fix suspend query ordering in SessionState for one shot

nbauernfeind · Nov 9, 2023 · 4563ed9 · 4563ed9
1 parent be34bb2
commit 4563ed9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/src/main/java/io/deephaven/server/session/SessionState.java b/server/src/main/java/io/deephaven/server/session/SessionState.java
@@ -1487,11 +1487,11 @@ public ExportBuilder<T> onSuccess(final Runnable successHandler) {
          * @return the submitted export object
          */
         public ExportObject<T> submit(final Callable<T> exportMain) {
-            export.setWork(exportMain, errorHandler, successHandler, requiresSerialQueue);
             if (export.queryPerformanceRecorder != null && !export.qprIsForBatch) {
-                // transfer ownership of the qpr to the export
+                // transfer ownership of the qpr to the export before it can be resumed by the scheduler
                 export.queryPerformanceRecorder.suspendQuery();
             }
+            export.setWork(exportMain, errorHandler, successHandler, requiresSerialQueue);
             return export;
         }