io.joshworks.snappy.maven.RunMojo Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012-2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.joshworks.snappy.maven;

import io.joshworks.snappy.maven.tools.JavaExecutable;
import io.joshworks.snappy.maven.tools.RunProcess;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugins.annotations.Execute;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.ResolutionScope;

import java.net.URLClassLoader;
import java.util.List;

/**
 * Run an executable archive application.
 *
 * @author Phillip Webb
 * @author Stephane Nicoll
 */
@Mojo(name = "run", requiresProject = true, defaultPhase = LifecyclePhase.VALIDATE, requiresDependencyResolution = ResolutionScope.TEST)
@Execute(phase = LifecyclePhase.TEST_COMPILE)
public class RunMojo extends AbstractRunMojo {

    @Override
    protected void runWithForkedJvm(List args) throws MojoExecutionException {
        try {
            int exitCode = new RunProcess(new JavaExecutable().toString()).run(true,
                    args.toArray(new String[args.size()]));
            if (exitCode != 0) {
                throw new MojoExecutionException(
                        "Application finished with non-zero exit code: " + exitCode);
            }
        } catch (Exception ex) {
            throw new MojoExecutionException("Could not exec java", ex);
        }
    }

    @Override
    protected void runWithMavenJvm(String startClassName, String... arguments)
            throws MojoExecutionException {
        IsolatedThreadGroup threadGroup = new IsolatedThreadGroup(startClassName);
        Thread launchThread = new Thread(threadGroup,
                new LaunchRunner(startClassName, arguments), "main");
        launchThread.setContextClassLoader(new URLClassLoader(getClassPathUrls()));
        launchThread.start();
        join(threadGroup);
        threadGroup.rethrowUncaughtException();
    }

    private void join(ThreadGroup threadGroup) {
        boolean hasNonDaemonThreads;
        do {
            hasNonDaemonThreads = false;
            Thread[] threads = new Thread[threadGroup.activeCount()];
            threadGroup.enumerate(threads);
            for (Thread thread : threads) {
                if (thread != null && !thread.isDaemon()) {
                    try {
                        hasNonDaemonThreads = true;
                        thread.join();
                    } catch (InterruptedException ex) {
                        Thread.currentThread().interrupt();
                    }
                }
            }
        }
        while (hasNonDaemonThreads);
    }

}