zed.dockerunit.DockerTester Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zed-dockerunit Show documentation
A application used as an example on how to set up pushing its components to the Central Repository.
The newest version!
package zed.dockerunit;

import com.github.dockerjava.api.DockerClient;
import com.github.dockerjava.api.model.ExposedPort;
import com.github.dockerjava.api.model.PortBinding;
import com.github.dockerjava.api.model.Ports;
import com.github.dockerjava.core.DockerClientConfig.DockerClientConfigBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import static com.github.dockerjava.core.DockerClientBuilder.getInstance;
import static com.github.dockerjava.core.DockerClientConfig.createDefaultConfigBuilder;
import static com.google.common.collect.Sets.newConcurrentHashSet;
import static java.lang.Runtime.getRuntime;

public class DockerTester {

    private static final Logger LOG = LoggerFactory.getLogger(DockerTester.class);

    private final DockerClient docker;

    private Set startedContainers = newConcurrentHashSet();

    public DockerTester(DockerClient docker) {
        this.docker = docker;
    }

    public static DockerTester dockerTester(String apiVersion) {
        DockerClientConfigBuilder config = createDefaultConfigBuilder();
        if (apiVersion != null) {
            config.withVersion(apiVersion);
        }
        DockerClient docker = getInstance(config.build()).build();
        return new DockerTester(docker);
    }

    public static DockerTester dockerTester() {
        return dockerTester(null);
    }

    public String run(String image, int... ports) {
        String containerId = docker.createContainerCmd(image).exec().getId();
        List portBindings = new ArrayList<>();
        if (ports != null) {
            for (int port : ports) {
                portBindings.add(new PortBinding(new Ports.Binding(port), new ExposedPort(port)));
            }
        }
        docker.startContainerCmd(containerId).withPortBindings(portBindings.toArray(new PortBinding[portBindings.size()])).exec();
        startedContainers.add(containerId);
        registerShutdown(containerId);
        return containerId;
    }

    public void stop(String containerId) {
        docker.stopContainerCmd(containerId).exec();
        startedContainers.remove(containerId);
    }

    public void stopAll() {
        for (String containerId : startedContainers) {
            stop(containerId);
        }
    }

    public void registerShutdown(String containerId) {
        getRuntime().addShutdownHook(new Thread() {
            @Override
            public void run() {
                try {
                    DockerTester.this.stop(containerId);
                } catch (IllegalStateException e) {
                    LOG.info("Container {} is not running: {}", containerId, e.getMessage());
                }
            }
        });
    }

    public DockerClient docker() {
        return docker;
    }

}