feat(toolkit-cleaner): customizable timeout for asset cleaning Lambda…

… functions (#291) * feat(toolkit-cleaner): customizable timeout for asset cleaning Lambda functions * chore: self mutation Signed-off-by: github-actions <[email protected]> --------- Signed-off-by: github-actions <[email protected]> Co-authored-by: github-actions <[email protected]>
jogold · Nov 14, 2024 · 19aa403 · 19aa403
1 parent 1b68b28
commit 19aa403
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/API.md b/API.md
diff --git a/src/toolkit-cleaner/index.ts b/src/toolkit-cleaner/index.ts
@@ -44,6 +44,13 @@ export interface ToolkitCleanerProps {
    * @default - all unused assets are removed
    */
   readonly retainAssetsNewerThan?: Duration;
+
+  /**
+   * The timeout for the Lambda functions that clean assets
+   *
+   * @default Duration.minutes(5)
+   */
+  readonly cleanAssetsTimeout?: Duration;
 }
 
 /**
@@ -102,7 +109,7 @@ export class ToolkitCleaner extends Construct {
     });
 
     const cleanObjectsFunction = new CleanObjectsFunction(this, 'CleanObjectsFunction', {
-      timeout: Duration.minutes(5),
+      timeout: props.cleanAssetsTimeout ?? Duration.minutes(5),
     });
     cleanObjectsFunction.addEnvironment('BUCKET_NAME', fileAsset.bucket.bucketName);
     fileAsset.bucket.grantRead(cleanObjectsFunction);
@@ -113,7 +120,7 @@ export class ToolkitCleaner extends Construct {
     });
 
     const cleanImagesFunction = new CleanImagesFunction(this, 'CleanImagesFunction', {
-      timeout: Duration.minutes(5),
+      timeout: props.cleanAssetsTimeout ?? Duration.minutes(5),
     });
     cleanImagesFunction.addEnvironment('REPOSITORY_NAME', dockerImageAsset.repository.repositoryName);
     dockerImageAsset.repository.grant(cleanImagesFunction, 'ecr:DescribeImages', 'ecr:BatchDeleteImage');