From 35fee1a5faa3e6b32cc157f873e9cde491d0024c Mon Sep 17 00:00:00 2001
From: Georgios Andrianakis <geoand@gmail.com>
Date: Tue, 12 Oct 2021 15:51:21 +0300
Subject: [PATCH] Ensure that async console logging does not prevent failed
 startup errors

Fixes: #20660
---
 .../runtime/ApplicationLifecycleManager.java  | 37 +++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/core/runtime/src/main/java/io/quarkus/runtime/ApplicationLifecycleManager.java b/core/runtime/src/main/java/io/quarkus/runtime/ApplicationLifecycleManager.java
index 01fbe2a50fb8f..0e43a2fee4998 100644
--- a/core/runtime/src/main/java/io/quarkus/runtime/ApplicationLifecycleManager.java
+++ b/core/runtime/src/main/java/io/quarkus/runtime/ApplicationLifecycleManager.java
@@ -8,6 +8,8 @@
 import java.util.concurrent.locks.Lock;
 import java.util.function.BiConsumer;
 import java.util.function.Consumer;
+import java.util.logging.ConsoleHandler;
+import java.util.logging.Handler;
 
 import javax.enterprise.context.spi.CreationalContext;
 import javax.enterprise.inject.Any;
@@ -19,8 +21,10 @@
 import org.eclipse.microprofile.config.spi.ConfigProviderResolver;
 import org.graalvm.nativeimage.ImageInfo;
 import org.jboss.logging.Logger;
+import org.jboss.logmanager.handlers.AsyncHandler;
 import org.wildfly.common.lock.Locks;
 
+import io.quarkus.bootstrap.logging.InitialConfigurator;
 import io.quarkus.bootstrap.runner.RunnerClassLoader;
 import io.quarkus.runtime.configuration.ConfigurationException;
 import io.quarkus.runtime.configuration.ProfileManager;
@@ -205,6 +209,7 @@ public static void run(Application application, Class<? extends QuarkusApplicati
                 } else {
                     applicationLogger.errorv(rootCause, "Failed to start application (with profile {0})",
                             ProfileManager.getActiveProfile());
+                    ensureConsoleLogsDrained();
                 }
             }
             stateLock.lock();
@@ -234,6 +239,38 @@ public static void run(Application application, Class<? extends QuarkusApplicati
         (exitCodeHandler == null ? defaultExitCodeHandler : exitCodeHandler).accept(getExitCode(), null); //this may not be called if shutdown was initiated by a signal
     }
 
+    // this is needed only when async console logging is enabled
+    private static void ensureConsoleLogsDrained() {
+        AsyncHandler asyncHandler = null;
+        for (Handler handler : InitialConfigurator.DELAYED_HANDLER.getHandlers()) {
+            if (handler instanceof AsyncHandler) {
+                asyncHandler = (AsyncHandler) handler;
+                Handler[] nestedHandlers = asyncHandler.getHandlers();
+                boolean foundNestedConsoleHandler = false;
+                for (Handler nestedHandler : nestedHandlers) {
+                    if (nestedHandler instanceof ConsoleHandler) {
+                        foundNestedConsoleHandler = true;
+                        break;
+                    }
+                }
+                if (!foundNestedConsoleHandler) {
+                    asyncHandler = null;
+                }
+            }
+            if (asyncHandler != null) {
+                break;
+            }
+        }
+        if (asyncHandler != null) {
+            try {
+                // all we can do is wait because the thread that takes records off the queue is a daemon thread and there is no way to interact with its lifecycle
+                Thread.sleep(200);
+            } catch (InterruptedException e) {
+                Thread.currentThread().interrupt();
+            }
+        }
+    }
+
     /**
      * Run some background cleanup once after the application has booted.
      * This will not be invoked for command mode, as it's not worth it for a short lived process.