org.arquillian.spacelift.execution.impl.ShutdownHooks Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of arquillian-spacelift Show documentation
Show all versions of arquillian-spacelift Show documentation
Arquillian Process and Package Manager
/*
* JBoss, Home of Professional Open Source
* Copyright 2014, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* 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 org.arquillian.spacelift.execution.impl;
import java.util.concurrent.atomic.AtomicInteger;
import org.arquillian.spacelift.execution.Execution;
/**
* Utility class that registers a shutdown hook for execution.
*
* @author Karel Piwko
*
*/
public class ShutdownHooks {
/**
* Register a shutdown hook for {@code execution} that is automatically called if JVM exits
*
* @param execution Execution to be terminated when JVM exits
*/
public static void addHookFor(final Execution execution) {
Thread shutdownThread = new SpaceliftShutDownHook(execution);
Runtime.getRuntime().addShutdownHook(shutdownThread);
}
/**
* A shutdown thread for Spacelift that terminates running execution.
*
* @author Karel Piwko
*
* @param
*/
public static class SpaceliftShutDownHook extends Thread {
private static AtomicInteger shutdownCounter = new AtomicInteger(1);
private final Execution execution;
public SpaceliftShutDownHook(Execution execution) {
super("spacelift-shutdown-hook-" + shutdownCounter.getAndIncrement());
this.execution = execution;
}
@Override
public void run() {
if (execution != null) {
if (!execution.isFinished() && !execution.isMarkedAsFinished()) {
execution.terminate();
}
}
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy