瀏覽代碼

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

StrayWarrior 3 周之前
父節點
當前提交
ee9adf6e49
共有 1 個文件被更改,包括 10 次插入7 次删除
  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'],
             group_id=config['mq']['receive_group'],
             topic_type='FIFO'
             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:
     else:
         receive_queue = MemoryQueueBackend()
         receive_queue = MemoryQueueBackend()
         send_queue = MemoryQueueBackend()
         send_queue = MemoryQueueBackend()