summaryrefslogtreecommitdiffstats
path: root/python/fatcat_tools/kafka.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/fatcat_tools/kafka.py')
-rw-r--r--python/fatcat_tools/kafka.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/python/fatcat_tools/kafka.py b/python/fatcat_tools/kafka.py
index 32749db2..2a4451ad 100644
--- a/python/fatcat_tools/kafka.py
+++ b/python/fatcat_tools/kafka.py
@@ -1,4 +1,3 @@
-
from confluent_kafka import KafkaException, Producer
@@ -9,14 +8,15 @@ def kafka_fail_fast(err, msg):
# TODO: should it be sys.exit(-1)?
raise KafkaException(err)
+
def simple_kafka_producer(kafka_hosts):
kafka_config = {
- 'bootstrap.servers': kafka_hosts,
- 'message.max.bytes': 20000000, # ~20 MBytes; broker-side max is ~50 MBytes
- 'delivery.report.only.error': True,
- 'default.topic.config': {
- 'request.required.acks': -1,
+ "bootstrap.servers": kafka_hosts,
+ "message.max.bytes": 20000000, # ~20 MBytes; broker-side max is ~50 MBytes
+ "delivery.report.only.error": True,
+ "default.topic.config": {
+ "request.required.acks": -1,
},
}
return Producer(kafka_config)