Fixes piranhacloud#3573 - Add CRaC support to Piranha Core Profile

mnriem · Nov 9, 2023 · ce5f2f7 · ce5f2f7
1 parent ada8145
commit ce5f2f7
Show file tree

Hide file tree

Showing 4 changed files with 66 additions and 0 deletions.
diff --git a/dist/coreprofile/pom.xml b/dist/coreprofile/pom.xml
@@ -52,6 +52,12 @@
             <version>${project.version}</version>
             <scope>compile</scope>
         </dependency>
+        <dependency>
+            <groupId>cloud.piranha.http</groupId>
+            <artifactId>piranha-http-crac</artifactId>
+            <version>${project.version}</version>
+            <scope>compile</scope>
+        </dependency>
     </dependencies>
 
     <build>

diff --git a/dist/coreprofile/src/main/java/cloud/piranha/dist/coreprofile/CoreProfilePiranha.java b/dist/coreprofile/src/main/java/cloud/piranha/dist/coreprofile/CoreProfilePiranha.java
@@ -39,13 +39,16 @@
 import cloud.piranha.feature.impl.DefaultFeatureManager;
 import cloud.piranha.feature.logging.LoggingFeature;
 import cloud.piranha.feature.webapp.WebAppFeature;
+import cloud.piranha.http.api.HttpServer;
 import java.io.File;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.io.PrintWriter;
 import java.lang.System.Logger;
+import static java.lang.System.Logger.Level.ERROR;
 import static java.lang.System.Logger.Level.INFO;
 import static java.lang.System.Logger.Level.WARNING;
+import java.lang.reflect.InvocationTargetException;
 
 /**
  * The Piranha Core Profile runtime.
@@ -146,6 +149,27 @@ public void run() {
             httpFeature.setPort(configuration.getInteger("httpPort"));
             httpFeature.init();
             httpFeature.getHttpServer().setHttpServerProcessor(webAppFeature.getHttpServerProcessor());
+
+            /*
+             * Enable Project CRaC.
+             */
+            if (configuration.getBoolean("cracEnabled", false)) {
+                HttpServer httpServer = httpFeature.getHttpServer();
+                try {
+                    HttpServer cracHttpServer = (HttpServer) Class
+                            .forName("cloud.piranha.http.crac.CracHttpServer")
+                            .getDeclaredConstructor(HttpServer.class)
+                            .newInstance(httpServer);
+                    httpServer = cracHttpServer;
+                } catch (ClassNotFoundException | IllegalAccessException
+                        | IllegalArgumentException | InstantiationException
+                        | NoSuchMethodException | SecurityException
+                        | InvocationTargetException t) {
+                    LOGGER.log(ERROR, "Unable to construct HTTP server", t);
+                }
+                httpFeature.setHttpServer(httpServer);
+            }
+
             httpFeature.start();
         }
 
@@ -163,6 +187,27 @@ public void run() {
             httpsFeature.setPort(configuration.getInteger("httpsPort"));
             httpsFeature.init();
             httpsFeature.getHttpsServer().setHttpServerProcessor(webAppFeature.getHttpServerProcessor());
+
+            /*
+             * Enable Project CRaC.
+             */
+            if (configuration.getBoolean("cracEnabled", false)) {
+                HttpServer httpServer = httpsFeature.getHttpsServer();
+                try {
+                    HttpServer cracHttpServer = (HttpServer) Class
+                            .forName("cloud.piranha.http.crac.CracHttpServer")
+                            .getDeclaredConstructor(HttpServer.class)
+                            .newInstance(httpServer);
+                    httpServer = cracHttpServer;
+                } catch (ClassNotFoundException | IllegalAccessException
+                        | IllegalArgumentException | InstantiationException
+                        | NoSuchMethodException | SecurityException
+                        | InvocationTargetException t) {
+                    LOGGER.log(ERROR, "Unable to construct HTTP server", t);
+                }
+                httpsFeature.setHttpsServer(httpServer);
+            }
+
             httpsFeature.start();
         }
 

diff --git a/dist/coreprofile/src/main/java/cloud/piranha/dist/coreprofile/CoreProfilePiranhaBuilder.java b/dist/coreprofile/src/main/java/cloud/piranha/dist/coreprofile/CoreProfilePiranhaBuilder.java
@@ -78,6 +78,17 @@ public CoreProfilePiranhaBuilder contextPath(String contextPath) {
         piranha.getConfiguration().setString("contextPath", contextPath);
         return this;
     }
+
+    /**
+     * Set the CRaC enabled flag.
+     *
+     * @param crac the CRaC enabled flag.
+     * @return the builder.
+     */
+    public CoreProfilePiranhaBuilder crac(boolean crac) {
+        piranha.getConfiguration().setBoolean("cracEnabled", crac);
+        return this;
+    }
 
     /**
      * Set the exit on stop flag.

diff --git a/dist/coreprofile/src/main/java/cloud/piranha/dist/coreprofile/CoreProfilePiranhaMain.java b/dist/coreprofile/src/main/java/cloud/piranha/dist/coreprofile/CoreProfilePiranhaMain.java
@@ -70,6 +70,9 @@ private CoreProfilePiranhaBuilder processArguments(String[] arguments) {
         int httpsPort = 0;
         if (arguments != null) {
             for (int i = 0; i < arguments.length; i++) {
+                if (arguments[i].equals("--enable-crac")) {
+                    builder = builder.crac(true);
+                }
                 if (arguments[i].equals("--extension-class")) {
                     builder = builder.extensionClass(arguments[i + 1]);
                 }
@@ -145,6 +148,7 @@ private static void showHelp() {
         LOGGER.log(Level.INFO, "");
         LOGGER.log(Level.INFO,
                 """
+  --enable-crac                        - Enable Project CRaC support
   --extension-class <className>        - Set the extension to use
   --context-path <string>              - Set the Servlet context path
   --help                               - Show this help