KafkaPush.py 387 B

123456789101112131415
  1. import json
  2. import sys
  3. from kafka import KafkaProducer
  4. producer = KafkaProducer(value_serializer=lambda v: json.dumps(v).encode('utf-8'),
  5. bootstrap_servers=sys.argv[1].split(","))
  6. msg = sys.argv[3]
  7. if msg.startswith("{"):
  8. msg = json.loads(msg)
  9. else:
  10. msg = {'message': msg}
  11. print(f"msg: {msg}")
  12. producer.send(sys.argv[2], value=msg, partition=0).get()