Amazon MWAA now lets you choose a worker replacement strategy when updating an environment. This release adds two worker replacement strategies: FORCED (default), which stops workers immediately, and GRACEFUL, which allows workers to finish current tasks before shutting down.

Author: aws-sdk-dotnet-automation

generator/ServiceModels/mwaa/mwaa-2020-07-01.api.json

@@ -568,7 +568,8 @@
         "Status":{"shape":"UpdateStatus"},
         "CreatedAt":{"shape":"UpdateCreatedAt"},
         "Error":{"shape":"UpdateError"},
-        "Source":{"shape":"UpdateSource"}
+        "Source":{"shape":"UpdateSource"},
+        "WorkerReplacementStrategy":{"shape":"WorkerReplacementStrategy"}
       }
     },
     "ListEnvironmentsInput":{
@@ -988,26 +989,27 @@
           "locationName":"Name"
         },
         "ExecutionRoleArn":{"shape":"IamRoleArn"},
+        "AirflowConfigurationOptions":{"shape":"AirflowConfigurationOptions"},
         "AirflowVersion":{"shape":"AirflowVersion"},
-        "SourceBucketArn":{"shape":"S3BucketArn"},
         "DagS3Path":{"shape":"RelativePath"},
+        "EnvironmentClass":{"shape":"EnvironmentClass"},
+        "LoggingConfiguration":{"shape":"LoggingConfigurationInput"},
+        "MaxWorkers":{"shape":"MaxWorkers"},
+        "MinWorkers":{"shape":"MinWorkers"},
+        "MaxWebservers":{"shape":"MaxWebservers"},
+        "MinWebservers":{"shape":"MinWebservers"},
+        "WorkerReplacementStrategy":{"shape":"WorkerReplacementStrategy"},
+        "NetworkConfiguration":{"shape":"UpdateNetworkConfigurationInput"},
         "PluginsS3Path":{"shape":"RelativePath"},
         "PluginsS3ObjectVersion":{"shape":"S3ObjectVersion"},
         "RequirementsS3Path":{"shape":"RelativePath"},
         "RequirementsS3ObjectVersion":{"shape":"S3ObjectVersion"},
+        "Schedulers":{"shape":"Schedulers"},
+        "SourceBucketArn":{"shape":"S3BucketArn"},
         "StartupScriptS3Path":{"shape":"RelativePath"},
         "StartupScriptS3ObjectVersion":{"shape":"S3ObjectVersion"},
-        "AirflowConfigurationOptions":{"shape":"AirflowConfigurationOptions"},
-        "EnvironmentClass":{"shape":"EnvironmentClass"},
-        "MaxWorkers":{"shape":"MaxWorkers"},
-        "NetworkConfiguration":{"shape":"UpdateNetworkConfigurationInput"},
-        "LoggingConfiguration":{"shape":"LoggingConfigurationInput"},
-        "WeeklyMaintenanceWindowStart":{"shape":"WeeklyMaintenanceWindowStart"},
         "WebserverAccessMode":{"shape":"WebserverAccessMode"},
-        "MinWorkers":{"shape":"MinWorkers"},
-        "Schedulers":{"shape":"Schedulers"},
-        "MinWebservers":{"shape":"MinWebservers"},
-        "MaxWebservers":{"shape":"MaxWebservers"}
+        "WeeklyMaintenanceWindowStart":{"shape":"WeeklyMaintenanceWindowStart"}
       }
     },
     "UpdateEnvironmentOutput":{
@@ -1079,6 +1081,13 @@
       "max":9,
       "min":1,
       "pattern":".*(MON|TUE|WED|THU|FRI|SAT|SUN):([01]\\d|2[0-3]):(00|30).*"
+    },
+    "WorkerReplacementStrategy":{
+      "type":"string",
+      "enum":[
+        "FORCED",
+        "GRACEFUL"
+      ]
     }
   }
 }

generator/ServiceModels/mwaa/mwaa-2020-07-01.docs.json

