Titus: set metatron sigs in both attributes and job labels

spinnaker · Mar 26, 2018 · d8c906c · d8c906c
1 parent 0dbb113
commit d8c906c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/groovy/com/netflix/spinnaker/clouddriver/titus/client/model/JobDescription.java b/src/main/groovy/com/netflix/spinnaker/clouddriver/titus/client/model/JobDescription.java
@@ -60,6 +60,7 @@ public class JobDescription {
   private String capacityGroup;
   private Efs efs;
   private MigrationPolicy migrationPolicy;
+  private Map<String, String> securityAttributes;
 
   //Soft/Hard constraints
 
@@ -103,6 +104,7 @@ public class JobDescription {
     gpu = request.getGpu();
     retries = request.getRetries();
     runtimeLimitSecs = request.getRuntimeLimitSecs();
+    securityAttributes = new HashMap<String, String>();
   }
 
   public String getName() {
@@ -364,6 +366,11 @@ public void setSecurityGroups(List<String> securityGroups) {
     this.securityGroups = securityGroups;
   }
 
+  @JsonIgnore
+  public Map<String, String> getSecurityAttributes() {
+    return securityAttributes;
+  }
+
   @JsonIgnore
   public JobDescriptor getGrpcJobDescriptor() {
 
@@ -428,6 +435,10 @@ public JobDescriptor getGrpcJobDescriptor() {
       securityProfile.setIamRole(iamProfile);
     }
 
+    if (!securityAttributes.isEmpty()){
+      securityProfile.putAllAttributes(securityAttributes);
+    }
+
     containerBuilder.setSecurityProfile(securityProfile);
     containerBuilder.setImage(Image.newBuilder().setName(applicationName).setTag(version));