X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=near-rt-ric-simulator%2Ftest%2Fcommon%2Fpublish_response_event_to_kafka_bus.py;fp=near-rt-ric-simulator%2Ftest%2Fcommon%2Fpublish_response_event_to_kafka_bus.py;h=251ba0937c1119a6b3e61e97a20238673e803a2b;hb=514637b303ec7dc6390007be1a046189f7a9d169;hp=0000000000000000000000000000000000000000;hpb=f6a48104c0a89a91dc36bb175c8ca301bfc9950d;p=sim%2Fa1-interface.git

diff --git a/near-rt-ric-simulator/test/common/publish_response_event_to_kafka_bus.py b/near-rt-ric-simulator/test/common/publish_response_event_to_kafka_bus.py
new file mode 100644
index 0000000..251ba09
--- /dev/null
+++ b/near-rt-ric-simulator/test/common/publish_response_event_to_kafka_bus.py
@@ -0,0 +1,88 @@
+#  ============LICENSE_START===============================================
+#  Copyright (C) 2022 Nordix Foundation. All rights reserved.
+#  ========================================================================
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+#  ============LICENSE_END=================================================
+#
+
+# This script publishes a response-event to a kafka bus
+# In order to use this script, you must have an venv for Python and kafka-python libs has to be installed
+# To instal kafka-python please use: pip install kafka-python
+# Example of an response-event json
+#{
+  #"response-code": "400",
+  #"error-info": "Bad format"
+#}
+
+
+import os
+import json
+import sys
+
+from kafka import KafkaProducer
+
+# Response string with JSON format
+response_data_JSON =  """
+{
+  "response-code": 200,
+  "error-info": ""
+}
+"""
+
+# Instantiate KafkaProducer with keyword arguments
+# https://kafka-python.readthedocs.io/en/master/apidoc/KafkaProducer.html
+def create_kafka_producer():
+
+  producer = KafkaProducer(
+    bootstrap_servers = ['localhost:9092'],
+    key_serializer = str.encode,
+    value_serializer = lambda m: json.dumps(m).encode('ascii'),
+  )
+  return producer
+
+# Helper: Publishes (to) the target broker and the topic in synch
+def publish(kafka_evet, req_id, targettopic):
+
+  # Instantiate KafkaProducer with keyword arguments
+  producer = create_kafka_producer()
+  # Assigns an id to each request that is supposed to get a result
+  # req_id  = 'Hll1EsycKLNRric7'
+
+  try:
+
+    # synch-publish
+    # KafkaProducer.send(topicname, value=broker_message, key=req_id, headers=None, partition=None, timestamp_ms=None)
+    fut_rec_metadata = producer.send(targettopic, kafka_evet, req_id)
+    return fut_rec_metadata.get()
+
+  except Exception as err:
+    print('Error while publish', err)
+  finally:
+    producer.close()
+
+if __name__ == '__main__':
+    try:
+
+        requestid = sys.argv[1]
+        targettopic = sys.argv[2]
+        # response_data_JSON is str
+        future = publish(response_data_JSON, requestid, targettopic)
+
+        if (future is not None):
+            print (0)
+        else:
+            print (1)
+
+    except Exception:
+        print (1)
+    sys.exit()