feat: rabbitmq get channel 제거 및 연결, 채널 응답하도록 변경

Author: ChaJunWoo-dev

src/config/constants.py

@@ -1,6 +1,9 @@
 class GCS:
     SIGNED_URL_EXPIRE = 60 * 24
 
+class RABBITMQ:
+    HEART_BEAT = 60
+
 class MESSAGES:
     class SUCCESS:
         ANALYZE = "영상 분석을 완료했습니다."

src/config/rabbitmq.py

@@ -2,31 +2,24 @@
 import pika
 import time
 
-from config.constants import MESSAGES
-
-_channel = None
-
-def connect(retry=0):
-    global _channel
+from config.constants import MESSAGES, RABBITMQ
 
+def create_connection(retry=0):
     if retry >= 5:
         raise RuntimeError(MESSAGES.ERROR.FAILED_CONNECT_CHANNEL)
 
     try:
         params = pika.URLParameters(environ["MQ_HOST"])
+        params.heartbeat = RABBITMQ.HEART_BEAT
+
         connection = pika.BlockingConnection(params)
-        _channel = connection.channel()
+        channel = connection.channel()
 
-        _channel.queue_declare(queue=environ["MQ_CONSUME_QUEUE"], durable=True)
-        _channel.queue_declare(queue=environ["MQ_PUBLISH_QUEUE"], durable=True)
+        channel.queue_declare(queue=environ["MQ_CONSUME_QUEUE"], durable=True)
+        channel.queue_declare(queue=environ["MQ_PUBLISH_QUEUE"], durable=True)
+
+        return connection, channel
     except:
         time.sleep(5)
-        connect(retry + 1)
-
-def get_channel():
-    global _channel
-
-    if _channel is None:
-        connect(retry=0)
 
-    return _channel
+        return create_connection(retry + 1)

src/rabbitmq/publish.py

@@ -1,13 +1,15 @@
 import json
 from os import environ
 
-from config.rabbitmq import get_channel
+from config.rabbitmq import create_connection
 
 def publish_message(message):
-    channel = get_channel()
+    connection, channel = create_connection()
 
     channel.basic_publish(
         exchange="",
         routing_key=environ["MQ_PUBLISH_QUEUE"],
         body=json.dumps(message)
     )
+
+    connection.close()