blob: 80d4aa7335b2a7053712d66bc2ed9b2d40a0ccb6 [file] [log] [blame]
tierno1d213f42020-04-24 14:02:51 +00001# -*- coding: utf-8 -*-
2
3##
4# Copyright 2020 Telefonica Investigacion y Desarrollo, S.A.U.
5#
6# Licensed under the Apache License, Version 2.0 (the "License"); you may
7# not use this file except in compliance with the License. You may obtain
8# a copy of the License at
9#
10# http://www.apache.org/licenses/LICENSE-2.0
11#
12# Unless required by applicable law or agreed to in writing, software
13# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
14# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
15# License for the specific language governing permissions and limitations
16# under the License.
17#
18##
19
20""""
21This is thread that interacts with a VIM. It processes TASKs sequentially against a single VIM.
22The tasks are stored at database in table ro_tasks
23A single ro_task refers to a VIM element (flavor, image, network, ...).
24A ro_task can contain several 'tasks', each one with a target, where to store the results
25"""
26
sousaedu80135b92021-02-17 15:05:18 +010027from copy import deepcopy
28from http import HTTPStatus
sousaedu049cbb12022-01-05 11:39:35 +000029import logging
sousaedu80135b92021-02-17 15:05:18 +010030from os import mkdir
sousaedu049cbb12022-01-05 11:39:35 +000031import queue
sousaedu80135b92021-02-17 15:05:18 +010032from shutil import rmtree
sousaedu049cbb12022-01-05 11:39:35 +000033import threading
34import time
aticige5d78422022-05-16 23:03:54 +030035import traceback
aticig973bed22022-06-30 19:29:04 +030036from typing import Dict
sousaedu80135b92021-02-17 15:05:18 +010037from unittest.mock import Mock
38
sousaedu049cbb12022-01-05 11:39:35 +000039from importlib_metadata import entry_points
tierno1d213f42020-04-24 14:02:51 +000040from osm_common.dbbase import DbException
tiernof1b640f2020-12-09 15:06:01 +000041from osm_ng_ro.vim_admin import LockRenew
sousaedu049cbb12022-01-05 11:39:35 +000042from osm_ro_plugin import sdnconn, vimconn
43from osm_ro_plugin.sdn_dummy import SdnDummyConnector
44from osm_ro_plugin.vim_dummy import VimDummyConnector
45import yaml
tierno1d213f42020-04-24 14:02:51 +000046
47__author__ = "Alfonso Tierno"
48__date__ = "$28-Sep-2017 12:07:15$"
49
50
51def deep_get(target_dict, *args, **kwargs):
52 """
53 Get a value from target_dict entering in the nested keys. If keys does not exist, it returns None
54 Example target_dict={a: {b: 5}}; key_list=[a,b] returns 5; both key_list=[a,b,c] and key_list=[f,h] return None
55 :param target_dict: dictionary to be read
56 :param args: list of keys to read from target_dict
57 :param kwargs: only can contain default=value to return if key is not present in the nested dictionary
58 :return: The wanted value if exist, None or default otherwise
59 """
60 for key in args:
61 if not isinstance(target_dict, dict) or key not in target_dict:
62 return kwargs.get("default")
63 target_dict = target_dict[key]
64 return target_dict
65
66
67class NsWorkerException(Exception):
68 pass
69
70
71class FailingConnector:
72 def __init__(self, error_msg):
73 self.error_msg = error_msg
sousaedu80135b92021-02-17 15:05:18 +010074
tierno1d213f42020-04-24 14:02:51 +000075 for method in dir(vimconn.VimConnector):
76 if method[0] != "_":
sousaedu80135b92021-02-17 15:05:18 +010077 setattr(
78 self, method, Mock(side_effect=vimconn.VimConnException(error_msg))
79 )
80
tierno70eeb182020-10-19 16:38:00 +000081 for method in dir(sdnconn.SdnConnectorBase):
82 if method[0] != "_":
sousaedu80135b92021-02-17 15:05:18 +010083 setattr(
84 self, method, Mock(side_effect=sdnconn.SdnConnectorError(error_msg))
85 )
tierno1d213f42020-04-24 14:02:51 +000086
87
88class NsWorkerExceptionNotFound(NsWorkerException):
89 pass
90
91
tierno70eeb182020-10-19 16:38:00 +000092class VimInteractionBase:
sousaedu80135b92021-02-17 15:05:18 +010093 """Base class to call VIM/SDN for creating, deleting and refresh networks, VMs, flavors, ...
tierno70eeb182020-10-19 16:38:00 +000094 It implements methods that does nothing and return ok"""
sousaedu80135b92021-02-17 15:05:18 +010095
tierno70eeb182020-10-19 16:38:00 +000096 def __init__(self, db, my_vims, db_vims, logger):
tierno1d213f42020-04-24 14:02:51 +000097 self.db = db
tierno70eeb182020-10-19 16:38:00 +000098 self.logger = logger
99 self.my_vims = my_vims
100 self.db_vims = db_vims
tierno1d213f42020-04-24 14:02:51 +0000101
tierno70eeb182020-10-19 16:38:00 +0000102 def new(self, ro_task, task_index, task_depends):
103 return "BUILD", {}
tierno1d213f42020-04-24 14:02:51 +0000104
tierno70eeb182020-10-19 16:38:00 +0000105 def refresh(self, ro_task):
106 """skip calling VIM to get image, flavor status. Assumes ok"""
tierno1d213f42020-04-24 14:02:51 +0000107 if ro_task["vim_info"]["vim_status"] == "VIM_ERROR":
108 return "FAILED", {}
sousaedu80135b92021-02-17 15:05:18 +0100109
tierno1d213f42020-04-24 14:02:51 +0000110 return "DONE", {}
111
tierno70eeb182020-10-19 16:38:00 +0000112 def delete(self, ro_task, task_index):
113 """skip calling VIM to delete image. Assumes ok"""
tierno1d213f42020-04-24 14:02:51 +0000114 return "DONE", {}
115
tierno70eeb182020-10-19 16:38:00 +0000116 def exec(self, ro_task, task_index, task_depends):
117 return "DONE", None, None
tierno1d213f42020-04-24 14:02:51 +0000118
tierno1d213f42020-04-24 14:02:51 +0000119
tierno70eeb182020-10-19 16:38:00 +0000120class VimInteractionNet(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +0000121 def new(self, ro_task, task_index, task_depends):
tierno1d213f42020-04-24 14:02:51 +0000122 vim_net_id = None
123 task = ro_task["tasks"][task_index]
124 task_id = task["task_id"]
125 created = False
126 created_items = {}
127 target_vim = self.my_vims[ro_task["target_id"]]
aticige9a26f22021-12-10 12:59:20 +0300128 mgmtnet = False
129 mgmtnet_defined_in_vim = False
sousaedu80135b92021-02-17 15:05:18 +0100130
tierno1d213f42020-04-24 14:02:51 +0000131 try:
132 # FIND
133 if task.get("find_params"):
134 # if management, get configuration of VIM
135 if task["find_params"].get("filter_dict"):
136 vim_filter = task["find_params"]["filter_dict"]
aticige9a26f22021-12-10 12:59:20 +0300137 # management network
sousaedu80135b92021-02-17 15:05:18 +0100138 elif task["find_params"].get("mgmt"):
aticige9a26f22021-12-10 12:59:20 +0300139 mgmtnet = True
sousaedu80135b92021-02-17 15:05:18 +0100140 if deep_get(
141 self.db_vims[ro_task["target_id"]],
142 "config",
143 "management_network_id",
144 ):
aticige9a26f22021-12-10 12:59:20 +0300145 mgmtnet_defined_in_vim = True
sousaedu80135b92021-02-17 15:05:18 +0100146 vim_filter = {
147 "id": self.db_vims[ro_task["target_id"]]["config"][
148 "management_network_id"
149 ]
150 }
151 elif deep_get(
152 self.db_vims[ro_task["target_id"]],
153 "config",
154 "management_network_name",
155 ):
aticige9a26f22021-12-10 12:59:20 +0300156 mgmtnet_defined_in_vim = True
sousaedu80135b92021-02-17 15:05:18 +0100157 vim_filter = {
158 "name": self.db_vims[ro_task["target_id"]]["config"][
159 "management_network_name"
160 ]
161 }
tierno1d213f42020-04-24 14:02:51 +0000162 else:
163 vim_filter = {"name": task["find_params"]["name"]}
164 else:
sousaedu80135b92021-02-17 15:05:18 +0100165 raise NsWorkerExceptionNotFound(
166 "Invalid find_params for new_net {}".format(task["find_params"])
167 )
tierno1d213f42020-04-24 14:02:51 +0000168
169 vim_nets = target_vim.get_network_list(vim_filter)
170 if not vim_nets and not task.get("params"):
aticige9a26f22021-12-10 12:59:20 +0300171 # If there is mgmt-network in the descriptor,
172 # there is no mapping of that network to a VIM network in the descriptor,
173 # also there is no mapping in the "--config" parameter or at VIM creation;
174 # that mgmt-network will be created.
175 if mgmtnet and not mgmtnet_defined_in_vim:
176 net_name = (
177 vim_filter.get("name")
178 if vim_filter.get("name")
179 else vim_filter.get("id")[:16]
sousaedu80135b92021-02-17 15:05:18 +0100180 )
aticige9a26f22021-12-10 12:59:20 +0300181 vim_net_id, created_items = target_vim.new_network(
182 net_name, None
183 )
184 self.logger.debug(
185 "Created mgmt network vim_net_id: {}".format(vim_net_id)
186 )
187 created = True
188 else:
189 raise NsWorkerExceptionNotFound(
190 "Network not found with this criteria: '{}'".format(
191 task.get("find_params")
192 )
193 )
tierno1d213f42020-04-24 14:02:51 +0000194 elif len(vim_nets) > 1:
195 raise NsWorkerException(
sousaedu80135b92021-02-17 15:05:18 +0100196 "More than one network found with this criteria: '{}'".format(
197 task["find_params"]
198 )
199 )
200
tierno1d213f42020-04-24 14:02:51 +0000201 if vim_nets:
202 vim_net_id = vim_nets[0]["id"]
203 else:
204 # CREATE
205 params = task["params"]
206 vim_net_id, created_items = target_vim.new_network(**params)
207 created = True
208
sousaedu80135b92021-02-17 15:05:18 +0100209 ro_vim_item_update = {
210 "vim_id": vim_net_id,
211 "vim_status": "BUILD",
212 "created": created,
213 "created_items": created_items,
214 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300215 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100216 }
tierno1d213f42020-04-24 14:02:51 +0000217 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100218 "task={} {} new-net={} created={}".format(
219 task_id, ro_task["target_id"], vim_net_id, created
220 )
221 )
222
tierno1d213f42020-04-24 14:02:51 +0000223 return "BUILD", ro_vim_item_update
224 except (vimconn.VimConnException, NsWorkerException) as e:
sousaedu80135b92021-02-17 15:05:18 +0100225 self.logger.error(
226 "task={} vim={} new-net: {}".format(task_id, ro_task["target_id"], e)
227 )
228 ro_vim_item_update = {
229 "vim_status": "VIM_ERROR",
230 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300231 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100232 }
233
tierno1d213f42020-04-24 14:02:51 +0000234 return "FAILED", ro_vim_item_update
235
tierno70eeb182020-10-19 16:38:00 +0000236 def refresh(self, ro_task):
tierno1d213f42020-04-24 14:02:51 +0000237 """Call VIM to get network status"""
238 ro_task_id = ro_task["_id"]
239 target_vim = self.my_vims[ro_task["target_id"]]
tierno1d213f42020-04-24 14:02:51 +0000240 vim_id = ro_task["vim_info"]["vim_id"]
241 net_to_refresh_list = [vim_id]
sousaedu80135b92021-02-17 15:05:18 +0100242
tierno1d213f42020-04-24 14:02:51 +0000243 try:
244 vim_dict = target_vim.refresh_nets_status(net_to_refresh_list)
245 vim_info = vim_dict[vim_id]
sousaedu80135b92021-02-17 15:05:18 +0100246
tierno1d213f42020-04-24 14:02:51 +0000247 if vim_info["status"] == "ACTIVE":
248 task_status = "DONE"
249 elif vim_info["status"] == "BUILD":
250 task_status = "BUILD"
251 else:
252 task_status = "FAILED"
253 except vimconn.VimConnException as e:
254 # Mark all tasks at VIM_ERROR status
sousaedu80135b92021-02-17 15:05:18 +0100255 self.logger.error(
256 "ro_task={} vim={} get-net={}: {}".format(
257 ro_task_id, ro_task["target_id"], vim_id, e
258 )
259 )
tierno1d213f42020-04-24 14:02:51 +0000260 vim_info = {"status": "VIM_ERROR", "error_msg": str(e)}
261 task_status = "FAILED"
262
263 ro_vim_item_update = {}
264 if ro_task["vim_info"]["vim_status"] != vim_info["status"]:
265 ro_vim_item_update["vim_status"] = vim_info["status"]
sousaedu80135b92021-02-17 15:05:18 +0100266
tierno1d213f42020-04-24 14:02:51 +0000267 if ro_task["vim_info"]["vim_name"] != vim_info.get("name"):
268 ro_vim_item_update["vim_name"] = vim_info.get("name")
sousaedu80135b92021-02-17 15:05:18 +0100269
tierno1d213f42020-04-24 14:02:51 +0000270 if vim_info["status"] in ("ERROR", "VIM_ERROR"):
aticig79ac6df2022-05-06 16:09:52 +0300271 if ro_task["vim_info"]["vim_message"] != vim_info.get("error_msg"):
272 ro_vim_item_update["vim_message"] = vim_info.get("error_msg")
tierno1d213f42020-04-24 14:02:51 +0000273 elif vim_info["status"] == "DELETED":
274 ro_vim_item_update["vim_id"] = None
aticig79ac6df2022-05-06 16:09:52 +0300275 ro_vim_item_update["vim_message"] = "Deleted externally"
tierno1d213f42020-04-24 14:02:51 +0000276 else:
277 if ro_task["vim_info"]["vim_details"] != vim_info["vim_info"]:
278 ro_vim_item_update["vim_details"] = vim_info["vim_info"]
sousaedu80135b92021-02-17 15:05:18 +0100279
tierno1d213f42020-04-24 14:02:51 +0000280 if ro_vim_item_update:
sousaedu80135b92021-02-17 15:05:18 +0100281 self.logger.debug(
282 "ro_task={} {} get-net={}: status={} {}".format(
283 ro_task_id,
284 ro_task["target_id"],
285 vim_id,
286 ro_vim_item_update.get("vim_status"),
aticig79ac6df2022-05-06 16:09:52 +0300287 ro_vim_item_update.get("vim_message")
sousaedu80135b92021-02-17 15:05:18 +0100288 if ro_vim_item_update.get("vim_status") != "ACTIVE"
289 else "",
290 )
291 )
292
tierno1d213f42020-04-24 14:02:51 +0000293 return task_status, ro_vim_item_update
294
tierno70eeb182020-10-19 16:38:00 +0000295 def delete(self, ro_task, task_index):
tierno1d213f42020-04-24 14:02:51 +0000296 task = ro_task["tasks"][task_index]
297 task_id = task["task_id"]
298 net_vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100299 ro_vim_item_update_ok = {
300 "vim_status": "DELETED",
301 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300302 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +0100303 "vim_id": None,
304 }
305
tierno1d213f42020-04-24 14:02:51 +0000306 try:
307 if net_vim_id or ro_task["vim_info"]["created_items"]:
308 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100309 target_vim.delete_network(
310 net_vim_id, ro_task["vim_info"]["created_items"]
311 )
tierno1d213f42020-04-24 14:02:51 +0000312 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300313 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno1d213f42020-04-24 14:02:51 +0000314 except vimconn.VimConnException as e:
sousaedu80135b92021-02-17 15:05:18 +0100315 self.logger.error(
316 "ro_task={} vim={} del-net={}: {}".format(
317 ro_task["_id"], ro_task["target_id"], net_vim_id, e
318 )
319 )
320 ro_vim_item_update = {
321 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300322 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +0100323 }
324
tierno1d213f42020-04-24 14:02:51 +0000325 return "FAILED", ro_vim_item_update
326
sousaedu80135b92021-02-17 15:05:18 +0100327 self.logger.debug(
328 "task={} {} del-net={} {}".format(
329 task_id,
330 ro_task["target_id"],
331 net_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300332 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +0100333 )
334 )
335
tierno1d213f42020-04-24 14:02:51 +0000336 return "DONE", ro_vim_item_update_ok
337
tierno70eeb182020-10-19 16:38:00 +0000338
339class VimInteractionVdu(VimInteractionBase):
sousaedu80135b92021-02-17 15:05:18 +0100340 max_retries_inject_ssh_key = 20 # 20 times
341 time_retries_inject_ssh_key = 30 # wevery 30 seconds
tierno70eeb182020-10-19 16:38:00 +0000342
343 def new(self, ro_task, task_index, task_depends):
tierno1d213f42020-04-24 14:02:51 +0000344 task = ro_task["tasks"][task_index]
345 task_id = task["task_id"]
346 created = False
347 created_items = {}
348 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100349
tierno1d213f42020-04-24 14:02:51 +0000350 try:
351 created = True
352 params = task["params"]
353 params_copy = deepcopy(params)
354 net_list = params_copy["net_list"]
sousaedu80135b92021-02-17 15:05:18 +0100355
tierno1d213f42020-04-24 14:02:51 +0000356 for net in net_list:
sousaedu80135b92021-02-17 15:05:18 +0100357 # change task_id into network_id
358 if "net_id" in net and net["net_id"].startswith("TASK-"):
tierno1d213f42020-04-24 14:02:51 +0000359 network_id = task_depends[net["net_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100360
tierno1d213f42020-04-24 14:02:51 +0000361 if not network_id:
sousaedu80135b92021-02-17 15:05:18 +0100362 raise NsWorkerException(
363 "Cannot create VM because depends on a network not created or found "
364 "for {}".format(net["net_id"])
365 )
366
tierno1d213f42020-04-24 14:02:51 +0000367 net["net_id"] = network_id
sousaedu80135b92021-02-17 15:05:18 +0100368
tierno1d213f42020-04-24 14:02:51 +0000369 if params_copy["image_id"].startswith("TASK-"):
370 params_copy["image_id"] = task_depends[params_copy["image_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100371
tierno1d213f42020-04-24 14:02:51 +0000372 if params_copy["flavor_id"].startswith("TASK-"):
373 params_copy["flavor_id"] = task_depends[params_copy["flavor_id"]]
374
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100375 affinity_group_list = params_copy["affinity_group_list"]
376 for affinity_group in affinity_group_list:
377 # change task_id into affinity_group_id
378 if "affinity_group_id" in affinity_group and affinity_group[
379 "affinity_group_id"
380 ].startswith("TASK-"):
381 affinity_group_id = task_depends[
382 affinity_group["affinity_group_id"]
383 ]
384
385 if not affinity_group_id:
386 raise NsWorkerException(
387 "found for {}".format(affinity_group["affinity_group_id"])
388 )
389
390 affinity_group["affinity_group_id"] = affinity_group_id
391
tierno1d213f42020-04-24 14:02:51 +0000392 vim_vm_id, created_items = target_vim.new_vminstance(**params_copy)
393 interfaces = [iface["vim_id"] for iface in params_copy["net_list"]]
394
palaciosj42eb06f2022-05-05 14:59:36 +0000395 # add to created items previous_created_volumes (healing)
396 if task.get("previous_created_volumes"):
397 for k, v in task["previous_created_volumes"].items():
398 created_items[k] = v
399
sousaedu80135b92021-02-17 15:05:18 +0100400 ro_vim_item_update = {
401 "vim_id": vim_vm_id,
402 "vim_status": "BUILD",
403 "created": created,
404 "created_items": created_items,
405 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300406 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100407 "interfaces_vim_ids": interfaces,
408 "interfaces": [],
aticig37ecec02022-05-25 03:12:36 +0300409 "interfaces_backup": [],
sousaedu80135b92021-02-17 15:05:18 +0100410 }
tierno1d213f42020-04-24 14:02:51 +0000411 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100412 "task={} {} new-vm={} created={}".format(
413 task_id, ro_task["target_id"], vim_vm_id, created
414 )
415 )
416
tierno1d213f42020-04-24 14:02:51 +0000417 return "BUILD", ro_vim_item_update
418 except (vimconn.VimConnException, NsWorkerException) as e:
aticigcf14bb12022-05-19 13:03:17 +0300419 self.logger.debug(traceback.format_exc())
sousaedu80135b92021-02-17 15:05:18 +0100420 self.logger.error(
421 "task={} {} new-vm: {}".format(task_id, ro_task["target_id"], e)
422 )
423 ro_vim_item_update = {
424 "vim_status": "VIM_ERROR",
425 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300426 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100427 }
428
tierno1d213f42020-04-24 14:02:51 +0000429 return "FAILED", ro_vim_item_update
430
tierno70eeb182020-10-19 16:38:00 +0000431 def delete(self, ro_task, task_index):
tierno1d213f42020-04-24 14:02:51 +0000432 task = ro_task["tasks"][task_index]
433 task_id = task["task_id"]
434 vm_vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100435 ro_vim_item_update_ok = {
436 "vim_status": "DELETED",
437 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300438 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +0100439 "vim_id": None,
440 }
441
tierno1d213f42020-04-24 14:02:51 +0000442 try:
palaciosj8f2060b2022-02-24 12:05:59 +0000443 self.logger.debug(
444 "delete_vminstance: vm_vim_id={} created_items={}".format(
445 vm_vim_id, ro_task["vim_info"]["created_items"]
446 )
447 )
tierno1d213f42020-04-24 14:02:51 +0000448 if vm_vim_id or ro_task["vim_info"]["created_items"]:
449 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100450 target_vim.delete_vminstance(
palaciosj8f2060b2022-02-24 12:05:59 +0000451 vm_vim_id,
452 ro_task["vim_info"]["created_items"],
453 ro_task["vim_info"].get("volumes_to_hold", []),
sousaedu80135b92021-02-17 15:05:18 +0100454 )
tierno1d213f42020-04-24 14:02:51 +0000455 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300456 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno1d213f42020-04-24 14:02:51 +0000457 except vimconn.VimConnException as e:
sousaedu80135b92021-02-17 15:05:18 +0100458 self.logger.error(
459 "ro_task={} vim={} del-vm={}: {}".format(
460 ro_task["_id"], ro_task["target_id"], vm_vim_id, e
461 )
462 )
463 ro_vim_item_update = {
464 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300465 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +0100466 }
467
tierno1d213f42020-04-24 14:02:51 +0000468 return "FAILED", ro_vim_item_update
469
sousaedu80135b92021-02-17 15:05:18 +0100470 self.logger.debug(
471 "task={} {} del-vm={} {}".format(
472 task_id,
473 ro_task["target_id"],
474 vm_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300475 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +0100476 )
477 )
478
tierno1d213f42020-04-24 14:02:51 +0000479 return "DONE", ro_vim_item_update_ok
480
tierno70eeb182020-10-19 16:38:00 +0000481 def refresh(self, ro_task):
tierno1d213f42020-04-24 14:02:51 +0000482 """Call VIM to get vm status"""
483 ro_task_id = ro_task["_id"]
484 target_vim = self.my_vims[ro_task["target_id"]]
tierno1d213f42020-04-24 14:02:51 +0000485 vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100486
tierno1d213f42020-04-24 14:02:51 +0000487 if not vim_id:
488 return None, None
sousaedu80135b92021-02-17 15:05:18 +0100489
tierno1d213f42020-04-24 14:02:51 +0000490 vm_to_refresh_list = [vim_id]
491 try:
492 vim_dict = target_vim.refresh_vms_status(vm_to_refresh_list)
493 vim_info = vim_dict[vim_id]
sousaedu80135b92021-02-17 15:05:18 +0100494
tierno1d213f42020-04-24 14:02:51 +0000495 if vim_info["status"] == "ACTIVE":
496 task_status = "DONE"
497 elif vim_info["status"] == "BUILD":
498 task_status = "BUILD"
499 else:
500 task_status = "FAILED"
sousaedu80135b92021-02-17 15:05:18 +0100501
tierno70eeb182020-10-19 16:38:00 +0000502 # try to load and parse vim_information
503 try:
504 vim_info_info = yaml.safe_load(vim_info["vim_info"])
505 if vim_info_info.get("name"):
506 vim_info["name"] = vim_info_info["name"]
aticig78ac0cf2022-07-15 00:43:09 +0300507 except Exception as vim_info_error:
508 self.logger.exception(
509 f"{vim_info_error} occured while getting the vim_info from yaml"
510 )
tierno1d213f42020-04-24 14:02:51 +0000511 except vimconn.VimConnException as e:
512 # Mark all tasks at VIM_ERROR status
sousaedu80135b92021-02-17 15:05:18 +0100513 self.logger.error(
514 "ro_task={} vim={} get-vm={}: {}".format(
515 ro_task_id, ro_task["target_id"], vim_id, e
516 )
517 )
tierno1d213f42020-04-24 14:02:51 +0000518 vim_info = {"status": "VIM_ERROR", "error_msg": str(e)}
519 task_status = "FAILED"
520
521 ro_vim_item_update = {}
sousaedu80135b92021-02-17 15:05:18 +0100522
tierno70eeb182020-10-19 16:38:00 +0000523 # Interfaces cannot be present if e.g. VM is not present, that is status=DELETED
tierno1d213f42020-04-24 14:02:51 +0000524 vim_interfaces = []
tierno70eeb182020-10-19 16:38:00 +0000525 if vim_info.get("interfaces"):
526 for vim_iface_id in ro_task["vim_info"]["interfaces_vim_ids"]:
sousaedu80135b92021-02-17 15:05:18 +0100527 iface = next(
528 (
529 iface
530 for iface in vim_info["interfaces"]
531 if vim_iface_id == iface["vim_interface_id"]
532 ),
533 None,
534 )
tierno70eeb182020-10-19 16:38:00 +0000535 # if iface:
536 # iface.pop("vim_info", None)
537 vim_interfaces.append(iface)
tierno1d213f42020-04-24 14:02:51 +0000538
sousaedu80135b92021-02-17 15:05:18 +0100539 task_create = next(
540 t
541 for t in ro_task["tasks"]
542 if t and t["action"] == "CREATE" and t["status"] != "FINISHED"
543 )
tierno70eeb182020-10-19 16:38:00 +0000544 if vim_interfaces and task_create.get("mgmt_vnf_interface") is not None:
sousaedu80135b92021-02-17 15:05:18 +0100545 vim_interfaces[task_create["mgmt_vnf_interface"]][
546 "mgmt_vnf_interface"
547 ] = True
548
549 mgmt_vdu_iface = task_create.get(
550 "mgmt_vdu_interface", task_create.get("mgmt_vnf_interface", 0)
551 )
tierno70eeb182020-10-19 16:38:00 +0000552 if vim_interfaces:
553 vim_interfaces[mgmt_vdu_iface]["mgmt_vdu_interface"] = True
tierno1d213f42020-04-24 14:02:51 +0000554
555 if ro_task["vim_info"]["interfaces"] != vim_interfaces:
556 ro_vim_item_update["interfaces"] = vim_interfaces
sousaedu80135b92021-02-17 15:05:18 +0100557
tierno1d213f42020-04-24 14:02:51 +0000558 if ro_task["vim_info"]["vim_status"] != vim_info["status"]:
559 ro_vim_item_update["vim_status"] = vim_info["status"]
sousaedu80135b92021-02-17 15:05:18 +0100560
tierno1d213f42020-04-24 14:02:51 +0000561 if ro_task["vim_info"]["vim_name"] != vim_info.get("name"):
562 ro_vim_item_update["vim_name"] = vim_info.get("name")
sousaedu80135b92021-02-17 15:05:18 +0100563
tierno1d213f42020-04-24 14:02:51 +0000564 if vim_info["status"] in ("ERROR", "VIM_ERROR"):
aticig79ac6df2022-05-06 16:09:52 +0300565 if ro_task["vim_info"]["vim_message"] != vim_info.get("error_msg"):
566 ro_vim_item_update["vim_message"] = vim_info.get("error_msg")
tierno1d213f42020-04-24 14:02:51 +0000567 elif vim_info["status"] == "DELETED":
568 ro_vim_item_update["vim_id"] = None
aticig79ac6df2022-05-06 16:09:52 +0300569 ro_vim_item_update["vim_message"] = "Deleted externally"
tierno1d213f42020-04-24 14:02:51 +0000570 else:
571 if ro_task["vim_info"]["vim_details"] != vim_info["vim_info"]:
572 ro_vim_item_update["vim_details"] = vim_info["vim_info"]
sousaedu80135b92021-02-17 15:05:18 +0100573
tierno1d213f42020-04-24 14:02:51 +0000574 if ro_vim_item_update:
sousaedu80135b92021-02-17 15:05:18 +0100575 self.logger.debug(
576 "ro_task={} {} get-vm={}: status={} {}".format(
577 ro_task_id,
578 ro_task["target_id"],
579 vim_id,
580 ro_vim_item_update.get("vim_status"),
aticig79ac6df2022-05-06 16:09:52 +0300581 ro_vim_item_update.get("vim_message")
sousaedu80135b92021-02-17 15:05:18 +0100582 if ro_vim_item_update.get("vim_status") != "ACTIVE"
583 else "",
584 )
585 )
586
tierno1d213f42020-04-24 14:02:51 +0000587 return task_status, ro_vim_item_update
588
tierno70eeb182020-10-19 16:38:00 +0000589 def exec(self, ro_task, task_index, task_depends):
tierno1d213f42020-04-24 14:02:51 +0000590 task = ro_task["tasks"][task_index]
591 task_id = task["task_id"]
592 target_vim = self.my_vims[ro_task["target_id"]]
tierno70eeb182020-10-19 16:38:00 +0000593 db_task_update = {"retries": 0}
594 retries = task.get("retries", 0)
sousaedu80135b92021-02-17 15:05:18 +0100595
tierno1d213f42020-04-24 14:02:51 +0000596 try:
597 params = task["params"]
598 params_copy = deepcopy(params)
sousaedu80135b92021-02-17 15:05:18 +0100599 params_copy["ro_key"] = self.db.decrypt(
600 params_copy.pop("private_key"),
601 params_copy.pop("schema_version"),
602 params_copy.pop("salt"),
603 )
tierno70eeb182020-10-19 16:38:00 +0000604 params_copy["ip_addr"] = params_copy.pop("ip_address")
tierno1d213f42020-04-24 14:02:51 +0000605 target_vim.inject_user_key(**params_copy)
606 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100607 "task={} {} action-vm=inject_key".format(task_id, ro_task["target_id"])
608 )
609
610 return (
611 "DONE",
612 None,
613 db_task_update,
614 ) # params_copy["key"]
tierno1d213f42020-04-24 14:02:51 +0000615 except (vimconn.VimConnException, NsWorkerException) as e:
tierno70eeb182020-10-19 16:38:00 +0000616 retries += 1
sousaedu80135b92021-02-17 15:05:18 +0100617
aticige5d78422022-05-16 23:03:54 +0300618 self.logger.debug(traceback.format_exc())
tierno70eeb182020-10-19 16:38:00 +0000619 if retries < self.max_retries_inject_ssh_key:
sousaedu80135b92021-02-17 15:05:18 +0100620 return (
621 "BUILD",
622 None,
623 {
624 "retries": retries,
625 "next_retry": self.time_retries_inject_ssh_key,
626 },
627 )
628
629 self.logger.error(
630 "task={} {} inject-ssh-key: {}".format(task_id, ro_task["target_id"], e)
631 )
aticig79ac6df2022-05-06 16:09:52 +0300632 ro_vim_item_update = {"vim_message": str(e)}
sousaedu80135b92021-02-17 15:05:18 +0100633
tierno70eeb182020-10-19 16:38:00 +0000634 return "FAILED", ro_vim_item_update, db_task_update
635
636
637class VimInteractionImage(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +0000638 def new(self, ro_task, task_index, task_depends):
639 task = ro_task["tasks"][task_index]
640 task_id = task["task_id"]
641 created = False
642 created_items = {}
643 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100644
tierno70eeb182020-10-19 16:38:00 +0000645 try:
646 # FIND
647 if task.get("find_params"):
648 vim_images = target_vim.get_image_list(**task["find_params"])
sousaedu80135b92021-02-17 15:05:18 +0100649
tierno70eeb182020-10-19 16:38:00 +0000650 if not vim_images:
sousaedu80135b92021-02-17 15:05:18 +0100651 raise NsWorkerExceptionNotFound(
652 "Image not found with this criteria: '{}'".format(
653 task["find_params"]
654 )
655 )
tierno70eeb182020-10-19 16:38:00 +0000656 elif len(vim_images) > 1:
657 raise NsWorkerException(
sousaeduee6a6202021-05-11 13:22:37 +0200658 "More than one image found with this criteria: '{}'".format(
sousaedu80135b92021-02-17 15:05:18 +0100659 task["find_params"]
660 )
661 )
tierno70eeb182020-10-19 16:38:00 +0000662 else:
663 vim_image_id = vim_images[0]["id"]
664
sousaedu80135b92021-02-17 15:05:18 +0100665 ro_vim_item_update = {
666 "vim_id": vim_image_id,
667 "vim_status": "DONE",
668 "created": created,
669 "created_items": created_items,
670 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300671 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100672 }
tierno70eeb182020-10-19 16:38:00 +0000673 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100674 "task={} {} new-image={} created={}".format(
675 task_id, ro_task["target_id"], vim_image_id, created
676 )
677 )
678
tierno70eeb182020-10-19 16:38:00 +0000679 return "DONE", ro_vim_item_update
680 except (NsWorkerException, vimconn.VimConnException) as e:
sousaedu80135b92021-02-17 15:05:18 +0100681 self.logger.error(
682 "task={} {} new-image: {}".format(task_id, ro_task["target_id"], e)
683 )
684 ro_vim_item_update = {
685 "vim_status": "VIM_ERROR",
686 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300687 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100688 }
689
tierno1d213f42020-04-24 14:02:51 +0000690 return "FAILED", ro_vim_item_update
691
tierno70eeb182020-10-19 16:38:00 +0000692
693class VimInteractionFlavor(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +0000694 def delete(self, ro_task, task_index):
695 task = ro_task["tasks"][task_index]
696 task_id = task["task_id"]
697 flavor_vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100698 ro_vim_item_update_ok = {
699 "vim_status": "DELETED",
700 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300701 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +0100702 "vim_id": None,
703 }
704
tierno70eeb182020-10-19 16:38:00 +0000705 try:
706 if flavor_vim_id:
707 target_vim = self.my_vims[ro_task["target_id"]]
708 target_vim.delete_flavor(flavor_vim_id)
tierno70eeb182020-10-19 16:38:00 +0000709 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300710 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno70eeb182020-10-19 16:38:00 +0000711 except vimconn.VimConnException as e:
sousaedu80135b92021-02-17 15:05:18 +0100712 self.logger.error(
713 "ro_task={} vim={} del-flavor={}: {}".format(
714 ro_task["_id"], ro_task["target_id"], flavor_vim_id, e
715 )
716 )
717 ro_vim_item_update = {
718 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300719 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +0100720 }
721
tierno70eeb182020-10-19 16:38:00 +0000722 return "FAILED", ro_vim_item_update
723
sousaedu80135b92021-02-17 15:05:18 +0100724 self.logger.debug(
725 "task={} {} del-flavor={} {}".format(
726 task_id,
727 ro_task["target_id"],
728 flavor_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300729 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +0100730 )
731 )
732
tierno70eeb182020-10-19 16:38:00 +0000733 return "DONE", ro_vim_item_update_ok
734
735 def new(self, ro_task, task_index, task_depends):
736 task = ro_task["tasks"][task_index]
737 task_id = task["task_id"]
738 created = False
739 created_items = {}
740 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100741
tierno70eeb182020-10-19 16:38:00 +0000742 try:
743 # FIND
744 vim_flavor_id = None
sousaedu80135b92021-02-17 15:05:18 +0100745
tierno70eeb182020-10-19 16:38:00 +0000746 if task.get("find_params"):
747 try:
748 flavor_data = task["find_params"]["flavor_data"]
749 vim_flavor_id = target_vim.get_flavor_id_from_data(flavor_data)
750 except vimconn.VimConnNotFoundException:
aticig78ac0cf2022-07-15 00:43:09 +0300751 self.logger.exception("VimConnNotFoundException occured.")
tierno70eeb182020-10-19 16:38:00 +0000752
753 if not vim_flavor_id and task.get("params"):
754 # CREATE
755 flavor_data = task["params"]["flavor_data"]
756 vim_flavor_id = target_vim.new_flavor(flavor_data)
757 created = True
758
sousaedu80135b92021-02-17 15:05:18 +0100759 ro_vim_item_update = {
760 "vim_id": vim_flavor_id,
761 "vim_status": "DONE",
762 "created": created,
763 "created_items": created_items,
764 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300765 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100766 }
tierno70eeb182020-10-19 16:38:00 +0000767 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100768 "task={} {} new-flavor={} created={}".format(
769 task_id, ro_task["target_id"], vim_flavor_id, created
770 )
771 )
772
tierno70eeb182020-10-19 16:38:00 +0000773 return "DONE", ro_vim_item_update
774 except (vimconn.VimConnException, NsWorkerException) as e:
sousaedu80135b92021-02-17 15:05:18 +0100775 self.logger.error(
776 "task={} vim={} new-flavor: {}".format(task_id, ro_task["target_id"], e)
777 )
778 ro_vim_item_update = {
779 "vim_status": "VIM_ERROR",
780 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300781 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100782 }
783
tierno70eeb182020-10-19 16:38:00 +0000784 return "FAILED", ro_vim_item_update
785
786
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100787class VimInteractionAffinityGroup(VimInteractionBase):
788 def delete(self, ro_task, task_index):
789 task = ro_task["tasks"][task_index]
790 task_id = task["task_id"]
791 affinity_group_vim_id = ro_task["vim_info"]["vim_id"]
792 ro_vim_item_update_ok = {
793 "vim_status": "DELETED",
794 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300795 "vim_message": "DELETED",
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100796 "vim_id": None,
797 }
798
799 try:
800 if affinity_group_vim_id:
801 target_vim = self.my_vims[ro_task["target_id"]]
802 target_vim.delete_affinity_group(affinity_group_vim_id)
803 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300804 ro_vim_item_update_ok["vim_message"] = "already deleted"
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100805 except vimconn.VimConnException as e:
806 self.logger.error(
807 "ro_task={} vim={} del-affinity-or-anti-affinity-group={}: {}".format(
808 ro_task["_id"], ro_task["target_id"], affinity_group_vim_id, e
809 )
810 )
811 ro_vim_item_update = {
812 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300813 "vim_message": "Error while deleting: {}".format(e),
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100814 }
815
816 return "FAILED", ro_vim_item_update
817
818 self.logger.debug(
819 "task={} {} del-affinity-or-anti-affinity-group={} {}".format(
820 task_id,
821 ro_task["target_id"],
822 affinity_group_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300823 ro_vim_item_update_ok.get("vim_message", ""),
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100824 )
825 )
826
827 return "DONE", ro_vim_item_update_ok
828
829 def new(self, ro_task, task_index, task_depends):
830 task = ro_task["tasks"][task_index]
831 task_id = task["task_id"]
832 created = False
833 created_items = {}
834 target_vim = self.my_vims[ro_task["target_id"]]
835
836 try:
837 affinity_group_vim_id = None
Alexis Romero123de182022-04-26 19:24:40 +0200838 affinity_group_data = None
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100839
840 if task.get("params"):
Alexis Romero123de182022-04-26 19:24:40 +0200841 affinity_group_data = task["params"].get("affinity_group_data")
842
843 if affinity_group_data and affinity_group_data.get("vim-affinity-group-id"):
844 try:
845 param_affinity_group_id = task["params"]["affinity_group_data"].get(
846 "vim-affinity-group-id"
847 )
848 affinity_group_vim_id = target_vim.get_affinity_group(
849 param_affinity_group_id
850 ).get("id")
851 except vimconn.VimConnNotFoundException:
852 self.logger.error(
853 "task={} {} new-affinity-or-anti-affinity-group. Provided VIM Affinity Group ID {}"
854 "could not be found at VIM. Creating a new one.".format(
855 task_id, ro_task["target_id"], param_affinity_group_id
856 )
857 )
858
859 if not affinity_group_vim_id and affinity_group_data:
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100860 affinity_group_vim_id = target_vim.new_affinity_group(
861 affinity_group_data
862 )
863 created = True
864
865 ro_vim_item_update = {
866 "vim_id": affinity_group_vim_id,
867 "vim_status": "DONE",
868 "created": created,
869 "created_items": created_items,
870 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300871 "vim_message": None,
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100872 }
873 self.logger.debug(
874 "task={} {} new-affinity-or-anti-affinity-group={} created={}".format(
875 task_id, ro_task["target_id"], affinity_group_vim_id, created
876 )
877 )
878
879 return "DONE", ro_vim_item_update
880 except (vimconn.VimConnException, NsWorkerException) as e:
881 self.logger.error(
882 "task={} vim={} new-affinity-or-anti-affinity-group:"
883 " {}".format(task_id, ro_task["target_id"], e)
884 )
885 ro_vim_item_update = {
886 "vim_status": "VIM_ERROR",
887 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300888 "vim_message": str(e),
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100889 }
890
891 return "FAILED", ro_vim_item_update
892
893
k4.rahul78f474e2022-05-02 15:47:57 +0000894class VimInteractionUpdateVdu(VimInteractionBase):
895 def exec(self, ro_task, task_index, task_depends):
896 task = ro_task["tasks"][task_index]
897 task_id = task["task_id"]
898 db_task_update = {"retries": 0}
899 created = False
900 created_items = {}
901 target_vim = self.my_vims[ro_task["target_id"]]
902
903 try:
904 if task.get("params"):
905 vim_vm_id = task["params"].get("vim_vm_id")
906 action = task["params"].get("action")
907 context = {action: action}
908 target_vim.action_vminstance(vim_vm_id, context)
909 # created = True
910 ro_vim_item_update = {
911 "vim_id": vim_vm_id,
912 "vim_status": "DONE",
913 "created": created,
914 "created_items": created_items,
915 "vim_details": None,
916 "vim_message": None,
917 }
918 self.logger.debug(
919 "task={} {} vm-migration done".format(task_id, ro_task["target_id"])
920 )
921 return "DONE", ro_vim_item_update, db_task_update
922 except (vimconn.VimConnException, NsWorkerException) as e:
923 self.logger.error(
924 "task={} vim={} VM Migration:"
925 " {}".format(task_id, ro_task["target_id"], e)
926 )
927 ro_vim_item_update = {
928 "vim_status": "VIM_ERROR",
929 "created": created,
930 "vim_message": str(e),
931 }
932
933 return "FAILED", ro_vim_item_update, db_task_update
934
935
tierno70eeb182020-10-19 16:38:00 +0000936class VimInteractionSdnNet(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +0000937 @staticmethod
938 def _match_pci(port_pci, mapping):
939 """
940 Check if port_pci matches with mapping
941 mapping can have brackets to indicate that several chars are accepted. e.g
942 pci '0000:af:10.1' matches with '0000:af:1[01].[1357]'
943 :param port_pci: text
944 :param mapping: text, can contain brackets to indicate several chars are available
945 :return: True if matches, False otherwise
946 """
947 if not port_pci or not mapping:
948 return False
949 if port_pci == mapping:
950 return True
951
952 mapping_index = 0
953 pci_index = 0
954 while True:
955 bracket_start = mapping.find("[", mapping_index)
sousaedu80135b92021-02-17 15:05:18 +0100956
tierno70eeb182020-10-19 16:38:00 +0000957 if bracket_start == -1:
958 break
sousaedu80135b92021-02-17 15:05:18 +0100959
tierno70eeb182020-10-19 16:38:00 +0000960 bracket_end = mapping.find("]", bracket_start)
961 if bracket_end == -1:
962 break
sousaedu80135b92021-02-17 15:05:18 +0100963
tierno70eeb182020-10-19 16:38:00 +0000964 length = bracket_start - mapping_index
sousaedu80135b92021-02-17 15:05:18 +0100965 if (
966 length
967 and port_pci[pci_index : pci_index + length]
968 != mapping[mapping_index:bracket_start]
969 ):
tierno70eeb182020-10-19 16:38:00 +0000970 return False
sousaedu80135b92021-02-17 15:05:18 +0100971
972 if (
973 port_pci[pci_index + length]
974 not in mapping[bracket_start + 1 : bracket_end]
975 ):
tierno70eeb182020-10-19 16:38:00 +0000976 return False
sousaedu80135b92021-02-17 15:05:18 +0100977
tierno70eeb182020-10-19 16:38:00 +0000978 pci_index += length + 1
979 mapping_index = bracket_end + 1
980
981 if port_pci[pci_index:] != mapping[mapping_index:]:
982 return False
sousaedu80135b92021-02-17 15:05:18 +0100983
tierno70eeb182020-10-19 16:38:00 +0000984 return True
985
986 def _get_interfaces(self, vlds_to_connect, vim_account_id):
987 """
988 :param vlds_to_connect: list with format vnfrs:<id>:vld.<vld_id> or nsrs:<id>:vld.<vld_id>
989 :param vim_account_id:
990 :return:
991 """
992 interfaces = []
sousaedu80135b92021-02-17 15:05:18 +0100993
tierno70eeb182020-10-19 16:38:00 +0000994 for vld in vlds_to_connect:
995 table, _, db_id = vld.partition(":")
996 db_id, _, vld = db_id.partition(":")
997 _, _, vld_id = vld.partition(".")
sousaedu80135b92021-02-17 15:05:18 +0100998
tierno70eeb182020-10-19 16:38:00 +0000999 if table == "vnfrs":
1000 q_filter = {"vim-account-id": vim_account_id, "_id": db_id}
1001 iface_key = "vnf-vld-id"
1002 else: # table == "nsrs"
1003 q_filter = {"vim-account-id": vim_account_id, "nsr-id-ref": db_id}
1004 iface_key = "ns-vld-id"
sousaedu80135b92021-02-17 15:05:18 +01001005
tierno70eeb182020-10-19 16:38:00 +00001006 db_vnfrs = self.db.get_list("vnfrs", q_filter=q_filter)
sousaedu80135b92021-02-17 15:05:18 +01001007
tierno70eeb182020-10-19 16:38:00 +00001008 for db_vnfr in db_vnfrs:
1009 for vdu_index, vdur in enumerate(db_vnfr.get("vdur", ())):
1010 for iface_index, interface in enumerate(vdur["interfaces"]):
sousaedu80135b92021-02-17 15:05:18 +01001011 if interface.get(iface_key) == vld_id and interface.get(
1012 "type"
1013 ) in ("SR-IOV", "PCI-PASSTHROUGH"):
tierno70eeb182020-10-19 16:38:00 +00001014 # only SR-IOV o PT
1015 interface_ = interface.copy()
sousaedu80135b92021-02-17 15:05:18 +01001016 interface_["id"] = "vnfrs:{}:vdu.{}.interfaces.{}".format(
1017 db_vnfr["_id"], vdu_index, iface_index
1018 )
1019
tierno70eeb182020-10-19 16:38:00 +00001020 if vdur.get("status") == "ERROR":
1021 interface_["status"] = "ERROR"
sousaedu80135b92021-02-17 15:05:18 +01001022
tierno70eeb182020-10-19 16:38:00 +00001023 interfaces.append(interface_)
sousaedu80135b92021-02-17 15:05:18 +01001024
tierno70eeb182020-10-19 16:38:00 +00001025 return interfaces
1026
1027 def refresh(self, ro_task):
1028 # look for task create
sousaedu80135b92021-02-17 15:05:18 +01001029 task_create_index, _ = next(
1030 i_t
1031 for i_t in enumerate(ro_task["tasks"])
1032 if i_t[1]
1033 and i_t[1]["action"] == "CREATE"
1034 and i_t[1]["status"] != "FINISHED"
1035 )
tierno70eeb182020-10-19 16:38:00 +00001036
1037 return self.new(ro_task, task_create_index, None)
1038
1039 def new(self, ro_task, task_index, task_depends):
1040
1041 task = ro_task["tasks"][task_index]
1042 task_id = task["task_id"]
1043 target_vim = self.my_vims[ro_task["target_id"]]
1044
1045 sdn_net_id = ro_task["vim_info"]["vim_id"]
1046
1047 created_items = ro_task["vim_info"].get("created_items")
1048 connected_ports = ro_task["vim_info"].get("connected_ports", [])
1049 new_connected_ports = []
1050 last_update = ro_task["vim_info"].get("last_update", 0)
1051 sdn_status = ro_task["vim_info"].get("vim_status", "BUILD") or "BUILD"
1052 error_list = []
1053 created = ro_task["vim_info"].get("created", False)
1054
1055 try:
tierno70eeb182020-10-19 16:38:00 +00001056 # CREATE
1057 params = task["params"]
gifrerenom4f5c1092022-03-07 17:55:20 +00001058 vlds_to_connect = params.get("vlds", [])
1059 associated_vim = params.get("target_vim")
sousaedu80135b92021-02-17 15:05:18 +01001060 # external additional ports
1061 additional_ports = params.get("sdn-ports") or ()
gifrerenom4f5c1092022-03-07 17:55:20 +00001062 _, _, vim_account_id = (
1063 (None, None, None)
1064 if associated_vim is None
1065 else associated_vim.partition(":")
1066 )
sousaedu80135b92021-02-17 15:05:18 +01001067
tierno70eeb182020-10-19 16:38:00 +00001068 if associated_vim:
1069 # get associated VIM
1070 if associated_vim not in self.db_vims:
sousaedu80135b92021-02-17 15:05:18 +01001071 self.db_vims[associated_vim] = self.db.get_one(
1072 "vim_accounts", {"_id": vim_account_id}
1073 )
1074
tierno70eeb182020-10-19 16:38:00 +00001075 db_vim = self.db_vims[associated_vim]
1076
1077 # look for ports to connect
1078 ports = self._get_interfaces(vlds_to_connect, vim_account_id)
1079 # print(ports)
1080
1081 sdn_ports = []
1082 pending_ports = error_ports = 0
1083 vlan_used = None
1084 sdn_need_update = False
sousaedu80135b92021-02-17 15:05:18 +01001085
tierno70eeb182020-10-19 16:38:00 +00001086 for port in ports:
1087 vlan_used = port.get("vlan") or vlan_used
sousaedu80135b92021-02-17 15:05:18 +01001088
tierno70eeb182020-10-19 16:38:00 +00001089 # TODO. Do not connect if already done
1090 if not port.get("compute_node") or not port.get("pci"):
1091 if port.get("status") == "ERROR":
1092 error_ports += 1
1093 else:
1094 pending_ports += 1
1095 continue
sousaedu80135b92021-02-17 15:05:18 +01001096
tierno70eeb182020-10-19 16:38:00 +00001097 pmap = None
sousaedu80135b92021-02-17 15:05:18 +01001098 compute_node_mappings = next(
1099 (
1100 c
1101 for c in db_vim["config"].get("sdn-port-mapping", ())
1102 if c and c["compute_node"] == port["compute_node"]
1103 ),
1104 None,
1105 )
1106
tierno70eeb182020-10-19 16:38:00 +00001107 if compute_node_mappings:
1108 # process port_mapping pci of type 0000:af:1[01].[1357]
sousaedu80135b92021-02-17 15:05:18 +01001109 pmap = next(
1110 (
1111 p
1112 for p in compute_node_mappings["ports"]
1113 if self._match_pci(port["pci"], p.get("pci"))
1114 ),
1115 None,
1116 )
1117
tierno70eeb182020-10-19 16:38:00 +00001118 if not pmap:
1119 if not db_vim["config"].get("mapping_not_needed"):
sousaedu80135b92021-02-17 15:05:18 +01001120 error_list.append(
1121 "Port mapping not found for compute_node={} pci={}".format(
1122 port["compute_node"], port["pci"]
1123 )
1124 )
tierno70eeb182020-10-19 16:38:00 +00001125 continue
sousaedu80135b92021-02-17 15:05:18 +01001126
tierno70eeb182020-10-19 16:38:00 +00001127 pmap = {}
1128
1129 service_endpoint_id = "{}:{}".format(port["compute_node"], port["pci"])
1130 new_port = {
sousaedu80135b92021-02-17 15:05:18 +01001131 "service_endpoint_id": pmap.get("service_endpoint_id")
1132 or service_endpoint_id,
1133 "service_endpoint_encapsulation_type": "dot1q"
1134 if port["type"] == "SR-IOV"
1135 else None,
tierno70eeb182020-10-19 16:38:00 +00001136 "service_endpoint_encapsulation_info": {
1137 "vlan": port.get("vlan"),
lloretgalleg160fcad2021-02-19 10:57:50 +00001138 "mac": port.get("mac-address"),
sousaedu80135b92021-02-17 15:05:18 +01001139 "device_id": pmap.get("device_id") or port["compute_node"],
1140 "device_interface_id": pmap.get("device_interface_id")
1141 or port["pci"],
tierno70eeb182020-10-19 16:38:00 +00001142 "switch_dpid": pmap.get("switch_id") or pmap.get("switch_dpid"),
1143 "switch_port": pmap.get("switch_port"),
1144 "service_mapping_info": pmap.get("service_mapping_info"),
sousaedu80135b92021-02-17 15:05:18 +01001145 },
tierno70eeb182020-10-19 16:38:00 +00001146 }
1147
1148 # TODO
1149 # if port["modified_at"] > last_update:
1150 # sdn_need_update = True
1151 new_connected_ports.append(port["id"]) # TODO
1152 sdn_ports.append(new_port)
1153
1154 if error_ports:
sousaedu80135b92021-02-17 15:05:18 +01001155 error_list.append(
1156 "{} interfaces have not been created as VDU is on ERROR status".format(
1157 error_ports
1158 )
1159 )
tierno70eeb182020-10-19 16:38:00 +00001160
1161 # connect external ports
1162 for index, additional_port in enumerate(additional_ports):
sousaedu80135b92021-02-17 15:05:18 +01001163 additional_port_id = additional_port.get(
1164 "service_endpoint_id"
1165 ) or "external-{}".format(index)
1166 sdn_ports.append(
1167 {
1168 "service_endpoint_id": additional_port_id,
1169 "service_endpoint_encapsulation_type": additional_port.get(
1170 "service_endpoint_encapsulation_type", "dot1q"
1171 ),
1172 "service_endpoint_encapsulation_info": {
1173 "vlan": additional_port.get("vlan") or vlan_used,
1174 "mac": additional_port.get("mac_address"),
1175 "device_id": additional_port.get("device_id"),
1176 "device_interface_id": additional_port.get(
1177 "device_interface_id"
1178 ),
1179 "switch_dpid": additional_port.get("switch_dpid")
1180 or additional_port.get("switch_id"),
1181 "switch_port": additional_port.get("switch_port"),
1182 "service_mapping_info": additional_port.get(
1183 "service_mapping_info"
1184 ),
1185 },
1186 }
1187 )
tierno70eeb182020-10-19 16:38:00 +00001188 new_connected_ports.append(additional_port_id)
1189 sdn_info = ""
sousaedu80135b92021-02-17 15:05:18 +01001190
tierno70eeb182020-10-19 16:38:00 +00001191 # if there are more ports to connect or they have been modified, call create/update
1192 if error_list:
1193 sdn_status = "ERROR"
1194 sdn_info = "; ".join(error_list)
1195 elif set(connected_ports) != set(new_connected_ports) or sdn_need_update:
1196 last_update = time.time()
sousaedu80135b92021-02-17 15:05:18 +01001197
tierno70eeb182020-10-19 16:38:00 +00001198 if not sdn_net_id:
1199 if len(sdn_ports) < 2:
1200 sdn_status = "ACTIVE"
sousaedu80135b92021-02-17 15:05:18 +01001201
tierno70eeb182020-10-19 16:38:00 +00001202 if not pending_ports:
sousaedu80135b92021-02-17 15:05:18 +01001203 self.logger.debug(
1204 "task={} {} new-sdn-net done, less than 2 ports".format(
1205 task_id, ro_task["target_id"]
1206 )
1207 )
tierno70eeb182020-10-19 16:38:00 +00001208 else:
1209 net_type = params.get("type") or "ELAN"
sousaedu80135b92021-02-17 15:05:18 +01001210 (
1211 sdn_net_id,
1212 created_items,
1213 ) = target_vim.create_connectivity_service(net_type, sdn_ports)
tierno70eeb182020-10-19 16:38:00 +00001214 created = True
sousaedu80135b92021-02-17 15:05:18 +01001215 self.logger.debug(
1216 "task={} {} new-sdn-net={} created={}".format(
1217 task_id, ro_task["target_id"], sdn_net_id, created
1218 )
1219 )
tierno70eeb182020-10-19 16:38:00 +00001220 else:
1221 created_items = target_vim.edit_connectivity_service(
sousaedu80135b92021-02-17 15:05:18 +01001222 sdn_net_id, conn_info=created_items, connection_points=sdn_ports
1223 )
tierno70eeb182020-10-19 16:38:00 +00001224 created = True
sousaedu80135b92021-02-17 15:05:18 +01001225 self.logger.debug(
1226 "task={} {} update-sdn-net={} created={}".format(
1227 task_id, ro_task["target_id"], sdn_net_id, created
1228 )
1229 )
1230
tierno70eeb182020-10-19 16:38:00 +00001231 connected_ports = new_connected_ports
1232 elif sdn_net_id:
sousaedu80135b92021-02-17 15:05:18 +01001233 wim_status_dict = target_vim.get_connectivity_service_status(
1234 sdn_net_id, conn_info=created_items
1235 )
tierno70eeb182020-10-19 16:38:00 +00001236 sdn_status = wim_status_dict["sdn_status"]
sousaedu80135b92021-02-17 15:05:18 +01001237
tierno70eeb182020-10-19 16:38:00 +00001238 if wim_status_dict.get("sdn_info"):
1239 sdn_info = str(wim_status_dict.get("sdn_info")) or ""
sousaedu80135b92021-02-17 15:05:18 +01001240
tierno70eeb182020-10-19 16:38:00 +00001241 if wim_status_dict.get("error_msg"):
1242 sdn_info = wim_status_dict.get("error_msg") or ""
1243
1244 if pending_ports:
1245 if sdn_status != "ERROR":
1246 sdn_info = "Waiting for getting interfaces location from VIM. Obtained '{}' of {}".format(
sousaedu80135b92021-02-17 15:05:18 +01001247 len(ports) - pending_ports, len(ports)
1248 )
1249
tierno70eeb182020-10-19 16:38:00 +00001250 if sdn_status == "ACTIVE":
1251 sdn_status = "BUILD"
1252
sousaedu80135b92021-02-17 15:05:18 +01001253 ro_vim_item_update = {
1254 "vim_id": sdn_net_id,
1255 "vim_status": sdn_status,
1256 "created": created,
1257 "created_items": created_items,
1258 "connected_ports": connected_ports,
1259 "vim_details": sdn_info,
aticig79ac6df2022-05-06 16:09:52 +03001260 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +01001261 "last_update": last_update,
1262 }
1263
tierno70eeb182020-10-19 16:38:00 +00001264 return sdn_status, ro_vim_item_update
1265 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001266 self.logger.error(
1267 "task={} vim={} new-net: {}".format(task_id, ro_task["target_id"], e),
1268 exc_info=not isinstance(
1269 e, (sdnconn.SdnConnectorError, vimconn.VimConnException)
1270 ),
1271 )
1272 ro_vim_item_update = {
1273 "vim_status": "VIM_ERROR",
1274 "created": created,
aticig79ac6df2022-05-06 16:09:52 +03001275 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +01001276 }
1277
tierno70eeb182020-10-19 16:38:00 +00001278 return "FAILED", ro_vim_item_update
1279
1280 def delete(self, ro_task, task_index):
1281 task = ro_task["tasks"][task_index]
1282 task_id = task["task_id"]
1283 sdn_vim_id = ro_task["vim_info"].get("vim_id")
sousaedu80135b92021-02-17 15:05:18 +01001284 ro_vim_item_update_ok = {
1285 "vim_status": "DELETED",
1286 "created": False,
aticig79ac6df2022-05-06 16:09:52 +03001287 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +01001288 "vim_id": None,
1289 }
1290
tierno70eeb182020-10-19 16:38:00 +00001291 try:
1292 if sdn_vim_id:
1293 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +01001294 target_vim.delete_connectivity_service(
1295 sdn_vim_id, ro_task["vim_info"].get("created_items")
1296 )
tierno70eeb182020-10-19 16:38:00 +00001297
1298 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001299 if (
1300 isinstance(e, sdnconn.SdnConnectorError)
1301 and e.http_code == HTTPStatus.NOT_FOUND.value
1302 ):
aticig79ac6df2022-05-06 16:09:52 +03001303 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno70eeb182020-10-19 16:38:00 +00001304 else:
sousaedu80135b92021-02-17 15:05:18 +01001305 self.logger.error(
1306 "ro_task={} vim={} del-sdn-net={}: {}".format(
1307 ro_task["_id"], ro_task["target_id"], sdn_vim_id, e
1308 ),
1309 exc_info=not isinstance(
1310 e, (sdnconn.SdnConnectorError, vimconn.VimConnException)
1311 ),
1312 )
1313 ro_vim_item_update = {
1314 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +03001315 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +01001316 }
1317
tierno70eeb182020-10-19 16:38:00 +00001318 return "FAILED", ro_vim_item_update
1319
sousaedu80135b92021-02-17 15:05:18 +01001320 self.logger.debug(
1321 "task={} {} del-sdn-net={} {}".format(
1322 task_id,
1323 ro_task["target_id"],
1324 sdn_vim_id,
aticig79ac6df2022-05-06 16:09:52 +03001325 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +01001326 )
1327 )
1328
tierno70eeb182020-10-19 16:38:00 +00001329 return "DONE", ro_vim_item_update_ok
1330
1331
elumalai8658c2c2022-04-28 19:09:31 +05301332class VimInteractionMigration(VimInteractionBase):
1333 def exec(self, ro_task, task_index, task_depends):
1334 task = ro_task["tasks"][task_index]
1335 task_id = task["task_id"]
1336 db_task_update = {"retries": 0}
1337 target_vim = self.my_vims[ro_task["target_id"]]
1338 vim_interfaces = []
1339 created = False
1340 created_items = {}
1341 refreshed_vim_info = {}
1342
1343 try:
1344 if task.get("params"):
1345 vim_vm_id = task["params"].get("vim_vm_id")
1346 migrate_host = task["params"].get("migrate_host")
1347 _, migrated_compute_node = target_vim.migrate_instance(
1348 vim_vm_id, migrate_host
1349 )
1350
1351 if migrated_compute_node:
1352 # When VM is migrated, vdu["vim_info"] needs to be updated
1353 vdu_old_vim_info = task["params"]["vdu_vim_info"].get(
1354 ro_task["target_id"]
1355 )
1356
1357 # Refresh VM to get new vim_info
1358 vm_to_refresh_list = [vim_vm_id]
1359 vim_dict = target_vim.refresh_vms_status(vm_to_refresh_list)
1360 refreshed_vim_info = vim_dict[vim_vm_id]
1361
1362 if refreshed_vim_info.get("interfaces"):
1363 for old_iface in vdu_old_vim_info.get("interfaces"):
1364 iface = next(
1365 (
1366 iface
1367 for iface in refreshed_vim_info["interfaces"]
1368 if old_iface["vim_interface_id"]
1369 == iface["vim_interface_id"]
1370 ),
1371 None,
1372 )
1373 vim_interfaces.append(iface)
1374
1375 ro_vim_item_update = {
1376 "vim_id": vim_vm_id,
1377 "vim_status": "ACTIVE",
1378 "created": created,
1379 "created_items": created_items,
1380 "vim_details": None,
1381 "vim_message": None,
1382 }
1383
1384 if refreshed_vim_info and refreshed_vim_info.get("status") not in (
1385 "ERROR",
1386 "VIM_ERROR",
1387 ):
1388 ro_vim_item_update["vim_details"] = refreshed_vim_info["vim_info"]
1389
1390 if vim_interfaces:
1391 ro_vim_item_update["interfaces"] = vim_interfaces
1392
1393 self.logger.debug(
1394 "task={} {} vm-migration done".format(task_id, ro_task["target_id"])
1395 )
1396
1397 return "DONE", ro_vim_item_update, db_task_update
1398
1399 except (vimconn.VimConnException, NsWorkerException) as e:
1400 self.logger.error(
1401 "task={} vim={} VM Migration:"
1402 " {}".format(task_id, ro_task["target_id"], e)
1403 )
1404 ro_vim_item_update = {
1405 "vim_status": "VIM_ERROR",
1406 "created": created,
1407 "vim_message": str(e),
1408 }
1409
1410 return "FAILED", ro_vim_item_update, db_task_update
1411
1412
sritharan29a4c1a2022-05-05 12:15:04 +00001413class VimInteractionResize(VimInteractionBase):
1414 def exec(self, ro_task, task_index, task_depends):
1415 task = ro_task["tasks"][task_index]
1416 task_id = task["task_id"]
1417 db_task_update = {"retries": 0}
1418 created = False
1419 target_flavor_uuid = None
1420 created_items = {}
1421 refreshed_vim_info = {}
1422 target_vim = self.my_vims[ro_task["target_id"]]
1423
1424 try:
1425 if task.get("params"):
1426 vim_vm_id = task["params"].get("vim_vm_id")
1427 flavor_dict = task["params"].get("flavor_dict")
1428 self.logger.info("flavor_dict %s", flavor_dict)
1429
1430 try:
1431 target_flavor_uuid = target_vim.get_flavor_id_from_data(flavor_dict)
1432 except Exception as e:
1433 self.logger.info("Cannot find any flavor matching %s.", str(e))
1434 try:
1435 target_flavor_uuid = target_vim.new_flavor(flavor_dict)
1436 except Exception as e:
1437 self.logger.error("Error creating flavor at VIM %s.", str(e))
1438
1439 if target_flavor_uuid is not None:
1440 resized_status = target_vim.resize_instance(
1441 vim_vm_id, target_flavor_uuid
1442 )
1443
1444 if resized_status:
1445 # Refresh VM to get new vim_info
1446 vm_to_refresh_list = [vim_vm_id]
1447 vim_dict = target_vim.refresh_vms_status(vm_to_refresh_list)
1448 refreshed_vim_info = vim_dict[vim_vm_id]
1449
1450 ro_vim_item_update = {
1451 "vim_id": vim_vm_id,
1452 "vim_status": "DONE",
1453 "created": created,
1454 "created_items": created_items,
1455 "vim_details": None,
1456 "vim_message": None,
1457 }
1458
1459 if refreshed_vim_info and refreshed_vim_info.get("status") not in (
1460 "ERROR",
1461 "VIM_ERROR",
1462 ):
1463 ro_vim_item_update["vim_details"] = refreshed_vim_info["vim_info"]
1464
1465 self.logger.debug(
1466 "task={} {} resize done".format(task_id, ro_task["target_id"])
1467 )
1468 return "DONE", ro_vim_item_update, db_task_update
1469 except (vimconn.VimConnException, NsWorkerException) as e:
1470 self.logger.error(
1471 "task={} vim={} Resize:" " {}".format(task_id, ro_task["target_id"], e)
1472 )
1473 ro_vim_item_update = {
1474 "vim_status": "VIM_ERROR",
1475 "created": created,
1476 "vim_message": str(e),
1477 }
1478
1479 return "FAILED", ro_vim_item_update, db_task_update
1480
1481
aticig973bed22022-06-30 19:29:04 +03001482class ConfigValidate:
1483 def __init__(self, config: Dict):
1484 self.conf = config
tierno70eeb182020-10-19 16:38:00 +00001485
aticig973bed22022-06-30 19:29:04 +03001486 @property
1487 def active(self):
1488 # default 1 min, allowed >= 60 or -1, -1 disables periodic checks
1489 if (
1490 self.conf["period"]["refresh_active"] >= 60
1491 or self.conf["period"]["refresh_active"] == -1
1492 ):
1493 return self.conf["period"]["refresh_active"]
1494
1495 return 60
1496
1497 @property
1498 def build(self):
1499 return self.conf["period"]["refresh_build"]
1500
1501 @property
1502 def image(self):
1503 return self.conf["period"]["refresh_image"]
1504
1505 @property
1506 def error(self):
1507 return self.conf["period"]["refresh_error"]
1508
1509 @property
1510 def queue_size(self):
1511 return self.conf["period"]["queue_size"]
1512
1513
1514class NsWorker(threading.Thread):
tierno70eeb182020-10-19 16:38:00 +00001515 def __init__(self, worker_index, config, plugins, db):
1516 """
tierno70eeb182020-10-19 16:38:00 +00001517 :param worker_index: thread index
1518 :param config: general configuration of RO, among others the process_id with the docker id where it runs
1519 :param plugins: global shared dict with the loaded plugins
1520 :param db: database class instance to use
1521 """
1522 threading.Thread.__init__(self)
1523 self.config = config
1524 self.plugins = plugins
1525 self.plugin_name = "unknown"
sousaedu80135b92021-02-17 15:05:18 +01001526 self.logger = logging.getLogger("ro.worker{}".format(worker_index))
tierno70eeb182020-10-19 16:38:00 +00001527 self.worker_index = worker_index
aticig973bed22022-06-30 19:29:04 +03001528 # refresh periods for created items
1529 self.refresh_config = ConfigValidate(config)
1530 self.task_queue = queue.Queue(self.refresh_config.queue_size)
sousaedu80135b92021-02-17 15:05:18 +01001531 # targetvim: vimplugin class
1532 self.my_vims = {}
1533 # targetvim: vim information from database
1534 self.db_vims = {}
1535 # targetvim list
1536 self.vim_targets = []
tierno70eeb182020-10-19 16:38:00 +00001537 self.my_id = config["process_id"] + ":" + str(worker_index)
1538 self.db = db
1539 self.item2class = {
1540 "net": VimInteractionNet(self.db, self.my_vims, self.db_vims, self.logger),
1541 "vdu": VimInteractionVdu(self.db, self.my_vims, self.db_vims, self.logger),
sousaedu80135b92021-02-17 15:05:18 +01001542 "image": VimInteractionImage(
1543 self.db, self.my_vims, self.db_vims, self.logger
1544 ),
1545 "flavor": VimInteractionFlavor(
1546 self.db, self.my_vims, self.db_vims, self.logger
1547 ),
1548 "sdn_net": VimInteractionSdnNet(
1549 self.db, self.my_vims, self.db_vims, self.logger
1550 ),
k4.rahul78f474e2022-05-02 15:47:57 +00001551 "update": VimInteractionUpdateVdu(
1552 self.db, self.my_vims, self.db_vims, self.logger
1553 ),
Alexis Romerob70f4ed2022-03-11 18:00:49 +01001554 "affinity-or-anti-affinity-group": VimInteractionAffinityGroup(
1555 self.db, self.my_vims, self.db_vims, self.logger
1556 ),
elumalai8658c2c2022-04-28 19:09:31 +05301557 "migrate": VimInteractionMigration(
1558 self.db, self.my_vims, self.db_vims, self.logger
1559 ),
sritharan29a4c1a2022-05-05 12:15:04 +00001560 "verticalscale": VimInteractionResize(
1561 self.db, self.my_vims, self.db_vims, self.logger
1562 ),
tierno70eeb182020-10-19 16:38:00 +00001563 }
1564 self.time_last_task_processed = None
sousaedu80135b92021-02-17 15:05:18 +01001565 # lists of tasks to delete because nsrs or vnfrs has been deleted from db
1566 self.tasks_to_delete = []
1567 # it is idle when there are not vim_targets associated
1568 self.idle = True
tiernof1b640f2020-12-09 15:06:01 +00001569 self.task_locked_time = config["global"]["task_locked_time"]
tierno70eeb182020-10-19 16:38:00 +00001570
1571 def insert_task(self, task):
1572 try:
1573 self.task_queue.put(task, False)
1574 return None
1575 except queue.Full:
1576 raise NsWorkerException("timeout inserting a task")
1577
1578 def terminate(self):
1579 self.insert_task("exit")
1580
1581 def del_task(self, task):
1582 with self.task_lock:
1583 if task["status"] == "SCHEDULED":
1584 task["status"] = "SUPERSEDED"
1585 return True
1586 else: # task["status"] == "processing"
1587 self.task_lock.release()
1588 return False
1589
1590 def _process_vim_config(self, target_id, db_vim):
1591 """
1592 Process vim config, creating vim configuration files as ca_cert
1593 :param target_id: vim/sdn/wim + id
1594 :param db_vim: Vim dictionary obtained from database
1595 :return: None. Modifies vim. Creates a folder target_id:worker_index and several files
1596 """
1597 if not db_vim.get("config"):
1598 return
sousaedu80135b92021-02-17 15:05:18 +01001599
tierno70eeb182020-10-19 16:38:00 +00001600 file_name = ""
sousaedu80135b92021-02-17 15:05:18 +01001601
tierno70eeb182020-10-19 16:38:00 +00001602 try:
1603 if db_vim["config"].get("ca_cert_content"):
1604 file_name = "{}:{}".format(target_id, self.worker_index)
sousaedu80135b92021-02-17 15:05:18 +01001605
tierno70eeb182020-10-19 16:38:00 +00001606 try:
1607 mkdir(file_name)
1608 except FileExistsError:
aticig78ac0cf2022-07-15 00:43:09 +03001609 self.logger.exception(
1610 "FileExistsError occured while processing vim_config."
1611 )
sousaedu80135b92021-02-17 15:05:18 +01001612
tierno70eeb182020-10-19 16:38:00 +00001613 file_name = file_name + "/ca_cert"
sousaedu80135b92021-02-17 15:05:18 +01001614
tierno70eeb182020-10-19 16:38:00 +00001615 with open(file_name, "w") as f:
1616 f.write(db_vim["config"]["ca_cert_content"])
1617 del db_vim["config"]["ca_cert_content"]
1618 db_vim["config"]["ca_cert"] = file_name
1619 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001620 raise NsWorkerException(
1621 "Error writing to file '{}': {}".format(file_name, e)
1622 )
tierno70eeb182020-10-19 16:38:00 +00001623
1624 def _load_plugin(self, name, type="vim"):
1625 # type can be vim or sdn
1626 if "rovim_dummy" not in self.plugins:
1627 self.plugins["rovim_dummy"] = VimDummyConnector
sousaedu80135b92021-02-17 15:05:18 +01001628
tierno70eeb182020-10-19 16:38:00 +00001629 if "rosdn_dummy" not in self.plugins:
1630 self.plugins["rosdn_dummy"] = SdnDummyConnector
sousaedu80135b92021-02-17 15:05:18 +01001631
tierno70eeb182020-10-19 16:38:00 +00001632 if name in self.plugins:
1633 return self.plugins[name]
sousaedu80135b92021-02-17 15:05:18 +01001634
tierno70eeb182020-10-19 16:38:00 +00001635 try:
sousaedubecd0832021-04-08 00:16:18 +02001636 for ep in entry_points(group="osm_ro{}.plugins".format(type), name=name):
1637 self.plugins[name] = ep.load()
tierno70eeb182020-10-19 16:38:00 +00001638 except Exception as e:
1639 raise NsWorkerException("Cannot load plugin osm_{}: {}".format(name, e))
sousaedu80135b92021-02-17 15:05:18 +01001640
tierno70eeb182020-10-19 16:38:00 +00001641 if name and name not in self.plugins:
sousaedu80135b92021-02-17 15:05:18 +01001642 raise NsWorkerException(
1643 "Plugin 'osm_{n}' has not been installed".format(n=name)
1644 )
1645
tierno70eeb182020-10-19 16:38:00 +00001646 return self.plugins[name]
1647
1648 def _unload_vim(self, target_id):
1649 """
1650 Unload a vim_account. Removes it from self db_vims dictionary, my_vims dictionary and vim_targets list
1651 :param target_id: Contains type:_id; where type can be 'vim', ...
1652 :return: None.
1653 """
1654 try:
tierno70eeb182020-10-19 16:38:00 +00001655 self.db_vims.pop(target_id, None)
1656 self.my_vims.pop(target_id, None)
sousaedu80135b92021-02-17 15:05:18 +01001657
tierno86153522020-12-06 18:27:16 +00001658 if target_id in self.vim_targets:
1659 self.vim_targets.remove(target_id)
sousaedu80135b92021-02-17 15:05:18 +01001660
tierno86153522020-12-06 18:27:16 +00001661 self.logger.info("Unloaded {}".format(target_id))
tierno70eeb182020-10-19 16:38:00 +00001662 rmtree("{}:{}".format(target_id, self.worker_index))
1663 except FileNotFoundError:
aticig78ac0cf2022-07-15 00:43:09 +03001664 # This is raised by rmtree if folder does not exist.
1665 self.logger.exception("FileNotFoundError occured while unloading VIM.")
tierno70eeb182020-10-19 16:38:00 +00001666 except Exception as e:
1667 self.logger.error("Cannot unload {}: {}".format(target_id, e))
1668
1669 def _check_vim(self, target_id):
1670 """
1671 Load a VIM/SDN/WIM (if not loaded) and check connectivity, updating database with ENABLE or ERROR
1672 :param target_id: Contains type:_id; type can be 'vim', 'sdn' or 'wim'
1673 :return: None.
1674 """
1675 target, _, _id = target_id.partition(":")
1676 now = time.time()
1677 update_dict = {}
1678 unset_dict = {}
1679 op_text = ""
1680 step = ""
tierno86153522020-12-06 18:27:16 +00001681 loaded = target_id in self.vim_targets
sousaedu80135b92021-02-17 15:05:18 +01001682 target_database = (
1683 "vim_accounts"
1684 if target == "vim"
1685 else "wim_accounts"
1686 if target == "wim"
1687 else "sdns"
1688 )
1689
tierno70eeb182020-10-19 16:38:00 +00001690 try:
1691 step = "Getting {} from db".format(target_id)
1692 db_vim = self.db.get_one(target_database, {"_id": _id})
sousaedu80135b92021-02-17 15:05:18 +01001693
1694 for op_index, operation in enumerate(
1695 db_vim["_admin"].get("operations", ())
1696 ):
tierno70eeb182020-10-19 16:38:00 +00001697 if operation["operationState"] != "PROCESSING":
1698 continue
sousaedu80135b92021-02-17 15:05:18 +01001699
tierno70eeb182020-10-19 16:38:00 +00001700 locked_at = operation.get("locked_at")
sousaedu80135b92021-02-17 15:05:18 +01001701
tiernof1b640f2020-12-09 15:06:01 +00001702 if locked_at is not None and locked_at >= now - self.task_locked_time:
tierno70eeb182020-10-19 16:38:00 +00001703 # some other thread is doing this operation
1704 return
sousaedu80135b92021-02-17 15:05:18 +01001705
tierno70eeb182020-10-19 16:38:00 +00001706 # lock
1707 op_text = "_admin.operations.{}.".format(op_index)
sousaedu80135b92021-02-17 15:05:18 +01001708
1709 if not self.db.set_one(
1710 target_database,
1711 q_filter={
1712 "_id": _id,
1713 op_text + "operationState": "PROCESSING",
1714 op_text + "locked_at": locked_at,
1715 },
1716 update_dict={
1717 op_text + "locked_at": now,
1718 "admin.current_operation": op_index,
1719 },
1720 fail_on_empty=False,
1721 ):
tierno70eeb182020-10-19 16:38:00 +00001722 return
sousaedu80135b92021-02-17 15:05:18 +01001723
tierno70eeb182020-10-19 16:38:00 +00001724 unset_dict[op_text + "locked_at"] = None
1725 unset_dict["current_operation"] = None
1726 step = "Loading " + target_id
1727 error_text = self._load_vim(target_id)
sousaedu80135b92021-02-17 15:05:18 +01001728
tierno70eeb182020-10-19 16:38:00 +00001729 if not error_text:
1730 step = "Checking connectivity"
sousaedu80135b92021-02-17 15:05:18 +01001731
1732 if target == "vim":
tierno70eeb182020-10-19 16:38:00 +00001733 self.my_vims[target_id].check_vim_connectivity()
1734 else:
1735 self.my_vims[target_id].check_credentials()
sousaedu80135b92021-02-17 15:05:18 +01001736
tierno70eeb182020-10-19 16:38:00 +00001737 update_dict["_admin.operationalState"] = "ENABLED"
1738 update_dict["_admin.detailed-status"] = ""
1739 unset_dict[op_text + "detailed-status"] = None
1740 update_dict[op_text + "operationState"] = "COMPLETED"
sousaedu80135b92021-02-17 15:05:18 +01001741
tierno70eeb182020-10-19 16:38:00 +00001742 return
1743
1744 except Exception as e:
1745 error_text = "{}: {}".format(step, e)
1746 self.logger.error("{} for {}: {}".format(step, target_id, e))
1747
1748 finally:
1749 if update_dict or unset_dict:
1750 if error_text:
1751 update_dict[op_text + "operationState"] = "FAILED"
1752 update_dict[op_text + "detailed-status"] = error_text
1753 unset_dict.pop(op_text + "detailed-status", None)
1754 update_dict["_admin.operationalState"] = "ERROR"
1755 update_dict["_admin.detailed-status"] = error_text
sousaedu80135b92021-02-17 15:05:18 +01001756
tierno70eeb182020-10-19 16:38:00 +00001757 if op_text:
1758 update_dict[op_text + "statusEnteredTime"] = now
sousaedu80135b92021-02-17 15:05:18 +01001759
1760 self.db.set_one(
1761 target_database,
1762 q_filter={"_id": _id},
1763 update_dict=update_dict,
1764 unset=unset_dict,
1765 fail_on_empty=False,
1766 )
1767
tierno70eeb182020-10-19 16:38:00 +00001768 if not loaded:
1769 self._unload_vim(target_id)
1770
1771 def _reload_vim(self, target_id):
1772 if target_id in self.vim_targets:
1773 self._load_vim(target_id)
1774 else:
1775 # if the vim is not loaded, but database information of VIM is cached at self.db_vims,
1776 # just remove it to force load again next time it is needed
1777 self.db_vims.pop(target_id, None)
1778
1779 def _load_vim(self, target_id):
1780 """
1781 Load or reload a vim_account, sdn_controller or wim_account.
1782 Read content from database, load the plugin if not loaded.
1783 In case of error loading the plugin, it load a failing VIM_connector
1784 It fills self db_vims dictionary, my_vims dictionary and vim_targets list
1785 :param target_id: Contains type:_id; where type can be 'vim', ...
1786 :return: None if ok, descriptive text if error
1787 """
1788 target, _, _id = target_id.partition(":")
sousaedu80135b92021-02-17 15:05:18 +01001789 target_database = (
1790 "vim_accounts"
1791 if target == "vim"
1792 else "wim_accounts"
1793 if target == "wim"
1794 else "sdns"
1795 )
tierno70eeb182020-10-19 16:38:00 +00001796 plugin_name = ""
1797 vim = None
sousaedu80135b92021-02-17 15:05:18 +01001798
tierno70eeb182020-10-19 16:38:00 +00001799 try:
1800 step = "Getting {}={} from db".format(target, _id)
1801 # TODO process for wim, sdnc, ...
1802 vim = self.db.get_one(target_database, {"_id": _id})
1803
1804 # if deep_get(vim, "config", "sdn-controller"):
1805 # step = "Getting sdn-controller-id='{}' from db".format(vim["config"]["sdn-controller"])
1806 # db_sdn = self.db.get_one("sdns", {"_id": vim["config"]["sdn-controller"]})
1807
1808 step = "Decrypting password"
1809 schema_version = vim.get("schema_version")
sousaedu80135b92021-02-17 15:05:18 +01001810 self.db.encrypt_decrypt_fields(
1811 vim,
1812 "decrypt",
1813 fields=("password", "secret"),
1814 schema_version=schema_version,
1815 salt=_id,
1816 )
tierno70eeb182020-10-19 16:38:00 +00001817 self._process_vim_config(target_id, vim)
sousaedu80135b92021-02-17 15:05:18 +01001818
tierno70eeb182020-10-19 16:38:00 +00001819 if target == "vim":
1820 plugin_name = "rovim_" + vim["vim_type"]
1821 step = "Loading plugin '{}'".format(plugin_name)
1822 vim_module_conn = self._load_plugin(plugin_name)
1823 step = "Loading {}'".format(target_id)
1824 self.my_vims[target_id] = vim_module_conn(
sousaedu80135b92021-02-17 15:05:18 +01001825 uuid=vim["_id"],
1826 name=vim["name"],
1827 tenant_id=vim.get("vim_tenant_id"),
1828 tenant_name=vim.get("vim_tenant_name"),
1829 url=vim["vim_url"],
1830 url_admin=None,
1831 user=vim["vim_user"],
1832 passwd=vim["vim_password"],
1833 config=vim.get("config") or {},
1834 persistent_info={},
tierno70eeb182020-10-19 16:38:00 +00001835 )
1836 else: # sdn
gifrerenom4f5c1092022-03-07 17:55:20 +00001837 plugin_name = "rosdn_" + (vim.get("type") or vim.get("wim_type"))
tierno70eeb182020-10-19 16:38:00 +00001838 step = "Loading plugin '{}'".format(plugin_name)
1839 vim_module_conn = self._load_plugin(plugin_name, "sdn")
1840 step = "Loading {}'".format(target_id)
1841 wim = deepcopy(vim)
1842 wim_config = wim.pop("config", {}) or {}
1843 wim["uuid"] = wim["_id"]
gifrerenom34d488c2022-03-07 18:00:01 +00001844 if "url" in wim and "wim_url" not in wim:
1845 wim["wim_url"] = wim["url"]
1846 elif "url" not in wim and "wim_url" in wim:
1847 wim["url"] = wim["wim_url"]
sousaedu80135b92021-02-17 15:05:18 +01001848
tierno70eeb182020-10-19 16:38:00 +00001849 if wim.get("dpid"):
1850 wim_config["dpid"] = wim.pop("dpid")
sousaedu80135b92021-02-17 15:05:18 +01001851
tierno70eeb182020-10-19 16:38:00 +00001852 if wim.get("switch_id"):
1853 wim_config["switch_id"] = wim.pop("switch_id")
sousaedu80135b92021-02-17 15:05:18 +01001854
1855 # wim, wim_account, config
1856 self.my_vims[target_id] = vim_module_conn(wim, wim, wim_config)
tierno70eeb182020-10-19 16:38:00 +00001857 self.db_vims[target_id] = vim
1858 self.error_status = None
sousaedu80135b92021-02-17 15:05:18 +01001859
1860 self.logger.info(
1861 "Connector loaded for {}, plugin={}".format(target_id, plugin_name)
1862 )
tierno70eeb182020-10-19 16:38:00 +00001863 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001864 self.logger.error(
1865 "Cannot load {} plugin={}: {} {}".format(
1866 target_id, plugin_name, step, e
1867 )
1868 )
1869
tierno70eeb182020-10-19 16:38:00 +00001870 self.db_vims[target_id] = vim or {}
1871 self.db_vims[target_id] = FailingConnector(str(e))
1872 error_status = "{} Error: {}".format(step, e)
sousaedu80135b92021-02-17 15:05:18 +01001873
tierno70eeb182020-10-19 16:38:00 +00001874 return error_status
1875 finally:
1876 if target_id not in self.vim_targets:
1877 self.vim_targets.append(target_id)
1878
1879 def _get_db_task(self):
1880 """
1881 Read actions from database and reload them at memory. Fill self.refresh_list, pending_list, vim_actions
1882 :return: None
1883 """
1884 now = time.time()
sousaedu80135b92021-02-17 15:05:18 +01001885
tierno70eeb182020-10-19 16:38:00 +00001886 if not self.time_last_task_processed:
1887 self.time_last_task_processed = now
sousaedu80135b92021-02-17 15:05:18 +01001888
tierno70eeb182020-10-19 16:38:00 +00001889 try:
1890 while True:
gallardo2f4aaaa2022-01-31 16:50:48 +00001891 """
1892 # Log RO tasks only when loglevel is DEBUG
1893 if self.logger.getEffectiveLevel() == logging.DEBUG:
1894 self._log_ro_task(
1895 None,
1896 None,
1897 None,
1898 "TASK_WF",
1899 "task_locked_time="
1900 + str(self.task_locked_time)
1901 + " "
1902 + "time_last_task_processed="
1903 + str(self.time_last_task_processed)
1904 + " "
1905 + "now="
1906 + str(now),
1907 )
1908 """
tierno70eeb182020-10-19 16:38:00 +00001909 locked = self.db.set_one(
1910 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01001911 q_filter={
1912 "target_id": self.vim_targets,
1913 "tasks.status": ["SCHEDULED", "BUILD", "DONE", "FAILED"],
1914 "locked_at.lt": now - self.task_locked_time,
1915 "to_check_at.lt": self.time_last_task_processed,
aticig973bed22022-06-30 19:29:04 +03001916 "to_check_at.gt": -1,
sousaedu80135b92021-02-17 15:05:18 +01001917 },
tierno70eeb182020-10-19 16:38:00 +00001918 update_dict={"locked_by": self.my_id, "locked_at": now},
sousaedu80135b92021-02-17 15:05:18 +01001919 fail_on_empty=False,
1920 )
1921
tierno70eeb182020-10-19 16:38:00 +00001922 if locked:
1923 # read and return
1924 ro_task = self.db.get_one(
1925 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01001926 q_filter={
1927 "target_id": self.vim_targets,
1928 "tasks.status": ["SCHEDULED", "BUILD", "DONE", "FAILED"],
1929 "locked_at": now,
1930 },
1931 )
tierno70eeb182020-10-19 16:38:00 +00001932 return ro_task
sousaedu80135b92021-02-17 15:05:18 +01001933
tierno70eeb182020-10-19 16:38:00 +00001934 if self.time_last_task_processed == now:
1935 self.time_last_task_processed = None
1936 return None
1937 else:
1938 self.time_last_task_processed = now
1939 # self.time_last_task_processed = min(self.time_last_task_processed + 1000, now)
1940
1941 except DbException as e:
1942 self.logger.error("Database exception at _get_db_task: {}".format(e))
1943 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001944 self.logger.critical(
1945 "Unexpected exception at _get_db_task: {}".format(e), exc_info=True
1946 )
1947
tierno70eeb182020-10-19 16:38:00 +00001948 return None
1949
gallardo2f4aaaa2022-01-31 16:50:48 +00001950 def _get_db_all_tasks(self):
1951 """
1952 Read all content of table ro_tasks to log it
1953 :return: None
1954 """
1955 try:
1956 # Checking the content of the BD:
1957
1958 # read and return
1959 ro_task = self.db.get_list("ro_tasks")
1960 for rt in ro_task:
1961 self._log_ro_task(rt, None, None, "TASK_WF", "GET_ALL_TASKS")
1962 return ro_task
1963
1964 except DbException as e:
1965 self.logger.error("Database exception at _get_db_all_tasks: {}".format(e))
1966 except Exception as e:
1967 self.logger.critical(
1968 "Unexpected exception at _get_db_all_tasks: {}".format(e), exc_info=True
1969 )
1970
1971 return None
1972
1973 def _log_ro_task(self, ro_task, db_ro_task_update, db_ro_task_delete, mark, event):
1974 """
1975 Generate a log with the following format:
1976
1977 Mark;Event;ro_task_id;locked_at;modified_at;created_at;to_check_at;locked_by;
1978 target_id;vim_info.refresh_at;vim_info;no_of_tasks;task_status;action_id;
1979 task_array_index;task_id;task_action;task_item;task_args
1980
1981 Example:
1982
1983 TASK_WF;GET_TASK;888f1864-749a-4fc2-bc1a-97c0fffd6a6f:2;1642158724.8210013;
1984 1642158640.7986135;1642158640.7986135;1642158640.7986135;b134c9494e75:0a
1985 ;vim:b7ff9e24-8868-4d68-8a57-a59dc11d0327;None;{'created': False,
1986 'created_items': None, 'vim_id': None, 'vim_name': None, 'vim_status': None,
aticig79ac6df2022-05-06 16:09:52 +03001987 'vim_details': None, 'vim_message': None, 'refresh_at': None};1;SCHEDULED;
gallardo2f4aaaa2022-01-31 16:50:48 +00001988 888f1864-749a-4fc2-bc1a-97c0fffd6a6f;0;888f1864-749a-4fc2-bc1a-97c0fffd6a6f:2;
1989 CREATE;image;{'filter_dict': {'name': 'ubuntu-os-cloud:image-family:ubuntu-1804-lts'}}
1990 """
1991 try:
1992 line = []
1993 i = 0
1994 if ro_task is not None and isinstance(ro_task, dict):
1995 for t in ro_task["tasks"]:
1996 line.clear()
1997 line.append(mark)
1998 line.append(event)
1999 line.append(ro_task.get("_id", ""))
2000 line.append(str(ro_task.get("locked_at", "")))
2001 line.append(str(ro_task.get("modified_at", "")))
2002 line.append(str(ro_task.get("created_at", "")))
2003 line.append(str(ro_task.get("to_check_at", "")))
2004 line.append(str(ro_task.get("locked_by", "")))
2005 line.append(str(ro_task.get("target_id", "")))
2006 line.append(str(ro_task.get("vim_info", {}).get("refresh_at", "")))
2007 line.append(str(ro_task.get("vim_info", "")))
2008 line.append(str(ro_task.get("tasks", "")))
2009 if isinstance(t, dict):
2010 line.append(str(t.get("status", "")))
2011 line.append(str(t.get("action_id", "")))
2012 line.append(str(i))
2013 line.append(str(t.get("task_id", "")))
2014 line.append(str(t.get("action", "")))
2015 line.append(str(t.get("item", "")))
2016 line.append(str(t.get("find_params", "")))
2017 line.append(str(t.get("params", "")))
2018 else:
2019 line.extend([""] * 2)
2020 line.append(str(i))
2021 line.extend([""] * 5)
2022
2023 i += 1
2024 self.logger.debug(";".join(line))
2025 elif db_ro_task_update is not None and isinstance(db_ro_task_update, dict):
2026 i = 0
2027 while True:
2028 st = "tasks.{}.status".format(i)
2029 if st not in db_ro_task_update:
2030 break
2031 line.clear()
2032 line.append(mark)
2033 line.append(event)
2034 line.append(db_ro_task_update.get("_id", ""))
2035 line.append(str(db_ro_task_update.get("locked_at", "")))
2036 line.append(str(db_ro_task_update.get("modified_at", "")))
2037 line.append("")
2038 line.append(str(db_ro_task_update.get("to_check_at", "")))
2039 line.append(str(db_ro_task_update.get("locked_by", "")))
2040 line.append("")
2041 line.append(str(db_ro_task_update.get("vim_info.refresh_at", "")))
2042 line.append("")
2043 line.append(str(db_ro_task_update.get("vim_info", "")))
2044 line.append(str(str(db_ro_task_update).count(".status")))
2045 line.append(db_ro_task_update.get(st, ""))
2046 line.append("")
2047 line.append(str(i))
2048 line.extend([""] * 3)
2049 i += 1
2050 self.logger.debug(";".join(line))
2051
2052 elif db_ro_task_delete is not None and isinstance(db_ro_task_delete, dict):
2053 line.clear()
2054 line.append(mark)
2055 line.append(event)
2056 line.append(db_ro_task_delete.get("_id", ""))
2057 line.append("")
2058 line.append(db_ro_task_delete.get("modified_at", ""))
2059 line.extend([""] * 13)
2060 self.logger.debug(";".join(line))
2061
2062 else:
2063 line.clear()
2064 line.append(mark)
2065 line.append(event)
2066 line.extend([""] * 16)
2067 self.logger.debug(";".join(line))
2068
2069 except Exception as e:
2070 self.logger.error("Error logging ro_task: {}".format(e))
2071
tierno70eeb182020-10-19 16:38:00 +00002072 def _delete_task(self, ro_task, task_index, task_depends, db_update):
2073 """
2074 Determine if this task need to be done or superseded
2075 :return: None
2076 """
2077 my_task = ro_task["tasks"][task_index]
2078 task_id = my_task["task_id"]
sousaedu80135b92021-02-17 15:05:18 +01002079 needed_delete = ro_task["vim_info"]["created"] or ro_task["vim_info"].get(
2080 "created_items", False
2081 )
2082
palaciosj8f2060b2022-02-24 12:05:59 +00002083 self.logger.warning("Needed delete: {}".format(needed_delete))
tierno70eeb182020-10-19 16:38:00 +00002084 if my_task["status"] == "FAILED":
2085 return None, None # TODO need to be retry??
sousaedu80135b92021-02-17 15:05:18 +01002086
tierno70eeb182020-10-19 16:38:00 +00002087 try:
2088 for index, task in enumerate(ro_task["tasks"]):
2089 if index == task_index or not task:
2090 continue # own task
sousaedu80135b92021-02-17 15:05:18 +01002091
2092 if (
2093 my_task["target_record"] == task["target_record"]
2094 and task["action"] == "CREATE"
2095 ):
tierno70eeb182020-10-19 16:38:00 +00002096 # set to finished
sousaedu80135b92021-02-17 15:05:18 +01002097 db_update["tasks.{}.status".format(index)] = task[
2098 "status"
2099 ] = "FINISHED"
2100 elif task["action"] == "CREATE" and task["status"] not in (
2101 "FINISHED",
2102 "SUPERSEDED",
2103 ):
tierno70eeb182020-10-19 16:38:00 +00002104 needed_delete = False
sousaedu80135b92021-02-17 15:05:18 +01002105
tierno70eeb182020-10-19 16:38:00 +00002106 if needed_delete:
aticig285185e2022-05-02 21:23:48 +03002107 self.logger.warning(
2108 "Deleting ro_task={} task_index={}".format(ro_task, task_index)
2109 )
tierno70eeb182020-10-19 16:38:00 +00002110 return self.item2class[my_task["item"]].delete(ro_task, task_index)
2111 else:
2112 return "SUPERSEDED", None
2113 except Exception as e:
2114 if not isinstance(e, NsWorkerException):
sousaedu80135b92021-02-17 15:05:18 +01002115 self.logger.critical(
2116 "Unexpected exception at _delete_task task={}: {}".format(
2117 task_id, e
2118 ),
2119 exc_info=True,
2120 )
2121
aticig79ac6df2022-05-06 16:09:52 +03002122 return "FAILED", {"vim_status": "VIM_ERROR", "vim_message": str(e)}
tierno70eeb182020-10-19 16:38:00 +00002123
2124 def _create_task(self, ro_task, task_index, task_depends, db_update):
2125 """
2126 Determine if this task need to create something at VIM
2127 :return: None
2128 """
2129 my_task = ro_task["tasks"][task_index]
2130 task_id = my_task["task_id"]
2131 task_status = None
sousaedu80135b92021-02-17 15:05:18 +01002132
tierno70eeb182020-10-19 16:38:00 +00002133 if my_task["status"] == "FAILED":
2134 return None, None # TODO need to be retry??
2135 elif my_task["status"] == "SCHEDULED":
2136 # check if already created by another task
2137 for index, task in enumerate(ro_task["tasks"]):
2138 if index == task_index or not task:
2139 continue # own task
sousaedu80135b92021-02-17 15:05:18 +01002140
2141 if task["action"] == "CREATE" and task["status"] not in (
2142 "SCHEDULED",
2143 "FINISHED",
2144 "SUPERSEDED",
2145 ):
tierno70eeb182020-10-19 16:38:00 +00002146 return task["status"], "COPY_VIM_INFO"
2147
2148 try:
2149 task_status, ro_vim_item_update = self.item2class[my_task["item"]].new(
sousaedu80135b92021-02-17 15:05:18 +01002150 ro_task, task_index, task_depends
2151 )
tierno70eeb182020-10-19 16:38:00 +00002152 # TODO update other CREATE tasks
2153 except Exception as e:
2154 if not isinstance(e, NsWorkerException):
sousaedu80135b92021-02-17 15:05:18 +01002155 self.logger.error(
2156 "Error executing task={}: {}".format(task_id, e), exc_info=True
2157 )
2158
tierno70eeb182020-10-19 16:38:00 +00002159 task_status = "FAILED"
aticig79ac6df2022-05-06 16:09:52 +03002160 ro_vim_item_update = {"vim_status": "VIM_ERROR", "vim_message": str(e)}
tierno70eeb182020-10-19 16:38:00 +00002161 # TODO update ro_vim_item_update
sousaedu80135b92021-02-17 15:05:18 +01002162
tierno70eeb182020-10-19 16:38:00 +00002163 return task_status, ro_vim_item_update
2164 else:
2165 return None, None
2166
2167 def _get_dependency(self, task_id, ro_task=None, target_id=None):
2168 """
2169 Look for dependency task
2170 :param task_id: Can be one of
2171 1. target_vim+blank+task.target_record_id: "(vim|sdn|wim):<id> (vnfrs|nsrs):(vld|vdu|flavor|image).<id>"
2172 2. task.target_record_id: "(vnfrs|nsrs):(vld|vdu|flavor|image).<id>"
2173 3. task.task_id: "<action_id>:number"
2174 :param ro_task:
2175 :param target_id:
2176 :return: database ro_task plus index of task
2177 """
sousaedu80135b92021-02-17 15:05:18 +01002178 if (
2179 task_id.startswith("vim:")
2180 or task_id.startswith("sdn:")
2181 or task_id.startswith("wim:")
2182 ):
tierno70eeb182020-10-19 16:38:00 +00002183 target_id, _, task_id = task_id.partition(" ")
2184
2185 if task_id.startswith("nsrs:") or task_id.startswith("vnfrs:"):
2186 ro_task_dependency = self.db.get_one(
2187 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01002188 q_filter={"target_id": target_id, "tasks.target_record_id": task_id},
2189 fail_on_empty=False,
2190 )
2191
tierno70eeb182020-10-19 16:38:00 +00002192 if ro_task_dependency:
2193 for task_index, task in enumerate(ro_task_dependency["tasks"]):
2194 if task["target_record_id"] == task_id:
2195 return ro_task_dependency, task_index
2196
2197 else:
2198 if ro_task:
2199 for task_index, task in enumerate(ro_task["tasks"]):
2200 if task and task["task_id"] == task_id:
2201 return ro_task, task_index
sousaedu80135b92021-02-17 15:05:18 +01002202
tierno70eeb182020-10-19 16:38:00 +00002203 ro_task_dependency = self.db.get_one(
2204 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01002205 q_filter={
2206 "tasks.ANYINDEX.task_id": task_id,
2207 "tasks.ANYINDEX.target_record.ne": None,
2208 },
2209 fail_on_empty=False,
2210 )
2211
palaciosj8f2060b2022-02-24 12:05:59 +00002212 self.logger.warning("ro_task_dependency={}".format(ro_task_dependency))
tierno70eeb182020-10-19 16:38:00 +00002213 if ro_task_dependency:
palaciosj8f2060b2022-02-24 12:05:59 +00002214 for task_index, task in enumerate(ro_task_dependency["tasks"]):
tierno70eeb182020-10-19 16:38:00 +00002215 if task["task_id"] == task_id:
2216 return ro_task_dependency, task_index
2217 raise NsWorkerException("Cannot get depending task {}".format(task_id))
2218
aticig973bed22022-06-30 19:29:04 +03002219 def update_vm_refresh(self):
2220 """Enables the VM status updates if self.refresh_config.active parameter
2221 is not -1 and than updates the DB accordingly
2222
2223 """
2224 try:
2225 self.logger.debug("Checking if VM status update config")
2226 next_refresh = time.time()
2227 if self.refresh_config.active == -1:
2228 next_refresh = -1
2229 else:
2230 next_refresh += self.refresh_config.active
2231
2232 if next_refresh != -1:
2233 db_ro_task_update = {}
2234 now = time.time()
2235 next_check_at = now + (24 * 60 * 60)
2236 next_check_at = min(next_check_at, next_refresh)
2237 db_ro_task_update["vim_info.refresh_at"] = next_refresh
2238 db_ro_task_update["to_check_at"] = next_check_at
2239
2240 self.logger.debug(
2241 "Finding tasks which to be updated to enable VM status updates"
2242 )
2243 refresh_tasks = self.db.get_list(
2244 "ro_tasks",
2245 q_filter={
2246 "tasks.status": "DONE",
2247 "to_check_at.lt": 0,
2248 },
2249 )
2250 self.logger.debug("Updating tasks to change the to_check_at status")
2251 for task in refresh_tasks:
2252 q_filter = {
2253 "_id": task["_id"],
2254 }
2255 self.db.set_one(
2256 "ro_tasks",
2257 q_filter=q_filter,
2258 update_dict=db_ro_task_update,
2259 fail_on_empty=True,
2260 )
2261
2262 except Exception as e:
2263 self.logger.error(f"Error updating tasks to enable VM status updates: {e}")
2264
tierno70eeb182020-10-19 16:38:00 +00002265 def _process_pending_tasks(self, ro_task):
2266 ro_task_id = ro_task["_id"]
2267 now = time.time()
sousaedu80135b92021-02-17 15:05:18 +01002268 # one day
2269 next_check_at = now + (24 * 60 * 60)
tierno70eeb182020-10-19 16:38:00 +00002270 db_ro_task_update = {}
2271
2272 def _update_refresh(new_status):
2273 # compute next_refresh
2274 nonlocal task
2275 nonlocal next_check_at
2276 nonlocal db_ro_task_update
2277 nonlocal ro_task
2278
2279 next_refresh = time.time()
sousaedu80135b92021-02-17 15:05:18 +01002280
tierno70eeb182020-10-19 16:38:00 +00002281 if task["item"] in ("image", "flavor"):
aticig973bed22022-06-30 19:29:04 +03002282 next_refresh += self.refresh_config.image
tierno70eeb182020-10-19 16:38:00 +00002283 elif new_status == "BUILD":
aticig973bed22022-06-30 19:29:04 +03002284 next_refresh += self.refresh_config.build
tierno70eeb182020-10-19 16:38:00 +00002285 elif new_status == "DONE":
aticig973bed22022-06-30 19:29:04 +03002286 if self.refresh_config.active == -1:
2287 next_refresh = -1
2288 else:
2289 next_refresh += self.refresh_config.active
tierno70eeb182020-10-19 16:38:00 +00002290 else:
aticig973bed22022-06-30 19:29:04 +03002291 next_refresh += self.refresh_config.error
sousaedu80135b92021-02-17 15:05:18 +01002292
tierno70eeb182020-10-19 16:38:00 +00002293 next_check_at = min(next_check_at, next_refresh)
2294 db_ro_task_update["vim_info.refresh_at"] = next_refresh
2295 ro_task["vim_info"]["refresh_at"] = next_refresh
2296
2297 try:
gallardo2f4aaaa2022-01-31 16:50:48 +00002298 """
2299 # Log RO tasks only when loglevel is DEBUG
2300 if self.logger.getEffectiveLevel() == logging.DEBUG:
2301 self._log_ro_task(ro_task, None, None, "TASK_WF", "GET_TASK")
2302 """
aticig973bed22022-06-30 19:29:04 +03002303 # Check if vim status refresh is enabled again
2304 self.update_vm_refresh()
tiernof1b640f2020-12-09 15:06:01 +00002305 # 0: get task_status_create
2306 lock_object = None
tierno70eeb182020-10-19 16:38:00 +00002307 task_status_create = None
sousaedu80135b92021-02-17 15:05:18 +01002308 task_create = next(
2309 (
2310 t
2311 for t in ro_task["tasks"]
2312 if t
2313 and t["action"] == "CREATE"
2314 and t["status"] in ("BUILD", "DONE")
2315 ),
2316 None,
2317 )
2318
tierno70eeb182020-10-19 16:38:00 +00002319 if task_create:
2320 task_status_create = task_create["status"]
sousaedu80135b92021-02-17 15:05:18 +01002321
tiernof1b640f2020-12-09 15:06:01 +00002322 # 1: look for tasks in status SCHEDULED, or in status CREATE if action is DONE or BUILD
tierno70eeb182020-10-19 16:38:00 +00002323 for task_action in ("DELETE", "CREATE", "EXEC"):
2324 db_vim_update = None
2325 new_status = None
sousaedu80135b92021-02-17 15:05:18 +01002326
tierno70eeb182020-10-19 16:38:00 +00002327 for task_index, task in enumerate(ro_task["tasks"]):
2328 if not task:
2329 continue # task deleted
sousaedu80135b92021-02-17 15:05:18 +01002330
tierno55fa0bb2020-12-08 23:11:53 +00002331 task_depends = {}
tierno70eeb182020-10-19 16:38:00 +00002332 target_update = None
sousaedu80135b92021-02-17 15:05:18 +01002333
2334 if (
2335 (
2336 task_action in ("DELETE", "EXEC")
2337 and task["status"] not in ("SCHEDULED", "BUILD")
2338 )
2339 or task["action"] != task_action
2340 or (
2341 task_action == "CREATE"
2342 and task["status"] in ("FINISHED", "SUPERSEDED")
2343 )
2344 ):
tierno70eeb182020-10-19 16:38:00 +00002345 continue
sousaedu80135b92021-02-17 15:05:18 +01002346
tierno70eeb182020-10-19 16:38:00 +00002347 task_path = "tasks.{}.status".format(task_index)
2348 try:
2349 db_vim_info_update = None
sousaedu80135b92021-02-17 15:05:18 +01002350
tierno70eeb182020-10-19 16:38:00 +00002351 if task["status"] == "SCHEDULED":
tierno70eeb182020-10-19 16:38:00 +00002352 # check if tasks that this depends on have been completed
2353 dependency_not_completed = False
sousaedu80135b92021-02-17 15:05:18 +01002354
2355 for dependency_task_id in task.get("depends_on") or ():
2356 (
2357 dependency_ro_task,
2358 dependency_task_index,
2359 ) = self._get_dependency(
2360 dependency_task_id, target_id=ro_task["target_id"]
2361 )
2362 dependency_task = dependency_ro_task["tasks"][
2363 dependency_task_index
2364 ]
aticig285185e2022-05-02 21:23:48 +03002365 self.logger.warning(
2366 "dependency_ro_task={} dependency_task_index={}".format(
2367 dependency_ro_task, dependency_task_index
2368 )
2369 )
sousaedu80135b92021-02-17 15:05:18 +01002370
tierno70eeb182020-10-19 16:38:00 +00002371 if dependency_task["status"] == "SCHEDULED":
2372 dependency_not_completed = True
sousaedu80135b92021-02-17 15:05:18 +01002373 next_check_at = min(
2374 next_check_at, dependency_ro_task["to_check_at"]
2375 )
lloretgalleg88486222021-02-19 12:06:52 +00002376 # must allow dependent task to be processed first
2377 # to do this set time after last_task_processed
2378 next_check_at = max(
2379 self.time_last_task_processed, next_check_at
2380 )
tierno70eeb182020-10-19 16:38:00 +00002381 break
2382 elif dependency_task["status"] == "FAILED":
2383 error_text = "Cannot {} {} because depends on failed {} {} id={}): {}".format(
sousaedu80135b92021-02-17 15:05:18 +01002384 task["action"],
2385 task["item"],
2386 dependency_task["action"],
2387 dependency_task["item"],
2388 dependency_task_id,
2389 dependency_ro_task["vim_info"].get(
aticig79ac6df2022-05-06 16:09:52 +03002390 "vim_message"
sousaedu80135b92021-02-17 15:05:18 +01002391 ),
2392 )
2393 self.logger.error(
2394 "task={} {}".format(task["task_id"], error_text)
2395 )
tierno70eeb182020-10-19 16:38:00 +00002396 raise NsWorkerException(error_text)
2397
sousaedu80135b92021-02-17 15:05:18 +01002398 task_depends[dependency_task_id] = dependency_ro_task[
2399 "vim_info"
2400 ]["vim_id"]
2401 task_depends[
2402 "TASK-{}".format(dependency_task_id)
2403 ] = dependency_ro_task["vim_info"]["vim_id"]
2404
tierno70eeb182020-10-19 16:38:00 +00002405 if dependency_not_completed:
aticig285185e2022-05-02 21:23:48 +03002406 self.logger.warning(
2407 "DEPENDENCY NOT COMPLETED {}".format(
2408 dependency_ro_task["vim_info"]["vim_id"]
2409 )
2410 )
tierno70eeb182020-10-19 16:38:00 +00002411 # TODO set at vim_info.vim_details that it is waiting
2412 continue
sousaedu80135b92021-02-17 15:05:18 +01002413
tiernof1b640f2020-12-09 15:06:01 +00002414 # before calling VIM-plugin as it can take more than task_locked_time, insert to LockRenew
2415 # the task of renew this locking. It will update database locket_at periodically
2416 if not lock_object:
sousaedu80135b92021-02-17 15:05:18 +01002417 lock_object = LockRenew.add_lock_object(
2418 "ro_tasks", ro_task, self
2419 )
2420
tierno70eeb182020-10-19 16:38:00 +00002421 if task["action"] == "DELETE":
sousaedu80135b92021-02-17 15:05:18 +01002422 (new_status, db_vim_info_update,) = self._delete_task(
2423 ro_task, task_index, task_depends, db_ro_task_update
2424 )
2425 new_status = (
2426 "FINISHED" if new_status == "DONE" else new_status
2427 )
tierno70eeb182020-10-19 16:38:00 +00002428 # ^with FINISHED instead of DONE it will not be refreshing
sousaedu80135b92021-02-17 15:05:18 +01002429
tierno70eeb182020-10-19 16:38:00 +00002430 if new_status in ("FINISHED", "SUPERSEDED"):
2431 target_update = "DELETE"
2432 elif task["action"] == "EXEC":
sousaedu80135b92021-02-17 15:05:18 +01002433 (
2434 new_status,
2435 db_vim_info_update,
2436 db_task_update,
2437 ) = self.item2class[task["item"]].exec(
2438 ro_task, task_index, task_depends
2439 )
2440 new_status = (
2441 "FINISHED" if new_status == "DONE" else new_status
2442 )
tierno70eeb182020-10-19 16:38:00 +00002443 # ^with FINISHED instead of DONE it will not be refreshing
sousaedu80135b92021-02-17 15:05:18 +01002444
tierno70eeb182020-10-19 16:38:00 +00002445 if db_task_update:
2446 # load into database the modified db_task_update "retries" and "next_retry"
2447 if db_task_update.get("retries"):
sousaedu80135b92021-02-17 15:05:18 +01002448 db_ro_task_update[
2449 "tasks.{}.retries".format(task_index)
2450 ] = db_task_update["retries"]
2451
2452 next_check_at = time.time() + db_task_update.get(
2453 "next_retry", 60
2454 )
tierno70eeb182020-10-19 16:38:00 +00002455 target_update = None
2456 elif task["action"] == "CREATE":
2457 if task["status"] == "SCHEDULED":
2458 if task_status_create:
2459 new_status = task_status_create
2460 target_update = "COPY_VIM_INFO"
2461 else:
sousaedu80135b92021-02-17 15:05:18 +01002462 new_status, db_vim_info_update = self.item2class[
2463 task["item"]
2464 ].new(ro_task, task_index, task_depends)
tierno70eeb182020-10-19 16:38:00 +00002465 # self._create_task(ro_task, task_index, task_depends, db_ro_task_update)
2466 _update_refresh(new_status)
2467 else:
aticig973bed22022-06-30 19:29:04 +03002468 refresh_at = ro_task["vim_info"]["refresh_at"]
2469 if refresh_at and refresh_at != -1 and now > refresh_at:
2470 (new_status, db_vim_info_update,) = self.item2class[
sousaedu80135b92021-02-17 15:05:18 +01002471 task["item"]
2472 ].refresh(ro_task)
tierno70eeb182020-10-19 16:38:00 +00002473 _update_refresh(new_status)
gallardo7788f692022-01-20 09:07:08 +00002474 else:
2475 # The refresh is updated to avoid set the value of "refresh_at" to
2476 # default value (next_check_at = now + (24 * 60 * 60)) when status is BUILD,
2477 # because it can happen that in this case the task is never processed
2478 _update_refresh(task["status"])
sousaedu80135b92021-02-17 15:05:18 +01002479
tierno70eeb182020-10-19 16:38:00 +00002480 except Exception as e:
2481 new_status = "FAILED"
sousaedu80135b92021-02-17 15:05:18 +01002482 db_vim_info_update = {
2483 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +03002484 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +01002485 }
2486
2487 if not isinstance(
2488 e, (NsWorkerException, vimconn.VimConnException)
2489 ):
2490 self.logger.error(
2491 "Unexpected exception at _delete_task task={}: {}".format(
2492 task["task_id"], e
2493 ),
2494 exc_info=True,
2495 )
tierno70eeb182020-10-19 16:38:00 +00002496
2497 try:
2498 if db_vim_info_update:
2499 db_vim_update = db_vim_info_update.copy()
sousaedu80135b92021-02-17 15:05:18 +01002500 db_ro_task_update.update(
2501 {
2502 "vim_info." + k: v
2503 for k, v in db_vim_info_update.items()
2504 }
2505 )
tierno70eeb182020-10-19 16:38:00 +00002506 ro_task["vim_info"].update(db_vim_info_update)
2507
2508 if new_status:
2509 if task_action == "CREATE":
2510 task_status_create = new_status
2511 db_ro_task_update[task_path] = new_status
tierno70eeb182020-10-19 16:38:00 +00002512
sousaedu80135b92021-02-17 15:05:18 +01002513 if target_update or db_vim_update:
tierno70eeb182020-10-19 16:38:00 +00002514 if target_update == "DELETE":
2515 self._update_target(task, None)
2516 elif target_update == "COPY_VIM_INFO":
2517 self._update_target(task, ro_task["vim_info"])
2518 else:
2519 self._update_target(task, db_vim_update)
2520
2521 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002522 if (
2523 isinstance(e, DbException)
2524 and e.http_code == HTTPStatus.NOT_FOUND
2525 ):
tierno70eeb182020-10-19 16:38:00 +00002526 # if the vnfrs or nsrs has been removed from database, this task must be removed
sousaedu80135b92021-02-17 15:05:18 +01002527 self.logger.debug(
2528 "marking to delete task={}".format(task["task_id"])
2529 )
tierno70eeb182020-10-19 16:38:00 +00002530 self.tasks_to_delete.append(task)
2531 else:
sousaedu80135b92021-02-17 15:05:18 +01002532 self.logger.error(
2533 "Unexpected exception at _update_target task={}: {}".format(
2534 task["task_id"], e
2535 ),
2536 exc_info=True,
2537 )
tierno70eeb182020-10-19 16:38:00 +00002538
tiernof1b640f2020-12-09 15:06:01 +00002539 locked_at = ro_task["locked_at"]
sousaedu80135b92021-02-17 15:05:18 +01002540
tiernof1b640f2020-12-09 15:06:01 +00002541 if lock_object:
sousaedu80135b92021-02-17 15:05:18 +01002542 locked_at = [
2543 lock_object["locked_at"],
2544 lock_object["locked_at"] + self.task_locked_time,
2545 ]
tiernof1b640f2020-12-09 15:06:01 +00002546 # locked_at contains two times to avoid race condition. In case the lock has been renew, it will
2547 # contain exactly locked_at + self.task_locked_time
2548 LockRenew.remove_lock_object(lock_object)
sousaedu80135b92021-02-17 15:05:18 +01002549
2550 q_filter = {
2551 "_id": ro_task["_id"],
2552 "to_check_at": ro_task["to_check_at"],
2553 "locked_at": locked_at,
2554 }
tierno70eeb182020-10-19 16:38:00 +00002555 # modify own task. Try filtering by to_next_check. For race condition if to_check_at has been modified,
2556 # outside this task (by ro_nbi) do not update it
2557 db_ro_task_update["locked_by"] = None
2558 # locked_at converted to int only for debugging. When has not decimals it means it has been unlocked
tiernof1b640f2020-12-09 15:06:01 +00002559 db_ro_task_update["locked_at"] = int(now - self.task_locked_time)
2560 db_ro_task_update["modified_at"] = now
tierno70eeb182020-10-19 16:38:00 +00002561 db_ro_task_update["to_check_at"] = next_check_at
sousaedu80135b92021-02-17 15:05:18 +01002562
gallardo2f4aaaa2022-01-31 16:50:48 +00002563 """
2564 # Log RO tasks only when loglevel is DEBUG
2565 if self.logger.getEffectiveLevel() == logging.DEBUG:
2566 db_ro_task_update_log = db_ro_task_update.copy()
2567 db_ro_task_update_log["_id"] = q_filter["_id"]
2568 self._log_ro_task(None, db_ro_task_update_log, None, "TASK_WF", "SET_TASK")
2569 """
2570
sousaedu80135b92021-02-17 15:05:18 +01002571 if not self.db.set_one(
2572 "ro_tasks",
2573 update_dict=db_ro_task_update,
2574 q_filter=q_filter,
2575 fail_on_empty=False,
2576 ):
tierno70eeb182020-10-19 16:38:00 +00002577 del db_ro_task_update["to_check_at"]
2578 del q_filter["to_check_at"]
gallardo2f4aaaa2022-01-31 16:50:48 +00002579 """
2580 # Log RO tasks only when loglevel is DEBUG
2581 if self.logger.getEffectiveLevel() == logging.DEBUG:
2582 self._log_ro_task(
2583 None,
2584 db_ro_task_update_log,
2585 None,
2586 "TASK_WF",
2587 "SET_TASK " + str(q_filter),
2588 )
2589 """
sousaedu80135b92021-02-17 15:05:18 +01002590 self.db.set_one(
2591 "ro_tasks",
2592 q_filter=q_filter,
2593 update_dict=db_ro_task_update,
2594 fail_on_empty=True,
2595 )
tierno70eeb182020-10-19 16:38:00 +00002596 except DbException as e:
sousaedu80135b92021-02-17 15:05:18 +01002597 self.logger.error(
2598 "ro_task={} Error updating database {}".format(ro_task_id, e)
2599 )
tierno70eeb182020-10-19 16:38:00 +00002600 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002601 self.logger.error(
2602 "Error executing ro_task={}: {}".format(ro_task_id, e), exc_info=True
2603 )
tierno70eeb182020-10-19 16:38:00 +00002604
2605 def _update_target(self, task, ro_vim_item_update):
2606 table, _, temp = task["target_record"].partition(":")
2607 _id, _, path_vim_status = temp.partition(":")
sousaedu80135b92021-02-17 15:05:18 +01002608 path_item = path_vim_status[: path_vim_status.rfind(".")]
2609 path_item = path_item[: path_item.rfind(".")]
tierno70eeb182020-10-19 16:38:00 +00002610 # path_vim_status: dot separated list targeting vim information, e.g. "vdur.10.vim_info.vim:id"
2611 # path_item: dot separated list targeting record information, e.g. "vdur.10"
sousaedu80135b92021-02-17 15:05:18 +01002612
tierno70eeb182020-10-19 16:38:00 +00002613 if ro_vim_item_update:
sousaedu80135b92021-02-17 15:05:18 +01002614 update_dict = {
2615 path_vim_status + "." + k: v
2616 for k, v in ro_vim_item_update.items()
2617 if k
aticig79ac6df2022-05-06 16:09:52 +03002618 in (
2619 "vim_id",
2620 "vim_details",
2621 "vim_message",
2622 "vim_name",
2623 "vim_status",
2624 "interfaces",
aticig37ecec02022-05-25 03:12:36 +03002625 "interfaces_backup",
aticig79ac6df2022-05-06 16:09:52 +03002626 )
sousaedu80135b92021-02-17 15:05:18 +01002627 }
2628
tierno70eeb182020-10-19 16:38:00 +00002629 if path_vim_status.startswith("vdur."):
2630 # for backward compatibility, add vdur.name apart from vdur.vim_name
2631 if ro_vim_item_update.get("vim_name"):
2632 update_dict[path_item + ".name"] = ro_vim_item_update["vim_name"]
sousaedu80135b92021-02-17 15:05:18 +01002633
tierno70eeb182020-10-19 16:38:00 +00002634 # for backward compatibility, add vdur.vim-id apart from vdur.vim_id
2635 if ro_vim_item_update.get("vim_id"):
2636 update_dict[path_item + ".vim-id"] = ro_vim_item_update["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +01002637
tierno70eeb182020-10-19 16:38:00 +00002638 # update general status
2639 if ro_vim_item_update.get("vim_status"):
sousaedu80135b92021-02-17 15:05:18 +01002640 update_dict[path_item + ".status"] = ro_vim_item_update[
2641 "vim_status"
2642 ]
2643
tierno70eeb182020-10-19 16:38:00 +00002644 if ro_vim_item_update.get("interfaces"):
2645 path_interfaces = path_item + ".interfaces"
sousaedu80135b92021-02-17 15:05:18 +01002646
tierno70eeb182020-10-19 16:38:00 +00002647 for i, iface in enumerate(ro_vim_item_update.get("interfaces")):
2648 if iface:
sousaedu80135b92021-02-17 15:05:18 +01002649 update_dict.update(
2650 {
2651 path_interfaces + ".{}.".format(i) + k: v
2652 for k, v in iface.items()
2653 if k in ("vlan", "compute_node", "pci")
2654 }
2655 )
2656
tierno70eeb182020-10-19 16:38:00 +00002657 # put ip_address and mac_address with ip-address and mac-address
sousaedu80135b92021-02-17 15:05:18 +01002658 if iface.get("ip_address"):
2659 update_dict[
2660 path_interfaces + ".{}.".format(i) + "ip-address"
2661 ] = iface["ip_address"]
2662
2663 if iface.get("mac_address"):
2664 update_dict[
2665 path_interfaces + ".{}.".format(i) + "mac-address"
2666 ] = iface["mac_address"]
2667
tierno70eeb182020-10-19 16:38:00 +00002668 if iface.get("mgmt_vnf_interface") and iface.get("ip_address"):
sousaedu80135b92021-02-17 15:05:18 +01002669 update_dict["ip-address"] = iface.get("ip_address").split(
2670 ";"
2671 )[0]
2672
tierno70eeb182020-10-19 16:38:00 +00002673 if iface.get("mgmt_vdu_interface") and iface.get("ip_address"):
sousaedu80135b92021-02-17 15:05:18 +01002674 update_dict[path_item + ".ip-address"] = iface.get(
2675 "ip_address"
2676 ).split(";")[0]
tierno70eeb182020-10-19 16:38:00 +00002677
2678 self.db.set_one(table, q_filter={"_id": _id}, update_dict=update_dict)
aticig37ecec02022-05-25 03:12:36 +03002679
2680 # If interfaces exists, it backups VDU interfaces in the DB for healing operations
2681 if ro_vim_item_update.get("interfaces"):
2682 search_key = path_vim_status + ".interfaces"
2683 if update_dict.get(search_key):
2684 interfaces_backup_update = {
2685 path_vim_status + ".interfaces_backup": update_dict[search_key]
2686 }
2687
2688 self.db.set_one(
2689 table,
2690 q_filter={"_id": _id},
2691 update_dict=interfaces_backup_update,
2692 )
2693
tierno70eeb182020-10-19 16:38:00 +00002694 else:
2695 update_dict = {path_item + ".status": "DELETED"}
sousaedu80135b92021-02-17 15:05:18 +01002696 self.db.set_one(
2697 table,
2698 q_filter={"_id": _id},
2699 update_dict=update_dict,
2700 unset={path_vim_status: None},
2701 )
tierno70eeb182020-10-19 16:38:00 +00002702
2703 def _process_delete_db_tasks(self):
2704 """
2705 Delete task from database because vnfrs or nsrs or both have been deleted
2706 :return: None. Uses and modify self.tasks_to_delete
2707 """
2708 while self.tasks_to_delete:
2709 task = self.tasks_to_delete[0]
2710 vnfrs_deleted = None
2711 nsr_id = task["nsr_id"]
sousaedu80135b92021-02-17 15:05:18 +01002712
tierno70eeb182020-10-19 16:38:00 +00002713 if task["target_record"].startswith("vnfrs:"):
2714 # check if nsrs is present
2715 if self.db.get_one("nsrs", {"_id": nsr_id}, fail_on_empty=False):
2716 vnfrs_deleted = task["target_record"].split(":")[1]
sousaedu80135b92021-02-17 15:05:18 +01002717
tierno70eeb182020-10-19 16:38:00 +00002718 try:
2719 self.delete_db_tasks(self.db, nsr_id, vnfrs_deleted)
2720 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002721 self.logger.error(
2722 "Error deleting task={}: {}".format(task["task_id"], e)
2723 )
tierno70eeb182020-10-19 16:38:00 +00002724 self.tasks_to_delete.pop(0)
2725
2726 @staticmethod
2727 def delete_db_tasks(db, nsr_id, vnfrs_deleted):
2728 """
2729 Static method because it is called from osm_ng_ro.ns
2730 :param db: instance of database to use
2731 :param nsr_id: affected nsrs id
2732 :param vnfrs_deleted: only tasks with this vnfr id. If None, all affected by nsr_id
2733 :return: None, exception is fails
2734 """
2735 retries = 5
2736 for retry in range(retries):
2737 ro_tasks = db.get_list("ro_tasks", {"tasks.nsr_id": nsr_id})
2738 now = time.time()
2739 conflict = False
sousaedu80135b92021-02-17 15:05:18 +01002740
tierno70eeb182020-10-19 16:38:00 +00002741 for ro_task in ro_tasks:
2742 db_update = {}
2743 to_delete_ro_task = True
sousaedu80135b92021-02-17 15:05:18 +01002744
tierno70eeb182020-10-19 16:38:00 +00002745 for index, task in enumerate(ro_task["tasks"]):
2746 if not task:
2747 pass
sousaedu80135b92021-02-17 15:05:18 +01002748 elif (not vnfrs_deleted and task["nsr_id"] == nsr_id) or (
2749 vnfrs_deleted
2750 and task["target_record"].startswith("vnfrs:" + vnfrs_deleted)
2751 ):
tierno70eeb182020-10-19 16:38:00 +00002752 db_update["tasks.{}".format(index)] = None
2753 else:
sousaedu80135b92021-02-17 15:05:18 +01002754 # used by other nsr, ro_task cannot be deleted
2755 to_delete_ro_task = False
2756
tierno70eeb182020-10-19 16:38:00 +00002757 # delete or update if nobody has changed ro_task meanwhile. Used modified_at for known if changed
2758 if to_delete_ro_task:
sousaedu80135b92021-02-17 15:05:18 +01002759 if not db.del_one(
2760 "ro_tasks",
2761 q_filter={
2762 "_id": ro_task["_id"],
2763 "modified_at": ro_task["modified_at"],
2764 },
2765 fail_on_empty=False,
2766 ):
tierno70eeb182020-10-19 16:38:00 +00002767 conflict = True
2768 elif db_update:
2769 db_update["modified_at"] = now
sousaedu80135b92021-02-17 15:05:18 +01002770 if not db.set_one(
2771 "ro_tasks",
2772 q_filter={
2773 "_id": ro_task["_id"],
2774 "modified_at": ro_task["modified_at"],
2775 },
2776 update_dict=db_update,
2777 fail_on_empty=False,
2778 ):
tierno70eeb182020-10-19 16:38:00 +00002779 conflict = True
2780 if not conflict:
2781 return
2782 else:
2783 raise NsWorkerException("Exceeded {} retries".format(retries))
2784
tierno1d213f42020-04-24 14:02:51 +00002785 def run(self):
2786 # load database
tierno86153522020-12-06 18:27:16 +00002787 self.logger.info("Starting")
tierno1d213f42020-04-24 14:02:51 +00002788 while True:
tierno70eeb182020-10-19 16:38:00 +00002789 # step 1: get commands from queue
tierno1d213f42020-04-24 14:02:51 +00002790 try:
tierno86153522020-12-06 18:27:16 +00002791 if self.vim_targets:
2792 task = self.task_queue.get(block=False)
2793 else:
2794 if not self.idle:
2795 self.logger.debug("enters in idle state")
2796 self.idle = True
2797 task = self.task_queue.get(block=True)
2798 self.idle = False
2799
tierno1d213f42020-04-24 14:02:51 +00002800 if task[0] == "terminate":
2801 break
tierno70eeb182020-10-19 16:38:00 +00002802 elif task[0] == "load_vim":
tierno86153522020-12-06 18:27:16 +00002803 self.logger.info("order to load vim {}".format(task[1]))
tierno1d213f42020-04-24 14:02:51 +00002804 self._load_vim(task[1])
tierno70eeb182020-10-19 16:38:00 +00002805 elif task[0] == "unload_vim":
tierno86153522020-12-06 18:27:16 +00002806 self.logger.info("order to unload vim {}".format(task[1]))
tierno70eeb182020-10-19 16:38:00 +00002807 self._unload_vim(task[1])
2808 elif task[0] == "reload_vim":
2809 self._reload_vim(task[1])
2810 elif task[0] == "check_vim":
tierno86153522020-12-06 18:27:16 +00002811 self.logger.info("order to check vim {}".format(task[1]))
tierno70eeb182020-10-19 16:38:00 +00002812 self._check_vim(task[1])
tierno1d213f42020-04-24 14:02:51 +00002813 continue
tierno70eeb182020-10-19 16:38:00 +00002814 except Exception as e:
2815 if isinstance(e, queue.Empty):
2816 pass
2817 else:
sousaedu80135b92021-02-17 15:05:18 +01002818 self.logger.critical(
2819 "Error processing task: {}".format(e), exc_info=True
2820 )
tierno1d213f42020-04-24 14:02:51 +00002821
tierno70eeb182020-10-19 16:38:00 +00002822 # step 2: process pending_tasks, delete not needed tasks
tierno1d213f42020-04-24 14:02:51 +00002823 try:
tierno70eeb182020-10-19 16:38:00 +00002824 if self.tasks_to_delete:
2825 self._process_delete_db_tasks()
tierno1d213f42020-04-24 14:02:51 +00002826 busy = False
gallardo2f4aaaa2022-01-31 16:50:48 +00002827 """
2828 # Log RO tasks only when loglevel is DEBUG
2829 if self.logger.getEffectiveLevel() == logging.DEBUG:
2830 _ = self._get_db_all_tasks()
2831 """
tierno1d213f42020-04-24 14:02:51 +00002832 ro_task = self._get_db_task()
2833 if ro_task:
palaciosj8f2060b2022-02-24 12:05:59 +00002834 self.logger.warning("Task to process: {}".format(ro_task))
2835 time.sleep(1)
tierno70eeb182020-10-19 16:38:00 +00002836 self._process_pending_tasks(ro_task)
tierno1d213f42020-04-24 14:02:51 +00002837 busy = True
2838 if not busy:
2839 time.sleep(5)
2840 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002841 self.logger.critical(
2842 "Unexpected exception at run: " + str(e), exc_info=True
2843 )
tierno1d213f42020-04-24 14:02:51 +00002844
tierno86153522020-12-06 18:27:16 +00002845 self.logger.info("Finishing")