@@ -692,6 +692,13 @@
         "Environment$WeeklyMaintenanceWindowStart": "<p>The day and time of the week in Coordinated Universal Time (UTC) 24-hour standard time that weekly maintenance updates are scheduled. For example: <code>TUE:03:30</code>.</p>",
         "UpdateEnvironmentInput$WeeklyMaintenanceWindowStart": "<p>The day and time of the week in Coordinated Universal Time (UTC) 24-hour standard time to start weekly maintenance updates of your environment in the following format: <code>DAY:HH:MM</code>. For example: <code>TUE:03:30</code>. You can specify a start time in 30 minute increments only.</p>"
       }
+    },
+    "WorkerReplacementStrategy": {
+      "base": null,
+      "refs": {
+        "LastUpdate$WorkerReplacementStrategy": "<p>The worker replacement strategy used in the last update of the environment.</p>",
+        "UpdateEnvironmentInput$WorkerReplacementStrategy": "<p>The worker replacement strategy to use when updating the environment.</p> <p>You can select one of the following strategies:</p> <ul> <li> <p> <b>Forced -</b> Stops and replaces Apache Airflow workers without waiting for tasks to complete before an update.</p> </li> <li> <p> <b>Graceful -</b> Allows Apache Airflow workers to complete running tasks for up to 12 hours during an update before they're stopped and replaced.</p> </li> </ul>"
+      }
     }
   }
 }

generator/ServiceModels/mwaa/mwaa-2020-07-01.normal.json

@@ -90,6 +90,12 @@ public LastUpdate Unmarshall(JsonUnmarshallerContext context)
                     unmarshalledObject.Status = unmarshaller.Unmarshall(context);
                     continue;
                 }
+                if (context.TestExpression("WorkerReplacementStrategy", targetDepth))
+                {
+                    var unmarshaller = StringUnmarshaller.Instance;
+                    unmarshalledObject.WorkerReplacementStrategy = unmarshaller.Unmarshall(context);
+                    continue;
+                }
             }
             return unmarshalledObject;
         }

sdk/src/Services/MWAA/Generated/Model/Internal/MarshallTransformations/LastUpdateUnmarshaller.cs

@@ -214,6 +214,12 @@ public IRequest Marshall(UpdateEnvironmentRequest publicRequest)
                     context.Writer.Write(publicRequest.WeeklyMaintenanceWindowStart);
                 }
 
+                if(publicRequest.IsSetWorkerReplacementStrategy())
+                {
+                    context.Writer.WritePropertyName("WorkerReplacementStrategy");
+                    context.Writer.Write(publicRequest.WorkerReplacementStrategy);
+                }
+
                 writer.WriteObjectEnd();
                 string snippet = stringWriter.ToString();
                 request.Content = System.Text.Encoding.UTF8.GetBytes(snippet);

sdk/src/Services/MWAA/Generated/Model/Internal/MarshallTransformations/UpdateEnvironmentRequestMarshaller.cs

@@ -39,6 +39,7 @@ public partial class LastUpdate
         private UpdateError _error;
         private string _source;
         private UpdateStatus _status;
+        private WorkerReplacementStrategy _workerReplacementStrategy;
 
         /// <summary>
         /// Gets and sets the property CreatedAt. 
@@ -114,5 +115,23 @@ internal bool IsSetStatus()
             return this._status != null;
         }
 
+        /// <summary>
+        /// Gets and sets the property WorkerReplacementStrategy. 
+        /// <para>
+        /// The worker replacement strategy used in the last update of the environment.
+        /// </para>
+        /// </summary>
+        public WorkerReplacementStrategy WorkerReplacementStrategy
+        {
+            get { return this._workerReplacementStrategy; }
+            set { this._workerReplacementStrategy = value; }
+        }
+
+        // Check to see if WorkerReplacementStrategy property is set
+        internal bool IsSetWorkerReplacementStrategy()
+        {
+            return this._workerReplacementStrategy != null;
+        }
+
     }
 }

