com.github.dockerjava.core.command.BuildImageResultCallback Maven / Gradle / Ivy

Go to download
/*
 * Created on 21.07.2015
 */
package com.github.dockerjava.core.command;

import java.util.concurrent.TimeUnit;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.github.dockerjava.api.DockerClientException;
import com.github.dockerjava.api.model.BuildResponseItem;
import com.github.dockerjava.core.async.ResultCallbackTemplate;

/**
 *
 * @author marcus
 *
 */
public class BuildImageResultCallback extends ResultCallbackTemplate {

    private final static Logger LOGGER = LoggerFactory.getLogger(BuildImageResultCallback.class);

    private BuildResponseItem latestItem = null;

    @Override
    public void onNext(BuildResponseItem item) {
        this.latestItem = item;
        LOGGER.debug(item.toString());
    }

    /**
     * Awaits the image id from the response stream.
     *
     * @throws DockerClientException
     *             if the build fails.
     */
    public String awaitImageId() {
        try {
            awaitCompletion();
        } catch (InterruptedException e) {
            throw new DockerClientException("", e);
        }

        return getImageId();
    }

    /**
     * Awaits the image id from the response stream.
     *
     * @throws DockerClientException
     *             if the build fails or the timeout occurs.
     */
    public String awaitImageId(long timeout, TimeUnit timeUnit) {
        try {
            awaitCompletion(timeout, timeUnit);
        } catch (InterruptedException e) {
            throw new DockerClientException("Awaiting image id interrupted: ", e);
        }

        return getImageId();
    }

    private String getImageId() {
        if (latestItem == null || !latestItem.isBuildSuccessIndicated()) {
            throw new DockerClientException("Could not build image: " + latestItem.getError());
        } else {
            return latestItem.getImageId();
        }
    }

}