From 3827ba09025df03448aa4b16d82ede5ce2c607ac Mon Sep 17 00:00:00 2001
From: Nate Bauernfeind <natebauernfeind@deephaven.io>
Date: Mon, 2 Dec 2024 14:04:26 -0700
Subject: [PATCH] fix: erroneous QueryCompiler retry when nothing unaccounted
 for (#6451)

Fixes #6216
---
 .../engine/context/TestQueryCompiler.java     | 65 +++++++++++++++----
 .../engine/context/QueryCompilerImpl.java     |  2 +-
 2 files changed, 52 insertions(+), 15 deletions(-)

diff --git a/engine/context/src/test/java/io/deephaven/engine/context/TestQueryCompiler.java b/engine/context/src/test/java/io/deephaven/engine/context/TestQueryCompiler.java
index d9bca87a113..cddb86f2291 100644
--- a/engine/context/src/test/java/io/deephaven/engine/context/TestQueryCompiler.java
+++ b/engine/context/src/test/java/io/deephaven/engine/context/TestQueryCompiler.java
@@ -264,18 +264,17 @@ public void testCollidingCompile() throws Exception {
 
     @Test
     public void testMultiCompileWithFailure() throws ExecutionException, InterruptedException {
-        final String goodProgram = String.join(
-                "\n",
+        final String goodProgram = String.join("\n",
                 "public class GoodTest {",
                 "   public static void main (String [] args) {",
                 "   }",
                 "}");
-        final String badProgram = String.join(
-                "\n",
-                "public class BadTest {",
-                "   public static void main (String [] args) {",
-                "   }",
-                "}}");
+        final String badProgram = String.join("\n",
+                "public class Formula {",
+                "    public Formula() {",
+                "        S.badCall(0);",
+                "    }",
+                "}");
 
         QueryCompilerRequest[] requests = new QueryCompilerRequest[] {
                 QueryCompilerRequest.builder()
@@ -299,15 +298,53 @@ public void testMultiCompileWithFailure() throws ExecutionException, Interrupted
                         CompletionStageFuture.make(),
                 };
 
-        try {
-            ExecutionContext.getContext().getQueryCompiler().compile(requests, resolvers);
-            // noinspection DataFlowIssue
-            throw Assert.statementNeverExecuted();
-        } catch (Exception ignored) {
-        }
+        ExecutionContext.getContext().getQueryCompiler().compile(requests, resolvers);
 
         Assert.eqTrue(resolvers[0].getFuture().isDone(), "resolvers[0].getFuture().isDone()");
         Assert.eqTrue(resolvers[1].getFuture().isDone(), "resolvers[0].getFuture().isDone()");
         Assert.neqNull(resolvers[1].getFuture().get(), "resolvers[1].getFuture().get()");
     }
+
+    @Test
+    public void testMultiCompileWithFailureSecond() throws ExecutionException, InterruptedException {
+        final String badProgram = String.join("\n",
+                "public class Formula {",
+                "    public Formula() {",
+                "        S.badCall(0);",
+                "    }",
+                "}");
+        final String goodProgram = String.join("\n",
+                "public class Formula {",
+                "   public static void main (String [] args) {",
+                "   }",
+                "}");
+
+        QueryCompilerRequest[] requests = new QueryCompilerRequest[] {
+                QueryCompilerRequest.builder()
+                        .description("Test Good Compile")
+                        .className("Formula")
+                        .classBody(goodProgram)
+                        .packageNameRoot("com.deephaven.test")
+                        .build(),
+                QueryCompilerRequest.builder()
+                        .description("Test Bad Compile")
+                        .className("Formula")
+                        .classBody(badProgram)
+                        .packageNameRoot("com.deephaven.test")
+                        .build(),
+        };
+
+        // noinspection unchecked
+        CompletionStageFuture.Resolver<Class<?>>[] resolvers =
+                (CompletionStageFuture.Resolver<Class<?>>[]) new CompletionStageFuture.Resolver[] {
+                        CompletionStageFuture.make(),
+                        CompletionStageFuture.make(),
+                };
+
+        ExecutionContext.getContext().getQueryCompiler().compile(requests, resolvers);
+
+        Assert.eqTrue(resolvers[1].getFuture().isDone(), "resolvers[0].getFuture().isDone()");
+        Assert.eqTrue(resolvers[0].getFuture().isDone(), "resolvers[0].getFuture().isDone()");
+        Assert.neqNull(resolvers[0].getFuture().get(), "resolvers[1].getFuture().get()");
+    }
 }
diff --git a/engine/table/src/main/java/io/deephaven/engine/context/QueryCompilerImpl.java b/engine/table/src/main/java/io/deephaven/engine/context/QueryCompilerImpl.java
index fd689835125..da433f5b6f4 100644
--- a/engine/table/src/main/java/io/deephaven/engine/context/QueryCompilerImpl.java
+++ b/engine/table/src/main/java/io/deephaven/engine/context/QueryCompilerImpl.java
@@ -1004,7 +1004,7 @@ private boolean doCreateClassesSingleRound(
             }
         });
 
-        return wantRetry;
+        return wantRetry && !toRetry.isEmpty();
     }
 
     /**