sdk/src/Services/MWAA/Generated/Model/LastUpdate.cs

@@ -57,6 +57,7 @@ public partial class UpdateEnvironmentRequest : AmazonMWAARequest
         private string _startupScriptS3Path;
         private WebserverAccessMode _webserverAccessMode;
         private string _weeklyMaintenanceWindowStart;
+        private WorkerReplacementStrategy _workerReplacementStrategy;
 
         /// <summary>
         /// Gets and sets the property AirflowConfigurationOptions. 
@@ -573,5 +574,38 @@ internal bool IsSetWeeklyMaintenanceWindowStart()
             return this._weeklyMaintenanceWindowStart != null;
         }
 
+        /// <summary>
+        /// Gets and sets the property WorkerReplacementStrategy. 
+        /// <para>
+        /// The worker replacement strategy to use when updating the environment.
+        /// </para>
+        ///  
+        /// <para>
+        /// You can select one of the following strategies:
+        /// </para>
+        ///  <ul> <li> 
+        /// <para>
+        ///  <b>Forced -</b> Stops and replaces Apache Airflow workers without waiting for tasks
+        /// to complete before an update.
+        /// </para>
+        ///  </li> <li> 
+        /// <para>
+        ///  <b>Graceful -</b> Allows Apache Airflow workers to complete running tasks for up
+        /// to 12 hours during an update before they're stopped and replaced.
+        /// </para>
+        ///  </li> </ul>
+        /// </summary>
+        public WorkerReplacementStrategy WorkerReplacementStrategy
+        {
+            get { return this._workerReplacementStrategy; }
+            set { this._workerReplacementStrategy = value; }
+        }
+
+        // Check to see if WorkerReplacementStrategy property is set
+        internal bool IsSetWorkerReplacementStrategy()
+        {
+            return this._workerReplacementStrategy != null;
+        }
+
     }
 }

sdk/src/Services/MWAA/Generated/Model/UpdateEnvironmentRequest.cs

@@ -541,4 +541,54 @@ public static implicit operator WebserverAccessMode(string value)
         }
     }
 
+
+    /// <summary>
+    /// Constants used for properties of type WorkerReplacementStrategy.
+    /// </summary>
+    public class WorkerReplacementStrategy : ConstantClass
+    {
+
+        /// <summary>
+        /// Constant FORCED for WorkerReplacementStrategy
+        /// </summary>
+        public static readonly WorkerReplacementStrategy FORCED = new WorkerReplacementStrategy("FORCED");
+        /// <summary>
+        /// Constant GRACEFUL for WorkerReplacementStrategy
+        /// </summary>
+        public static readonly WorkerReplacementStrategy GRACEFUL = new WorkerReplacementStrategy("GRACEFUL");
+
+        /// <summary>
+        /// This constant constructor does not need to be called if the constant
+        /// you are attempting to use is already defined as a static instance of 
+        /// this class.
+        /// This constructor should be used to construct constants that are not
+        /// defined as statics, for instance if attempting to use a feature that is
+        /// newer than the current version of the SDK.
+        /// </summary>
+        public WorkerReplacementStrategy(string value)
+            : base(value)
+        {
+        }
+
+        /// <summary>
+        /// Finds the constant for the unique value.
+        /// </summary>
+        /// <param name="value">The unique value for the constant</param>
+        /// <returns>The constant for the unique value</returns>
+        public static WorkerReplacementStrategy FindValue(string value)
+        {
+            return FindValue<WorkerReplacementStrategy>(value);
+        }
+
+        /// <summary>
+        /// Utility method to convert strings to the constant class.
+        /// </summary>
+        /// <param name="value">The string value to convert to the constant class.</param>
+        /// <returns></returns>
+        public static implicit operator WorkerReplacementStrategy(string value)
+        {
+            return FindValue(value);
+        }
+    }
+
 }