diff --git a/main.py b/main.py index 4343d05..12ddbb8 100644 --- a/main.py +++ b/main.py @@ -31,7 +31,7 @@ def main(): sasl_plain_username=conf['sasl_plain_username'], sasl_plain_password=conf['sasl_plain_password'], value_deserializer=json.loads) - consumer.subscribe((conf['topic_name'],)) + consumer.subscribe(conf['topic_name']) for msg in consumer: st = time.time() * 1000 key = msg.key diff --git a/setting.py b/setting.py index b43506d..e5712c0 100644 --- a/setting.py +++ b/setting.py @@ -13,8 +13,8 @@ class Config: 'sasl_plain_username': 'legu666666', 'sasl_plain_password': '5EINbjyI', 'bootstrap_servers': ["39.104.71.242:9093", "39.104.71.78:9093", "39.104.67.122:9093"], - 'topic_name': 'legu_geshouccs', - 'consumer_id': 'geshouccs_consumer' + 'topic_name': os.environ.get('topic_name').split(','), + 'consumer_id': os.environ.get('consumer_id') } local_mongo_uri = os.environ["local_mongo_uri"]