fix: server not supported

ReaJason · Dec 10, 2024 · f1bb363 · f1bb363
1 parent 3b4a353
commit f1bb363
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/generator/src/main/java/com/reajason/javaweb/GeneratorMain.java b/generator/src/main/java/com/reajason/javaweb/GeneratorMain.java
@@ -1,9 +1,11 @@
 package com.reajason.javaweb;
 
 import com.reajason.javaweb.config.*;
+import com.reajason.javaweb.memsell.jboss.JbossShell;
 import com.reajason.javaweb.memsell.jetty.JettyShell;
 import com.reajason.javaweb.memsell.packer.Packer;
 import com.reajason.javaweb.memsell.tomcat.TomcatShell;
+import com.reajason.javaweb.memsell.undertow.UndertowShell;
 import lombok.SneakyThrows;
 import org.apache.commons.codec.binary.Base64;
 
@@ -19,6 +21,8 @@
 public class GeneratorMain {
     static TomcatShell tomcatShell = new TomcatShell();
     static JettyShell jettyShell = new JettyShell();
+    static JbossShell jbossAsShell = new JbossShell();
+    static UndertowShell undertowShell = new UndertowShell();
 
     public static void main(String[] args) throws IOException {
         ShellConfig shellConfig = ShellConfig.builder()
@@ -47,6 +51,10 @@ public static GenerateResult generate(ShellConfig shellConfig, InjectorConfig in
                 return tomcatShell.generate(shellConfig, injectorConfig, shellToolConfig);
             case JETTY:
                 return jettyShell.generate(shellConfig, injectorConfig, shellToolConfig);
+            case JBOSS:
+                return jbossAsShell.generate(shellConfig, injectorConfig, shellToolConfig);
+            case UNDERTOW:
+                return undertowShell.generate(shellConfig, injectorConfig, shellToolConfig);
             case BES:
                 break;
             case RESIN: