| #!/usr/bin/python3 |
| # -*- coding: utf-8 -*- |
| |
| import asyncio |
| import aiohttp |
| import yaml |
| import ROclient |
| import time |
| import dbmemory |
| import logging |
| |
| from copy import deepcopy |
| from uuid import uuid4 |
| |
| #streamformat = "%(asctime)s %(name)s %(levelname)s: %(message)s" |
| streamformat = "%(name)s %(levelname)s: %(message)s" |
| logging.basicConfig(format=streamformat, level=logging.DEBUG) |
| logger = logging.getLogger('lcm') |
| |
| ro_account = { |
| "url": "http://localhost:9090/openmano", |
| "tenant": "osm" |
| } |
| |
| vca_account = { |
| # TODO |
| } |
| |
| # conains created tasks/futures to be able to cancel |
| lcm_tasks = {} |
| |
| headers_req = {'Accept': 'application/yaml', 'content-type': 'application/yaml'} |
| ns_status = ("CREATION-SCHEDULED", "DEPLOYING", "CONFIGURING", "DELETION-SCHEDULED", "UN-CONFIGURING", "UNDEPLOYING") |
| |
| # TODO replace with database calls |
| db = dbmemory.dbmemory() |
| |
| |
| class RO_Exception(Exception): |
| pass |
| |
| |
| async def CreateNS(loop, nsr_id): |
| logger.debug("CreateNS task nsr_id={} Enter".format(nsr_id)) |
| nsr_lcm = { |
| "id": nsr_id, |
| "RO": {"vnfd_id": {}, "nsd_id": None, "nsr_id": None, "nsr_status": "SCHEDULED"}, |
| "nsr_ip": {}, |
| "VCA": {"TODO"}, |
| "status": "BUILD", |
| "status_detailed": "", |
| } |
| |
| deloyment_timeout = 120 |
| try: |
| ns_request = db.get_one("ns_request", {"id": nsr_id}) |
| nsd = db.get_one("nsd", {"id": ns_request["nsd_id"]}) |
| RO = ROclient.ROClient(loop, endpoint_url=ro_account["url"], tenant=ro_account["tenant"], |
| datacenter=ns_request["vim"]) |
| nsr_lcm["status_detailed"] = "Creating vnfd at RO" |
| # ns_request["constituent-vnfr-ref"] = [] |
| |
| db.create("nsr_lcm", nsr_lcm) |
| |
| # get vnfds, instantiate at RO |
| logger.debug("CreateNS task nsr_id={} RO VNFD".format(nsr_id)) |
| for c_vnf in nsd["constituent-vnfd"]: |
| vnfd_id = c_vnf["vnfd-id-ref"] |
| vnfd = db.get_one("vnfd", {"id": vnfd_id}) |
| vnfd.pop("_admin", None) |
| vnfd.pop("_id", None) |
| # vnfr = deepcopy(vnfd) |
| # vnfr["member-vnf-index"] = c_vnf["member-vnf-index"] |
| # vnfr["nsr-id"] = nsr_id |
| # vnfr["id"] = uuid4() |
| # vnfr["vnf-id"] = vnfd["id"] |
| # ns_request["constituent-vnfr-ref"],append(vnfd_id) |
| |
| # TODO change id for RO in case it is present |
| try: |
| desc = await RO.create("vnfd", descriptor=vnfd) |
| nsr_lcm["RO"]["vnfd_id"][vnfd_id] = desc["uuid"] |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| except ROclient.ROClientException as e: |
| if e.http_code == 409: # conflict, vnfd already present |
| print("debug", e) |
| else: |
| raise |
| |
| # db_new("vnfr", vnfr) |
| # db_update("ns_request", nsr_id, ns_request) |
| |
| # create nsd at RO |
| logger.debug("CreateNS task nsr_id={} RO NSD".format(nsr_id)) |
| nsr_lcm["status_detailed"] = "Creating nsd at RO" |
| nsd_id = ns_request["nsd_id"] |
| nsd = db.get_one("nsd", {"id": nsd_id}) |
| nsd.pop("_admin", None) |
| nsd.pop("_id", None) |
| try: |
| desc = await RO.create("nsd", descriptor=nsd) |
| nsr_lcm["RO"]["nsd_id"] = desc["uuid"] |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| except ROclient.ROClientException as e: |
| if e.http_code == 409: # conflict, nsd already present |
| print("debug", e) |
| else: |
| raise |
| |
| # Crate ns at RO |
| logger.debug("CreateNS task nsr_id={} RO NS".format(nsr_id)) |
| nsr_lcm["status_detailed"] = "Creating ns at RO" |
| desc = await RO.create("ns", name=ns_request["name"], datacenter=ns_request["vim"], scenario=nsr_lcm["RO"]["nsd_id"]) |
| RO_nsr_id = desc["uuid"] |
| nsr_lcm["RO"]["nsr_id"] = RO_nsr_id |
| nsr_lcm["RO"]["nsr_status"] = "BUILD" |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| |
| # wait until NS is ready |
| deloyment_timeout = 600 |
| while deloyment_timeout > 0: |
| ns_status_detailed = "Waiting ns ready at RO" |
| nsr_lcm["status_detailed"] = ns_status_detailed |
| desc = await RO.show("ns", RO_nsr_id) |
| ns_status, ns_status_info = RO.check_ns_status(desc) |
| nsr_lcm["RO"]["nsr_status"] = ns_status |
| if ns_status == "ERROR": |
| raise ROclient.ROClientException(ns_status_info) |
| elif ns_status == "BUILD": |
| nsr_lcm["status_detailed"] = ns_status_detailed + "; nsr_id: '{}', {}".format(nsr_id, ns_status_info) |
| elif ns_status == "ACTIVE": |
| nsr_lcm["nsr_ip"] = RO.get_ns_vnf_ip(desc) |
| break |
| else: |
| assert False, "ROclient.check_ns_status returns unknown {}".format(ns_status) |
| |
| await asyncio.sleep(5, loop=loop) |
| deloyment_timeout -= 5 |
| if deloyment_timeout <= 0: |
| raise ROclient.ROClientException("Timeot wating ns to be ready") |
| nsr_lcm["status_detailed"] = "Configuring vnfr" |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| |
| #for nsd in nsr_lcm["descriptors"]["nsd"]: |
| |
| logger.debug("CreateNS task nsr_id={} VCA look for".format(nsr_id)) |
| for c_vnf in nsd["constituent-vnfd"]: |
| vnfd_id = c_vnf["vnfd-id-ref"] |
| vnfd_index = int(c_vnf["member-vnf-index"]) |
| vnfd = db.get_one("vnfd", {"id": vnfd_id}) |
| if vnfd.get("vnf-configuration") and vnfd["vnf-configuration"].get("juju"): |
| proxy_charm = vnfd["vnf-configuration"]["juju"]["charm"] |
| config_primitive = vnfd["vnf-configuration"].get("config-primitive") |
| # get parameters for juju charm |
| base_folder = vnfd["_admin"]["storage"] |
| path = base_folder + "/charms/" + proxy_charm |
| mgmt_ip = nsr_lcm['nsr_ip'][vnfd_index] |
| <<<<<<< HEAD |
| # task = asyncio.ensure_future(DeployCharm(loop, path, mgmt_ip, config_primitive)) |
| pass |
| # TODO launch VCA charm |
| ======= |
| # TODO launch VCA charm |
| # task = asyncio.ensure_future(DeployCharm(loop, path, mgmt_ip, config_primitive)) |
| >>>>>>> 6ce7cc879dda3b729b10d563bd26df613dc9f70f |
| nsr_lcm["status"] = "DONE" |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| |
| return nsr_lcm |
| |
| except (ROclient.ROClientException, Exception) as e: |
| logger.debug("CreateNS nsr_id={} Exception {}".format(nsr_id, e), exc_info=True) |
| nsr_lcm["status"] = "ERROR" |
| nsr_lcm["status_detailed"] += ": ERROR {}".format(e) |
| finally: |
| logger.debug("CreateNS task nsr_id={} Exit".format(nsr_id)) |
| |
| |
| async def DestroyNS(loop, nsr_id): |
| logger.debug("DestroyNS task nsr_id={} Enter".format(nsr_id)) |
| nsr_lcm = db.get_one("nsr_lcm", {"id": nsr_id}) |
| ns_request = db.get_one("ns_request", {"id": nsr_id}) |
| |
| nsr_lcm["status"] = "DELETING" |
| nsr_lcm["status_detailed"] = "Deleting charms" |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| <<<<<<< HEAD |
| # TODO destroy charms |
| ======= |
| # TODO destroy VCA charm |
| >>>>>>> 6ce7cc879dda3b729b10d563bd26df613dc9f70f |
| |
| # remove from RO |
| RO = ROclient.ROClient(loop, endpoint_url=ro_account["url"], tenant=ro_account["tenant"], |
| datacenter=ns_request["vim"]) |
| # Delete ns |
| try: |
| RO_nsr_id = nsr_lcm["RO"]["nsr_id"] |
| if RO_nsr_id: |
| nsr_lcm["status_detailed"] = "Deleting ns at RO" |
| desc = await RO.delete("ns", RO_nsr_id) |
| print("debug", "deleted RO ns {}".format(RO_nsr_id)) |
| nsr_lcm["RO"]["nsr_id"] = None |
| nsr_lcm["RO"]["nsr_status"] = "DELETED" |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| except ROclient.ROClientException as e: |
| if e.http_code == 404: |
| nsr_lcm["RO"]["nsr_id"] = None |
| nsr_lcm["RO"]["nsr_status"] = "DELETED" |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| print("warning", e) |
| else: |
| print("error", e) |
| |
| # Delete nsd |
| try: |
| RO_nsd_id = nsr_lcm["RO"]["nsd_id"] |
| if RO_nsd_id: |
| nsr_lcm["status_detailed"] = "Deleting nsd at RO" |
| desc = await RO.delete("nsd", RO_nsd_id) |
| print("debug", "deleted RO nsd {}".format(RO_nsd_id)) |
| nsr_lcm["RO"]["nsd_id"] = None |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| except ROclient.ROClientException as e: |
| if e.http_code == 404: |
| nsr_lcm["RO"]["nsd_id"] = None |
| print("warning", e) |
| else: |
| print("error", e) |
| |
| for vnf_id, RO_vnfd_id in nsr_lcm["RO"]["vnfd_id"].items(): |
| try: |
| if RO_vnfd_id: |
| nsr_lcm["status_detailed"] = "Deleting vnfd at RO" |
| desc = await RO.delete("vnfd", RO_vnfd_id) |
| print("debug", "deleted RO vnfd {}".format(RO_vnfd_id)) |
| nsr_lcm["RO"]["vnfd_id"][vnf_id] = None |
| db.replace("nsr_lcm", {"id": nsr_id}, nsr_lcm) |
| except ROclient.ROClientException as e: |
| if e.http_code == 404: |
| nsr_lcm["RO"]["vnfd_id"][vnf_id] = None |
| print("warning", e) |
| else: |
| print("error", e) |
| logger.debug("DestroyNS task nsr_id={} Exit".format(nsr_id)) |
| |
| |
| async def test(loop, param=None): |
| logger.debug("Starting/Ending test task: {}".format(param)) |
| |
| |
| def cancel_tasks(loop, nsr_id): |
| """ |
| Cancel all active tasks of a concrete nsr identified for nsr_id |
| :param loop: loop |
| :param nsr_id: nsr identity |
| :return: None, or raises an exception if not possible |
| """ |
| global lcm_tasks |
| if not lcm_tasks.get(nsr_id): |
| return |
| for order_id, tasks_set in lcm_tasks[nsr_id].items(): |
| for task_name, task in tasks_set.items(): |
| result = task.cancel() |
| if result: |
| logger.debug("nsr_id={} order_id={} task={} cancelled".format(nsr_id, order_id, task_name)) |
| lcm_tasks[nsr_id] = {} |
| |
| |
| |
| async def read_kafka(loop, bus_info): |
| global lcm_tasks |
| logger.debug("kafka task Enter") |
| order_id = 1 |
| # future = asyncio.Future() |
| with open(bus_info["file"]) as f: |
| |
| # ignore old orders. Read file |
| command = "fake" |
| while command: |
| command = f.read() |
| |
| while True: |
| command = f.read() |
| if not command: |
| await asyncio.sleep(2, loop=loop) |
| continue |
| order_id += 1 |
| command = command.strip() |
| command, _, params = command.partition(" ") |
| if command == "exit": |
| print("Bye!") |
| break |
| elif command.startswith("#"): |
| continue |
| elif command == "echo": |
| print(params) |
| elif command == "test": |
| asyncio.Task(test(loop, params), loop=loop) |
| elif command == "break": |
| print("put a break in this line of code") |
| elif command == "new-ns": |
| nsr_id = params.strip() |
| logger.debug("Deploying NS {}".format(nsr_id)) |
| task = asyncio.ensure_future(CreateNS(loop, nsr_id)) |
| if nsr_id not in lcm_tasks: |
| lcm_tasks[nsr_id] = {} |
| lcm_tasks[nsr_id][order_id] = {"CreateNS": task} |
| elif command == "del-ns": |
| nsr_id = params.strip() |
| logger.debug("Deleting NS {}".format(nsr_id)) |
| cancel_tasks(loop, nsr_id) |
| task = asyncio.ensure_future(DestroyNS(loop, nsr_id)) |
| if nsr_id not in lcm_tasks: |
| lcm_tasks[nsr_id] = {} |
| lcm_tasks[nsr_id][order_id] = {"DestroyNS": task} |
| elif command == "get-ns": |
| nsr_id = params.strip() |
| nsr_lcm = db.get_one("nsr_lcm", {"id": nsr_id}) |
| print("nsr_lcm", nsr_lcm) |
| print("lcm_tasks", lcm_tasks.get(nsr_id)) |
| else: |
| logger.debug("unknown command '{}'".format(command)) |
| print("Usage:\n echo <>\n new-ns <ns1|ns2>\n del-ns <ns1|ns2>\n get-ns <ns1|ns2>") |
| logger.debug("kafka task Exit") |
| |
| |
| def lcm(): |
| loop = asyncio.get_event_loop() |
| loop.run_until_complete(read_kafka(loop, {"file": "/home/atierno/OSM/osm/NBI/kafka"})) |
| return |
| |
| |
| def lcm2(): |
| loop = asyncio.get_event_loop() |
| # asyncio.ensure_future(CreateNS, loop) |
| try: |
| content = loop.run_until_complete(CreateNS(loop, "ns1")) |
| print("Done: {}".format(content)) |
| except ROclient.ROClientException as e: |
| print("Error {}".format(e)) |
| |
| time.sleep(10) |
| |
| content = loop.run_until_complete(DestroyNS(loop, "ns1")) |
| print(content) |
| |
| loop.close() |
| |
| |
| if __name__ == '__main__': |
| |
| # FOR TEST |
| RO_VIM = "OST2_MRT" |
| |
| #FILL DATABASE |
| with open("/home/atierno/OSM/osm/devops/descriptor-packages/vnfd/ping_vnf/src/ping_vnfd.yaml") as f: |
| vnfd = yaml.load(f) |
| vnfd_clean, _ = ROclient.remove_envelop("vnfd", vnfd) |
| vnfd_clean["_admin"] = {"storage": "/home/atierno/OSM/osm/devops/descriptor-packages/vnfd/ping_vnf"} |
| db.create("vnfd", vnfd_clean) |
| with open("/home/atierno/OSM/osm/devops/descriptor-packages/vnfd/pong_vnf/src/pong_vnfd.yaml") as f: |
| vnfd = yaml.load(f) |
| vnfd_clean, _ = ROclient.remove_envelop("vnfd", vnfd) |
| vnfd_clean["_admin"] = {"storage": "/home/atierno/OSM/osm/devops/descriptor-packages/vnfd/pong_vnf"} |
| db.create("vnfd", vnfd_clean) |
| with open("/home/atierno/OSM/osm/devops/descriptor-packages/nsd/ping_pong_ns/src/ping_pong_nsd.yaml") as f: |
| nsd = yaml.load(f) |
| nsd_clean, _ = ROclient.remove_envelop("nsd", nsd) |
| nsd_clean["_admin"] = {"storage": "/home/atierno/OSM/osm/devops/descriptor-packages/nsd/ping_pong_ns"} |
| db.create("nsd", nsd_clean) |
| |
| ns_request = { |
| "id": "ns1", |
| "nsr_id": "ns1", |
| "name": "pingpongOne", |
| "vim": RO_VIM, |
| "nsd_id": nsd_clean["id"], # nsd_ping_pong |
| } |
| db.create("ns_request", ns_request) |
| ns_request = { |
| "id": "ns2", |
| "nsr_id": "ns2", |
| "name": "pingpongTwo", |
| "vim": RO_VIM, |
| "nsd_id": nsd_clean["id"], # nsd_ping_pong |
| } |
| db.create("ns_request", ns_request) |
| # lcm2() |
| lcm() |