X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Fusecases%2Flinkfailure%2Fapp%2Fmain.py;h=30541d4c51097c926f321bb4a99b61bc8f6da516;hb=330db2ab9b62777a96450527eb1978e15986a0d4;hp=3629ea8376581ad46853838346a08ea87eaf4603;hpb=00fd1e7bdea32ecef53f99b5e94541350af27504;p=nonrtric.git diff --git a/test/usecases/linkfailure/app/main.py b/test/usecases/linkfailure/app/main.py index 3629ea83..30541d4c 100644 --- a/test/usecases/linkfailure/app/main.py +++ b/test/usecases/linkfailure/app/main.py @@ -18,11 +18,13 @@ import argparse import ast -import requests import json +import os +import requests import time -SDNR_PATH = "/rests/data/network-topology:network-topology/topology=topology-netconf/node=O-RAN-DU-01/yang-ext:mount/o-ran-sc-du-hello-world:network-function/du-to-ru-connection=" +MR_PATH = "/events/[TOPIC]/users/test/" +SDNR_PATH = "/rests/data/network-topology:network-topology/topology=topology-netconf/node=[O-DU-ID]/yang-ext:mount/o-ran-sc-du-hello-world:network-function/du-to-ru-connection=[O-RU-ID]" UNLOCK_MESSAGE = { "o-ran-sc-du-hello-world:du-to-ru-connection": [ @@ -41,7 +43,7 @@ def is_message_new_link_failure(message): link_failure = False if (event_headers["domain"] == "fault"): fault_fields = msg_as_json["event"]["faultFields"] - link_failure = fault_fields["specificProblem"] == "CUS Link Failure" and fault_fields["eventSeverity"] == "CRITICAL" + link_failure = fault_fields["alarmCondition"] == "30" and fault_fields["eventSeverity"] != "NORMAL" return link_failure @@ -53,7 +55,7 @@ def is_message_clear_link_failure(message): link_failure_clear = False if (event_headers["domain"] == "fault"): fault_fields = msg_as_json["event"]["faultFields"] - link_failure_clear = fault_fields["specificProblem"] == "CUS Link Failure" and fault_fields["eventSeverity"] == "NORMAL" + link_failure_clear = fault_fields["alarmCondition"] == "30" and fault_fields["eventSeverity"] == "NORMAL" return link_failure_clear @@ -62,20 +64,20 @@ def handle_link_failure(message, o_ru_to_o_du_map, sdnr_address): verboseprint("Got a link failure: ") alarm_msg_as_json = json.loads(message) event_headers = alarm_msg_as_json["event"]["commonEventHeader"] - o_ru_id = event_headers["reportingEntityId"] + o_ru_id = event_headers["sourceName"] verboseprint("O-RU ID: " + o_ru_id) o_du_id = o_ru_to_o_du_map[o_ru_id] verboseprint("O-DU ID: " + o_du_id) unlock_msg = json.loads(json.dumps(UNLOCK_MESSAGE)) - unlock_msg["o-ran-sc-du-hello-world:du-to-ru-connection"][0]["name"] = o_du_id - response = requests.post(sdnr_address + SDNR_PATH + o_du_id, json=unlock_msg) - print(response) + unlock_msg["o-ran-sc-du-hello-world:du-to-ru-connection"][0]["name"] = o_ru_id + send_path = SDNR_PATH.replace("[O-DU-ID]", o_du_id).replace("[O-RU-ID]", o_ru_id) + requests.post(sdnr_address + send_path, json=unlock_msg) def handle_clear_link_failure(message): msg_as_json = json.loads(message) event_headers = msg_as_json["event"]["commonEventHeader"] - o_ru_id = event_headers["reportingEntityId"] + o_ru_id = event_headers["sourceName"] verboseprint("Cleared Link Failure for O-RU ID: " + o_ru_id) @@ -89,25 +91,37 @@ def read_o_ru_to_o_du_map_from_file(map_file): if __name__ == '__main__': parser = argparse.ArgumentParser(prog='PROG') - parser.add_argument('--mrHost', help='The URL of the MR host', default="http://message-router.onap") - parser.add_argument('--mrPort', help='The port of the MR host', type=int, default=3904) - parser.add_argument('--mrTopic', help='The topic to poll messages from', default="ALARMS-WRITE") - parser.add_argument('--sdnrHost', help='The URL of the SNDR host', default="http://localhost") - parser.add_argument('--sdnrPort', help='The port of the SDNR host', type=int, default=9990) - parser.add_argument('--oRuTooDuMapFile', help='A file with the mapping between O-RU ID and O-DU ID as a dictionary', default="o-ru-to-o-du-map.txt") - parser.add_argument('--pollTime', help='The time between polls', type=int, default=10) + parser.add_argument('--mrHost', help='The URL of the MR host (default: %(default)s)', default="http://message-router.onap") + parser.add_argument('--mrPort', help='The port of the MR host (default: %(default)d)', type=int, default=3904) + parser.add_argument('--mrTopic', help='The topic to poll messages from (default: %(default)s)', default="unauthenticated.SEC_FAULT_OUTPUT") + parser.add_argument('--sdnrHost', help='The URL of the SNDR host (default: %(default)s)', default="http://localhost") + parser.add_argument('--sdnrPort', help='The port of the SDNR host (default: %(default)d)', type=int, default=9990) + parser.add_argument('--oRuTooDuMapFile', help='A file with the mapping between O-RU ID and O-DU ID as a dictionary (default: %(default)s)', default="o-ru-to-o-du-map.txt") + parser.add_argument('--pollTime', help='The time between polls (default: %(default)d)', type=int, default=10) parser.add_argument('-v', '--verbose', action='store_true', help='Turn on verbose printing') parser.add_argument('--version', action='version', version='%(prog)s 1.0') args = vars(parser.parse_args()) mr_host = args["mrHost"] + if os.getenv("MR-HOST") is not None: + mr_host = os.getenv("MR-HOST") + print("Using MR Host from os: " + mr_host) mr_port = args["mrPort"] + if os.getenv("MR-PORT") is not None: + mr_port = os.getenv("MR-PORT") + print("Using MR Port from os: " + mr_port) mr_topic = args["mrTopic"] sdnr_host = args["sdnrHost"] + if os.getenv("SDNR-HOST") is not None: + sdnr_host = os.getenv("SDNR-HOST") + print("Using SNDR Host from os: " + sdnr_host) sdnr_port = args["sdnrPort"] + if os.getenv("SDNR-PORT") is not None: + sdnr_port = os.getenv("SDNR-PORT") + print("Using SNDR Host from os: " + sdnr_port) o_ru_to_o_du_map = read_o_ru_to_o_du_map_from_file(args["oRuTooDuMapFile"]) pollTime = args["pollTime"] - if args["verbose"]: + if os.getenv("VERBOSE") is not None or args["verbose"]: def verboseprint(*args, **kwargs): print(*args, **kwargs) @@ -118,17 +132,20 @@ if __name__ == '__main__': verboseprint("Using MR address: " + mr_host + ":" + str(mr_port) + " and topic: " + mr_topic) verboseprint("Using SDNR address: " + sdnr_host + ":" + str(sdnr_port)) verboseprint("Starting with " + str(pollTime) + " seconds between polls") - mr_address = mr_host + ":" + str(mr_port) + "/events/" + mr_topic + "/users/test/" + mr_address = mr_host + ":" + str(mr_port) + MR_PATH.replace("[TOPIC]", mr_topic) sdnr_address = sdnr_host + ":" + str(sdnr_port) while True: - response = requests.get(mr_address) - verboseprint("Polling") - messages = response.json() - for message in messages: - if (is_message_new_link_failure(message)): - handle_link_failure(message, o_ru_to_o_du_map, sdnr_address) - elif (is_message_clear_link_failure(message)): - handle_clear_link_failure(message) + try: + verboseprint("Polling") + response = requests.get(mr_address) + messages = response.json() + for message in messages: + if (is_message_new_link_failure(message)): + handle_link_failure(message, o_ru_to_o_du_map, sdnr_address) + elif (is_message_clear_link_failure(message)): + handle_clear_link_failure(message) + except Exception as inst: + print(inst) time.sleep(pollTime)