Titus: record migration policy set in description

spinnaker · Mar 26, 2018 · 752e92d · 752e92d
1 parent 1c96361
commit 752e92d
Showing 1 changed file with 9 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
@@ -467,9 +467,18 @@ public JobDescriptor getGrpcJobDescriptor() {
         inService = true;
       }
 
+      com.netflix.titus.grpc.protogen.MigrationPolicy serviceMigrationPolicy;
+
+      if (migrationPolicy != null && migrationPolicy.getType().equals("userDefined")) {
+        serviceMigrationPolicy = com.netflix.titus.grpc.protogen.MigrationPolicy.newBuilder().setSelfManaged(com.netflix.titus.grpc.protogen.MigrationPolicy.SelfManaged.newBuilder().build()).build();
+      } else {
+        serviceMigrationPolicy = com.netflix.titus.grpc.protogen.MigrationPolicy.newBuilder().setSystemDefault(com.netflix.titus.grpc.protogen.MigrationPolicy.SystemDefault.newBuilder().build()).build();
+      }
+
       jobDescriptorBuilder.setService(
         ServiceJobSpec.newBuilder().setEnabled(inService)
           .setCapacity(jobCapacity)
+          .setMigrationPolicy(serviceMigrationPolicy)
           .setRetryPolicy(RetryPolicy.newBuilder().setExponentialBackOff(RetryPolicy.ExponentialBackOff.newBuilder().setInitialDelayMs(5000).setMaxDelayIntervalMs(300000))));
     }