🔊 logging 모듈을 이용해서 live_check 로깅

lambdas/live_check/main.py

@@ -12,6 +12,10 @@
     from layers.common.python.common import *
 
 import boto3
+import logging
+
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
 
 dynamodb = boto3.client('dynamodb')
 
@@ -22,6 +26,16 @@ def middleware(event, context):
     트리거마다 처리하는 데이터가 다르게 분산화합니다.
     """
 
+    logger.info(
+        json.dumps(
+            {
+                "type": "START",
+                "index": index
+            },
+            ensure_ascii=False
+        )
+    )
+
     result = {
         "total_channel_count": 0,
         "live_channel_count": 0,
@@ -41,13 +55,50 @@ def middleware(event, context):
         }
     )
 
+    logger.info(
+        json.dumps(
+            {
+                "type": "CHANNEL_QUERY_END",
+                "total_channel_count": res["Count"],
+                "index": index
+            },
+            ensure_ascii=False
+        )
+    )
+
     result["total_channel_count"] = res["Count"]
 
     for item in res["Items"]:
         channel_id = item["PK"]["S"].split("#")[1]
         last_live_id = item["lastLiveId"]["N"]
 
-        chzzk = get_chzzk(channel_id)
+        try:
+            start_time = datetime.now()
+            chzzk = get_chzzk(channel_id)
+            end_time = datetime.now()
+            logger.info(
+                json.dumps(
+                    {
+                        "type": "GET_CHZZK_END",
+                        "channel_id": channel_id,
+                        "index": index,
+                        "time": (end_time - start_time).total_seconds(),
+                    },
+                    ensure_ascii=False
+                )
+            )
+        except Exception as e:
+            logger.error(
+                json.dumps(
+                    {
+                        "type": "GET_CHZZK_ERROR",
+                        "channel_id": channel_id,
+                        "index": index,
+                        "error": str(e)
+                    }
+                )
+            )
+            continue
 
         if not chzzk:
             continue
@@ -58,6 +109,18 @@ def middleware(event, context):
         if chzzk['status'] != "OPEN":
             continue
 
+        logger.info(
+            json.dumps(
+                {
+                    "type": "LIVE_START",
+                    "chzzk_id": channel_id,
+                    "index": index,
+                    "liveId": chzzk['liveId'],
+                    "liveTitle": chzzk['liveTitle']
+                }
+            )
+        )
+
         result["live_channel_count"] += 1
 
         dynamodb.update_item(
@@ -73,9 +136,6 @@ def middleware(event, context):
             }
         )
 
-        print(
-            f"LIVE_START {chzzk['channel']['channelName']}, {chzzk['liveId']}, {chzzk['liveTitle']}, {index=}")
-
         res = dynamodb.query(
             TableName='chzzk-bot-db',
             KeyConditionExpression='PK = :pk_val AND begins_with(SK, :sk_val)',
@@ -85,6 +145,18 @@ def middleware(event, context):
             }
         )
 
+        logger.info(
+            json.dumps(
+                {
+                    "type": "NOTI_QUERY_END",
+                    "chzzk_id": channel_id,
+                    "index": index,
+                    "noti_count": res["Count"]
+                }
+            ),
+            ensure_ascii=False
+        )
+
         for noti in res["Items"]:
             discord_channel_id = noti["channel_id"]["S"]
 
@@ -140,7 +212,17 @@ def middleware(event, context):
 
             # 채널이 존재하지 않는 경우
             if res.status_code == 404:
-                print("channel not found")
+                logging.error(
+                    json.dumps(
+                        {
+                            "type": "DISCORD_CHANNEL_NOT_FOUND",
+                            "chzzk_id": channel_id,
+                            "index": index,
+                            "discord_channel_id": discord_channel_id
+                        },
+                        ensure_ascii=False
+                    )
+                )
                 dynamodb.delete_item(
                     TableName='chzzk-bot-db',
                     Key={
@@ -153,7 +235,20 @@ def middleware(event, context):
 
             # 메시지 전송에 실패한 경우
             if res.status_code != 200:
-                print("send message fail", res.status_code)
+
+                logging.error(
+                    json.dumps(
+                        {
+                            "type": "DISCORD_MESSAGE_SEND_FAIL",
+                            "chzzk_id": channel_id,
+                            "index": index,
+                            "discord_channel_id": discord_channel_id,
+                            "status_code": res.status_code,
+                            "response": res.text
+                        },
+                        ensure_ascii=False
+                    ),
+                )
                 dynamodb.update_item(
                     TableName='chzzk-bot-db',
                     Key={
@@ -167,18 +262,21 @@ def middleware(event, context):
                         ':noti_fail_count': {'N': str(int(noti.get("noti_fail_count", {"N": "0"})["N"]) + 1)}
                     }
                 )
-                try:
-                    print(res.json())
-                except:
-                    try:
-                        print(res.text)
-                    except:
-                        print("unknown error")
                 result["fail_send_count"] += 1
                 continue
 
             # 성공
-            print("send message success")
+            logging.info(
+                json.dumps(
+                    {
+                        "type": "DISCORD_MESSAGE_SEND_SUCCESS",
+                        "chzzk_id": channel_id,
+                        "index": index,
+                        "discord_channel_id": discord_channel_id
+                    },
+                    ensure_ascii=False
+                )
+            )
             result["total_send_count"] += 1
             dynamodb.update_item(
                 TableName='chzzk-bot-db',
@@ -200,7 +298,12 @@ def middleware(event, context):
 def lambda_handler(event, context):
     res = middleware(event, context)
 
-    print("=== res ===")
-    print(json.dumps(res))
-
-    return res
+    logger.info(
+        json.dumps(
+            {
+                "type": "FINAL_RESULT",
+                "result": res
+            },
+            ensure_ascii=False
+        )
+    )