io.github.mike10004.containment.lifecycle.LocalContainerCreatorFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of containment-lifecycle Show documentation

There is a newer version: 0.6

Show newest version

package io.github.mike10004.containment.lifecycle;

import com.github.dockerjava.core.DefaultDockerClientConfig;
import com.github.dockerjava.core.DockerClientConfig;
import io.github.mike10004.containment.ContainerCreator;
import io.github.mike10004.containment.ContainmentException;
import io.github.mike10004.containment.dockerjava.DefaultDjDockerManager;
import io.github.mike10004.containment.dockerjava.DjContainerMonitor;
import io.github.mike10004.containment.dockerjava.DjDockerManager;

import java.util.function.Function;

/**
 * Constructor of creators for local containers. Cleanup of these containers is managed
 * by the caller, but a global monitor adds a shutdown hook to clean up ones the caller forgot.
 */
class LocalContainerCreatorFactory implements ContainerCreatorFactory {

    private final Function djConstructor;
    private final Function shutdownHookMonitorFactory;

    public LocalContainerCreatorFactory(Function djConstructor, Function shutdownHookMonitorFactory) {
        this.djConstructor = djConstructor;
        this.shutdownHookMonitorFactory = shutdownHookMonitorFactory;
    }

    @Override
    public final ContainerCreator instantiate() throws ContainmentException {
        DockerClientConfig config = DefaultDockerClientConfig.createDefaultConfigBuilder().build();
        DjDockerManager manager = new DefaultDjDockerManager(config, shutdownHookMonitorFactory.apply(config));
        return djConstructor.apply(manager);
    }
}