org.arquillian.cube.impl.client.BeforeStopContainerObserver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arquillian-cube-docker Show documentation

There is a newer version: 2.0.0.Alpha1

Show newest version

package org.arquillian.cube.impl.client;

import java.io.IOException;
import java.util.List;
import java.util.Map;

import org.arquillian.cube.impl.docker.DockerClientExecutor;
import org.arquillian.cube.spi.Cube;
import org.arquillian.cube.spi.CubeRegistry;
import org.arquillian.cube.spi.event.lifecycle.BeforeStop;
import org.jboss.arquillian.core.api.annotation.Observes;

public class BeforeStopContainerObserver {

    private static final String BEFORE_STOP = "beforeStop";
    private static final String COPY = "copy";
    private static final String LOG = "log";

    @SuppressWarnings("unchecked")
    public void processCommands(@Observes BeforeStop beforeStop, CubeRegistry cubeRegistry,
            DockerClientExecutor dockerClientExecutor) throws IOException {

        Cube cube = cubeRegistry.getCube(beforeStop.getCubeId());
        Map configuration = cube.configuration();

        if (configuration.containsKey(BEFORE_STOP)) {
            List> beforeStopConfiguration = (List>) configuration
                    .get(BEFORE_STOP);

            for (Map map : beforeStopConfiguration) {
                if (map.containsKey(COPY)) {
                    Map copyConfiguration = (Map) map.get(COPY);
                    dockerClientExecutor.copyFromContainer(beforeStop.getCubeId(), copyConfiguration);
                } else {
                    if (map.containsKey(LOG)) {
                        Map logConfiguration = (Map) map.get(LOG);
                        dockerClientExecutor.copyLog(beforeStop.getCubeId(), logConfiguration);
                    }
                }
            }
        }
    }
}