add soft and hard container constraints data model to titus batch job…

… request
Netflix · Aug 23, 2022 · 44258eb · 44258eb
1 parent abe4f23
commit 44258eb
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/genie-web/src/main/java/com/netflix/genie/web/agent/launchers/dtos/TitusBatchJobRequest.java b/genie-web/src/main/java/com/netflix/genie/web/agent/launchers/dtos/TitusBatchJobRequest.java
@@ -20,7 +20,9 @@
 import lombok.Builder;
 import lombok.Data;
 import lombok.NonNull;
+import org.apache.logging.log4j.util.Strings;
 
+import javax.annotation.Nullable;
 import javax.validation.constraints.Min;
 import javax.validation.constraints.NotEmpty;
 import javax.validation.constraints.NotNull;
@@ -108,6 +110,10 @@ public static class Container {
         @NotNull
         @NonNull
         private Map<String, String> attributes;
+        @Nullable
+        private ContainerConstraints softConstraints;
+        @Nullable
+        private ContainerConstraints hardConstraints;
     }
 
     /**
@@ -140,6 +146,22 @@ public static class SecurityProfile {
         private String iamRole;
     }
 
+    /**
+     * Titus job container constraints.
+     */
+    @Data
+    @Builder
+    public static class ContainerConstraints {
+        @NotNull
+        @NonNull
+        private Map<String, String> constraints;
+
+        @NotNull
+        @NonNull
+        @Builder.Default()
+        private String expression = Strings.EMPTY;
+    }
+
     /**
      * Titus job container image.
      */