add support for DeviceRequests parameter (#1303)

hturki · web-flow · commit 6aa95c350650 · 2020-02-17T22:10:52.000+01:00
diff --git a/docker-java-api/src/main/java/com/github/dockerjava/api/model/DeviceRequest.java b/docker-java-api/src/main/java/com/github/dockerjava/api/model/DeviceRequest.java
@@ -0,0 +1,75 @@
+package com.github.dockerjava.api.model;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.EqualsAndHashCode;
+import lombok.ToString;
+
+import java.io.Serializable;
+import java.util.List;
+import java.util.Map;
+
+@EqualsAndHashCode
+@ToString
+public class DeviceRequest implements Serializable {
+    public static final long serialVersionUID = 1L;
+
+    @JsonProperty("Driver")
+    private String driver;
+
+    @JsonProperty("Count")
+    private Integer count;
+
+    @JsonProperty("DeviceIDs")
+    private List<String> deviceIds;
+
+    @JsonProperty("Capabilities")
+    private List<List<String>> capabilities;
+
+    @JsonProperty("Options")
+    private Map<String, String> options;
+
+    public String getDriver() {
+        return driver;
+    }
+
+    public DeviceRequest withDriver(String driver) {
+        this.driver = driver;
+        return this;
+    }
+
+    public Integer getCount() {
+        return count;
+    }
+
+    public DeviceRequest withCount(Integer count) {
+        this.count = count;
+        return this;
+    }
+
+    public List<String> getDeviceIds() {
+        return deviceIds;
+    }
+
+    public DeviceRequest withDeviceIds(List<String> deviceIds) {
+        this.deviceIds = deviceIds;
+        return this;
+    }
+
+    public List<List<String>> getCapabilities() {
+        return capabilities;
+    }
+
+    public DeviceRequest withCapabilities(List<List<String>> capabilities) {
+        this.capabilities = capabilities;
+        return this;
+    }
+
+    public Map<String, String> getOptions() {
+        return options;
+    }
+
+    public DeviceRequest withOptions(Map<String, String> options) {
+        this.options = options;
+        return this;
+    }
+}
diff --git a/docker-java-api/src/main/java/com/github/dockerjava/api/model/HostConfig.java b/docker-java-api/src/main/java/com/github/dockerjava/api/model/HostConfig.java
@@ -112,6 +112,12 @@ public static HostConfig newHostConfig() {
     @JsonProperty("DeviceCgroupRules")
     private List<String> deviceCgroupRules;
 
+    /**
+     * @since {@link com.github.dockerjava.core.RemoteApiVersion#VERSION_1_40}
+     */
+    @JsonProperty("DeviceRequests")
+    private List<DeviceRequest> deviceRequests;
+
     /**
      * @since {@link RemoteApiVersion#VERSION_1_25}
      */
@@ -1037,6 +1043,16 @@ public HostConfig withDeviceCgroupRules(List<String> deviceCgroupRules) {
         return this;
     }
 
+    @CheckForNull
+    public List<DeviceRequest> getDeviceRequests() {
+        return deviceRequests;
+    }
+
+    public HostConfig withDeviceRequests(List<DeviceRequest> deviceRequests) {
+        this.deviceRequests = deviceRequests;
+        return this;
+    }
+
     @CheckForNull
     public Long getNanoCPUs() {
         return nanoCPUs;
diff --git a/docker-java-core/src/main/java/com/github/dockerjava/core/RemoteApiVersion.java b/docker-java-core/src/main/java/com/github/dockerjava/core/RemoteApiVersion.java
@@ -89,6 +89,7 @@ public class RemoteApiVersion implements Serializable {
     public static final RemoteApiVersion VERSION_1_36 = RemoteApiVersion.create(1, 36);
     public static final RemoteApiVersion VERSION_1_37 = RemoteApiVersion.create(1, 37);
     public static final RemoteApiVersion VERSION_1_38 = RemoteApiVersion.create(1, 38);
+    public static final RemoteApiVersion VERSION_1_40 = RemoteApiVersion.create(1, 40);
 
 
     /**
