Restart feature

src/main/java/org/jboss/as/plugin/common/PropertyNames.java

@@ -79,6 +79,10 @@ public interface PropertyNames {
 
     String RELOAD_TIMEOUT = "jboss-as.reload.timeout";
 
+    String RESTART = "jboss-as.restart";
+
+    String RESTART_TIMEOUT = "jboss-as.restart.timeout";
+
     String SERVER_CONFIG = "jboss-as.serverConfig";
 
     String SKIP = "jboss-as.skip";

src/main/java/org/jboss/as/plugin/server/Shutdown.java

@@ -41,7 +41,7 @@
 /**
  * Shuts down a running JBoss Application Server.
  * <p/>
- * Can also be used to issue a reload instead of a full shutdown.
+ * Can also be used to issue a reload or a restart instead of a full shutdown.
  *
  * @author <a href="mailto:[email protected]">James R. Perkins</a>
  */
@@ -60,27 +60,40 @@ public class Shutdown extends AbstractServerMojo {
     @Parameter(defaultValue = "30", property = PropertyNames.RELOAD_TIMEOUT, alias = "reload-timeout")
     private int reloadTimeout;
 
+    /**
+     * Set to {@code true} if a {@code restart} operation should be invoked instead of a {@code shutdown}.
+     */
+    @Parameter(defaultValue = "false", property = PropertyNames.RESTART)
+    private boolean restart;
+
+    /**
+     * The maximum time, in seconds, to wait for a live server after a restart.
+     */
+    @Parameter(defaultValue = "30", property = PropertyNames.RESTART_TIMEOUT, alias = "restart-timeout")
+    private int restartTimeout;
+
     @Override
     public void execute() throws MojoExecutionException, MojoFailureException {
         if (isSkip()) {
-            getLog().debug("Skipping server start");
+            getLog().debug("Skipping server shutdown");
             return;
         }
         try {
             final ModelControllerClient client = getClient();
             if (reload) {
                 client.execute(ServerOperations.createOperation(ServerOperations.RELOAD));
                 waitForStandalone(client, reloadTimeout);
+            } else if (restart) {
+                getLog().debug("Restarting");
+                ModelNode modelNode = ServerOperations.createOperation(ServerOperations.SHUTDOWN);
+                modelNode.get("restart").set(true);
+                client.execute(modelNode);
+                waitForStandalone(client, restartTimeout);
+                getLog().debug("We are back!");
             } else {
                 client.execute(ServerOperations.createOperation(ServerOperations.SHUTDOWN));
             }
-            // Bad hack to get maven to complete it's message output
-            try {
-                TimeUnit.MILLISECONDS.sleep(500L);
-            } catch (InterruptedException ignore) {
-                ignore.printStackTrace();
-                // no-op
-            }
+            waitForMessageOutput();
         } catch (Exception e) {
             throw new MojoExecutionException(String.format("Could not execute goal %s. Reason: %s", goal(), e.getMessage()), e);
         } finally {
@@ -93,6 +106,17 @@ public String goal() {
         return "shutdown";
     }
 
+
+    private void waitForMessageOutput() {
+        // Bad hack to get maven to complete it's message output
+        try {
+            TimeUnit.MILLISECONDS.sleep(500L);
+        } catch (InterruptedException ignore) {
+            // ignore.printStackTrace();
+            // no-op
+        }
+    }
+
     private void waitForStandalone(final ModelControllerClient client, final int startupTimeout) throws InterruptedException, IOException {
         long timeout = startupTimeout * 1000;
         final long sleep = 100L;