Jelajahi Sumber

Update agent_server: thx to some stupid naming, never use aliyun mq for sending if env != prod

StrayWarrior 3 minggu lalu
induk
melakukan
ee9adf6e49
1 mengubah file dengan 10 tambahan dan 7 penghapusan
  1. 10 7
      pqai_agent_server/agent_server.py

+ 10 - 7
pqai_agent_server/agent_server.py

@@ -32,13 +32,16 @@ if __name__ == "__main__":
             group_id=config['mq']['receive_group'],
             topic_type='FIFO'
         )
-        send_queue = AliyunRocketMQQueueBackend(
-            config['mq']['endpoints'],
-            config['mq']['instance_id'],
-            config['mq']['send_topic'],
-            has_consumer=False, has_producer=True,
-            topic_type='FIFO'
-        )
+        if configs.get_env() == 'prod':
+            send_queue = AliyunRocketMQQueueBackend(
+                config['mq']['endpoints'],
+                config['mq']['instance_id'],
+                config['mq']['send_topic'],
+                has_consumer=False, has_producer=True,
+                topic_type='FIFO'
+            )
+        else:
+            send_queue = MemoryQueueBackend()
     else:
         receive_queue = MemoryQueueBackend()
         send_queue = MemoryQueueBackend()