csit/resources/scripts/kafka_producer.py

   1 #!/usr/bin/env python3
   2 #
   3 # ============LICENSE_START====================================================
   4 #  Copyright (C) 2023 Nordix Foundation.
   5 # =============================================================================
   6 # Licensed under the Apache License, Version 2.0 (the "License");
   7 # you may not use this file except in compliance with the License.
   8 # You may obtain a copy of the License at
   9 #
  10 #      http://www.apache.org/licenses/LICENSE-2.0
  11 #
  12 # Unless required by applicable law or agreed to in writing, software
  13 # distributed under the License is distributed on an "AS IS" BASIS,
  14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15 # See the License for the specific language governing permissions and
  16 # limitations under the License.
  17 #
  18 # SPDX-License-Identifier: Apache-2.0
  19 # ============LICENSE_END======================================================
  20
  21 # Python utility to produce a message on a kafka topic
  22 # Accepts the arguments {topic_name} and {message}
  23
  24 from confluent_kafka import Producer
  25 import sys
  26
  27 def post_to_kafka(topic, message):
  28     conf = {'bootstrap.servers': 'localhost:29092'}
  29
  30     producer = Producer(conf)
  31     try:
  32         producer.produce(topic, value=message.encode('utf-8'))
  33         producer.flush()
  34         print('Message posted to Kafka topic: {}'.format(topic))
  35     except Exception as e:
  36         print('Failed to post message: {}'.format(str(e)))
  37     finally:
  38         producer.flush()
  39
  40 if __name__ == '__main__':
  41     post_to_kafka(sys.argv[1], sys.argv[2])