-
Notifications
You must be signed in to change notification settings - Fork 3.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(scheduler-targets): Add step function start execution target
- Loading branch information
Showing
11 changed files
with
35,016 additions
and
0 deletions.
There are no files selected for viewing
35 changes: 35 additions & 0 deletions
35
packages/@aws-cdk/aws-scheduler-targets-alpha/lib/step-functions-start-execution.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,35 @@ | ||
import { IScheduleTarget, ISchedule } from '@aws-cdk/aws-scheduler-alpha'; | ||
import { Names } from 'aws-cdk-lib'; | ||
import { IRole } from 'aws-cdk-lib/aws-iam'; | ||
import { IStateMachine } from 'aws-cdk-lib/aws-stepfunctions'; | ||
import { ScheduleTargetBase, ScheduleTargetBaseProps } from './target'; | ||
import { sameEnvDimension } from './util'; | ||
|
||
/** | ||
* Use an AWS Step function as a target for AWS EventBridge Scheduler. | ||
*/ | ||
export class StepFunctionsStartExecution extends ScheduleTargetBase implements IScheduleTarget { | ||
constructor( | ||
private readonly stateMachine: IStateMachine, | ||
private readonly props: ScheduleTargetBaseProps, | ||
) { | ||
super(props, stateMachine.stateMachineArn); | ||
} | ||
|
||
protected addTargetActionToRole(schedule: ISchedule, role: IRole): void { | ||
const stateMachineEnv = this.stateMachine.env; | ||
if (!sameEnvDimension(stateMachineEnv.region, schedule.env.region)) { | ||
throw new Error(`Cannot assign stateMachine in region ${stateMachineEnv.region} to the schedule ${Names.nodeUniqueId(schedule.node)} in region ${schedule.env.region}. Both the schedule and the stateMachine must be in the same region.`); | ||
} | ||
|
||
if (!sameEnvDimension(stateMachineEnv.account, schedule.env.account)) { | ||
throw new Error(`Cannot assign stateMachine in account ${stateMachineEnv.account} to the schedule ${Names.nodeUniqueId(schedule.node)} in account ${schedule.env.region}. Both the schedule and the stateMachine must be in the same account.`); | ||
} | ||
|
||
if (this.props.role && !sameEnvDimension(this.props.role.env.account, stateMachineEnv.account)) { | ||
throw new Error(`Cannot grant permission to execution role in account ${this.props.role.env.account} to invoke target ${Names.nodeUniqueId(this.stateMachine.node)} in account ${stateMachineEnv.account}. Both the target and the execution role must be in the same account.`); | ||
} | ||
|
||
this.stateMachine.grantStartExecution(role); | ||
} | ||
} |
Oops, something went wrong.