lcm/ns/views.py

   1 # Copyright 2016-2017 ZTE Corporation.
   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 import json
  15 import logging
  16 import os
  17 import traceback
  18
  19 from rest_framework import status
  20 from rest_framework.response import Response
  21 from rest_framework.views import APIView
  22 from drf_yasg.utils import swagger_auto_schema
  23
  24 from lcm.ns.ns_create import CreateNSService
  25 from lcm.ns.ns_delete import DeleteNsService
  26 from lcm.ns.ns_get import GetNSInfoService
  27 from lcm.ns.ns_heal import NSHealService
  28 from lcm.ns.ns_instant import InstantNSService
  29 from lcm.ns.ns_manual_scale import NSManualScaleService
  30 from lcm.ns.ns_terminate import TerminateNsService
  31 from lcm.pub.database.models import NSInstModel, ServiceBaseInfoModel
  32 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
  33 from lcm.pub.utils.restcall import req_by_msb
  34 from lcm.pub.utils.values import ignore_case_get
  35 from lcm.ns.serializers import CreateNsReqSerializer, CreateNsRespSerializer
  36 from lcm.ns.serializers import QueryNsRespSerializer
  37 from lcm.pub.exceptions import NSLCMException
  38
  39 logger = logging.getLogger(__name__)
  40
  41
  42 class CreateNSView(APIView):
  43     @swagger_auto_schema(
  44         request_body=None,
  45         responses={
  46             status.HTTP_200_OK: QueryNsRespSerializer(help_text="NS instances", many=True),
  47             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
  48         }
  49     )
  50     def get(self, request):
  51         try:
  52             logger.debug("CreateNSView::get")
  53             filter = None
  54             csarId = ignore_case_get(request.META, 'csarId')
  55             if csarId:
  56                 filter = {"csarId": csarId}
  57
  58             ret = GetNSInfoService(filter).get_ns_info()
  59             logger.debug("CreateNSView::get::ret=%s", ret)
  60             resp_serializer = QueryNsRespSerializer(data=ret, many=True)
  61             if not resp_serializer.is_valid():
  62                 raise NSLCMException(resp_serializer.errors)
  63             return Response(data=resp_serializer.data, status=status.HTTP_200_OK)
  64         except Exception as e:
  65             logger.error(traceback.format_exc())
  66             logger.error("Exception in GetNS: %s", e.message)
  67             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
  68
  69     @swagger_auto_schema(
  70         request_body=CreateNsReqSerializer(),
  71         responses={
  72             status.HTTP_201_CREATED: CreateNsRespSerializer(),
  73             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
  74         }
  75     )
  76     def post(self, request):
  77         logger.debug("Enter CreateNS: %s", request.data)
  78         try:
  79             req_serializer = CreateNsReqSerializer(data=request.data)
  80             if not req_serializer.is_valid():
  81                 raise NSLCMException(req_serializer.errors)
  82
  83             if ignore_case_get(request.data, 'test') == "test":
  84                 return Response(data={'nsInstanceId': "test"}, status=status.HTTP_201_CREATED)
  85             csar_id = ignore_case_get(request.data, 'csarId')
  86             ns_name = ignore_case_get(request.data, 'nsName')
  87             description = ignore_case_get(request.data, 'description')
  88             context = ignore_case_get(request.data, 'context')
  89             ns_inst_id = CreateNSService(csar_id, ns_name, description, context).do_biz()
  90
  91             logger.debug("CreateNSView::post::ret={'nsInstanceId':%s}", ns_inst_id)
  92             resp_serializer = CreateNsRespSerializer(data={'nsInstanceId': ns_inst_id})
  93             if not resp_serializer.is_valid():
  94                 raise NSLCMException(resp_serializer.errors)
  95             return Response(data=resp_serializer.data, status=status.HTTP_201_CREATED)
  96         except Exception as e:
  97             logger.error(traceback.format_exc())
  98             logger.error("Exception in CreateNS: %s", e.message)
  99             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 100
 101
 102 class NSInstView(APIView):
 103     def post(self, request, ns_instance_id):
 104         ack = InstantNSService(ns_instance_id, request.data).do_biz()
 105         logger.debug("Leave NSInstView::post::ack=%s", ack)
 106         return Response(data=ack['data'], status=ack['status'])
 107
 108
 109 class TerminateNSView(APIView):
 110     def post(self, request, ns_instance_id):
 111         logger.debug("Enter TerminateNSView::post %s", request.data)
 112         termination_type = ignore_case_get(request.data, 'terminationType')
 113         graceful_termination_timeout = ignore_case_get(request.data, 'gracefulTerminationTimeout')
 114         job_id = JobUtil.create_job("VNF", JOB_TYPE.TERMINATE_VNF, ns_instance_id)
 115         try:
 116             TerminateNsService(ns_instance_id, termination_type, graceful_termination_timeout, job_id).start()
 117         except Exception as e:
 118             logger.error("Exception in CreateNS: %s", e.message)
 119             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 120         ret = {'jobId': job_id}
 121         logger.debug("Leave TerminateNSView::post ret=%s", ret)
 122         return Response(data=ret, status=status.HTTP_202_ACCEPTED)
 123
 124
 125 class NSHealView(APIView):
 126     def post(self, request, ns_instance_id):
 127         logger.debug("Enter HealNSView::post %s", request.data)
 128         job_id = JobUtil.create_job("VNF", JOB_TYPE.HEAL_VNF, ns_instance_id)
 129         try:
 130             NSHealService(ns_instance_id, request.data, job_id).start()
 131         except Exception as e:
 132             logger.error("Exception in HealNSView: %s", e.message)
 133             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 134         ret = {'jobId': job_id}
 135         logger.debug("Leave HealNSView::post ret=%s", ret)
 136         return Response(data=ret, status=status.HTTP_202_ACCEPTED)
 137
 138
 139 class NSDetailView(APIView):
 140     def get(self, request, ns_instance_id):
 141         logger.debug("Enter NSDetailView::get ns(%s)", ns_instance_id)
 142         ns_filter = {"ns_inst_id": ns_instance_id}
 143         ret = GetNSInfoService(ns_filter).get_ns_info()
 144         if not ret:
 145             return Response(status=status.HTTP_404_NOT_FOUND)
 146         logger.debug("Leave NSDetailView::get::ret=%s", ret)
 147         return Response(data=ret, status=status.HTTP_200_OK)
 148
 149     def delete(self, request, ns_instance_id):
 150         logger.debug("Enter NSDetailView::delete ns(%s)", ns_instance_id)
 151         DeleteNsService(ns_instance_id).do_biz()
 152         return Response(data={}, status=status.HTTP_204_NO_CONTENT)
 153
 154
 155 class SwaggerJsonView(APIView):
 156     def get(self, request):
 157         json_file = os.path.join(os.path.dirname(__file__), 'swagger.json')
 158         f = open(json_file)
 159         json_data = json.JSONDecoder().decode(f.read())
 160         f.close()
 161         return Response(json_data)
 162
 163
 164 class NSInstPostDealView(APIView):
 165     def post(self, request, ns_instance_id):
 166         logger.debug("Enter NSInstPostDealView::post %s, %s", request.data, ns_instance_id)
 167         ns_post_status = ignore_case_get(request.data, 'status')
 168         ns_status = 'ACTIVE' if ns_post_status == 'true' else 'FAILED'
 169         ns_opr_status = 'success' if ns_post_status == 'true' else 'failed'
 170         try:
 171             NSInstModel.objects.filter(id=ns_instance_id).update(status=ns_status)
 172             ServiceBaseInfoModel.objects.filter(service_id=ns_instance_id).update(
 173                 active_status=ns_status, status=ns_opr_status)
 174             nsd_info = NSInstModel.objects.filter(id=ns_instance_id)
 175             nsd_id = nsd_info[0].nsd_id
 176             nsd_model = json.loads(nsd_info[0].nsd_model)
 177             if "policies" in nsd_model and nsd_model["policies"]:
 178                 policy = nsd_model["policies"][0]
 179                 if "properties" in policy and policy["properties"]:
 180                     file_url = ignore_case_get(policy["properties"][0], "drl_file_url")
 181                 else:
 182                     file_url = ""
 183                 self.send_policy_request(ns_instance_id, nsd_id, file_url)
 184         except:
 185             logger.error(traceback.format_exc())
 186             return Response(data={'error': 'Failed to update status of NS(%s)' % ns_instance_id},
 187                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 188         logger.debug("*****NS INST %s, %s******", ns_status, ns_opr_status)
 189         return Response(data={'success': 'Update status of NS(%s) to %s' % (ns_instance_id, ns_status)},
 190                         status=status.HTTP_202_ACCEPTED)
 191
 192     def send_policy_request(self, ns_instance_id, nsd_id, file_url):
 193         input_data = {
 194             "nsid": ns_instance_id,
 195             "nsdid": nsd_id,
 196             "fileUri": file_url
 197         }
 198         req_param = json.JSONEncoder().encode(input_data)
 199         policy_engine_url = 'api/polengine/v1/policyinfo'
 200         ret = req_by_msb(policy_engine_url, "POST", req_param)
 201         if ret[0] != 0:
 202             logger.error("Failed to send ns policy req")
 203
 204
 205 class NSManualScaleView(APIView):
 206     def post(self, request, ns_instance_id):
 207         logger.debug("Enter NSManualScaleView::post %s, %s", request.data, ns_instance_id)
 208         job_id = JobUtil.create_job("NS", JOB_TYPE.MANUAL_SCALE_VNF, ns_instance_id)
 209         try:
 210             NSManualScaleService(ns_instance_id, request.data, job_id).start()
 211         except Exception as e:
 212             logger.error(traceback.format_exc())
 213             JobUtil.add_job_status(job_id, 255, 'NS scale failed: %s' % e.message)
 214             return Response(data={'error': 'NS scale failed: %s' % ns_instance_id},
 215                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 216         return Response(data={'jobId': job_id}, status=status.HTTP_202_ACCEPTED)