diff options
Diffstat (limited to 'kafka-send/kafka-send.py')
-rwxr-xr-x | kafka-send/kafka-send.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/kafka-send/kafka-send.py b/kafka-send/kafka-send.py new file mode 100755 index 0000000..b8ef414 --- /dev/null +++ b/kafka-send/kafka-send.py @@ -0,0 +1,16 @@ +#!/usr/bin/env python3 +import argparse +import kafka + +parser = argparse.ArgumentParser() +parser.add_argument('-s', '--server', default='localhost:9092', help='kafka bootstrap server (default localhost:9092)') +parser.add_argument('topic', help='kafka topic') +parser.add_argument('filename', help='file containing the message to be sent') +args = parser.parse_args() + +with open(args.filename, 'r') as file: + msg = ''.join(file.readlines()).encode() + +producer = kafka.KafkaProducer(bootstrap_servers=args.server) +producer.send(args.topic, msg) +producer.flush() |