Add scale vnf code framework

Change-Id: I71e1ea63bb6766832c234f4f70980ade97b3c369
Issue-ID: VFC-1306
Signed-off-by: fujinhua <fu.jinhua@zte.com.cn>

2 files changed, 85 insertions, 0 deletions

diff --git a/lcm/lcm/nf/urls.py b/lcm/lcm/nf/urls.py
index fd0aaae9..c056eed7 100644
--- a/lcm/lcm/nf/urls.py
+++ b/lcm/lcm/nf/urls.py
@@ -20,6 +20,7 @@ from lcm.nf.views.terminate_vnf_view import TerminateVnfView
 from lcm.nf.views.subscriptions_view import SubscriptionsView
 from lcm.nf.views.heal_vnf_view import HealVnfView
 from lcm.nf.views.operate_vnf_view import OperateVnfView
+from lcm.nf.views.scale_vnf_view import ScaleVnfView
 from lcm.nf.views.lcm_op_occs_view import QueryMultiVnfLcmOpOccs, QuerySingleVnfLcmOpOcc
 
 urlpatterns = [
@@ -30,6 +31,7 @@ urlpatterns = [
     url(r'^api/vnflcm/v1/vnf_instances/(?P<instanceid>[0-9a-zA-Z_-]+)/terminate$', TerminateVnfView.as_view()),
     url(r'^api/vnflcm/v1/vnf_instances/(?P<instanceid>[0-9a-zA-Z_-]+)/heal$', HealVnfView.as_view()),
     url(r'^api/vnflcm/v1/vnf_instances/(?P<instanceid>[0-9a-zA-Z_-]+)/operate$', OperateVnfView.as_view()),
+    url(r'^vnf_instances/(?P<instanceid>[0-9a-zA-Z_-]+)/scale$', ScaleVnfView.as_view()),
     url(r'^api/vnflcm/v1/vnf_lcm_op_occs$', QueryMultiVnfLcmOpOccs.as_view()),
     url(r'^api/vnflcm/v1/vnf_lcm_op_occs/(?P<lcmopoccid>[0-9a-zA-Z_-]+)$', QuerySingleVnfLcmOpOcc.as_view()),
 ]
diff --git a/lcm/lcm/nf/views/scale_vnf_view.py b/lcm/lcm/nf/views/scale_vnf_view.py
new file mode 100644
index 00000000..20ef95d0
--- /dev/null
+++ b/lcm/lcm/nf/views/scale_vnf_view.py
@@ -0,0 +1,83 @@
+# Copyright (C) 2019 ZTE. 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 logging
+import traceback
+
+from drf_yasg.utils import swagger_auto_schema
+from rest_framework import status
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from lcm.nf.serializers.scale_vnf_request import ScaleVnfRequestSerializer
+from lcm.nf.serializers.response import ProblemDetailsSerializer
+from lcm.pub.exceptions import NFLCMException, NFLCMExceptionNotFound, NFLCMExceptionConflict
+from lcm.pub.utils.jobutil import JobUtil
+from lcm.pub.database.models import NfInstModel
+from lcm.nf.const import VNF_STATUS
+
+logger = logging.getLogger(__name__)
+
+
+class ScaleVnfView(APIView):
+    @swagger_auto_schema(
+        request_body=ScaleVnfRequestSerializer(),
+        responses={
+            status.HTTP_202_ACCEPTED: "Success",
+            status.HTTP_404_NOT_FOUND: ProblemDetailsSerializer(),
+            status.HTTP_409_CONFLICT: ProblemDetailsSerializer(),
+            status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
+        }
+    )
+    def post(self, request, instanceid):
+        logger.debug("ScaleVnf--post::> %s" % request.data)
+        try:
+            scale_vnf_request_serializer = ScaleVnfRequestSerializer(data=request.data)
+            if not scale_vnf_request_serializer.is_valid():
+                raise NFLCMException(scale_vnf_request_serializer.errors)
+
+            job_id = JobUtil.create_job('NF', 'SCALE', instanceid)
+            JobUtil.add_job_status(job_id, 0, "SCALE_VNF_READY")
+            self.scale_pre_check(instanceid, job_id)
+            # TODO: do scale logic
+            response = Response(data={"jobId": job_id}, status=status.HTTP_202_ACCEPTED)
+            return response
+        except NFLCMExceptionNotFound as e:
+            probDetail = ProblemDetailsSerializer(data={"status": status.HTTP_404_NOT_FOUND, "detail": "VNF Instance not found"})
+            resp_isvalid = probDetail.is_valid()
+            if not resp_isvalid:
+                raise NFLCMException(probDetail.errors)
+            return Response(data=probDetail.data, status=status.HTTP_404_NOT_FOUND)
+        except NFLCMExceptionConflict as e:
+            probDetail = ProblemDetailsSerializer(data={"status": status.HTTP_409_CONFLICT, "detail": "VNF Instance not in Instantiated State"})
+            resp_isvalid = probDetail.is_valid()
+            if not resp_isvalid:
+                raise NFLCMException(probDetail.errors)
+            return Response(data=probDetail.data, status=status.HTTP_409_CONFLICT)
+        except NFLCMException as e:
+            logger.error(e.message)
+            return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        except Exception as e:
+            logger.error(e.message)
+            logger.error(traceback.format_exc())
+            return Response(data={'error': 'unexpected exception'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+    def scale_pre_check(self, nf_inst_id, job_id):
+        vnf_insts = NfInstModel.objects.filter(nfinstid=nf_inst_id)
+        if not vnf_insts.exists():
+            raise NFLCMExceptionNotFound("VNF nf_inst_id does not exist.")
+
+        NfInstModel.objects.filter(nfinstid=nf_inst_id).update(status=VNF_STATUS.SCALING)
+        JobUtil.add_job_status(job_id, 15, 'Nf scaling pre-check finish')
+        logger.info("Nf scaling pre-check finish")