refactor: 조건부 로직 적용

techeer-sv · Feb 11, 2024 · 38b77d1 · 38b77d1
1 parent eaaa7d4
commit 38b77d1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/server/types/AwsEvent.d.ts b/server/types/AwsEvent.d.ts
@@ -0,0 +1,5 @@
+import { AwsEvent, AwsCallback, AwsResponse } from "@slack/bolt/dist/receivers/AwsLambdaReceiver";
+
+interface CustomAwsEvent extends AwsEvent {
+    source?: string;
+}
diff --git a/server/utils/function/lambdaHandler/handler.ts b/server/utils/function/lambdaHandler/handler.ts
@@ -4,14 +4,20 @@ import { submitNotionCommandHandler } from "../../../notion/blog/handler/submitN
 import { uploadNotionBlogViewHandler } from "../../../notion/blog/handler/uploadNotionBlogViewHandler";
 import { app } from "../../appModule/slack/slack";
 import { notion } from "../../appModule/notion/notion";
+import { CustomAwsEvent } from "../../../types/AwsEvent";
 
 let handlerRegistered = false;
 
 export const handler = async(
-    event: AwsEvent,
+    event: CustomAwsEvent,
     context: any,
     callback: AwsCallback
 ): Promise<AwsResponse> => {
+    if(event.source === 'aws.events'){
+        console.log('Warm Start 설정');
+        return {statusCode:200, body:'Warm Start 완료'};
+    }
+
     if(!handlerRegistered){
         submitNotionCommandHandler(app);
         uploadNotionBlogViewHandler(app, notion);