support pull image

Author: gy2006

docker/src/main/java/com/flowci/docker/DockerManager.java

@@ -5,4 +5,6 @@ public interface DockerManager {
     String DockerLocalHost = "unix:///var/run/docker.sock";
 
     ContainerManager getContainerManager();
+
+    ImageManager getImageManager();
 }

docker/src/main/java/com/flowci/docker/DockerSDKManager.java

@@ -1,21 +1,31 @@
 package com.flowci.docker;
 
+import com.flowci.docker.domain.DockerCallback;
 import com.flowci.docker.domain.DockerStartOption;
 import com.flowci.util.ObjectsHelper;
 import com.github.dockerjava.api.DockerClient;
 import com.github.dockerjava.api.command.*;
+import com.github.dockerjava.api.exception.DockerException;
 import com.github.dockerjava.api.model.Container;
+import com.github.dockerjava.api.model.Image;
+import com.github.dockerjava.api.model.PullResponseItem;
 import com.github.dockerjava.core.DefaultDockerClientConfig;
 import com.github.dockerjava.core.DockerClientBuilder;
+import lombok.extern.log4j.Log4j2;
 
 import java.util.List;
+import java.util.concurrent.TimeUnit;
+import java.util.function.Consumer;
 
+@Log4j2
 public class DockerSDKManager implements DockerManager {
 
     private final String dockerHost;
 
     private final ContainerManager containerManager = new ContainerManagerImpl();
 
+    private final ImageManager imageManager = new ImageMangerImpl();
+
     public DockerSDKManager(String dockerHost) {
         this.dockerHost = dockerHost;
     }
@@ -25,6 +35,59 @@ public ContainerManager getContainerManager() {
         return containerManager;
     }
 
+    @Override
+    public ImageManager getImageManager() {
+        return imageManager;
+    }
+
+    private class ImageMangerImpl implements ImageManager {
+
+        @Override
+        public void pull(String image, int timeoutInSeconds, Consumer<PullResponseItem> progress) throws Exception {
+            String imageUrl = "docker.io/library/" + image;
+            if (image.contains("/")) {
+                imageUrl = "docker.io/" + image;
+            }
+
+            try (DockerClient client = newClient()) {
+                if (findImage(client, image).size() >= 1) {
+                    return;
+                }
+
+                PullImageCallback callback = client.pullImageCmd(imageUrl).exec(new PullImageCallback(progress));
+                boolean await = callback.getCounter().await(timeoutInSeconds, TimeUnit.SECONDS);
+
+                if (!await) {
+                    throw new DockerException(String.format("Timeout when pull image %s", image), 0);
+                }
+
+                if (findImage(client, image).isEmpty()) {
+                    throw new DockerException(String.format("Failed on pull image %s", image), 0);
+                }
+            }
+        }
+
+        private List<Image> findImage(DockerClient client, String image) {
+            return client.listImagesCmd().withImageNameFilter(image).exec();
+        }
+    }
+
+    private static class PullImageCallback extends DockerCallback<PullResponseItem> {
+
+        private final Consumer<PullResponseItem> progress;
+
+        private PullImageCallback(Consumer<PullResponseItem> progress) {
+            this.progress = progress;
+        }
+
+        @Override
+        public void onNext(PullResponseItem item) {
+            if (progress != null && item != null) {
+                progress.accept(item);
+            }
+        }
+    }
+
     private class ContainerManagerImpl implements ContainerManager {
 
         @Override

docker/src/main/java/com/flowci/docker/ImageManager.java

@@ -0,0 +1,13 @@
+package com.flowci.docker;
+
+import com.github.dockerjava.api.model.PullResponseItem;
+
+import java.util.function.Consumer;
+
+public interface ImageManager {
+
+    /**
+     * Sync call
+     */
+    void pull(String image, int timeoutInSeconds, Consumer<PullResponseItem> progress) throws Exception;
+}

docker/src/main/java/com/flowci/docker/domain/DockerCallback.java

@@ -0,0 +1,34 @@
+package com.flowci.docker.domain;
+
+import com.github.dockerjava.api.async.ResultCallback;
+import lombok.Getter;
+
+import java.io.Closeable;
+import java.io.IOException;
+import java.util.concurrent.CountDownLatch;
+
+public abstract class DockerCallback<T> implements ResultCallback<T> {
+
+    @Getter
+    protected final CountDownLatch counter = new CountDownLatch(1);
+
+    @Override
+    public void onStart(Closeable closeable) {
+
+    }
+
+    @Override
+    public void onError(Throwable throwable) {
+
+    }
+
+    @Override
+    public void onComplete() {
+        counter.countDown();
+    }
+
+    @Override
+    public void close() throws IOException {
+
+    }
+}

docker/src/test/java/com/flowci/docker/test/DockerSDKManagerTest.java

@@ -4,6 +4,7 @@
 import com.flowci.docker.DockerManager;
 import com.flowci.docker.DockerSDKManager;
 import com.flowci.docker.domain.DockerStartOption;
+import com.github.dockerjava.api.exception.NotFoundException;
 import com.github.dockerjava.api.model.Container;
 import org.junit.After;
 import org.junit.Assert;
@@ -26,6 +27,13 @@ public void clean() {
         manager = null;
     }
 
+    @Test
+    public void should_pull_image() throws Exception {
+        manager.getImageManager().pull("hello-world", 60, (item) -> {
+            System.out.println(item.getStatus());
+        });
+    }
+
     @Test
     public void should_list_container() throws Exception {
         List<Container> containers = manager.getContainerManager().list(null, null);
@@ -45,4 +53,19 @@ public void should_create_and_start_container() throws Exception {
         cm.stop(cid);
         cm.delete(cid);
     }
+
+    @Test(expected = NotFoundException.class)
+    public void should_throw_exception_when_resume_cid_not_exist() throws Exception {
+        manager.getContainerManager().resume("1231231");
+    }
+
+    @Test(expected = NotFoundException.class)
+    public void should_throw_exception_when_stop_cid_not_exist() throws Exception {
+        manager.getContainerManager().stop("1231231");
+    }
+
+    @Test(expected = NotFoundException.class)
+    public void should_throw_exception_when_delete_cid_not_exist() throws Exception {
+        manager.getContainerManager().delete("1231231");
+    }
 }