Add Heal Api in GVNFM

Change-Id: If825c7a03cd38a6694d2cc0d28fbff391a3f4a95
Signed-off-by: Shobana Jothi<shobana.jothi@verizon.com>
Issue-ID: VFC-995

1 files changed, 117 insertions, 0 deletions

diff --git a/lcm/lcm/nf/biz/heal_vnf.py b/lcm/lcm/nf/biz/heal_vnf.py
new file mode 100644
index 00000000..bf8e34a0
--- /dev/null
+++ b/lcm/lcm/nf/biz/heal_vnf.py
@@ -0,0 +1,117 @@
+# Copyright (C) 2018 Verizon. 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.
+
+import json
+import logging
+import traceback
+from threading import Thread
+
+from lcm.pub.database.models import NfInstModel, VmInstModel
+from lcm.pub.exceptions import NFLCMException
+from lcm.pub.msapi.gvnfmdriver import notify_lcm_to_nfvo, prepare_notification_data
+from lcm.pub.utils.jobutil import JobUtil
+from lcm.pub.utils.timeutil import now_time
+from lcm.pub.utils.values import ignore_case_get
+from lcm.pub.vimapi import adaptor
+from lcm.nf.biz.grant_vnf import grant_resource
+from lcm.nf.const import VNF_STATUS, GRANT_TYPE, HEAL_ACTION_TYPE, CHANGE_TYPE, OPERATION_TYPE
+from lcm.nf.biz import common
+
+logger = logging.getLogger(__name__)
+
+
+class HealVnf(Thread):
+    def __init__(self, data, nf_inst_id, job_id):
+        super(HealVnf, self).__init__()
+        self.data = data
+        self.nf_inst_id = nf_inst_id
+        self.job_id = job_id
+        self.affectedvm = ignore_case_get(ignore_case_get(self.data, "additionalParams"), "affectedvm")
+        # TODO: Check if we could move the action param into the list of affectedvm structure
+        self.action = ignore_case_get(ignore_case_get(self.data, "additionalParams"), "action")
+        self.grant_type = ""
+        if self.action == HEAL_ACTION_TYPE.START:
+            self.grant_type = GRANT_TYPE.HEAL_CREATE
+        elif self.action == HEAL_ACTION_TYPE.RESTART:
+            self.grant_type = GRANT_TYPE.HEAL_RESTART
+
+    def run(self):
+        try:
+            self.heal_pre()
+            self.apply_grant()
+            self.heal_resource()
+            JobUtil.add_job_status(self.job_id, 100, "Heal Vnf success.")
+            NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status='INSTANTIATED', lastuptime=now_time())
+            self.lcm_notify()
+        except NFLCMException as e:
+            logger.error(e.message)
+            self.vnf_heal_failed_handle(e.message)
+        except Exception as e:
+            logger.error(e.message)
+            self.vnf_heal_failed_handle(traceback.format_exc())
+
+    def heal_pre(self):
+        if self.action not in (HEAL_ACTION_TYPE.START, HEAL_ACTION_TYPE.RESTART):
+            raise NFLCMException("Action type in Request in invalid. Should be %s or %s" % (HEAL_ACTION_TYPE.START, HEAL_ACTION_TYPE.RESTART))
+
+        self.vm_id = ignore_case_get(self.affectedvm, "vmid")
+        self.vdu_id = ignore_case_get(self.affectedvm, "vduid")
+        self.vm_name = ignore_case_get(self.affectedvm, "vmname")
+        if not (self.vm_id and self.vdu_id and self.vm_name):
+            raise NFLCMException("VM identifiers is not present in request.")
+
+        self.vnf_insts = NfInstModel.objects.filter(nfinstid=self.nf_inst_id)
+        self.vnfd_info = json.loads(self.vnf_insts[0].vnfd_model)
+
+    def apply_grant(self):
+        if self.action == HEAL_ACTION_TYPE.RESTART:
+            self.vdu = VmInstModel.objects.filter(instid=self.nf_inst_id, is_predefined=1, vmid=self.vm_id, vmname=self.vm_name)
+            if not self.vdu:
+                raise NFLCMException("VNF Vm does not exist.")
+            self.vimid = self.vdu[0].vimid
+            self.tenant = self.vdu[0].tenant
+        elif self.action == HEAL_ACTION_TYPE.START:
+            vdus = ignore_case_get(self.vnfd_info, "vdus")
+            self.vdu = [elem for elem in vdus if ignore_case_get(elem, "vdu_id") == self.vdu_id]
+            if not self.vdu:
+                raise NFLCMException("VNF Vm does not exist.")
+        apply_result = grant_resource(data=self.data, nf_inst_id=self.nf_inst_id, job_id=self.job_id,
+                                      grant_type=self.grant_type, vdus=self.vdu)
+        if self.action == HEAL_ACTION_TYPE.START:
+            self.vimid = ignore_case_get(apply_result, "vimid"),
+            self.tenant = ignore_case_get(apply_result, "tenant")
+        logger.info("Grant resource, response: %s" % apply_result)
+        JobUtil.add_job_status(self.job_id, 20, 'Nf Healing grant_resource finish')
+
+    def heal_resource(self):
+        logger.info('Heal resource begin')
+        data = {'action': self.action, 'vimid': self.vimid, 'tenant': self.tenant}
+        adaptor.heal_vim_res(self.vdu, self.vnfd_info, self.do_notify, data, json.loads(self.vnf_insts[0].vimInfo), json.loads(self.vnf_insts[0].resInfo))
+        logger.info('Heal resource complete')
+
+    def do_notify(self, res_type, ret):
+        logger.info('Creating [%s] resource' % res_type)
+        resource_save_method = getattr(common, res_type + '_save')
+        resource_save_method(self.job_id, self.nf_inst_id, ret)
+
+    def lcm_notify(self):
+        notification_content = prepare_notification_data(self.nf_inst_id, self.job_id, CHANGE_TYPE.MODIFIED, OPERATION_TYPE.HEAL)
+        logger.info('Notify request data = %s' % notification_content)
+        resp = notify_lcm_to_nfvo(json.dumps(notification_content))
+        logger.info('Lcm notify end, response %s' % resp)
+
+    def vnf_heal_failed_handle(self, error_msg):
+        logger.error('VNF Healing failed, detail message: %s' % error_msg)
+        NfInstModel.objects.filter(nfinstid=self.nf_inst_id).update(status=VNF_STATUS.FAILED, lastuptime=now_time())
+        JobUtil.add_job_status(self.job_id, 255, error_msg)