Improve PullImageCmd for pulling an image through docker swarm.

hawky-4s- · KostyaSha · commit 4e66441a57e6 · 2017-05-05T05:40:55.000+03:00
diff --git a/src/main/java/com/github/dockerjava/core/command/PullImageResultCallback.java b/src/main/java/com/github/dockerjava/core/command/PullImageResultCallback.java
@@ -11,6 +11,11 @@
 import com.github.dockerjava.api.exception.DockerClientException;
 import com.github.dockerjava.api.model.PullResponseItem;
 import com.github.dockerjava.core.async.ResultCallbackTemplate;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.HashMap;
+import java.util.Map;
 
 /**
  *
@@ -21,15 +26,74 @@ public class PullImageResultCallback extends ResultCallbackTemplate<PullImageRes
 
     private static final Logger LOGGER = LoggerFactory.getLogger(PullImageResultCallback.class);
 
+    private boolean isSwarm = false;
+    private Map<String, PullResponseItem> results = null;
+
     @CheckForNull
     private PullResponseItem latestItem = null;
 
     @Override
     public void onNext(PullResponseItem item) {
-        this.latestItem = item;
+        // only do it once
+        if (results == null && latestItem == null) {
+            checkForDockerSwarmResponse(item);
+        }
+
+        if (isSwarm) {
+            handleDockerSwarmResponse(item);
+        } else {
+            handleDockerClientResponse(item);
+        }
         LOGGER.debug(item.toString());
     }
 
+    private void checkForDockerSwarmResponse(PullResponseItem item) {
+        if (item.getStatus().matches("Pulling\\s.+\\.{3}$")) {
+            isSwarm = true;
+            LOGGER.debug("Communicating with Docker Swarm.");
+        }
+    }
+
+    private void handleDockerSwarmResponse(PullResponseItem item) {
+        if (results == null) {
+            results = new HashMap<String, PullResponseItem>();
+        }
+        results.put(item.getId(), item);
+    }
+
+    private void checkDockerSwarmPullSuccessful() {
+        if (results.isEmpty()) {
+            throw new DockerClientException("Could not pull image");
+        } else {
+            boolean pullFailed = false;
+            StringBuilder sb = new StringBuilder();
+
+            for (PullResponseItem pullResponseItem : results.values()) {
+                if (!pullResponseItem.isPullSuccessIndicated()) {
+                    pullFailed = true;
+                    sb.append("[" + pullResponseItem.getId() + ";" + pullResponseItem.getError() + "]");
+                }
+            }
+
+            if (pullFailed) {
+                throw new DockerClientException("Could not pull image: " + sb.toString());
+            }
+        }
+    }
+
+    private void handleDockerClientResponse(PullResponseItem item) {
+        latestItem = item;
+    }
+
+    private void checkDockerClientPullSuccessful() {
+        if (latestItem == null) {
+            throw new DockerClientException("Could not pull image");
+        } else if (!latestItem.isPullSuccessIndicated()) {
+            String message = (latestItem.getError() != null) ? latestItem.getError() : latestItem.getStatus();
+            throw new DockerClientException("Could not pull image: " + message);
+        }
+    }
+
     /**
      * Awaits the image to be pulled successful.
      *
@@ -43,11 +107,10 @@ public void awaitSuccess() {
             throw new DockerClientException("", e);
         }
 
-        if (latestItem == null) {
-            throw new DockerClientException("Could not pull image");
-        } else if (!latestItem.isPullSuccessIndicated()) {
-            String message = (latestItem.getError() != null) ? latestItem.getError() : latestItem.getStatus();
-            throw new DockerClientException("Could not pull image: " + message);
+        if (isSwarm) {
+            checkDockerSwarmPullSuccessful();
+        } else {
+            checkDockerClientPullSuccessful();
         }
     }
 }
