org.ehoffman.test.junit.ShutdownHook Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of JunitAopAllianceRunner Show documentation

The newest version!

package org.ehoffman.test.junit;

import java.io.Closeable;
import java.io.IOException;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

/**
 * Closes on jvm shutdown.
 * 
 * @author rexhoffman
 */
public class ShutdownHook implements Runnable {

    private Set closeables = Collections.synchronizedSet(new HashSet());

    public void addClosable(final Closeable closeable) {
        closeables.add(closeable);
    }

    @Override
    public void run() {
        for (Closeable closable : closeables) {
            try {
                closable.close();
            } catch (final IOException e) {
                e.printStackTrace();
            }
        }
    }
}