Add scale vnf code framework
[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / views / scale_vnf_view.py
1 # Copyright (C) 2019 ZTE. All Rights Reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #       http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 import logging
16 import traceback
17
18 from drf_yasg.utils import swagger_auto_schema
19 from rest_framework import status
20 from rest_framework.response import Response
21 from rest_framework.views import APIView
22
23 from lcm.nf.serializers.scale_vnf_request import ScaleVnfRequestSerializer
24 from lcm.nf.serializers.response import ProblemDetailsSerializer
25 from lcm.pub.exceptions import NFLCMException, NFLCMExceptionNotFound, NFLCMExceptionConflict
26 from lcm.pub.utils.jobutil import JobUtil
27 from lcm.pub.database.models import NfInstModel
28 from lcm.nf.const import VNF_STATUS
29
30 logger = logging.getLogger(__name__)
31
32
33 class ScaleVnfView(APIView):
34     @swagger_auto_schema(
35         request_body=ScaleVnfRequestSerializer(),
36         responses={
37             status.HTTP_202_ACCEPTED: "Success",
38             status.HTTP_404_NOT_FOUND: ProblemDetailsSerializer(),
39             status.HTTP_409_CONFLICT: ProblemDetailsSerializer(),
40             status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
41         }
42     )
43     def post(self, request, instanceid):
44         logger.debug("ScaleVnf--post::> %s" % request.data)
45         try:
46             scale_vnf_request_serializer = ScaleVnfRequestSerializer(data=request.data)
47             if not scale_vnf_request_serializer.is_valid():
48                 raise NFLCMException(scale_vnf_request_serializer.errors)
49
50             job_id = JobUtil.create_job('NF', 'SCALE', instanceid)
51             JobUtil.add_job_status(job_id, 0, "SCALE_VNF_READY")
52             self.scale_pre_check(instanceid, job_id)
53             # TODO: do scale logic
54             response = Response(data={"jobId": job_id}, status=status.HTTP_202_ACCEPTED)
55             return response
56         except NFLCMExceptionNotFound as e:
57             probDetail = ProblemDetailsSerializer(data={"status": status.HTTP_404_NOT_FOUND, "detail": "VNF Instance not found"})
58             resp_isvalid = probDetail.is_valid()
59             if not resp_isvalid:
60                 raise NFLCMException(probDetail.errors)
61             return Response(data=probDetail.data, status=status.HTTP_404_NOT_FOUND)
62         except NFLCMExceptionConflict as e:
63             probDetail = ProblemDetailsSerializer(data={"status": status.HTTP_409_CONFLICT, "detail": "VNF Instance not in Instantiated State"})
64             resp_isvalid = probDetail.is_valid()
65             if not resp_isvalid:
66                 raise NFLCMException(probDetail.errors)
67             return Response(data=probDetail.data, status=status.HTTP_409_CONFLICT)
68         except NFLCMException as e:
69             logger.error(e.message)
70             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
71         except Exception as e:
72             logger.error(e.message)
73             logger.error(traceback.format_exc())
74             return Response(data={'error': 'unexpected exception'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
75
76     def scale_pre_check(self, nf_inst_id, job_id):
77         vnf_insts = NfInstModel.objects.filter(nfinstid=nf_inst_id)
78         if not vnf_insts.exists():
79             raise NFLCMExceptionNotFound("VNF nf_inst_id does not exist.")
80
81         NfInstModel.objects.filter(nfinstid=nf_inst_id).update(status=VNF_STATUS.SCALING)
82         JobUtil.add_job_status(job_id, 15, 'Nf scaling pre-check finish')
83         logger.info("Nf scaling pre-check finish")