diff --git a/examples/kafka/__init__.py b/examples/kafka/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/examples/kafka/kafka.py b/examples/kafka/kafka.py new file mode 100644 index 00000000..3ec8ca27 --- /dev/null +++ b/examples/kafka/kafka.py @@ -0,0 +1,25 @@ +import fire +from kafka import KafkaProducer +from kafka import KafkaConsumer + + +class KafkaFireExample: + def __init__(self): + self.producer = KafkaProducer(bootstrap_servers='localhost:9092') + + def kafka_producer(self, topic_name='Hello Topic'): + while True: + self.producer.send(topic_name, b'Hello, World!') + + def kafka_consumer(self, topic_name='Hello Topic'): + consumer = KafkaConsumer(topic_name) + for message in consumer: + return message + + +def main(): + fire.Fire(KafkaFireExample) + + +if __name__ == '__main__': + main() \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 9c558e35..1de34d82 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,2 @@ . +kafka-python==1.4.4 \ No newline at end of file