Allow filtering by event type

docker-java-api/src/main/java/com/github/dockerjava/api/command/EventsCmd.java

@@ -2,10 +2,12 @@
 
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Stream;
 
 import javax.annotation.CheckForNull;
 
 import com.github.dockerjava.api.model.Event;
+import com.github.dockerjava.api.model.EventType;
 
 /**
  * Get events
@@ -33,6 +35,24 @@ public interface EventsCmd extends AsyncDockerCmd<EventsCmd, Event> {
      */
     EventsCmd withEventFilter(String... event);
 
+    /**
+     * @param eventTypes event types to filter
+     */
+    EventsCmd withEventTypeFilter(String... eventTypes);
+
+    /**
+     * This provides a type safe version of {@link #withEventTypeFilter(String...)}.
+     *
+     * @param eventTypes event types to filter
+     */
+    default EventsCmd withEventTypeFilter(EventType... eventTypes) {
+        return withEventTypeFilter(
+            Stream.of(eventTypes)
+                .map(EventType::getValue)
+                .toArray(String[]::new)
+        );
+    }
+
     /**
      * @param image
      *            - image to filter

docker-java-core/src/main/java/com/github/dockerjava/core/command/EventsCmdImpl.java

@@ -57,6 +57,13 @@ public EventsCmd withEventFilter(String... event) {
         return this;
     }
 
+    @Override
+    public EventsCmd withEventTypeFilter(String... eventTypes) {
+        checkNotNull(eventTypes, "event types have not been specified");
+        this.filters.withEventTypes(eventTypes);
+        return this;
+    }
+
     @Override
     public EventsCmd withLabelFilter(String... label) {
         checkNotNull(label, "label have not been specified");

docker-java-core/src/main/java/com/github/dockerjava/core/util/FiltersBuilder.java

@@ -10,6 +10,8 @@
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.regex.Pattern;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 /**
  * Representation of Docker filters.
@@ -64,6 +66,16 @@ public List<String> getContainer() {
         return getFilter("container");
     }
 
+    /**
+     * Filter by event types
+     *
+     * @param eventTypes an array of event types
+     */
+    public FiltersBuilder withEventTypes(String... eventTypes) {
+        withFilter("type", Stream.of(eventTypes).collect(Collectors.toList()));
+        return this;
+    }
+
     /**
      * Filter by labels
      *

docker-java/src/test/java/com/github/dockerjava/cmd/EventsCmdIT.java

@@ -3,6 +3,7 @@
 import com.github.dockerjava.api.async.ResultCallback;
 import com.github.dockerjava.api.command.CreateContainerResponse;
 import com.github.dockerjava.api.model.Event;
+import com.github.dockerjava.api.model.EventType;
 import com.github.dockerjava.utils.TestUtils;
 import org.junit.Test;
 import org.slf4j.Logger;
@@ -113,6 +114,57 @@ public void testEventStreamingWithFilter() throws Exception {
         }
     }
 
+    @Test
+    public void testEventStreamingWithContainerEventTypeFilter() throws Exception {
+        assumeNotSwarm("", dockerRule);
+
+        String startTime = getEpochTime();
+        int expectedEvents = 1;
+
+        EventsTestCallback eventCallback = new EventsTestCallback(expectedEvents);
+
+        dockerRule.getClient().eventsCmd()
+            .withSince(startTime)
+            .withEventTypeFilter(EventType.CONTAINER)
+            .withEventFilter("start")
+            .exec(eventCallback);
+
+        generateEvents();
+
+        List<Event> events = eventCallback.awaitExpectedEvents(30, TimeUnit.SECONDS);
+
+        // we should only get "start" events here
+        for (Event event : events) {
+            assertThat("Received event: " + event, event.getAction(), is("start"));
+            assertThat("Received event: " + event, event.getType(), is(EventType.CONTAINER));
+        }
+    }
+
+    @Test
+    public void testEventStreamingWithImageEventTypeFilter() throws Exception {
+        assumeNotSwarm("", dockerRule);
+
+        String startTime = getEpochTime();
+        int expectedEvents = 1;
+
+        EventsTestCallback eventCallback = new EventsTestCallback(expectedEvents);
+
+        dockerRule.getClient().eventsCmd()
+            .withSince(startTime)
+            .withEventTypeFilter(EventType.IMAGE)
+            .exec(eventCallback);
+
+        generateEvents();
+
+        List<Event> events = eventCallback.awaitExpectedEvents(30, TimeUnit.SECONDS);
+
+        // we should only get "pull" events here
+        for (Event event : events) {
+            assertThat("Received event: " + event, event.getAction(), is("pull"));
+            assertThat("Received event: " + event, event.getType(), is(EventType.IMAGE));
+        }
+    }
+
     /**
      * This method generates some events and returns the number of events being generated
      */