feat(codepipeline): auto rollback for pipeline

packages/aws-cdk-lib/aws-codepipeline/lib/pipeline.ts

@@ -60,6 +60,29 @@ export interface StagePlacement {
   readonly justAfter?: IStage;
 }
 
+/**
+ * Result type for failure conditions.
+ */
+export enum FailureConditionsResult {
+  /**
+   * ROLLBACK
+   */
+  ROLLBACK = 'ROLLBACK',
+}
+
+/**
+ * Failure conditions for Stage props.
+ */
+export interface FailureConditions {
+  /**
+   * The specified result for when the failure conditions are met,
+   * such as rolling back the stage.
+   *
+   * @default ROLLBACK
+   */
+  readonly result?: FailureConditionsResult;
+}
+
 /**
  * Construction properties of a Pipeline Stage.
  */
@@ -89,6 +112,16 @@ export interface StageProps {
    * @default 'Transition disabled'
    */
   readonly transitionDisabledReason?: string;
+
+  /**
+   * The method to use when a stage has not completed successfully.
+   *
+   * For example, configuring this field for rollback will roll back a failed
+   * stage automatically to the last successful pipeline execution in the stage.
+   *
+   * @default - No failure conditions.
+   */
+  readonly onFailure?: FailureConditions;
 }
 
 export interface StageOptions extends StageProps {

packages/aws-cdk-lib/aws-codepipeline/lib/private/stage.ts

@@ -7,7 +7,7 @@ import { Token } from '../../../core';
 import { IAction, IPipeline, IStage } from '../action';
 import { Artifact } from '../artifact';
 import { CfnPipeline } from '../codepipeline.generated';
-import { Pipeline, StageProps } from '../pipeline';
+import { FailureConditions, FailureConditionsResult, Pipeline, StageProps } from '../pipeline';
 
 /**
  * A Stage in a Pipeline.
@@ -27,6 +27,7 @@ export class Stage implements IStage {
   private readonly scope: Construct;
   private readonly _pipeline: Pipeline;
   private readonly _actions = new Array<FullActionDescriptor>();
+  private readonly _onFailure?: FailureConditions;
 
   /**
    * Create a new Stage.
@@ -39,6 +40,7 @@ export class Stage implements IStage {
     this.transitionDisabledReason = props.transitionDisabledReason ?? 'Transition disabled';
     this._pipeline = pipeline;
     this.scope = new Construct(pipeline, this.stageName);
+    this._onFailure = props.onFailure;
 
     for (const action of props.actions || []) {
       this.addAction(action);
@@ -81,6 +83,9 @@ export class Stage implements IStage {
     return {
       name: this.stageName,
       actions: this._actions.map(action => this.renderAction(action)),
+      onFailure: this._onFailure ? {
+        result: this._onFailure?.result ?? FailureConditionsResult.ROLLBACK,
+      } : undefined,
     };
   }