Merge pull request #226 from gammarers/feature/add-state-machine-timeout

feat: add state machine timeout

Author: yicr

.projen/deps.json

@@ -4,8 +4,8 @@ const project = new awscdk.AwsCdkConstructLibrary({
   authorAddress: '[email protected]',
   authorOrganization: true,
   cdkVersion: '2.120.0',
-  typescriptVersion: '5.7.x',
-  jsiiVersion: '5.7.x',
+  typescriptVersion: '5.8.x',
+  jsiiVersion: '5.8.x',
   defaultReleaseBranch: 'main',
   name: '@gammarers/aws-ec2-instance-running-schedule-stack',
   description: 'AWS EC2 Instance Running Scheduler',

.projenrc.ts

@@ -1,4 +1,4 @@
-import { Stack } from 'aws-cdk-lib';
+import { Duration, Stack } from 'aws-cdk-lib';
 import * as iam from 'aws-cdk-lib/aws-iam';
 import * as logs from 'aws-cdk-lib/aws-logs';
 import * as scheduler from 'aws-cdk-lib/aws-scheduler';
@@ -32,12 +32,17 @@ export interface LogOption {
   readonly machineLogLevel?: EC2InstanceRunningScheduleStackMachineLogLevel;
 }
 
+export interface Timeout {
+  readonly stateMachineTimeout?: Duration;
+}
+
 export interface EC2InstanceRunningScheduleStackProps {
   readonly targetResource: TargetResource;
   readonly stopSchedule?: Schedule;
   readonly startSchedule?: Schedule;
   readonly notifications?: Notifications;
   readonly logOption?: LogOption;
+  readonly timeout?: Timeout;
 }
 
 export class EC2InstanceRunningScheduleStack extends Stack {
@@ -62,6 +67,12 @@ export class EC2InstanceRunningScheduleStack extends Stack {
     const machine = new RunningControlStateMachine(this, 'StateMachine', {
       stateMachineName: undefined,
       notificationTopic: topic,
+      timeout: (() => {
+        if (props.timeout?.stateMachineTimeout) {
+          return props.timeout?.stateMachineTimeout;
+        }
+        return Duration.hours(1);
+      })(),
       logs: (() => {
         if (props.logOption?.machineLogLevel) {
           return {

API.md

@@ -1,4 +1,4 @@
-import { App } from 'aws-cdk-lib';
+import { App, Duration } from 'aws-cdk-lib';
 import { Template } from 'aws-cdk-lib/assertions';
 import { EC2InstanceRunningScheduleStack, EC2InstanceRunningScheduleStackMachineLogLevel } from '../src';
 
@@ -29,6 +29,9 @@ describe('Ec2InstanceRunningScheduleStack specific Testing', () => {
         '[email protected]',
       ],
     },
+    timeout: {
+      stateMachineTimeout: Duration.minutes(30),
+    },
     logOption: {
       machineLogLevel: EC2InstanceRunningScheduleStackMachineLogLevel.ALL,
     },