Add post deal of build-in workflow
[vfc/nfvo/lcm.git] / lcm / workflows / build_in.py
1 # Copyright 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 logging
15 import traceback
16 from threading import Thread
17 import time
18
19 from lcm.pub.utils.syscomm import fun_name
20 from lcm.pub.utils.values import ignore_case_get
21 from lcm.pub.utils import restcall
22 from lcm.pub.exceptions import NSLCMException
23
24 logger = logging.getLogger(__name__)
25
26 RESULT_OK, RESULT_NG = "0", "1"
27 JOB_ERROR = 255
28
29 g_jobs_status = {}
30
31 """
32 format of input_data
33 {
34     "jobId": uuid of job, 
35     "nsInstanceId": id of ns instance,
36     "object_context": json format of nsd,
37     "object_additionalParamForNs": json format of additional parameters for ns,
38     "object_additionalParamForVnf": json format of additional parameters for vnf,
39     "vlCount": int type of VL count,
40     "vnfCount: int type of VNF count,
41     "sfcCount": int type of SFC count, 
42     "sdnControllerId": uuid of SDN controller
43 }
44 """
45 def run_ns_instantiate(input_data):
46     logger.debug("Enter %s, input_data is %s", fun_name(), input_data)
47     job_id = ignore_case_get(input_data, "jobId")
48     ns_inst_id = ignore_case_get(input_data, "nsInstanceId")
49     nsd_json = ignore_case_get(input_data, "object_context")
50     ns_param_json = ignore_case_get(input_data, "object_additionalParamForNs")
51     vnf_param_json = ignore_case_get(input_data, "object_additionalParamForVnf")
52     vl_count = ignore_case_get(input_data, "vlCount")
53     vnf_count = ignore_case_get(input_data, "vnfCount")
54     sfc_count = ignore_case_get(input_data, "sfcCount")
55     sdnc_id = ignore_case_get(input_data, "sdnControllerId")
56     g_jobs_status[job_id] = [1 for i in range(vnf_count)]
57     try:
58         update_job(job_id, 10, "0", "Start to create VL")
59         for i in range(vl_count):
60             create_vl(ns_inst_id, i + 1, nsd_json, ns_param_json)
61
62         update_job(job_id, 30, "0", "Start to create VNF")
63         jobs = [create_vnf(ns_inst_id, i + 1, vnf_param_json) for i in range(vnf_count)] 
64         wait_until_jobs_done(job_id, jobs)
65
66         update_job(job_id, 70, "0", "Start to create SFC")
67         jobs = [create_sfc(ns_inst_id, i + 1, nsd_json, sdnc_id) for i in range(sfc_count)] 
68         wait_until_jobs_done(job_id, jobs)
69
70         update_job(job_id, 90, "0", "Start to post deal")
71         post_deal(ns_inst_id, "true")
72
73         update_job(job_id, 100, "0", "Create NS successfully.")
74     except NSLCMException as e:
75         logger.error("Failded to Create NS: %s", e.message)
76         update_job(job_id, JOB_ERROR, "255", "Failded to Create NS.")
77         post_deal(ns_inst_id, "false")
78     except:
79         logger.error(traceback.format_exc())
80         update_job(job_id, JOB_ERROR, "255", "Failded to Create NS.")
81         post_deal(ns_inst_id, "false")
82     finally:
83         g_jobs_status.pop(job_id)
84
85
86 def create_vl(ns_inst_id, vl_index, nsd, ns_param):
87     uri = "api/nslcm/v1/ns/vls"
88     data = json.JSONEncoder().encode({
89         "nsInstanceId": ns_inst_id,
90         "vlIndex": vl_index,
91         "context": nsd,
92         "additionalParamForNs": ns_param
93     })
94
95     ret = restcall.req_by_msb(uri, "POST", data)
96     if ret[0] != 0:
97         logger.error("Failed to call create_vl(%s): %s", vl_index, ret[1])
98         raise NSLCMException("Failed to call create_vl(index is %s)" % vl_index)
99
100     result = str(ret[1]["result"])
101     detail = ret[1]["detail"]
102     vl_id = ret[1]["vlId"]
103     if result != RESULT_OK:
104         logger.error("Failed to create VL(%s): %s", vl_id, detail)
105         raise NSLCMException("Failed to create VL(%s)" % vl_id)
106
107     logger.debug("Create VL(%s) successfully.", vl_id)
108
109 def create_vnf(ns_inst_id, vnf_index, nf_param):
110     uri = "/ns/vnfs"
111     data = json.JSONEncoder().encode({
112         "nsInstanceId": ns_inst_id,
113         "vnfIndex": vnf_index,
114         "additionalParamForVnf": nf_param
115     })
116
117     ret = restcall.req_by_msb(uri, "POST", data)
118     if ret[0] != 0:
119         logger.error("Failed to call create_vnf(%s): %s", vnf_index, ret[1])
120         raise NSLCMException("Failed to call create_vnf(index is %s)" % vnf_index)
121
122     vnf_inst_id = ret[1]["vnfInstId"]
123     job_id = ret[1]["jobId"]
124     logger.debug("Create VNF(%s) started.", vnf_inst_id)
125     return vnf_inst_id, job_id, vnf_index - 1
126
127 def create_sfc(ns_inst_id, fp_index, nsd_json, sdnc_id):
128     uri = "/ns/sfcs"
129     data = json.JSONEncoder().encode({
130         "nsInstanceId": ns_inst_id,
131         "context": nsd_json,
132         "fpindex": fp_index,
133         "sdnControllerId": sdnc_id
134     })
135
136     ret = restcall.req_by_msb(uri, "POST", data)
137     if ret[0] != 0:
138         logger.error("Failed to call create_sfc(%s): %s", fp_index, ret[1])
139         raise NSLCMException("Failed to call create_sfc(index is %s)" % fp_index)
140
141     sfc_inst_id = ret[1]["sfcInstId"]
142     job_id = ret[1]["jobId"]
143     logger.debug("Create SFC(%s) started.", sfc_inst_id)
144     return sfc_inst_id, job_id, fp_index - 1
145
146 def post_deal(ns_inst_id, status):
147     uri = "/ns/{nsInstanceId}/postdeal".format(nsInstanceId=ns_inst_id) 
148     data = json.JSONEncoder().encode({
149         "status": status
150     })
151
152     ret = restcall.req_by_msb(uri, "POST", data)
153     if ret[0] != 0:
154         logger.error("Failed to call post_deal(%s): %s", ns_inst_id, ret[1])
155     logger.debug("Call post_deal(%s) successfully.", ns_inst_id)
156
157 def update_job(job_id, progress, errcode, desc):
158     uri = "api/nslcm/v1/jobs/{jobId}".format(jobId=job_id)
159     data = json.JSONEncoder().encode({
160         "progress": progress,
161         "errcode": errcode,
162         "desc": desc
163     })
164     restcall.req_by_msb(uri, "POST", data)  
165
166 class JobWaitThread(Thread):
167     """
168     Job Wait 
169     """
170
171     def __init__(self, inst_id, job_id, index):
172         Thread.__init__(self)
173         self.inst_id = inst_id
174         self.job_id = job_id
175         self.index = index
176         self.retry_count = 60
177         self.interval_second = 3
178
179     def run(self):
180         count = 0
181         response_id, new_response_id = 0, 0
182         job_end_normal, job_timeout = False, True
183         while count < self.retry_count:
184             count = count + 1
185             time.sleep(self.interval_second)
186             uri = "/api/nslcm/v1/jobs/%s?responseId=%s" % (self.job_id, response_id)
187             ret = restcall.req_by_msb(uri, "GET")
188             if ret[0] != 0:
189                 logger.error("Failed to query job: %s:%s", ret[2], ret[1])
190                 continue
191             job_result = json.JSONDecoder().decode(ret[1])
192             if "responseDescriptor" not in job_result:
193                 logger.error("Job(%s) does not exist.", self.job_id)
194                 continue
195             progress = job_result["responseDescriptor"]["progress"]
196             new_response_id = job_result["responseDescriptor"]["responseId"]
197             job_desc = job_result["responseDescriptor"]["statusDescription"]
198             if new_response_id != response_id:
199                 logger.debug("%s:%s:%s", progress, new_response_id, job_desc)
200                 response_id = new_response_id
201                 count = 0
202             if progress == JOB_ERROR:
203                 job_timeout = False
204                 logger.error("Job(%s) failed: %s", self.job_id, job_desc)
205                 break
206             elif progress == 100:
207                 job_end_normal, job_timeout = True, False
208                 logger.info("Job(%s) ended normally", self.job_id)
209                 break
210         if job_timeout:
211             logger.error("Job(%s) timeout", self.job_id)
212         if self.job_id in g_jobs_status:
213             if job_end_normal:
214                 g_jobs_status[self.job_id][self.index] = 0
215
216 def wait_until_jobs_done(g_job_id, jobs):
217     job_threads = []
218     for inst_id, job_id, index in jobs:
219         job_threads.append(JobWaitThread(inst_id, job_id, index))
220     for t in job_threads:
221         t.start()
222     for t in job_threads:
223         t.join()
224     if g_job_id in g_jobs_status:
225         if sum(g_jobs_status[g_job_id]) > 0:
226             raise NSLCMException("Some jobs failed!")
227       
228