blob: e64db287d206e97a9a5ad874b02486de38148ec3 [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
Gulsum Atici1f58a1a2023-06-19 13:59:46 +030020"""
tierno1d213f42020-04-24 14:02:51 +000021This 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
Guillermo Calvinoffee6602022-08-19 13:01:06 +020030from os import makedirs
31from os import path
sousaedu049cbb12022-01-05 11:39:35 +000032import queue
sousaedu049cbb12022-01-05 11:39:35 +000033import threading
34import time
aticige5d78422022-05-16 23:03:54 +030035import traceback
aticig1ac189e2022-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
Guillermo Calvinoffee6602022-08-19 13:01:06 +020042from osm_ro_plugin import sdnconn
43from osm_ro_plugin import vimconn
sousaedu049cbb12022-01-05 11:39:35 +000044from osm_ro_plugin.sdn_dummy import SdnDummyConnector
45from osm_ro_plugin.vim_dummy import VimDummyConnector
46import yaml
tierno1d213f42020-04-24 14:02:51 +000047
48__author__ = "Alfonso Tierno"
49__date__ = "$28-Sep-2017 12:07:15$"
50
51
52def deep_get(target_dict, *args, **kwargs):
53 """
54 Get a value from target_dict entering in the nested keys. If keys does not exist, it returns None
55 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
56 :param target_dict: dictionary to be read
57 :param args: list of keys to read from target_dict
58 :param kwargs: only can contain default=value to return if key is not present in the nested dictionary
Gulsum Atici1f58a1a2023-06-19 13:59:46 +030059 :return: The wanted value if exists, None or default otherwise
tierno1d213f42020-04-24 14:02:51 +000060 """
61 for key in args:
62 if not isinstance(target_dict, dict) or key not in target_dict:
63 return kwargs.get("default")
64 target_dict = target_dict[key]
65 return target_dict
66
67
68class NsWorkerException(Exception):
69 pass
70
71
72class FailingConnector:
73 def __init__(self, error_msg):
74 self.error_msg = error_msg
sousaedu80135b92021-02-17 15:05:18 +010075
tierno1d213f42020-04-24 14:02:51 +000076 for method in dir(vimconn.VimConnector):
77 if method[0] != "_":
sousaedu80135b92021-02-17 15:05:18 +010078 setattr(
79 self, method, Mock(side_effect=vimconn.VimConnException(error_msg))
80 )
81
tierno70eeb182020-10-19 16:38:00 +000082 for method in dir(sdnconn.SdnConnectorBase):
83 if method[0] != "_":
sousaedu80135b92021-02-17 15:05:18 +010084 setattr(
85 self, method, Mock(side_effect=sdnconn.SdnConnectorError(error_msg))
86 )
tierno1d213f42020-04-24 14:02:51 +000087
88
89class NsWorkerExceptionNotFound(NsWorkerException):
90 pass
91
92
tierno70eeb182020-10-19 16:38:00 +000093class VimInteractionBase:
sousaedu80135b92021-02-17 15:05:18 +010094 """Base class to call VIM/SDN for creating, deleting and refresh networks, VMs, flavors, ...
tierno70eeb182020-10-19 16:38:00 +000095 It implements methods that does nothing and return ok"""
sousaedu80135b92021-02-17 15:05:18 +010096
tierno70eeb182020-10-19 16:38:00 +000097 def __init__(self, db, my_vims, db_vims, logger):
tierno1d213f42020-04-24 14:02:51 +000098 self.db = db
tierno70eeb182020-10-19 16:38:00 +000099 self.logger = logger
100 self.my_vims = my_vims
101 self.db_vims = db_vims
tierno1d213f42020-04-24 14:02:51 +0000102
tierno70eeb182020-10-19 16:38:00 +0000103 def new(self, ro_task, task_index, task_depends):
104 return "BUILD", {}
tierno1d213f42020-04-24 14:02:51 +0000105
tierno70eeb182020-10-19 16:38:00 +0000106 def refresh(self, ro_task):
107 """skip calling VIM to get image, flavor status. Assumes ok"""
tierno1d213f42020-04-24 14:02:51 +0000108 if ro_task["vim_info"]["vim_status"] == "VIM_ERROR":
109 return "FAILED", {}
sousaedu80135b92021-02-17 15:05:18 +0100110
tierno1d213f42020-04-24 14:02:51 +0000111 return "DONE", {}
112
tierno70eeb182020-10-19 16:38:00 +0000113 def delete(self, ro_task, task_index):
114 """skip calling VIM to delete image. Assumes ok"""
tierno1d213f42020-04-24 14:02:51 +0000115 return "DONE", {}
116
tierno70eeb182020-10-19 16:38:00 +0000117 def exec(self, ro_task, task_index, task_depends):
118 return "DONE", None, None
tierno1d213f42020-04-24 14:02:51 +0000119
tierno1d213f42020-04-24 14:02:51 +0000120
tierno70eeb182020-10-19 16:38:00 +0000121class VimInteractionNet(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +0000122 def new(self, ro_task, task_index, task_depends):
tierno1d213f42020-04-24 14:02:51 +0000123 vim_net_id = None
124 task = ro_task["tasks"][task_index]
125 task_id = task["task_id"]
126 created = False
127 created_items = {}
128 target_vim = self.my_vims[ro_task["target_id"]]
aticige9a26f22021-12-10 12:59:20 +0300129 mgmtnet = False
130 mgmtnet_defined_in_vim = False
sousaedu80135b92021-02-17 15:05:18 +0100131
tierno1d213f42020-04-24 14:02:51 +0000132 try:
133 # FIND
134 if task.get("find_params"):
135 # if management, get configuration of VIM
136 if task["find_params"].get("filter_dict"):
137 vim_filter = task["find_params"]["filter_dict"]
aticige9a26f22021-12-10 12:59:20 +0300138 # management network
sousaedu80135b92021-02-17 15:05:18 +0100139 elif task["find_params"].get("mgmt"):
aticige9a26f22021-12-10 12:59:20 +0300140 mgmtnet = True
sousaedu80135b92021-02-17 15:05:18 +0100141 if deep_get(
142 self.db_vims[ro_task["target_id"]],
143 "config",
144 "management_network_id",
145 ):
aticige9a26f22021-12-10 12:59:20 +0300146 mgmtnet_defined_in_vim = True
sousaedu80135b92021-02-17 15:05:18 +0100147 vim_filter = {
148 "id": self.db_vims[ro_task["target_id"]]["config"][
149 "management_network_id"
150 ]
151 }
152 elif deep_get(
153 self.db_vims[ro_task["target_id"]],
154 "config",
155 "management_network_name",
156 ):
aticige9a26f22021-12-10 12:59:20 +0300157 mgmtnet_defined_in_vim = True
sousaedu80135b92021-02-17 15:05:18 +0100158 vim_filter = {
159 "name": self.db_vims[ro_task["target_id"]]["config"][
160 "management_network_name"
161 ]
162 }
tierno1d213f42020-04-24 14:02:51 +0000163 else:
164 vim_filter = {"name": task["find_params"]["name"]}
165 else:
sousaedu80135b92021-02-17 15:05:18 +0100166 raise NsWorkerExceptionNotFound(
167 "Invalid find_params for new_net {}".format(task["find_params"])
168 )
tierno1d213f42020-04-24 14:02:51 +0000169
170 vim_nets = target_vim.get_network_list(vim_filter)
171 if not vim_nets and not task.get("params"):
aticige9a26f22021-12-10 12:59:20 +0300172 # If there is mgmt-network in the descriptor,
173 # there is no mapping of that network to a VIM network in the descriptor,
174 # also there is no mapping in the "--config" parameter or at VIM creation;
175 # that mgmt-network will be created.
176 if mgmtnet and not mgmtnet_defined_in_vim:
177 net_name = (
178 vim_filter.get("name")
179 if vim_filter.get("name")
180 else vim_filter.get("id")[:16]
sousaedu80135b92021-02-17 15:05:18 +0100181 )
aticige9a26f22021-12-10 12:59:20 +0300182 vim_net_id, created_items = target_vim.new_network(
183 net_name, None
184 )
185 self.logger.debug(
186 "Created mgmt network vim_net_id: {}".format(vim_net_id)
187 )
188 created = True
189 else:
190 raise NsWorkerExceptionNotFound(
191 "Network not found with this criteria: '{}'".format(
192 task.get("find_params")
193 )
194 )
tierno1d213f42020-04-24 14:02:51 +0000195 elif len(vim_nets) > 1:
196 raise NsWorkerException(
sousaedu80135b92021-02-17 15:05:18 +0100197 "More than one network found with this criteria: '{}'".format(
198 task["find_params"]
199 )
200 )
201
tierno1d213f42020-04-24 14:02:51 +0000202 if vim_nets:
203 vim_net_id = vim_nets[0]["id"]
204 else:
205 # CREATE
206 params = task["params"]
207 vim_net_id, created_items = target_vim.new_network(**params)
208 created = True
209
sousaedu80135b92021-02-17 15:05:18 +0100210 ro_vim_item_update = {
211 "vim_id": vim_net_id,
212 "vim_status": "BUILD",
213 "created": created,
214 "created_items": created_items,
215 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300216 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100217 }
tierno1d213f42020-04-24 14:02:51 +0000218 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100219 "task={} {} new-net={} created={}".format(
220 task_id, ro_task["target_id"], vim_net_id, created
221 )
222 )
223
tierno1d213f42020-04-24 14:02:51 +0000224 return "BUILD", ro_vim_item_update
225 except (vimconn.VimConnException, NsWorkerException) as e:
sousaedu80135b92021-02-17 15:05:18 +0100226 self.logger.error(
227 "task={} vim={} new-net: {}".format(task_id, ro_task["target_id"], e)
228 )
229 ro_vim_item_update = {
230 "vim_status": "VIM_ERROR",
231 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300232 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100233 }
234
tierno1d213f42020-04-24 14:02:51 +0000235 return "FAILED", ro_vim_item_update
236
tierno70eeb182020-10-19 16:38:00 +0000237 def refresh(self, ro_task):
tierno1d213f42020-04-24 14:02:51 +0000238 """Call VIM to get network status"""
239 ro_task_id = ro_task["_id"]
240 target_vim = self.my_vims[ro_task["target_id"]]
tierno1d213f42020-04-24 14:02:51 +0000241 vim_id = ro_task["vim_info"]["vim_id"]
242 net_to_refresh_list = [vim_id]
sousaedu80135b92021-02-17 15:05:18 +0100243
tierno1d213f42020-04-24 14:02:51 +0000244 try:
245 vim_dict = target_vim.refresh_nets_status(net_to_refresh_list)
246 vim_info = vim_dict[vim_id]
sousaedu80135b92021-02-17 15:05:18 +0100247
tierno1d213f42020-04-24 14:02:51 +0000248 if vim_info["status"] == "ACTIVE":
249 task_status = "DONE"
250 elif vim_info["status"] == "BUILD":
251 task_status = "BUILD"
252 else:
253 task_status = "FAILED"
254 except vimconn.VimConnException as e:
255 # Mark all tasks at VIM_ERROR status
sousaedu80135b92021-02-17 15:05:18 +0100256 self.logger.error(
257 "ro_task={} vim={} get-net={}: {}".format(
258 ro_task_id, ro_task["target_id"], vim_id, e
259 )
260 )
tierno1d213f42020-04-24 14:02:51 +0000261 vim_info = {"status": "VIM_ERROR", "error_msg": str(e)}
262 task_status = "FAILED"
263
264 ro_vim_item_update = {}
265 if ro_task["vim_info"]["vim_status"] != vim_info["status"]:
266 ro_vim_item_update["vim_status"] = vim_info["status"]
sousaedu80135b92021-02-17 15:05:18 +0100267
tierno1d213f42020-04-24 14:02:51 +0000268 if ro_task["vim_info"]["vim_name"] != vim_info.get("name"):
269 ro_vim_item_update["vim_name"] = vim_info.get("name")
sousaedu80135b92021-02-17 15:05:18 +0100270
tierno1d213f42020-04-24 14:02:51 +0000271 if vim_info["status"] in ("ERROR", "VIM_ERROR"):
aticig79ac6df2022-05-06 16:09:52 +0300272 if ro_task["vim_info"]["vim_message"] != vim_info.get("error_msg"):
273 ro_vim_item_update["vim_message"] = vim_info.get("error_msg")
tierno1d213f42020-04-24 14:02:51 +0000274 elif vim_info["status"] == "DELETED":
275 ro_vim_item_update["vim_id"] = None
aticig79ac6df2022-05-06 16:09:52 +0300276 ro_vim_item_update["vim_message"] = "Deleted externally"
tierno1d213f42020-04-24 14:02:51 +0000277 else:
278 if ro_task["vim_info"]["vim_details"] != vim_info["vim_info"]:
279 ro_vim_item_update["vim_details"] = vim_info["vim_info"]
sousaedu80135b92021-02-17 15:05:18 +0100280
tierno1d213f42020-04-24 14:02:51 +0000281 if ro_vim_item_update:
sousaedu80135b92021-02-17 15:05:18 +0100282 self.logger.debug(
283 "ro_task={} {} get-net={}: status={} {}".format(
284 ro_task_id,
285 ro_task["target_id"],
286 vim_id,
287 ro_vim_item_update.get("vim_status"),
aticig79ac6df2022-05-06 16:09:52 +0300288 ro_vim_item_update.get("vim_message")
sousaedu80135b92021-02-17 15:05:18 +0100289 if ro_vim_item_update.get("vim_status") != "ACTIVE"
290 else "",
291 )
292 )
293
tierno1d213f42020-04-24 14:02:51 +0000294 return task_status, ro_vim_item_update
295
tierno70eeb182020-10-19 16:38:00 +0000296 def delete(self, ro_task, task_index):
tierno1d213f42020-04-24 14:02:51 +0000297 task = ro_task["tasks"][task_index]
298 task_id = task["task_id"]
299 net_vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100300 ro_vim_item_update_ok = {
301 "vim_status": "DELETED",
302 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300303 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +0100304 "vim_id": None,
305 }
306
tierno1d213f42020-04-24 14:02:51 +0000307 try:
308 if net_vim_id or ro_task["vim_info"]["created_items"]:
309 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100310 target_vim.delete_network(
311 net_vim_id, ro_task["vim_info"]["created_items"]
312 )
tierno1d213f42020-04-24 14:02:51 +0000313 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300314 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno1d213f42020-04-24 14:02:51 +0000315 except vimconn.VimConnException as e:
sousaedu80135b92021-02-17 15:05:18 +0100316 self.logger.error(
317 "ro_task={} vim={} del-net={}: {}".format(
318 ro_task["_id"], ro_task["target_id"], net_vim_id, e
319 )
320 )
321 ro_vim_item_update = {
322 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300323 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +0100324 }
325
tierno1d213f42020-04-24 14:02:51 +0000326 return "FAILED", ro_vim_item_update
327
sousaedu80135b92021-02-17 15:05:18 +0100328 self.logger.debug(
329 "task={} {} del-net={} {}".format(
330 task_id,
331 ro_task["target_id"],
332 net_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300333 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +0100334 )
335 )
336
tierno1d213f42020-04-24 14:02:51 +0000337 return "DONE", ro_vim_item_update_ok
338
tierno70eeb182020-10-19 16:38:00 +0000339
340class VimInteractionVdu(VimInteractionBase):
sousaedu80135b92021-02-17 15:05:18 +0100341 max_retries_inject_ssh_key = 20 # 20 times
342 time_retries_inject_ssh_key = 30 # wevery 30 seconds
tierno70eeb182020-10-19 16:38:00 +0000343
344 def new(self, ro_task, task_index, task_depends):
tierno1d213f42020-04-24 14:02:51 +0000345 task = ro_task["tasks"][task_index]
346 task_id = task["task_id"]
347 created = False
tierno1d213f42020-04-24 14:02:51 +0000348 target_vim = self.my_vims[ro_task["target_id"]]
349 try:
350 created = True
351 params = task["params"]
352 params_copy = deepcopy(params)
353 net_list = params_copy["net_list"]
sousaedu80135b92021-02-17 15:05:18 +0100354
tierno1d213f42020-04-24 14:02:51 +0000355 for net in net_list:
sousaedu80135b92021-02-17 15:05:18 +0100356 # change task_id into network_id
357 if "net_id" in net and net["net_id"].startswith("TASK-"):
tierno1d213f42020-04-24 14:02:51 +0000358 network_id = task_depends[net["net_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100359
tierno1d213f42020-04-24 14:02:51 +0000360 if not network_id:
sousaedu80135b92021-02-17 15:05:18 +0100361 raise NsWorkerException(
362 "Cannot create VM because depends on a network not created or found "
363 "for {}".format(net["net_id"])
364 )
365
tierno1d213f42020-04-24 14:02:51 +0000366 net["net_id"] = network_id
sousaedu80135b92021-02-17 15:05:18 +0100367
tierno1d213f42020-04-24 14:02:51 +0000368 if params_copy["image_id"].startswith("TASK-"):
369 params_copy["image_id"] = task_depends[params_copy["image_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100370
tierno1d213f42020-04-24 14:02:51 +0000371 if params_copy["flavor_id"].startswith("TASK-"):
372 params_copy["flavor_id"] = task_depends[params_copy["flavor_id"]]
373
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100374 affinity_group_list = params_copy["affinity_group_list"]
375 for affinity_group in affinity_group_list:
376 # change task_id into affinity_group_id
377 if "affinity_group_id" in affinity_group and affinity_group[
378 "affinity_group_id"
379 ].startswith("TASK-"):
380 affinity_group_id = task_depends[
381 affinity_group["affinity_group_id"]
382 ]
383
384 if not affinity_group_id:
385 raise NsWorkerException(
386 "found for {}".format(affinity_group["affinity_group_id"])
387 )
388
389 affinity_group["affinity_group_id"] = affinity_group_id
tierno1d213f42020-04-24 14:02:51 +0000390 vim_vm_id, created_items = target_vim.new_vminstance(**params_copy)
391 interfaces = [iface["vim_id"] for iface in params_copy["net_list"]]
392
palaciosj42eb06f2022-05-05 14:59:36 +0000393 # add to created items previous_created_volumes (healing)
394 if task.get("previous_created_volumes"):
395 for k, v in task["previous_created_volumes"].items():
396 created_items[k] = v
397
sousaedu80135b92021-02-17 15:05:18 +0100398 ro_vim_item_update = {
399 "vim_id": vim_vm_id,
400 "vim_status": "BUILD",
401 "created": created,
402 "created_items": created_items,
403 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300404 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100405 "interfaces_vim_ids": interfaces,
406 "interfaces": [],
aticig37ecec02022-05-25 03:12:36 +0300407 "interfaces_backup": [],
sousaedu80135b92021-02-17 15:05:18 +0100408 }
tierno1d213f42020-04-24 14:02:51 +0000409 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100410 "task={} {} new-vm={} created={}".format(
411 task_id, ro_task["target_id"], vim_vm_id, created
412 )
413 )
414
tierno1d213f42020-04-24 14:02:51 +0000415 return "BUILD", ro_vim_item_update
416 except (vimconn.VimConnException, NsWorkerException) as e:
aticigcf14bb12022-05-19 13:03:17 +0300417 self.logger.debug(traceback.format_exc())
sousaedu80135b92021-02-17 15:05:18 +0100418 self.logger.error(
419 "task={} {} new-vm: {}".format(task_id, ro_task["target_id"], e)
420 )
421 ro_vim_item_update = {
422 "vim_status": "VIM_ERROR",
423 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300424 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100425 }
426
tierno1d213f42020-04-24 14:02:51 +0000427 return "FAILED", ro_vim_item_update
428
tierno70eeb182020-10-19 16:38:00 +0000429 def delete(self, ro_task, task_index):
tierno1d213f42020-04-24 14:02:51 +0000430 task = ro_task["tasks"][task_index]
431 task_id = task["task_id"]
432 vm_vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100433 ro_vim_item_update_ok = {
434 "vim_status": "DELETED",
435 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300436 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +0100437 "vim_id": None,
438 }
439
tierno1d213f42020-04-24 14:02:51 +0000440 try:
palaciosj8f2060b2022-02-24 12:05:59 +0000441 self.logger.debug(
442 "delete_vminstance: vm_vim_id={} created_items={}".format(
443 vm_vim_id, ro_task["vim_info"]["created_items"]
444 )
445 )
tierno1d213f42020-04-24 14:02:51 +0000446 if vm_vim_id or ro_task["vim_info"]["created_items"]:
447 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100448 target_vim.delete_vminstance(
palaciosj8f2060b2022-02-24 12:05:59 +0000449 vm_vim_id,
450 ro_task["vim_info"]["created_items"],
451 ro_task["vim_info"].get("volumes_to_hold", []),
sousaedu80135b92021-02-17 15:05:18 +0100452 )
tierno1d213f42020-04-24 14:02:51 +0000453 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300454 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno1d213f42020-04-24 14:02:51 +0000455 except vimconn.VimConnException as e:
sousaedu80135b92021-02-17 15:05:18 +0100456 self.logger.error(
457 "ro_task={} vim={} del-vm={}: {}".format(
458 ro_task["_id"], ro_task["target_id"], vm_vim_id, e
459 )
460 )
461 ro_vim_item_update = {
462 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300463 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +0100464 }
465
tierno1d213f42020-04-24 14:02:51 +0000466 return "FAILED", ro_vim_item_update
467
sousaedu80135b92021-02-17 15:05:18 +0100468 self.logger.debug(
469 "task={} {} del-vm={} {}".format(
470 task_id,
471 ro_task["target_id"],
472 vm_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300473 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +0100474 )
475 )
476
tierno1d213f42020-04-24 14:02:51 +0000477 return "DONE", ro_vim_item_update_ok
478
tierno70eeb182020-10-19 16:38:00 +0000479 def refresh(self, ro_task):
tierno1d213f42020-04-24 14:02:51 +0000480 """Call VIM to get vm status"""
481 ro_task_id = ro_task["_id"]
482 target_vim = self.my_vims[ro_task["target_id"]]
tierno1d213f42020-04-24 14:02:51 +0000483 vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100484
tierno1d213f42020-04-24 14:02:51 +0000485 if not vim_id:
486 return None, None
sousaedu80135b92021-02-17 15:05:18 +0100487
tierno1d213f42020-04-24 14:02:51 +0000488 vm_to_refresh_list = [vim_id]
489 try:
490 vim_dict = target_vim.refresh_vms_status(vm_to_refresh_list)
491 vim_info = vim_dict[vim_id]
sousaedu80135b92021-02-17 15:05:18 +0100492
tierno1d213f42020-04-24 14:02:51 +0000493 if vim_info["status"] == "ACTIVE":
494 task_status = "DONE"
495 elif vim_info["status"] == "BUILD":
496 task_status = "BUILD"
497 else:
498 task_status = "FAILED"
sousaedu80135b92021-02-17 15:05:18 +0100499
tierno70eeb182020-10-19 16:38:00 +0000500 # try to load and parse vim_information
501 try:
502 vim_info_info = yaml.safe_load(vim_info["vim_info"])
503 if vim_info_info.get("name"):
504 vim_info["name"] = vim_info_info["name"]
aticig7b521f72022-07-15 00:43:09 +0300505 except Exception as vim_info_error:
506 self.logger.exception(
507 f"{vim_info_error} occured while getting the vim_info from yaml"
508 )
tierno1d213f42020-04-24 14:02:51 +0000509 except vimconn.VimConnException as e:
510 # Mark all tasks at VIM_ERROR status
sousaedu80135b92021-02-17 15:05:18 +0100511 self.logger.error(
512 "ro_task={} vim={} get-vm={}: {}".format(
513 ro_task_id, ro_task["target_id"], vim_id, e
514 )
515 )
tierno1d213f42020-04-24 14:02:51 +0000516 vim_info = {"status": "VIM_ERROR", "error_msg": str(e)}
517 task_status = "FAILED"
518
519 ro_vim_item_update = {}
sousaedu80135b92021-02-17 15:05:18 +0100520
tierno70eeb182020-10-19 16:38:00 +0000521 # Interfaces cannot be present if e.g. VM is not present, that is status=DELETED
tierno1d213f42020-04-24 14:02:51 +0000522 vim_interfaces = []
tierno70eeb182020-10-19 16:38:00 +0000523 if vim_info.get("interfaces"):
524 for vim_iface_id in ro_task["vim_info"]["interfaces_vim_ids"]:
sousaedu80135b92021-02-17 15:05:18 +0100525 iface = next(
526 (
527 iface
528 for iface in vim_info["interfaces"]
529 if vim_iface_id == iface["vim_interface_id"]
530 ),
531 None,
532 )
tierno70eeb182020-10-19 16:38:00 +0000533 # if iface:
534 # iface.pop("vim_info", None)
535 vim_interfaces.append(iface)
tierno1d213f42020-04-24 14:02:51 +0000536
sousaedu80135b92021-02-17 15:05:18 +0100537 task_create = next(
538 t
539 for t in ro_task["tasks"]
540 if t and t["action"] == "CREATE" and t["status"] != "FINISHED"
541 )
tierno70eeb182020-10-19 16:38:00 +0000542 if vim_interfaces and task_create.get("mgmt_vnf_interface") is not None:
sousaedu80135b92021-02-17 15:05:18 +0100543 vim_interfaces[task_create["mgmt_vnf_interface"]][
544 "mgmt_vnf_interface"
545 ] = True
546
547 mgmt_vdu_iface = task_create.get(
548 "mgmt_vdu_interface", task_create.get("mgmt_vnf_interface", 0)
549 )
tierno70eeb182020-10-19 16:38:00 +0000550 if vim_interfaces:
551 vim_interfaces[mgmt_vdu_iface]["mgmt_vdu_interface"] = True
tierno1d213f42020-04-24 14:02:51 +0000552
553 if ro_task["vim_info"]["interfaces"] != vim_interfaces:
554 ro_vim_item_update["interfaces"] = vim_interfaces
sousaedu80135b92021-02-17 15:05:18 +0100555
tierno1d213f42020-04-24 14:02:51 +0000556 if ro_task["vim_info"]["vim_status"] != vim_info["status"]:
557 ro_vim_item_update["vim_status"] = vim_info["status"]
sousaedu80135b92021-02-17 15:05:18 +0100558
tierno1d213f42020-04-24 14:02:51 +0000559 if ro_task["vim_info"]["vim_name"] != vim_info.get("name"):
560 ro_vim_item_update["vim_name"] = vim_info.get("name")
sousaedu80135b92021-02-17 15:05:18 +0100561
tierno1d213f42020-04-24 14:02:51 +0000562 if vim_info["status"] in ("ERROR", "VIM_ERROR"):
aticig79ac6df2022-05-06 16:09:52 +0300563 if ro_task["vim_info"]["vim_message"] != vim_info.get("error_msg"):
564 ro_vim_item_update["vim_message"] = vim_info.get("error_msg")
tierno1d213f42020-04-24 14:02:51 +0000565 elif vim_info["status"] == "DELETED":
566 ro_vim_item_update["vim_id"] = None
aticig79ac6df2022-05-06 16:09:52 +0300567 ro_vim_item_update["vim_message"] = "Deleted externally"
tierno1d213f42020-04-24 14:02:51 +0000568 else:
569 if ro_task["vim_info"]["vim_details"] != vim_info["vim_info"]:
570 ro_vim_item_update["vim_details"] = vim_info["vim_info"]
sousaedu80135b92021-02-17 15:05:18 +0100571
tierno1d213f42020-04-24 14:02:51 +0000572 if ro_vim_item_update:
sousaedu80135b92021-02-17 15:05:18 +0100573 self.logger.debug(
574 "ro_task={} {} get-vm={}: status={} {}".format(
575 ro_task_id,
576 ro_task["target_id"],
577 vim_id,
578 ro_vim_item_update.get("vim_status"),
aticig79ac6df2022-05-06 16:09:52 +0300579 ro_vim_item_update.get("vim_message")
sousaedu80135b92021-02-17 15:05:18 +0100580 if ro_vim_item_update.get("vim_status") != "ACTIVE"
581 else "",
582 )
583 )
584
tierno1d213f42020-04-24 14:02:51 +0000585 return task_status, ro_vim_item_update
586
tierno70eeb182020-10-19 16:38:00 +0000587 def exec(self, ro_task, task_index, task_depends):
tierno1d213f42020-04-24 14:02:51 +0000588 task = ro_task["tasks"][task_index]
589 task_id = task["task_id"]
590 target_vim = self.my_vims[ro_task["target_id"]]
tierno70eeb182020-10-19 16:38:00 +0000591 db_task_update = {"retries": 0}
592 retries = task.get("retries", 0)
sousaedu80135b92021-02-17 15:05:18 +0100593
tierno1d213f42020-04-24 14:02:51 +0000594 try:
595 params = task["params"]
596 params_copy = deepcopy(params)
sousaedu80135b92021-02-17 15:05:18 +0100597 params_copy["ro_key"] = self.db.decrypt(
598 params_copy.pop("private_key"),
599 params_copy.pop("schema_version"),
600 params_copy.pop("salt"),
601 )
tierno70eeb182020-10-19 16:38:00 +0000602 params_copy["ip_addr"] = params_copy.pop("ip_address")
tierno1d213f42020-04-24 14:02:51 +0000603 target_vim.inject_user_key(**params_copy)
604 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100605 "task={} {} action-vm=inject_key".format(task_id, ro_task["target_id"])
606 )
607
608 return (
609 "DONE",
610 None,
611 db_task_update,
612 ) # params_copy["key"]
tierno1d213f42020-04-24 14:02:51 +0000613 except (vimconn.VimConnException, NsWorkerException) as e:
tierno70eeb182020-10-19 16:38:00 +0000614 retries += 1
sousaedu80135b92021-02-17 15:05:18 +0100615
aticige5d78422022-05-16 23:03:54 +0300616 self.logger.debug(traceback.format_exc())
tierno70eeb182020-10-19 16:38:00 +0000617 if retries < self.max_retries_inject_ssh_key:
sousaedu80135b92021-02-17 15:05:18 +0100618 return (
619 "BUILD",
620 None,
621 {
622 "retries": retries,
623 "next_retry": self.time_retries_inject_ssh_key,
624 },
625 )
626
627 self.logger.error(
628 "task={} {} inject-ssh-key: {}".format(task_id, ro_task["target_id"], e)
629 )
aticig79ac6df2022-05-06 16:09:52 +0300630 ro_vim_item_update = {"vim_message": str(e)}
sousaedu80135b92021-02-17 15:05:18 +0100631
tierno70eeb182020-10-19 16:38:00 +0000632 return "FAILED", ro_vim_item_update, db_task_update
633
634
635class VimInteractionImage(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +0000636 def new(self, ro_task, task_index, task_depends):
637 task = ro_task["tasks"][task_index]
638 task_id = task["task_id"]
639 created = False
640 created_items = {}
641 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +0100642
tierno70eeb182020-10-19 16:38:00 +0000643 try:
644 # FIND
Gulsum Atici1f58a1a2023-06-19 13:59:46 +0300645 vim_image_id = ""
tierno70eeb182020-10-19 16:38:00 +0000646 if task.get("find_params"):
gatici335a06a2023-07-26 00:34:04 +0300647 vim_images = target_vim.get_image_list(
648 task["find_params"].get("filter_dict", {})
649 )
sousaedu80135b92021-02-17 15:05:18 +0100650
tierno70eeb182020-10-19 16:38:00 +0000651 if not vim_images:
sousaedu80135b92021-02-17 15:05:18 +0100652 raise NsWorkerExceptionNotFound(
653 "Image not found with this criteria: '{}'".format(
654 task["find_params"]
655 )
656 )
tierno70eeb182020-10-19 16:38:00 +0000657 elif len(vim_images) > 1:
658 raise NsWorkerException(
sousaeduee6a6202021-05-11 13:22:37 +0200659 "More than one image found with this criteria: '{}'".format(
sousaedu80135b92021-02-17 15:05:18 +0100660 task["find_params"]
661 )
662 )
tierno70eeb182020-10-19 16:38:00 +0000663 else:
664 vim_image_id = vim_images[0]["id"]
665
sousaedu80135b92021-02-17 15:05:18 +0100666 ro_vim_item_update = {
667 "vim_id": vim_image_id,
Gulsum Atici1f58a1a2023-06-19 13:59:46 +0300668 "vim_status": "ACTIVE",
sousaedu80135b92021-02-17 15:05:18 +0100669 "created": created,
670 "created_items": created_items,
671 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300672 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100673 }
tierno70eeb182020-10-19 16:38:00 +0000674 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100675 "task={} {} new-image={} created={}".format(
676 task_id, ro_task["target_id"], vim_image_id, created
677 )
678 )
679
tierno70eeb182020-10-19 16:38:00 +0000680 return "DONE", ro_vim_item_update
681 except (NsWorkerException, vimconn.VimConnException) as e:
sousaedu80135b92021-02-17 15:05:18 +0100682 self.logger.error(
683 "task={} {} new-image: {}".format(task_id, ro_task["target_id"], e)
684 )
685 ro_vim_item_update = {
686 "vim_status": "VIM_ERROR",
687 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300688 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100689 }
690
tierno1d213f42020-04-24 14:02:51 +0000691 return "FAILED", ro_vim_item_update
692
tierno70eeb182020-10-19 16:38:00 +0000693
vegall364627c2023-03-17 15:09:50 +0000694class VimInteractionSharedVolume(VimInteractionBase):
695 def delete(self, ro_task, task_index):
696 task = ro_task["tasks"][task_index]
697 task_id = task["task_id"]
698 shared_volume_vim_id = ro_task["vim_info"]["vim_id"]
Gabriel Cuba1fd411b2023-06-14 00:50:57 -0500699 created_items = ro_task["vim_info"]["created_items"]
vegall364627c2023-03-17 15:09:50 +0000700 ro_vim_item_update_ok = {
701 "vim_status": "DELETED",
702 "created": False,
703 "vim_message": "DELETED",
704 "vim_id": None,
705 }
Gabriel Cuba1fd411b2023-06-14 00:50:57 -0500706 if created_items and created_items.get(shared_volume_vim_id).get("keep"):
707 ro_vim_item_update_ok = {
708 "vim_status": "ACTIVE",
709 "created": False,
710 "vim_message": None,
711 }
712 return "DONE", ro_vim_item_update_ok
vegall364627c2023-03-17 15:09:50 +0000713 try:
714 if shared_volume_vim_id:
715 target_vim = self.my_vims[ro_task["target_id"]]
716 target_vim.delete_shared_volumes(shared_volume_vim_id)
717 except vimconn.VimConnNotFoundException:
718 ro_vim_item_update_ok["vim_message"] = "already deleted"
719 except vimconn.VimConnException as e:
720 self.logger.error(
721 "ro_task={} vim={} del-shared-volume={}: {}".format(
722 ro_task["_id"], ro_task["target_id"], shared_volume_vim_id, e
723 )
724 )
725 ro_vim_item_update = {
726 "vim_status": "VIM_ERROR",
727 "vim_message": "Error while deleting: {}".format(e),
728 }
729
730 return "FAILED", ro_vim_item_update
731
732 self.logger.debug(
733 "task={} {} del-shared-volume={} {}".format(
734 task_id,
735 ro_task["target_id"],
736 shared_volume_vim_id,
737 ro_vim_item_update_ok.get("vim_message", ""),
738 )
739 )
740
741 return "DONE", ro_vim_item_update_ok
742
743 def new(self, ro_task, task_index, task_depends):
744 task = ro_task["tasks"][task_index]
745 task_id = task["task_id"]
746 created = False
747 created_items = {}
748 target_vim = self.my_vims[ro_task["target_id"]]
749
750 try:
vegall364627c2023-03-17 15:09:50 +0000751 shared_volume_vim_id = None
752 shared_volume_data = None
753
754 if task.get("params"):
755 shared_volume_data = task["params"]
756
757 if shared_volume_data:
758 self.logger.info(
759 f"Creating the new shared_volume for {shared_volume_data}\n"
760 )
761 (
762 shared_volume_name,
763 shared_volume_vim_id,
764 ) = target_vim.new_shared_volumes(shared_volume_data)
765 created = True
Gabriel Cuba1fd411b2023-06-14 00:50:57 -0500766 created_items[shared_volume_vim_id] = {
767 "name": shared_volume_name,
768 "keep": shared_volume_data.get("keep"),
769 }
vegall364627c2023-03-17 15:09:50 +0000770
771 ro_vim_item_update = {
772 "vim_id": shared_volume_vim_id,
Gabriel Cuba1fd411b2023-06-14 00:50:57 -0500773 "vim_status": "ACTIVE",
vegall364627c2023-03-17 15:09:50 +0000774 "created": created,
775 "created_items": created_items,
776 "vim_details": None,
777 "vim_message": None,
778 }
779 self.logger.debug(
780 "task={} {} new-shared-volume={} created={}".format(
781 task_id, ro_task["target_id"], shared_volume_vim_id, created
782 )
783 )
784
785 return "DONE", ro_vim_item_update
786 except (vimconn.VimConnException, NsWorkerException) as e:
787 self.logger.error(
788 "task={} vim={} new-shared-volume:"
789 " {}".format(task_id, ro_task["target_id"], e)
790 )
791 ro_vim_item_update = {
792 "vim_status": "VIM_ERROR",
793 "created": created,
794 "vim_message": str(e),
795 }
796
797 return "FAILED", ro_vim_item_update
798
799
tierno70eeb182020-10-19 16:38:00 +0000800class VimInteractionFlavor(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +0000801 def delete(self, ro_task, task_index):
802 task = ro_task["tasks"][task_index]
803 task_id = task["task_id"]
804 flavor_vim_id = ro_task["vim_info"]["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +0100805 ro_vim_item_update_ok = {
806 "vim_status": "DELETED",
807 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300808 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +0100809 "vim_id": None,
810 }
811
tierno70eeb182020-10-19 16:38:00 +0000812 try:
813 if flavor_vim_id:
814 target_vim = self.my_vims[ro_task["target_id"]]
815 target_vim.delete_flavor(flavor_vim_id)
tierno70eeb182020-10-19 16:38:00 +0000816 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300817 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno70eeb182020-10-19 16:38:00 +0000818 except vimconn.VimConnException as e:
sousaedu80135b92021-02-17 15:05:18 +0100819 self.logger.error(
820 "ro_task={} vim={} del-flavor={}: {}".format(
821 ro_task["_id"], ro_task["target_id"], flavor_vim_id, e
822 )
823 )
824 ro_vim_item_update = {
825 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300826 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +0100827 }
828
tierno70eeb182020-10-19 16:38:00 +0000829 return "FAILED", ro_vim_item_update
830
sousaedu80135b92021-02-17 15:05:18 +0100831 self.logger.debug(
832 "task={} {} del-flavor={} {}".format(
833 task_id,
834 ro_task["target_id"],
835 flavor_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300836 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +0100837 )
838 )
839
tierno70eeb182020-10-19 16:38:00 +0000840 return "DONE", ro_vim_item_update_ok
841
842 def new(self, ro_task, task_index, task_depends):
843 task = ro_task["tasks"][task_index]
844 task_id = task["task_id"]
845 created = False
846 created_items = {}
847 target_vim = self.my_vims[ro_task["target_id"]]
848 try:
849 # FIND
850 vim_flavor_id = None
sousaedu80135b92021-02-17 15:05:18 +0100851
Gabriel Cubaa5233f82023-08-07 18:43:26 -0500852 if task.get("find_params", {}).get("vim_flavor_id"):
853 vim_flavor_id = task["find_params"]["vim_flavor_id"]
854 elif task.get("find_params", {}).get("flavor_data"):
tierno70eeb182020-10-19 16:38:00 +0000855 try:
856 flavor_data = task["find_params"]["flavor_data"]
857 vim_flavor_id = target_vim.get_flavor_id_from_data(flavor_data)
Gulsum Atici5da710e2022-10-25 22:45:07 +0300858 except vimconn.VimConnNotFoundException as flavor_not_found_msg:
859 self.logger.warning(
860 f"VimConnNotFoundException occured: {flavor_not_found_msg}"
861 )
tierno70eeb182020-10-19 16:38:00 +0000862
863 if not vim_flavor_id and task.get("params"):
864 # CREATE
865 flavor_data = task["params"]["flavor_data"]
866 vim_flavor_id = target_vim.new_flavor(flavor_data)
867 created = True
868
sousaedu80135b92021-02-17 15:05:18 +0100869 ro_vim_item_update = {
870 "vim_id": vim_flavor_id,
Gulsum Atici1f58a1a2023-06-19 13:59:46 +0300871 "vim_status": "ACTIVE",
sousaedu80135b92021-02-17 15:05:18 +0100872 "created": created,
873 "created_items": created_items,
874 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300875 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +0100876 }
tierno70eeb182020-10-19 16:38:00 +0000877 self.logger.debug(
sousaedu80135b92021-02-17 15:05:18 +0100878 "task={} {} new-flavor={} created={}".format(
879 task_id, ro_task["target_id"], vim_flavor_id, created
880 )
881 )
882
tierno70eeb182020-10-19 16:38:00 +0000883 return "DONE", ro_vim_item_update
884 except (vimconn.VimConnException, NsWorkerException) as e:
sousaedu80135b92021-02-17 15:05:18 +0100885 self.logger.error(
886 "task={} vim={} new-flavor: {}".format(task_id, ro_task["target_id"], e)
887 )
888 ro_vim_item_update = {
889 "vim_status": "VIM_ERROR",
890 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300891 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +0100892 }
893
tierno70eeb182020-10-19 16:38:00 +0000894 return "FAILED", ro_vim_item_update
895
896
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100897class VimInteractionAffinityGroup(VimInteractionBase):
898 def delete(self, ro_task, task_index):
899 task = ro_task["tasks"][task_index]
900 task_id = task["task_id"]
901 affinity_group_vim_id = ro_task["vim_info"]["vim_id"]
902 ro_vim_item_update_ok = {
903 "vim_status": "DELETED",
904 "created": False,
aticig79ac6df2022-05-06 16:09:52 +0300905 "vim_message": "DELETED",
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100906 "vim_id": None,
907 }
908
909 try:
910 if affinity_group_vim_id:
911 target_vim = self.my_vims[ro_task["target_id"]]
912 target_vim.delete_affinity_group(affinity_group_vim_id)
913 except vimconn.VimConnNotFoundException:
aticig79ac6df2022-05-06 16:09:52 +0300914 ro_vim_item_update_ok["vim_message"] = "already deleted"
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100915 except vimconn.VimConnException as e:
916 self.logger.error(
917 "ro_task={} vim={} del-affinity-or-anti-affinity-group={}: {}".format(
918 ro_task["_id"], ro_task["target_id"], affinity_group_vim_id, e
919 )
920 )
921 ro_vim_item_update = {
922 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +0300923 "vim_message": "Error while deleting: {}".format(e),
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100924 }
925
926 return "FAILED", ro_vim_item_update
927
928 self.logger.debug(
929 "task={} {} del-affinity-or-anti-affinity-group={} {}".format(
930 task_id,
931 ro_task["target_id"],
932 affinity_group_vim_id,
aticig79ac6df2022-05-06 16:09:52 +0300933 ro_vim_item_update_ok.get("vim_message", ""),
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100934 )
935 )
936
937 return "DONE", ro_vim_item_update_ok
938
939 def new(self, ro_task, task_index, task_depends):
940 task = ro_task["tasks"][task_index]
941 task_id = task["task_id"]
942 created = False
943 created_items = {}
944 target_vim = self.my_vims[ro_task["target_id"]]
945
946 try:
947 affinity_group_vim_id = None
Alexis Romero123de182022-04-26 19:24:40 +0200948 affinity_group_data = None
Gulsum Atici1f58a1a2023-06-19 13:59:46 +0300949 param_affinity_group_id = ""
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100950
951 if task.get("params"):
Alexis Romero123de182022-04-26 19:24:40 +0200952 affinity_group_data = task["params"].get("affinity_group_data")
953
954 if affinity_group_data and affinity_group_data.get("vim-affinity-group-id"):
955 try:
956 param_affinity_group_id = task["params"]["affinity_group_data"].get(
957 "vim-affinity-group-id"
958 )
959 affinity_group_vim_id = target_vim.get_affinity_group(
960 param_affinity_group_id
961 ).get("id")
962 except vimconn.VimConnNotFoundException:
963 self.logger.error(
964 "task={} {} new-affinity-or-anti-affinity-group. Provided VIM Affinity Group ID {}"
965 "could not be found at VIM. Creating a new one.".format(
966 task_id, ro_task["target_id"], param_affinity_group_id
967 )
968 )
969
970 if not affinity_group_vim_id and affinity_group_data:
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100971 affinity_group_vim_id = target_vim.new_affinity_group(
972 affinity_group_data
973 )
974 created = True
975
976 ro_vim_item_update = {
977 "vim_id": affinity_group_vim_id,
Gulsum Atici1f58a1a2023-06-19 13:59:46 +0300978 "vim_status": "ACTIVE",
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100979 "created": created,
980 "created_items": created_items,
981 "vim_details": None,
aticig79ac6df2022-05-06 16:09:52 +0300982 "vim_message": None,
Alexis Romerob70f4ed2022-03-11 18:00:49 +0100983 }
984 self.logger.debug(
985 "task={} {} new-affinity-or-anti-affinity-group={} created={}".format(
986 task_id, ro_task["target_id"], affinity_group_vim_id, created
987 )
988 )
989
990 return "DONE", ro_vim_item_update
991 except (vimconn.VimConnException, NsWorkerException) as e:
992 self.logger.error(
993 "task={} vim={} new-affinity-or-anti-affinity-group:"
994 " {}".format(task_id, ro_task["target_id"], e)
995 )
996 ro_vim_item_update = {
997 "vim_status": "VIM_ERROR",
998 "created": created,
aticig79ac6df2022-05-06 16:09:52 +0300999 "vim_message": str(e),
Alexis Romerob70f4ed2022-03-11 18:00:49 +01001000 }
1001
1002 return "FAILED", ro_vim_item_update
1003
1004
k4.rahul78f474e2022-05-02 15:47:57 +00001005class VimInteractionUpdateVdu(VimInteractionBase):
1006 def exec(self, ro_task, task_index, task_depends):
1007 task = ro_task["tasks"][task_index]
1008 task_id = task["task_id"]
1009 db_task_update = {"retries": 0}
1010 created = False
1011 created_items = {}
1012 target_vim = self.my_vims[ro_task["target_id"]]
1013
1014 try:
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001015 vim_vm_id = ""
k4.rahul78f474e2022-05-02 15:47:57 +00001016 if task.get("params"):
1017 vim_vm_id = task["params"].get("vim_vm_id")
1018 action = task["params"].get("action")
1019 context = {action: action}
1020 target_vim.action_vminstance(vim_vm_id, context)
1021 # created = True
1022 ro_vim_item_update = {
1023 "vim_id": vim_vm_id,
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001024 "vim_status": "ACTIVE",
k4.rahul78f474e2022-05-02 15:47:57 +00001025 "created": created,
1026 "created_items": created_items,
1027 "vim_details": None,
1028 "vim_message": None,
1029 }
1030 self.logger.debug(
1031 "task={} {} vm-migration done".format(task_id, ro_task["target_id"])
1032 )
1033 return "DONE", ro_vim_item_update, db_task_update
1034 except (vimconn.VimConnException, NsWorkerException) as e:
1035 self.logger.error(
1036 "task={} vim={} VM Migration:"
1037 " {}".format(task_id, ro_task["target_id"], e)
1038 )
1039 ro_vim_item_update = {
1040 "vim_status": "VIM_ERROR",
1041 "created": created,
1042 "vim_message": str(e),
1043 }
1044
1045 return "FAILED", ro_vim_item_update, db_task_update
1046
1047
tierno70eeb182020-10-19 16:38:00 +00001048class VimInteractionSdnNet(VimInteractionBase):
tierno70eeb182020-10-19 16:38:00 +00001049 @staticmethod
1050 def _match_pci(port_pci, mapping):
1051 """
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001052 Check if port_pci matches with mapping.
1053 The mapping can have brackets to indicate that several chars are accepted. e.g
tierno70eeb182020-10-19 16:38:00 +00001054 pci '0000:af:10.1' matches with '0000:af:1[01].[1357]'
1055 :param port_pci: text
1056 :param mapping: text, can contain brackets to indicate several chars are available
1057 :return: True if matches, False otherwise
1058 """
1059 if not port_pci or not mapping:
1060 return False
1061 if port_pci == mapping:
1062 return True
1063
1064 mapping_index = 0
1065 pci_index = 0
1066 while True:
1067 bracket_start = mapping.find("[", mapping_index)
sousaedu80135b92021-02-17 15:05:18 +01001068
tierno70eeb182020-10-19 16:38:00 +00001069 if bracket_start == -1:
1070 break
sousaedu80135b92021-02-17 15:05:18 +01001071
tierno70eeb182020-10-19 16:38:00 +00001072 bracket_end = mapping.find("]", bracket_start)
1073 if bracket_end == -1:
1074 break
sousaedu80135b92021-02-17 15:05:18 +01001075
tierno70eeb182020-10-19 16:38:00 +00001076 length = bracket_start - mapping_index
sousaedu80135b92021-02-17 15:05:18 +01001077 if (
1078 length
1079 and port_pci[pci_index : pci_index + length]
1080 != mapping[mapping_index:bracket_start]
1081 ):
tierno70eeb182020-10-19 16:38:00 +00001082 return False
sousaedu80135b92021-02-17 15:05:18 +01001083
1084 if (
1085 port_pci[pci_index + length]
1086 not in mapping[bracket_start + 1 : bracket_end]
1087 ):
tierno70eeb182020-10-19 16:38:00 +00001088 return False
sousaedu80135b92021-02-17 15:05:18 +01001089
tierno70eeb182020-10-19 16:38:00 +00001090 pci_index += length + 1
1091 mapping_index = bracket_end + 1
1092
1093 if port_pci[pci_index:] != mapping[mapping_index:]:
1094 return False
sousaedu80135b92021-02-17 15:05:18 +01001095
tierno70eeb182020-10-19 16:38:00 +00001096 return True
1097
1098 def _get_interfaces(self, vlds_to_connect, vim_account_id):
1099 """
1100 :param vlds_to_connect: list with format vnfrs:<id>:vld.<vld_id> or nsrs:<id>:vld.<vld_id>
1101 :param vim_account_id:
1102 :return:
1103 """
1104 interfaces = []
sousaedu80135b92021-02-17 15:05:18 +01001105
tierno70eeb182020-10-19 16:38:00 +00001106 for vld in vlds_to_connect:
1107 table, _, db_id = vld.partition(":")
1108 db_id, _, vld = db_id.partition(":")
1109 _, _, vld_id = vld.partition(".")
sousaedu80135b92021-02-17 15:05:18 +01001110
tierno70eeb182020-10-19 16:38:00 +00001111 if table == "vnfrs":
1112 q_filter = {"vim-account-id": vim_account_id, "_id": db_id}
1113 iface_key = "vnf-vld-id"
1114 else: # table == "nsrs"
1115 q_filter = {"vim-account-id": vim_account_id, "nsr-id-ref": db_id}
1116 iface_key = "ns-vld-id"
sousaedu80135b92021-02-17 15:05:18 +01001117
tierno70eeb182020-10-19 16:38:00 +00001118 db_vnfrs = self.db.get_list("vnfrs", q_filter=q_filter)
sousaedu80135b92021-02-17 15:05:18 +01001119
tierno70eeb182020-10-19 16:38:00 +00001120 for db_vnfr in db_vnfrs:
1121 for vdu_index, vdur in enumerate(db_vnfr.get("vdur", ())):
1122 for iface_index, interface in enumerate(vdur["interfaces"]):
sousaedu80135b92021-02-17 15:05:18 +01001123 if interface.get(iface_key) == vld_id and interface.get(
1124 "type"
1125 ) in ("SR-IOV", "PCI-PASSTHROUGH"):
tierno70eeb182020-10-19 16:38:00 +00001126 # only SR-IOV o PT
1127 interface_ = interface.copy()
sousaedu80135b92021-02-17 15:05:18 +01001128 interface_["id"] = "vnfrs:{}:vdu.{}.interfaces.{}".format(
1129 db_vnfr["_id"], vdu_index, iface_index
1130 )
1131
tierno70eeb182020-10-19 16:38:00 +00001132 if vdur.get("status") == "ERROR":
1133 interface_["status"] = "ERROR"
sousaedu80135b92021-02-17 15:05:18 +01001134
tierno70eeb182020-10-19 16:38:00 +00001135 interfaces.append(interface_)
sousaedu80135b92021-02-17 15:05:18 +01001136
tierno70eeb182020-10-19 16:38:00 +00001137 return interfaces
1138
1139 def refresh(self, ro_task):
1140 # look for task create
sousaedu80135b92021-02-17 15:05:18 +01001141 task_create_index, _ = next(
1142 i_t
1143 for i_t in enumerate(ro_task["tasks"])
1144 if i_t[1]
1145 and i_t[1]["action"] == "CREATE"
1146 and i_t[1]["status"] != "FINISHED"
1147 )
tierno70eeb182020-10-19 16:38:00 +00001148
1149 return self.new(ro_task, task_create_index, None)
1150
1151 def new(self, ro_task, task_index, task_depends):
tierno70eeb182020-10-19 16:38:00 +00001152 task = ro_task["tasks"][task_index]
1153 task_id = task["task_id"]
1154 target_vim = self.my_vims[ro_task["target_id"]]
1155
1156 sdn_net_id = ro_task["vim_info"]["vim_id"]
1157
1158 created_items = ro_task["vim_info"].get("created_items")
1159 connected_ports = ro_task["vim_info"].get("connected_ports", [])
1160 new_connected_ports = []
1161 last_update = ro_task["vim_info"].get("last_update", 0)
1162 sdn_status = ro_task["vim_info"].get("vim_status", "BUILD") or "BUILD"
1163 error_list = []
1164 created = ro_task["vim_info"].get("created", False)
1165
1166 try:
tierno70eeb182020-10-19 16:38:00 +00001167 # CREATE
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001168 db_vim = {}
tierno70eeb182020-10-19 16:38:00 +00001169 params = task["params"]
gifrerenombf3e9a82022-03-07 17:55:20 +00001170 vlds_to_connect = params.get("vlds", [])
1171 associated_vim = params.get("target_vim")
sousaedu80135b92021-02-17 15:05:18 +01001172 # external additional ports
1173 additional_ports = params.get("sdn-ports") or ()
gifrerenombf3e9a82022-03-07 17:55:20 +00001174 _, _, vim_account_id = (
1175 (None, None, None)
1176 if associated_vim is None
1177 else associated_vim.partition(":")
1178 )
sousaedu80135b92021-02-17 15:05:18 +01001179
tierno70eeb182020-10-19 16:38:00 +00001180 if associated_vim:
1181 # get associated VIM
1182 if associated_vim not in self.db_vims:
sousaedu80135b92021-02-17 15:05:18 +01001183 self.db_vims[associated_vim] = self.db.get_one(
1184 "vim_accounts", {"_id": vim_account_id}
1185 )
1186
tierno70eeb182020-10-19 16:38:00 +00001187 db_vim = self.db_vims[associated_vim]
1188
1189 # look for ports to connect
1190 ports = self._get_interfaces(vlds_to_connect, vim_account_id)
1191 # print(ports)
1192
1193 sdn_ports = []
1194 pending_ports = error_ports = 0
1195 vlan_used = None
1196 sdn_need_update = False
sousaedu80135b92021-02-17 15:05:18 +01001197
tierno70eeb182020-10-19 16:38:00 +00001198 for port in ports:
1199 vlan_used = port.get("vlan") or vlan_used
sousaedu80135b92021-02-17 15:05:18 +01001200
tierno70eeb182020-10-19 16:38:00 +00001201 # TODO. Do not connect if already done
1202 if not port.get("compute_node") or not port.get("pci"):
1203 if port.get("status") == "ERROR":
1204 error_ports += 1
1205 else:
1206 pending_ports += 1
1207 continue
sousaedu80135b92021-02-17 15:05:18 +01001208
tierno70eeb182020-10-19 16:38:00 +00001209 pmap = None
sousaedu80135b92021-02-17 15:05:18 +01001210 compute_node_mappings = next(
1211 (
1212 c
1213 for c in db_vim["config"].get("sdn-port-mapping", ())
1214 if c and c["compute_node"] == port["compute_node"]
1215 ),
1216 None,
1217 )
1218
tierno70eeb182020-10-19 16:38:00 +00001219 if compute_node_mappings:
1220 # process port_mapping pci of type 0000:af:1[01].[1357]
sousaedu80135b92021-02-17 15:05:18 +01001221 pmap = next(
1222 (
1223 p
1224 for p in compute_node_mappings["ports"]
1225 if self._match_pci(port["pci"], p.get("pci"))
1226 ),
1227 None,
1228 )
1229
tierno70eeb182020-10-19 16:38:00 +00001230 if not pmap:
1231 if not db_vim["config"].get("mapping_not_needed"):
sousaedu80135b92021-02-17 15:05:18 +01001232 error_list.append(
1233 "Port mapping not found for compute_node={} pci={}".format(
1234 port["compute_node"], port["pci"]
1235 )
1236 )
tierno70eeb182020-10-19 16:38:00 +00001237 continue
sousaedu80135b92021-02-17 15:05:18 +01001238
tierno70eeb182020-10-19 16:38:00 +00001239 pmap = {}
1240
1241 service_endpoint_id = "{}:{}".format(port["compute_node"], port["pci"])
1242 new_port = {
sousaedu80135b92021-02-17 15:05:18 +01001243 "service_endpoint_id": pmap.get("service_endpoint_id")
1244 or service_endpoint_id,
1245 "service_endpoint_encapsulation_type": "dot1q"
1246 if port["type"] == "SR-IOV"
1247 else None,
tierno70eeb182020-10-19 16:38:00 +00001248 "service_endpoint_encapsulation_info": {
1249 "vlan": port.get("vlan"),
lloretgalleg160fcad2021-02-19 10:57:50 +00001250 "mac": port.get("mac-address"),
sousaedu80135b92021-02-17 15:05:18 +01001251 "device_id": pmap.get("device_id") or port["compute_node"],
1252 "device_interface_id": pmap.get("device_interface_id")
1253 or port["pci"],
tierno70eeb182020-10-19 16:38:00 +00001254 "switch_dpid": pmap.get("switch_id") or pmap.get("switch_dpid"),
1255 "switch_port": pmap.get("switch_port"),
1256 "service_mapping_info": pmap.get("service_mapping_info"),
sousaedu80135b92021-02-17 15:05:18 +01001257 },
tierno70eeb182020-10-19 16:38:00 +00001258 }
1259
1260 # TODO
1261 # if port["modified_at"] > last_update:
1262 # sdn_need_update = True
1263 new_connected_ports.append(port["id"]) # TODO
1264 sdn_ports.append(new_port)
1265
1266 if error_ports:
sousaedu80135b92021-02-17 15:05:18 +01001267 error_list.append(
1268 "{} interfaces have not been created as VDU is on ERROR status".format(
1269 error_ports
1270 )
1271 )
tierno70eeb182020-10-19 16:38:00 +00001272
1273 # connect external ports
1274 for index, additional_port in enumerate(additional_ports):
sousaedu80135b92021-02-17 15:05:18 +01001275 additional_port_id = additional_port.get(
1276 "service_endpoint_id"
1277 ) or "external-{}".format(index)
1278 sdn_ports.append(
1279 {
1280 "service_endpoint_id": additional_port_id,
1281 "service_endpoint_encapsulation_type": additional_port.get(
1282 "service_endpoint_encapsulation_type", "dot1q"
1283 ),
1284 "service_endpoint_encapsulation_info": {
1285 "vlan": additional_port.get("vlan") or vlan_used,
1286 "mac": additional_port.get("mac_address"),
1287 "device_id": additional_port.get("device_id"),
1288 "device_interface_id": additional_port.get(
1289 "device_interface_id"
1290 ),
1291 "switch_dpid": additional_port.get("switch_dpid")
1292 or additional_port.get("switch_id"),
1293 "switch_port": additional_port.get("switch_port"),
1294 "service_mapping_info": additional_port.get(
1295 "service_mapping_info"
1296 ),
1297 },
1298 }
1299 )
tierno70eeb182020-10-19 16:38:00 +00001300 new_connected_ports.append(additional_port_id)
1301 sdn_info = ""
sousaedu80135b92021-02-17 15:05:18 +01001302
tierno70eeb182020-10-19 16:38:00 +00001303 # if there are more ports to connect or they have been modified, call create/update
1304 if error_list:
1305 sdn_status = "ERROR"
1306 sdn_info = "; ".join(error_list)
1307 elif set(connected_ports) != set(new_connected_ports) or sdn_need_update:
1308 last_update = time.time()
sousaedu80135b92021-02-17 15:05:18 +01001309
tierno70eeb182020-10-19 16:38:00 +00001310 if not sdn_net_id:
1311 if len(sdn_ports) < 2:
1312 sdn_status = "ACTIVE"
sousaedu80135b92021-02-17 15:05:18 +01001313
tierno70eeb182020-10-19 16:38:00 +00001314 if not pending_ports:
sousaedu80135b92021-02-17 15:05:18 +01001315 self.logger.debug(
1316 "task={} {} new-sdn-net done, less than 2 ports".format(
1317 task_id, ro_task["target_id"]
1318 )
1319 )
tierno70eeb182020-10-19 16:38:00 +00001320 else:
1321 net_type = params.get("type") or "ELAN"
sousaedu80135b92021-02-17 15:05:18 +01001322 (
1323 sdn_net_id,
1324 created_items,
1325 ) = target_vim.create_connectivity_service(net_type, sdn_ports)
tierno70eeb182020-10-19 16:38:00 +00001326 created = True
sousaedu80135b92021-02-17 15:05:18 +01001327 self.logger.debug(
1328 "task={} {} new-sdn-net={} created={}".format(
1329 task_id, ro_task["target_id"], sdn_net_id, created
1330 )
1331 )
tierno70eeb182020-10-19 16:38:00 +00001332 else:
1333 created_items = target_vim.edit_connectivity_service(
sousaedu80135b92021-02-17 15:05:18 +01001334 sdn_net_id, conn_info=created_items, connection_points=sdn_ports
1335 )
tierno70eeb182020-10-19 16:38:00 +00001336 created = True
sousaedu80135b92021-02-17 15:05:18 +01001337 self.logger.debug(
1338 "task={} {} update-sdn-net={} created={}".format(
1339 task_id, ro_task["target_id"], sdn_net_id, created
1340 )
1341 )
1342
tierno70eeb182020-10-19 16:38:00 +00001343 connected_ports = new_connected_ports
1344 elif sdn_net_id:
sousaedu80135b92021-02-17 15:05:18 +01001345 wim_status_dict = target_vim.get_connectivity_service_status(
1346 sdn_net_id, conn_info=created_items
1347 )
tierno70eeb182020-10-19 16:38:00 +00001348 sdn_status = wim_status_dict["sdn_status"]
sousaedu80135b92021-02-17 15:05:18 +01001349
tierno70eeb182020-10-19 16:38:00 +00001350 if wim_status_dict.get("sdn_info"):
1351 sdn_info = str(wim_status_dict.get("sdn_info")) or ""
sousaedu80135b92021-02-17 15:05:18 +01001352
tierno70eeb182020-10-19 16:38:00 +00001353 if wim_status_dict.get("error_msg"):
1354 sdn_info = wim_status_dict.get("error_msg") or ""
1355
1356 if pending_ports:
1357 if sdn_status != "ERROR":
1358 sdn_info = "Waiting for getting interfaces location from VIM. Obtained '{}' of {}".format(
sousaedu80135b92021-02-17 15:05:18 +01001359 len(ports) - pending_ports, len(ports)
1360 )
1361
tierno70eeb182020-10-19 16:38:00 +00001362 if sdn_status == "ACTIVE":
1363 sdn_status = "BUILD"
1364
sousaedu80135b92021-02-17 15:05:18 +01001365 ro_vim_item_update = {
1366 "vim_id": sdn_net_id,
1367 "vim_status": sdn_status,
1368 "created": created,
1369 "created_items": created_items,
1370 "connected_ports": connected_ports,
1371 "vim_details": sdn_info,
aticig79ac6df2022-05-06 16:09:52 +03001372 "vim_message": None,
sousaedu80135b92021-02-17 15:05:18 +01001373 "last_update": last_update,
1374 }
1375
tierno70eeb182020-10-19 16:38:00 +00001376 return sdn_status, ro_vim_item_update
1377 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001378 self.logger.error(
1379 "task={} vim={} new-net: {}".format(task_id, ro_task["target_id"], e),
1380 exc_info=not isinstance(
1381 e, (sdnconn.SdnConnectorError, vimconn.VimConnException)
1382 ),
1383 )
1384 ro_vim_item_update = {
1385 "vim_status": "VIM_ERROR",
1386 "created": created,
aticig79ac6df2022-05-06 16:09:52 +03001387 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +01001388 }
1389
tierno70eeb182020-10-19 16:38:00 +00001390 return "FAILED", ro_vim_item_update
1391
1392 def delete(self, ro_task, task_index):
1393 task = ro_task["tasks"][task_index]
1394 task_id = task["task_id"]
1395 sdn_vim_id = ro_task["vim_info"].get("vim_id")
sousaedu80135b92021-02-17 15:05:18 +01001396 ro_vim_item_update_ok = {
1397 "vim_status": "DELETED",
1398 "created": False,
aticig79ac6df2022-05-06 16:09:52 +03001399 "vim_message": "DELETED",
sousaedu80135b92021-02-17 15:05:18 +01001400 "vim_id": None,
1401 }
1402
tierno70eeb182020-10-19 16:38:00 +00001403 try:
1404 if sdn_vim_id:
1405 target_vim = self.my_vims[ro_task["target_id"]]
sousaedu80135b92021-02-17 15:05:18 +01001406 target_vim.delete_connectivity_service(
1407 sdn_vim_id, ro_task["vim_info"].get("created_items")
1408 )
tierno70eeb182020-10-19 16:38:00 +00001409
1410 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001411 if (
1412 isinstance(e, sdnconn.SdnConnectorError)
1413 and e.http_code == HTTPStatus.NOT_FOUND.value
1414 ):
aticig79ac6df2022-05-06 16:09:52 +03001415 ro_vim_item_update_ok["vim_message"] = "already deleted"
tierno70eeb182020-10-19 16:38:00 +00001416 else:
sousaedu80135b92021-02-17 15:05:18 +01001417 self.logger.error(
1418 "ro_task={} vim={} del-sdn-net={}: {}".format(
1419 ro_task["_id"], ro_task["target_id"], sdn_vim_id, e
1420 ),
1421 exc_info=not isinstance(
1422 e, (sdnconn.SdnConnectorError, vimconn.VimConnException)
1423 ),
1424 )
1425 ro_vim_item_update = {
1426 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +03001427 "vim_message": "Error while deleting: {}".format(e),
sousaedu80135b92021-02-17 15:05:18 +01001428 }
1429
tierno70eeb182020-10-19 16:38:00 +00001430 return "FAILED", ro_vim_item_update
1431
sousaedu80135b92021-02-17 15:05:18 +01001432 self.logger.debug(
1433 "task={} {} del-sdn-net={} {}".format(
1434 task_id,
1435 ro_task["target_id"],
1436 sdn_vim_id,
aticig79ac6df2022-05-06 16:09:52 +03001437 ro_vim_item_update_ok.get("vim_message", ""),
sousaedu80135b92021-02-17 15:05:18 +01001438 )
1439 )
1440
tierno70eeb182020-10-19 16:38:00 +00001441 return "DONE", ro_vim_item_update_ok
1442
1443
elumalai8658c2c2022-04-28 19:09:31 +05301444class VimInteractionMigration(VimInteractionBase):
1445 def exec(self, ro_task, task_index, task_depends):
1446 task = ro_task["tasks"][task_index]
1447 task_id = task["task_id"]
1448 db_task_update = {"retries": 0}
1449 target_vim = self.my_vims[ro_task["target_id"]]
1450 vim_interfaces = []
1451 created = False
1452 created_items = {}
1453 refreshed_vim_info = {}
1454
1455 try:
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001456 vim_vm_id = ""
elumalai8658c2c2022-04-28 19:09:31 +05301457 if task.get("params"):
1458 vim_vm_id = task["params"].get("vim_vm_id")
1459 migrate_host = task["params"].get("migrate_host")
1460 _, migrated_compute_node = target_vim.migrate_instance(
1461 vim_vm_id, migrate_host
1462 )
1463
1464 if migrated_compute_node:
1465 # When VM is migrated, vdu["vim_info"] needs to be updated
1466 vdu_old_vim_info = task["params"]["vdu_vim_info"].get(
1467 ro_task["target_id"]
1468 )
1469
1470 # Refresh VM to get new vim_info
1471 vm_to_refresh_list = [vim_vm_id]
1472 vim_dict = target_vim.refresh_vms_status(vm_to_refresh_list)
1473 refreshed_vim_info = vim_dict[vim_vm_id]
1474
1475 if refreshed_vim_info.get("interfaces"):
1476 for old_iface in vdu_old_vim_info.get("interfaces"):
1477 iface = next(
1478 (
1479 iface
1480 for iface in refreshed_vim_info["interfaces"]
1481 if old_iface["vim_interface_id"]
1482 == iface["vim_interface_id"]
1483 ),
1484 None,
1485 )
1486 vim_interfaces.append(iface)
1487
1488 ro_vim_item_update = {
1489 "vim_id": vim_vm_id,
1490 "vim_status": "ACTIVE",
1491 "created": created,
1492 "created_items": created_items,
1493 "vim_details": None,
1494 "vim_message": None,
1495 }
1496
1497 if refreshed_vim_info and refreshed_vim_info.get("status") not in (
1498 "ERROR",
1499 "VIM_ERROR",
1500 ):
1501 ro_vim_item_update["vim_details"] = refreshed_vim_info["vim_info"]
1502
1503 if vim_interfaces:
1504 ro_vim_item_update["interfaces"] = vim_interfaces
1505
1506 self.logger.debug(
1507 "task={} {} vm-migration done".format(task_id, ro_task["target_id"])
1508 )
1509
1510 return "DONE", ro_vim_item_update, db_task_update
1511
1512 except (vimconn.VimConnException, NsWorkerException) as e:
1513 self.logger.error(
1514 "task={} vim={} VM Migration:"
1515 " {}".format(task_id, ro_task["target_id"], e)
1516 )
1517 ro_vim_item_update = {
1518 "vim_status": "VIM_ERROR",
1519 "created": created,
1520 "vim_message": str(e),
1521 }
1522
1523 return "FAILED", ro_vim_item_update, db_task_update
1524
1525
sritharan29a4c1a2022-05-05 12:15:04 +00001526class VimInteractionResize(VimInteractionBase):
1527 def exec(self, ro_task, task_index, task_depends):
1528 task = ro_task["tasks"][task_index]
1529 task_id = task["task_id"]
1530 db_task_update = {"retries": 0}
1531 created = False
1532 target_flavor_uuid = None
1533 created_items = {}
1534 refreshed_vim_info = {}
1535 target_vim = self.my_vims[ro_task["target_id"]]
1536
1537 try:
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001538 vim_vm_id = ""
sritharan29a4c1a2022-05-05 12:15:04 +00001539 if task.get("params"):
1540 vim_vm_id = task["params"].get("vim_vm_id")
1541 flavor_dict = task["params"].get("flavor_dict")
1542 self.logger.info("flavor_dict %s", flavor_dict)
1543
1544 try:
1545 target_flavor_uuid = target_vim.get_flavor_id_from_data(flavor_dict)
1546 except Exception as e:
1547 self.logger.info("Cannot find any flavor matching %s.", str(e))
1548 try:
1549 target_flavor_uuid = target_vim.new_flavor(flavor_dict)
1550 except Exception as e:
1551 self.logger.error("Error creating flavor at VIM %s.", str(e))
1552
1553 if target_flavor_uuid is not None:
1554 resized_status = target_vim.resize_instance(
1555 vim_vm_id, target_flavor_uuid
1556 )
1557
1558 if resized_status:
1559 # Refresh VM to get new vim_info
1560 vm_to_refresh_list = [vim_vm_id]
1561 vim_dict = target_vim.refresh_vms_status(vm_to_refresh_list)
1562 refreshed_vim_info = vim_dict[vim_vm_id]
1563
1564 ro_vim_item_update = {
1565 "vim_id": vim_vm_id,
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001566 "vim_status": "ACTIVE",
sritharan29a4c1a2022-05-05 12:15:04 +00001567 "created": created,
1568 "created_items": created_items,
1569 "vim_details": None,
1570 "vim_message": None,
1571 }
1572
1573 if refreshed_vim_info and refreshed_vim_info.get("status") not in (
1574 "ERROR",
1575 "VIM_ERROR",
1576 ):
1577 ro_vim_item_update["vim_details"] = refreshed_vim_info["vim_info"]
1578
1579 self.logger.debug(
1580 "task={} {} resize done".format(task_id, ro_task["target_id"])
1581 )
1582 return "DONE", ro_vim_item_update, db_task_update
1583 except (vimconn.VimConnException, NsWorkerException) as e:
1584 self.logger.error(
1585 "task={} vim={} Resize:" " {}".format(task_id, ro_task["target_id"], e)
1586 )
1587 ro_vim_item_update = {
1588 "vim_status": "VIM_ERROR",
1589 "created": created,
1590 "vim_message": str(e),
1591 }
1592
1593 return "FAILED", ro_vim_item_update, db_task_update
1594
1595
aticig1ac189e2022-06-30 19:29:04 +03001596class ConfigValidate:
1597 def __init__(self, config: Dict):
1598 self.conf = config
tierno70eeb182020-10-19 16:38:00 +00001599
aticig1ac189e2022-06-30 19:29:04 +03001600 @property
1601 def active(self):
1602 # default 1 min, allowed >= 60 or -1, -1 disables periodic checks
1603 if (
1604 self.conf["period"]["refresh_active"] >= 60
1605 or self.conf["period"]["refresh_active"] == -1
1606 ):
1607 return self.conf["period"]["refresh_active"]
1608
1609 return 60
1610
1611 @property
1612 def build(self):
1613 return self.conf["period"]["refresh_build"]
1614
1615 @property
1616 def image(self):
1617 return self.conf["period"]["refresh_image"]
1618
1619 @property
1620 def error(self):
1621 return self.conf["period"]["refresh_error"]
1622
1623 @property
1624 def queue_size(self):
1625 return self.conf["period"]["queue_size"]
1626
1627
1628class NsWorker(threading.Thread):
tierno70eeb182020-10-19 16:38:00 +00001629 def __init__(self, worker_index, config, plugins, db):
1630 """
tierno70eeb182020-10-19 16:38:00 +00001631 :param worker_index: thread index
1632 :param config: general configuration of RO, among others the process_id with the docker id where it runs
1633 :param plugins: global shared dict with the loaded plugins
1634 :param db: database class instance to use
1635 """
1636 threading.Thread.__init__(self)
1637 self.config = config
1638 self.plugins = plugins
1639 self.plugin_name = "unknown"
sousaedu80135b92021-02-17 15:05:18 +01001640 self.logger = logging.getLogger("ro.worker{}".format(worker_index))
tierno70eeb182020-10-19 16:38:00 +00001641 self.worker_index = worker_index
aticig1ac189e2022-06-30 19:29:04 +03001642 # refresh periods for created items
1643 self.refresh_config = ConfigValidate(config)
1644 self.task_queue = queue.Queue(self.refresh_config.queue_size)
sousaedu80135b92021-02-17 15:05:18 +01001645 # targetvim: vimplugin class
1646 self.my_vims = {}
1647 # targetvim: vim information from database
1648 self.db_vims = {}
1649 # targetvim list
1650 self.vim_targets = []
tierno70eeb182020-10-19 16:38:00 +00001651 self.my_id = config["process_id"] + ":" + str(worker_index)
1652 self.db = db
1653 self.item2class = {
1654 "net": VimInteractionNet(self.db, self.my_vims, self.db_vims, self.logger),
vegall364627c2023-03-17 15:09:50 +00001655 "shared-volumes": VimInteractionSharedVolume(
1656 self.db, self.my_vims, self.db_vims, self.logger
1657 ),
tierno70eeb182020-10-19 16:38:00 +00001658 "vdu": VimInteractionVdu(self.db, self.my_vims, self.db_vims, self.logger),
sousaedu80135b92021-02-17 15:05:18 +01001659 "image": VimInteractionImage(
1660 self.db, self.my_vims, self.db_vims, self.logger
1661 ),
1662 "flavor": VimInteractionFlavor(
1663 self.db, self.my_vims, self.db_vims, self.logger
1664 ),
1665 "sdn_net": VimInteractionSdnNet(
1666 self.db, self.my_vims, self.db_vims, self.logger
1667 ),
k4.rahul78f474e2022-05-02 15:47:57 +00001668 "update": VimInteractionUpdateVdu(
1669 self.db, self.my_vims, self.db_vims, self.logger
1670 ),
Alexis Romerob70f4ed2022-03-11 18:00:49 +01001671 "affinity-or-anti-affinity-group": VimInteractionAffinityGroup(
1672 self.db, self.my_vims, self.db_vims, self.logger
1673 ),
elumalai8658c2c2022-04-28 19:09:31 +05301674 "migrate": VimInteractionMigration(
1675 self.db, self.my_vims, self.db_vims, self.logger
1676 ),
sritharan29a4c1a2022-05-05 12:15:04 +00001677 "verticalscale": VimInteractionResize(
1678 self.db, self.my_vims, self.db_vims, self.logger
1679 ),
tierno70eeb182020-10-19 16:38:00 +00001680 }
1681 self.time_last_task_processed = None
sousaedu80135b92021-02-17 15:05:18 +01001682 # lists of tasks to delete because nsrs or vnfrs has been deleted from db
1683 self.tasks_to_delete = []
1684 # it is idle when there are not vim_targets associated
1685 self.idle = True
tiernof1b640f2020-12-09 15:06:01 +00001686 self.task_locked_time = config["global"]["task_locked_time"]
tierno70eeb182020-10-19 16:38:00 +00001687
1688 def insert_task(self, task):
1689 try:
1690 self.task_queue.put(task, False)
1691 return None
1692 except queue.Full:
1693 raise NsWorkerException("timeout inserting a task")
1694
1695 def terminate(self):
1696 self.insert_task("exit")
1697
1698 def del_task(self, task):
1699 with self.task_lock:
1700 if task["status"] == "SCHEDULED":
1701 task["status"] = "SUPERSEDED"
1702 return True
1703 else: # task["status"] == "processing"
1704 self.task_lock.release()
1705 return False
1706
Guillermo Calvinoffee6602022-08-19 13:01:06 +02001707 def _process_vim_config(self, target_id: str, db_vim: dict) -> None:
tierno70eeb182020-10-19 16:38:00 +00001708 """
1709 Process vim config, creating vim configuration files as ca_cert
1710 :param target_id: vim/sdn/wim + id
1711 :param db_vim: Vim dictionary obtained from database
1712 :return: None. Modifies vim. Creates a folder target_id:worker_index and several files
1713 """
1714 if not db_vim.get("config"):
1715 return
sousaedu80135b92021-02-17 15:05:18 +01001716
tierno70eeb182020-10-19 16:38:00 +00001717 file_name = ""
Guillermo Calvinoffee6602022-08-19 13:01:06 +02001718 work_dir = "/app/osm_ro/certs"
sousaedu80135b92021-02-17 15:05:18 +01001719
tierno70eeb182020-10-19 16:38:00 +00001720 try:
1721 if db_vim["config"].get("ca_cert_content"):
Guillermo Calvinoffee6602022-08-19 13:01:06 +02001722 file_name = f"{work_dir}/{target_id}:{self.worker_index}"
sousaedu80135b92021-02-17 15:05:18 +01001723
Guillermo Calvinoffee6602022-08-19 13:01:06 +02001724 if not path.isdir(file_name):
1725 makedirs(file_name)
sousaedu80135b92021-02-17 15:05:18 +01001726
tierno70eeb182020-10-19 16:38:00 +00001727 file_name = file_name + "/ca_cert"
sousaedu80135b92021-02-17 15:05:18 +01001728
tierno70eeb182020-10-19 16:38:00 +00001729 with open(file_name, "w") as f:
1730 f.write(db_vim["config"]["ca_cert_content"])
1731 del db_vim["config"]["ca_cert_content"]
1732 db_vim["config"]["ca_cert"] = file_name
1733 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001734 raise NsWorkerException(
1735 "Error writing to file '{}': {}".format(file_name, e)
1736 )
tierno70eeb182020-10-19 16:38:00 +00001737
1738 def _load_plugin(self, name, type="vim"):
1739 # type can be vim or sdn
1740 if "rovim_dummy" not in self.plugins:
1741 self.plugins["rovim_dummy"] = VimDummyConnector
sousaedu80135b92021-02-17 15:05:18 +01001742
tierno70eeb182020-10-19 16:38:00 +00001743 if "rosdn_dummy" not in self.plugins:
1744 self.plugins["rosdn_dummy"] = SdnDummyConnector
sousaedu80135b92021-02-17 15:05:18 +01001745
tierno70eeb182020-10-19 16:38:00 +00001746 if name in self.plugins:
1747 return self.plugins[name]
sousaedu80135b92021-02-17 15:05:18 +01001748
tierno70eeb182020-10-19 16:38:00 +00001749 try:
sousaedubecd0832021-04-08 00:16:18 +02001750 for ep in entry_points(group="osm_ro{}.plugins".format(type), name=name):
1751 self.plugins[name] = ep.load()
tierno70eeb182020-10-19 16:38:00 +00001752 except Exception as e:
1753 raise NsWorkerException("Cannot load plugin osm_{}: {}".format(name, e))
sousaedu80135b92021-02-17 15:05:18 +01001754
tierno70eeb182020-10-19 16:38:00 +00001755 if name and name not in self.plugins:
sousaedu80135b92021-02-17 15:05:18 +01001756 raise NsWorkerException(
1757 "Plugin 'osm_{n}' has not been installed".format(n=name)
1758 )
1759
tierno70eeb182020-10-19 16:38:00 +00001760 return self.plugins[name]
1761
1762 def _unload_vim(self, target_id):
1763 """
1764 Unload a vim_account. Removes it from self db_vims dictionary, my_vims dictionary and vim_targets list
1765 :param target_id: Contains type:_id; where type can be 'vim', ...
1766 :return: None.
1767 """
1768 try:
tierno70eeb182020-10-19 16:38:00 +00001769 self.db_vims.pop(target_id, None)
1770 self.my_vims.pop(target_id, None)
sousaedu80135b92021-02-17 15:05:18 +01001771
tierno86153522020-12-06 18:27:16 +00001772 if target_id in self.vim_targets:
1773 self.vim_targets.remove(target_id)
sousaedu80135b92021-02-17 15:05:18 +01001774
tierno86153522020-12-06 18:27:16 +00001775 self.logger.info("Unloaded {}".format(target_id))
tierno70eeb182020-10-19 16:38:00 +00001776 except Exception as e:
1777 self.logger.error("Cannot unload {}: {}".format(target_id, e))
1778
1779 def _check_vim(self, target_id):
1780 """
1781 Load a VIM/SDN/WIM (if not loaded) and check connectivity, updating database with ENABLE or ERROR
1782 :param target_id: Contains type:_id; type can be 'vim', 'sdn' or 'wim'
1783 :return: None.
1784 """
1785 target, _, _id = target_id.partition(":")
1786 now = time.time()
1787 update_dict = {}
1788 unset_dict = {}
1789 op_text = ""
1790 step = ""
tierno86153522020-12-06 18:27:16 +00001791 loaded = target_id in self.vim_targets
sousaedu80135b92021-02-17 15:05:18 +01001792 target_database = (
1793 "vim_accounts"
1794 if target == "vim"
1795 else "wim_accounts"
1796 if target == "wim"
1797 else "sdns"
1798 )
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001799 error_text = ""
sousaedu80135b92021-02-17 15:05:18 +01001800
tierno70eeb182020-10-19 16:38:00 +00001801 try:
1802 step = "Getting {} from db".format(target_id)
1803 db_vim = self.db.get_one(target_database, {"_id": _id})
sousaedu80135b92021-02-17 15:05:18 +01001804
1805 for op_index, operation in enumerate(
1806 db_vim["_admin"].get("operations", ())
1807 ):
tierno70eeb182020-10-19 16:38:00 +00001808 if operation["operationState"] != "PROCESSING":
1809 continue
sousaedu80135b92021-02-17 15:05:18 +01001810
tierno70eeb182020-10-19 16:38:00 +00001811 locked_at = operation.get("locked_at")
sousaedu80135b92021-02-17 15:05:18 +01001812
tiernof1b640f2020-12-09 15:06:01 +00001813 if locked_at is not None and locked_at >= now - self.task_locked_time:
tierno70eeb182020-10-19 16:38:00 +00001814 # some other thread is doing this operation
1815 return
sousaedu80135b92021-02-17 15:05:18 +01001816
tierno70eeb182020-10-19 16:38:00 +00001817 # lock
1818 op_text = "_admin.operations.{}.".format(op_index)
sousaedu80135b92021-02-17 15:05:18 +01001819
1820 if not self.db.set_one(
1821 target_database,
1822 q_filter={
1823 "_id": _id,
1824 op_text + "operationState": "PROCESSING",
1825 op_text + "locked_at": locked_at,
1826 },
1827 update_dict={
1828 op_text + "locked_at": now,
1829 "admin.current_operation": op_index,
1830 },
1831 fail_on_empty=False,
1832 ):
tierno70eeb182020-10-19 16:38:00 +00001833 return
sousaedu80135b92021-02-17 15:05:18 +01001834
tierno70eeb182020-10-19 16:38:00 +00001835 unset_dict[op_text + "locked_at"] = None
1836 unset_dict["current_operation"] = None
1837 step = "Loading " + target_id
1838 error_text = self._load_vim(target_id)
sousaedu80135b92021-02-17 15:05:18 +01001839
tierno70eeb182020-10-19 16:38:00 +00001840 if not error_text:
1841 step = "Checking connectivity"
sousaedu80135b92021-02-17 15:05:18 +01001842
1843 if target == "vim":
tierno70eeb182020-10-19 16:38:00 +00001844 self.my_vims[target_id].check_vim_connectivity()
1845 else:
1846 self.my_vims[target_id].check_credentials()
sousaedu80135b92021-02-17 15:05:18 +01001847
tierno70eeb182020-10-19 16:38:00 +00001848 update_dict["_admin.operationalState"] = "ENABLED"
1849 update_dict["_admin.detailed-status"] = ""
1850 unset_dict[op_text + "detailed-status"] = None
1851 update_dict[op_text + "operationState"] = "COMPLETED"
sousaedu80135b92021-02-17 15:05:18 +01001852
tierno70eeb182020-10-19 16:38:00 +00001853 return
1854
1855 except Exception as e:
1856 error_text = "{}: {}".format(step, e)
1857 self.logger.error("{} for {}: {}".format(step, target_id, e))
1858
1859 finally:
1860 if update_dict or unset_dict:
1861 if error_text:
1862 update_dict[op_text + "operationState"] = "FAILED"
1863 update_dict[op_text + "detailed-status"] = error_text
1864 unset_dict.pop(op_text + "detailed-status", None)
1865 update_dict["_admin.operationalState"] = "ERROR"
1866 update_dict["_admin.detailed-status"] = error_text
sousaedu80135b92021-02-17 15:05:18 +01001867
tierno70eeb182020-10-19 16:38:00 +00001868 if op_text:
1869 update_dict[op_text + "statusEnteredTime"] = now
sousaedu80135b92021-02-17 15:05:18 +01001870
1871 self.db.set_one(
1872 target_database,
1873 q_filter={"_id": _id},
1874 update_dict=update_dict,
1875 unset=unset_dict,
1876 fail_on_empty=False,
1877 )
1878
tierno70eeb182020-10-19 16:38:00 +00001879 if not loaded:
1880 self._unload_vim(target_id)
1881
1882 def _reload_vim(self, target_id):
1883 if target_id in self.vim_targets:
1884 self._load_vim(target_id)
1885 else:
1886 # if the vim is not loaded, but database information of VIM is cached at self.db_vims,
1887 # just remove it to force load again next time it is needed
1888 self.db_vims.pop(target_id, None)
1889
1890 def _load_vim(self, target_id):
1891 """
1892 Load or reload a vim_account, sdn_controller or wim_account.
1893 Read content from database, load the plugin if not loaded.
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03001894 In case of error loading the plugin, it loads a failing VIM_connector
tierno70eeb182020-10-19 16:38:00 +00001895 It fills self db_vims dictionary, my_vims dictionary and vim_targets list
1896 :param target_id: Contains type:_id; where type can be 'vim', ...
1897 :return: None if ok, descriptive text if error
1898 """
1899 target, _, _id = target_id.partition(":")
sousaedu80135b92021-02-17 15:05:18 +01001900 target_database = (
1901 "vim_accounts"
1902 if target == "vim"
1903 else "wim_accounts"
1904 if target == "wim"
1905 else "sdns"
1906 )
tierno70eeb182020-10-19 16:38:00 +00001907 plugin_name = ""
1908 vim = None
Gulsum Atici2f995052022-11-23 16:06:40 +03001909 step = "Getting {}={} from db".format(target, _id)
sousaedu80135b92021-02-17 15:05:18 +01001910
tierno70eeb182020-10-19 16:38:00 +00001911 try:
tierno70eeb182020-10-19 16:38:00 +00001912 # TODO process for wim, sdnc, ...
1913 vim = self.db.get_one(target_database, {"_id": _id})
1914
1915 # if deep_get(vim, "config", "sdn-controller"):
1916 # step = "Getting sdn-controller-id='{}' from db".format(vim["config"]["sdn-controller"])
1917 # db_sdn = self.db.get_one("sdns", {"_id": vim["config"]["sdn-controller"]})
1918
1919 step = "Decrypting password"
1920 schema_version = vim.get("schema_version")
sousaedu80135b92021-02-17 15:05:18 +01001921 self.db.encrypt_decrypt_fields(
1922 vim,
1923 "decrypt",
1924 fields=("password", "secret"),
1925 schema_version=schema_version,
1926 salt=_id,
1927 )
tierno70eeb182020-10-19 16:38:00 +00001928 self._process_vim_config(target_id, vim)
sousaedu80135b92021-02-17 15:05:18 +01001929
tierno70eeb182020-10-19 16:38:00 +00001930 if target == "vim":
1931 plugin_name = "rovim_" + vim["vim_type"]
1932 step = "Loading plugin '{}'".format(plugin_name)
1933 vim_module_conn = self._load_plugin(plugin_name)
1934 step = "Loading {}'".format(target_id)
1935 self.my_vims[target_id] = vim_module_conn(
sousaedu80135b92021-02-17 15:05:18 +01001936 uuid=vim["_id"],
1937 name=vim["name"],
1938 tenant_id=vim.get("vim_tenant_id"),
1939 tenant_name=vim.get("vim_tenant_name"),
1940 url=vim["vim_url"],
1941 url_admin=None,
1942 user=vim["vim_user"],
1943 passwd=vim["vim_password"],
1944 config=vim.get("config") or {},
1945 persistent_info={},
tierno70eeb182020-10-19 16:38:00 +00001946 )
1947 else: # sdn
gifrerenombf3e9a82022-03-07 17:55:20 +00001948 plugin_name = "rosdn_" + (vim.get("type") or vim.get("wim_type"))
tierno70eeb182020-10-19 16:38:00 +00001949 step = "Loading plugin '{}'".format(plugin_name)
1950 vim_module_conn = self._load_plugin(plugin_name, "sdn")
1951 step = "Loading {}'".format(target_id)
1952 wim = deepcopy(vim)
1953 wim_config = wim.pop("config", {}) or {}
1954 wim["uuid"] = wim["_id"]
gifrerenom3ba07142022-03-07 18:00:01 +00001955 if "url" in wim and "wim_url" not in wim:
1956 wim["wim_url"] = wim["url"]
1957 elif "url" not in wim and "wim_url" in wim:
1958 wim["url"] = wim["wim_url"]
sousaedu80135b92021-02-17 15:05:18 +01001959
tierno70eeb182020-10-19 16:38:00 +00001960 if wim.get("dpid"):
1961 wim_config["dpid"] = wim.pop("dpid")
sousaedu80135b92021-02-17 15:05:18 +01001962
tierno70eeb182020-10-19 16:38:00 +00001963 if wim.get("switch_id"):
1964 wim_config["switch_id"] = wim.pop("switch_id")
sousaedu80135b92021-02-17 15:05:18 +01001965
1966 # wim, wim_account, config
1967 self.my_vims[target_id] = vim_module_conn(wim, wim, wim_config)
tierno70eeb182020-10-19 16:38:00 +00001968 self.db_vims[target_id] = vim
1969 self.error_status = None
sousaedu80135b92021-02-17 15:05:18 +01001970
1971 self.logger.info(
1972 "Connector loaded for {}, plugin={}".format(target_id, plugin_name)
1973 )
tierno70eeb182020-10-19 16:38:00 +00001974 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01001975 self.logger.error(
1976 "Cannot load {} plugin={}: {} {}".format(
1977 target_id, plugin_name, step, e
1978 )
1979 )
1980
tierno70eeb182020-10-19 16:38:00 +00001981 self.db_vims[target_id] = vim or {}
1982 self.db_vims[target_id] = FailingConnector(str(e))
1983 error_status = "{} Error: {}".format(step, e)
sousaedu80135b92021-02-17 15:05:18 +01001984
tierno70eeb182020-10-19 16:38:00 +00001985 return error_status
1986 finally:
1987 if target_id not in self.vim_targets:
1988 self.vim_targets.append(target_id)
1989
1990 def _get_db_task(self):
1991 """
1992 Read actions from database and reload them at memory. Fill self.refresh_list, pending_list, vim_actions
1993 :return: None
1994 """
1995 now = time.time()
sousaedu80135b92021-02-17 15:05:18 +01001996
tierno70eeb182020-10-19 16:38:00 +00001997 if not self.time_last_task_processed:
1998 self.time_last_task_processed = now
sousaedu80135b92021-02-17 15:05:18 +01001999
tierno70eeb182020-10-19 16:38:00 +00002000 try:
2001 while True:
gallardo2f4aaaa2022-01-31 16:50:48 +00002002 """
2003 # Log RO tasks only when loglevel is DEBUG
2004 if self.logger.getEffectiveLevel() == logging.DEBUG:
2005 self._log_ro_task(
2006 None,
2007 None,
2008 None,
2009 "TASK_WF",
2010 "task_locked_time="
2011 + str(self.task_locked_time)
2012 + " "
2013 + "time_last_task_processed="
2014 + str(self.time_last_task_processed)
2015 + " "
2016 + "now="
2017 + str(now),
2018 )
2019 """
tierno70eeb182020-10-19 16:38:00 +00002020 locked = self.db.set_one(
2021 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01002022 q_filter={
2023 "target_id": self.vim_targets,
2024 "tasks.status": ["SCHEDULED", "BUILD", "DONE", "FAILED"],
2025 "locked_at.lt": now - self.task_locked_time,
2026 "to_check_at.lt": self.time_last_task_processed,
aticig1ac189e2022-06-30 19:29:04 +03002027 "to_check_at.gt": -1,
sousaedu80135b92021-02-17 15:05:18 +01002028 },
tierno70eeb182020-10-19 16:38:00 +00002029 update_dict={"locked_by": self.my_id, "locked_at": now},
sousaedu80135b92021-02-17 15:05:18 +01002030 fail_on_empty=False,
2031 )
2032
tierno70eeb182020-10-19 16:38:00 +00002033 if locked:
2034 # read and return
2035 ro_task = self.db.get_one(
2036 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01002037 q_filter={
2038 "target_id": self.vim_targets,
2039 "tasks.status": ["SCHEDULED", "BUILD", "DONE", "FAILED"],
2040 "locked_at": now,
2041 },
2042 )
tierno70eeb182020-10-19 16:38:00 +00002043 return ro_task
sousaedu80135b92021-02-17 15:05:18 +01002044
tierno70eeb182020-10-19 16:38:00 +00002045 if self.time_last_task_processed == now:
2046 self.time_last_task_processed = None
2047 return None
2048 else:
2049 self.time_last_task_processed = now
2050 # self.time_last_task_processed = min(self.time_last_task_processed + 1000, now)
2051
2052 except DbException as e:
2053 self.logger.error("Database exception at _get_db_task: {}".format(e))
2054 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002055 self.logger.critical(
2056 "Unexpected exception at _get_db_task: {}".format(e), exc_info=True
2057 )
2058
tierno70eeb182020-10-19 16:38:00 +00002059 return None
2060
2061 def _delete_task(self, ro_task, task_index, task_depends, db_update):
2062 """
2063 Determine if this task need to be done or superseded
2064 :return: None
2065 """
2066 my_task = ro_task["tasks"][task_index]
2067 task_id = my_task["task_id"]
sousaedu80135b92021-02-17 15:05:18 +01002068 needed_delete = ro_task["vim_info"]["created"] or ro_task["vim_info"].get(
2069 "created_items", False
2070 )
2071
aticigddff2b02022-09-03 23:06:37 +03002072 self.logger.debug("Needed delete: {}".format(needed_delete))
tierno70eeb182020-10-19 16:38:00 +00002073 if my_task["status"] == "FAILED":
2074 return None, None # TODO need to be retry??
sousaedu80135b92021-02-17 15:05:18 +01002075
tierno70eeb182020-10-19 16:38:00 +00002076 try:
2077 for index, task in enumerate(ro_task["tasks"]):
2078 if index == task_index or not task:
2079 continue # own task
sousaedu80135b92021-02-17 15:05:18 +01002080
2081 if (
2082 my_task["target_record"] == task["target_record"]
2083 and task["action"] == "CREATE"
2084 ):
tierno70eeb182020-10-19 16:38:00 +00002085 # set to finished
sousaedu80135b92021-02-17 15:05:18 +01002086 db_update["tasks.{}.status".format(index)] = task[
2087 "status"
2088 ] = "FINISHED"
2089 elif task["action"] == "CREATE" and task["status"] not in (
2090 "FINISHED",
2091 "SUPERSEDED",
2092 ):
tierno70eeb182020-10-19 16:38:00 +00002093 needed_delete = False
sousaedu80135b92021-02-17 15:05:18 +01002094
tierno70eeb182020-10-19 16:38:00 +00002095 if needed_delete:
aticigddff2b02022-09-03 23:06:37 +03002096 self.logger.debug(
aticig285185e2022-05-02 21:23:48 +03002097 "Deleting ro_task={} task_index={}".format(ro_task, task_index)
2098 )
tierno70eeb182020-10-19 16:38:00 +00002099 return self.item2class[my_task["item"]].delete(ro_task, task_index)
2100 else:
2101 return "SUPERSEDED", None
2102 except Exception as e:
2103 if not isinstance(e, NsWorkerException):
sousaedu80135b92021-02-17 15:05:18 +01002104 self.logger.critical(
2105 "Unexpected exception at _delete_task task={}: {}".format(
2106 task_id, e
2107 ),
2108 exc_info=True,
2109 )
2110
aticig79ac6df2022-05-06 16:09:52 +03002111 return "FAILED", {"vim_status": "VIM_ERROR", "vim_message": str(e)}
tierno70eeb182020-10-19 16:38:00 +00002112
2113 def _create_task(self, ro_task, task_index, task_depends, db_update):
2114 """
2115 Determine if this task need to create something at VIM
2116 :return: None
2117 """
2118 my_task = ro_task["tasks"][task_index]
2119 task_id = my_task["task_id"]
sousaedu80135b92021-02-17 15:05:18 +01002120
tierno70eeb182020-10-19 16:38:00 +00002121 if my_task["status"] == "FAILED":
2122 return None, None # TODO need to be retry??
2123 elif my_task["status"] == "SCHEDULED":
2124 # check if already created by another task
2125 for index, task in enumerate(ro_task["tasks"]):
2126 if index == task_index or not task:
2127 continue # own task
sousaedu80135b92021-02-17 15:05:18 +01002128
2129 if task["action"] == "CREATE" and task["status"] not in (
2130 "SCHEDULED",
2131 "FINISHED",
2132 "SUPERSEDED",
2133 ):
tierno70eeb182020-10-19 16:38:00 +00002134 return task["status"], "COPY_VIM_INFO"
2135
2136 try:
2137 task_status, ro_vim_item_update = self.item2class[my_task["item"]].new(
sousaedu80135b92021-02-17 15:05:18 +01002138 ro_task, task_index, task_depends
2139 )
tierno70eeb182020-10-19 16:38:00 +00002140 # TODO update other CREATE tasks
2141 except Exception as e:
2142 if not isinstance(e, NsWorkerException):
sousaedu80135b92021-02-17 15:05:18 +01002143 self.logger.error(
2144 "Error executing task={}: {}".format(task_id, e), exc_info=True
2145 )
2146
tierno70eeb182020-10-19 16:38:00 +00002147 task_status = "FAILED"
aticig79ac6df2022-05-06 16:09:52 +03002148 ro_vim_item_update = {"vim_status": "VIM_ERROR", "vim_message": str(e)}
tierno70eeb182020-10-19 16:38:00 +00002149 # TODO update ro_vim_item_update
sousaedu80135b92021-02-17 15:05:18 +01002150
tierno70eeb182020-10-19 16:38:00 +00002151 return task_status, ro_vim_item_update
2152 else:
2153 return None, None
2154
2155 def _get_dependency(self, task_id, ro_task=None, target_id=None):
2156 """
2157 Look for dependency task
2158 :param task_id: Can be one of
2159 1. target_vim+blank+task.target_record_id: "(vim|sdn|wim):<id> (vnfrs|nsrs):(vld|vdu|flavor|image).<id>"
2160 2. task.target_record_id: "(vnfrs|nsrs):(vld|vdu|flavor|image).<id>"
2161 3. task.task_id: "<action_id>:number"
2162 :param ro_task:
2163 :param target_id:
2164 :return: database ro_task plus index of task
2165 """
sousaedu80135b92021-02-17 15:05:18 +01002166 if (
2167 task_id.startswith("vim:")
2168 or task_id.startswith("sdn:")
2169 or task_id.startswith("wim:")
2170 ):
tierno70eeb182020-10-19 16:38:00 +00002171 target_id, _, task_id = task_id.partition(" ")
2172
2173 if task_id.startswith("nsrs:") or task_id.startswith("vnfrs:"):
2174 ro_task_dependency = self.db.get_one(
2175 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01002176 q_filter={"target_id": target_id, "tasks.target_record_id": task_id},
2177 fail_on_empty=False,
2178 )
2179
tierno70eeb182020-10-19 16:38:00 +00002180 if ro_task_dependency:
2181 for task_index, task in enumerate(ro_task_dependency["tasks"]):
2182 if task["target_record_id"] == task_id:
2183 return ro_task_dependency, task_index
2184
2185 else:
2186 if ro_task:
2187 for task_index, task in enumerate(ro_task["tasks"]):
2188 if task and task["task_id"] == task_id:
2189 return ro_task, task_index
sousaedu80135b92021-02-17 15:05:18 +01002190
tierno70eeb182020-10-19 16:38:00 +00002191 ro_task_dependency = self.db.get_one(
2192 "ro_tasks",
sousaedu80135b92021-02-17 15:05:18 +01002193 q_filter={
2194 "tasks.ANYINDEX.task_id": task_id,
2195 "tasks.ANYINDEX.target_record.ne": None,
2196 },
2197 fail_on_empty=False,
2198 )
2199
aticigddff2b02022-09-03 23:06:37 +03002200 self.logger.debug("ro_task_dependency={}".format(ro_task_dependency))
tierno70eeb182020-10-19 16:38:00 +00002201 if ro_task_dependency:
palaciosj8f2060b2022-02-24 12:05:59 +00002202 for task_index, task in enumerate(ro_task_dependency["tasks"]):
tierno70eeb182020-10-19 16:38:00 +00002203 if task["task_id"] == task_id:
2204 return ro_task_dependency, task_index
2205 raise NsWorkerException("Cannot get depending task {}".format(task_id))
2206
Gulsum Aticid586d892023-02-13 18:40:03 +03002207 def update_vm_refresh(self, ro_task):
aticig1ac189e2022-06-30 19:29:04 +03002208 """Enables the VM status updates if self.refresh_config.active parameter
Gulsum Aticid586d892023-02-13 18:40:03 +03002209 is not -1 and then updates the DB accordingly
aticig1ac189e2022-06-30 19:29:04 +03002210
2211 """
2212 try:
2213 self.logger.debug("Checking if VM status update config")
2214 next_refresh = time.time()
Gulsum Aticid586d892023-02-13 18:40:03 +03002215 next_refresh = self._get_next_refresh(ro_task, next_refresh)
aticig1ac189e2022-06-30 19:29:04 +03002216
2217 if next_refresh != -1:
2218 db_ro_task_update = {}
2219 now = time.time()
2220 next_check_at = now + (24 * 60 * 60)
2221 next_check_at = min(next_check_at, next_refresh)
2222 db_ro_task_update["vim_info.refresh_at"] = next_refresh
2223 db_ro_task_update["to_check_at"] = next_check_at
2224
2225 self.logger.debug(
2226 "Finding tasks which to be updated to enable VM status updates"
2227 )
2228 refresh_tasks = self.db.get_list(
2229 "ro_tasks",
2230 q_filter={
2231 "tasks.status": "DONE",
2232 "to_check_at.lt": 0,
2233 },
2234 )
2235 self.logger.debug("Updating tasks to change the to_check_at status")
2236 for task in refresh_tasks:
2237 q_filter = {
2238 "_id": task["_id"],
2239 }
2240 self.db.set_one(
2241 "ro_tasks",
2242 q_filter=q_filter,
2243 update_dict=db_ro_task_update,
2244 fail_on_empty=True,
2245 )
2246
2247 except Exception as e:
2248 self.logger.error(f"Error updating tasks to enable VM status updates: {e}")
2249
Gulsum Aticid586d892023-02-13 18:40:03 +03002250 def _get_next_refresh(self, ro_task: dict, next_refresh: float):
2251 """Decide the next_refresh according to vim type and refresh config period.
2252 Args:
2253 ro_task (dict): ro_task details
2254 next_refresh (float): next refresh time as epoch format
2255
2256 Returns:
2257 next_refresh (float) -1 if vm updates are disabled or vim type is openstack.
2258 """
2259 target_vim = ro_task["target_id"]
2260 vim_type = self.db_vims[target_vim]["vim_type"]
2261 if self.refresh_config.active == -1 or vim_type == "openstack":
2262 next_refresh = -1
2263 else:
2264 next_refresh += self.refresh_config.active
2265 return next_refresh
2266
tierno70eeb182020-10-19 16:38:00 +00002267 def _process_pending_tasks(self, ro_task):
2268 ro_task_id = ro_task["_id"]
2269 now = time.time()
sousaedu80135b92021-02-17 15:05:18 +01002270 # one day
2271 next_check_at = now + (24 * 60 * 60)
tierno70eeb182020-10-19 16:38:00 +00002272 db_ro_task_update = {}
2273
2274 def _update_refresh(new_status):
2275 # compute next_refresh
2276 nonlocal task
2277 nonlocal next_check_at
2278 nonlocal db_ro_task_update
2279 nonlocal ro_task
2280
2281 next_refresh = time.time()
sousaedu80135b92021-02-17 15:05:18 +01002282
tierno70eeb182020-10-19 16:38:00 +00002283 if task["item"] in ("image", "flavor"):
aticig1ac189e2022-06-30 19:29:04 +03002284 next_refresh += self.refresh_config.image
tierno70eeb182020-10-19 16:38:00 +00002285 elif new_status == "BUILD":
aticig1ac189e2022-06-30 19:29:04 +03002286 next_refresh += self.refresh_config.build
tierno70eeb182020-10-19 16:38:00 +00002287 elif new_status == "DONE":
Gulsum Aticid586d892023-02-13 18:40:03 +03002288 next_refresh = self._get_next_refresh(ro_task, next_refresh)
tierno70eeb182020-10-19 16:38:00 +00002289 else:
aticig1ac189e2022-06-30 19:29:04 +03002290 next_refresh += self.refresh_config.error
sousaedu80135b92021-02-17 15:05:18 +01002291
tierno70eeb182020-10-19 16:38:00 +00002292 next_check_at = min(next_check_at, next_refresh)
2293 db_ro_task_update["vim_info.refresh_at"] = next_refresh
2294 ro_task["vim_info"]["refresh_at"] = next_refresh
2295
2296 try:
gallardo2f4aaaa2022-01-31 16:50:48 +00002297 """
2298 # Log RO tasks only when loglevel is DEBUG
2299 if self.logger.getEffectiveLevel() == logging.DEBUG:
2300 self._log_ro_task(ro_task, None, None, "TASK_WF", "GET_TASK")
2301 """
aticig1ac189e2022-06-30 19:29:04 +03002302 # Check if vim status refresh is enabled again
Gulsum Aticid586d892023-02-13 18:40:03 +03002303 self.update_vm_refresh(ro_task)
tiernof1b640f2020-12-09 15:06:01 +00002304 # 0: get task_status_create
2305 lock_object = None
tierno70eeb182020-10-19 16:38:00 +00002306 task_status_create = None
sousaedu80135b92021-02-17 15:05:18 +01002307 task_create = next(
2308 (
2309 t
2310 for t in ro_task["tasks"]
2311 if t
2312 and t["action"] == "CREATE"
2313 and t["status"] in ("BUILD", "DONE")
2314 ),
2315 None,
2316 )
2317
tierno70eeb182020-10-19 16:38:00 +00002318 if task_create:
2319 task_status_create = task_create["status"]
sousaedu80135b92021-02-17 15:05:18 +01002320
tiernof1b640f2020-12-09 15:06:01 +00002321 # 1: look for tasks in status SCHEDULED, or in status CREATE if action is DONE or BUILD
tierno70eeb182020-10-19 16:38:00 +00002322 for task_action in ("DELETE", "CREATE", "EXEC"):
2323 db_vim_update = None
2324 new_status = None
sousaedu80135b92021-02-17 15:05:18 +01002325
tierno70eeb182020-10-19 16:38:00 +00002326 for task_index, task in enumerate(ro_task["tasks"]):
2327 if not task:
2328 continue # task deleted
sousaedu80135b92021-02-17 15:05:18 +01002329
tierno55fa0bb2020-12-08 23:11:53 +00002330 task_depends = {}
tierno70eeb182020-10-19 16:38:00 +00002331 target_update = None
sousaedu80135b92021-02-17 15:05:18 +01002332
2333 if (
2334 (
2335 task_action in ("DELETE", "EXEC")
2336 and task["status"] not in ("SCHEDULED", "BUILD")
2337 )
2338 or task["action"] != task_action
2339 or (
2340 task_action == "CREATE"
2341 and task["status"] in ("FINISHED", "SUPERSEDED")
2342 )
2343 ):
tierno70eeb182020-10-19 16:38:00 +00002344 continue
sousaedu80135b92021-02-17 15:05:18 +01002345
tierno70eeb182020-10-19 16:38:00 +00002346 task_path = "tasks.{}.status".format(task_index)
2347 try:
2348 db_vim_info_update = None
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03002349 dependency_ro_task = {}
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 ]
aticigddff2b02022-09-03 23:06:37 +03002365 self.logger.debug(
aticig285185e2022-05-02 21:23:48 +03002366 "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 )
tierno70eeb182020-10-19 16:38:00 +00002420 if task["action"] == "DELETE":
Gulsum Atici21c55d62023-02-02 20:41:00 +03002421 (
2422 new_status,
2423 db_vim_info_update,
2424 ) = self._delete_task(
sousaedu80135b92021-02-17 15:05:18 +01002425 ro_task, task_index, task_depends, db_ro_task_update
2426 )
2427 new_status = (
2428 "FINISHED" if new_status == "DONE" else new_status
2429 )
tierno70eeb182020-10-19 16:38:00 +00002430 # ^with FINISHED instead of DONE it will not be refreshing
sousaedu80135b92021-02-17 15:05:18 +01002431
tierno70eeb182020-10-19 16:38:00 +00002432 if new_status in ("FINISHED", "SUPERSEDED"):
2433 target_update = "DELETE"
2434 elif task["action"] == "EXEC":
sousaedu80135b92021-02-17 15:05:18 +01002435 (
2436 new_status,
2437 db_vim_info_update,
2438 db_task_update,
2439 ) = self.item2class[task["item"]].exec(
2440 ro_task, task_index, task_depends
2441 )
2442 new_status = (
2443 "FINISHED" if new_status == "DONE" else new_status
2444 )
tierno70eeb182020-10-19 16:38:00 +00002445 # ^with FINISHED instead of DONE it will not be refreshing
sousaedu80135b92021-02-17 15:05:18 +01002446
tierno70eeb182020-10-19 16:38:00 +00002447 if db_task_update:
2448 # load into database the modified db_task_update "retries" and "next_retry"
2449 if db_task_update.get("retries"):
sousaedu80135b92021-02-17 15:05:18 +01002450 db_ro_task_update[
2451 "tasks.{}.retries".format(task_index)
2452 ] = db_task_update["retries"]
2453
2454 next_check_at = time.time() + db_task_update.get(
2455 "next_retry", 60
2456 )
tierno70eeb182020-10-19 16:38:00 +00002457 target_update = None
2458 elif task["action"] == "CREATE":
2459 if task["status"] == "SCHEDULED":
2460 if task_status_create:
2461 new_status = task_status_create
2462 target_update = "COPY_VIM_INFO"
2463 else:
sousaedu80135b92021-02-17 15:05:18 +01002464 new_status, db_vim_info_update = self.item2class[
2465 task["item"]
2466 ].new(ro_task, task_index, task_depends)
tierno70eeb182020-10-19 16:38:00 +00002467 _update_refresh(new_status)
2468 else:
aticig1ac189e2022-06-30 19:29:04 +03002469 refresh_at = ro_task["vim_info"]["refresh_at"]
2470 if refresh_at and refresh_at != -1 and now > refresh_at:
Gulsum Atici21c55d62023-02-02 20:41:00 +03002471 (
2472 new_status,
2473 db_vim_info_update,
2474 ) = self.item2class[
sousaedu80135b92021-02-17 15:05:18 +01002475 task["item"]
2476 ].refresh(ro_task)
tierno70eeb182020-10-19 16:38:00 +00002477 _update_refresh(new_status)
gallardo7788f692022-01-20 09:07:08 +00002478 else:
2479 # The refresh is updated to avoid set the value of "refresh_at" to
2480 # default value (next_check_at = now + (24 * 60 * 60)) when status is BUILD,
2481 # because it can happen that in this case the task is never processed
2482 _update_refresh(task["status"])
sousaedu80135b92021-02-17 15:05:18 +01002483
tierno70eeb182020-10-19 16:38:00 +00002484 except Exception as e:
2485 new_status = "FAILED"
sousaedu80135b92021-02-17 15:05:18 +01002486 db_vim_info_update = {
2487 "vim_status": "VIM_ERROR",
aticig79ac6df2022-05-06 16:09:52 +03002488 "vim_message": str(e),
sousaedu80135b92021-02-17 15:05:18 +01002489 }
2490
2491 if not isinstance(
2492 e, (NsWorkerException, vimconn.VimConnException)
2493 ):
2494 self.logger.error(
2495 "Unexpected exception at _delete_task task={}: {}".format(
2496 task["task_id"], e
2497 ),
2498 exc_info=True,
2499 )
tierno70eeb182020-10-19 16:38:00 +00002500
2501 try:
2502 if db_vim_info_update:
2503 db_vim_update = db_vim_info_update.copy()
sousaedu80135b92021-02-17 15:05:18 +01002504 db_ro_task_update.update(
2505 {
2506 "vim_info." + k: v
2507 for k, v in db_vim_info_update.items()
2508 }
2509 )
tierno70eeb182020-10-19 16:38:00 +00002510 ro_task["vim_info"].update(db_vim_info_update)
2511
2512 if new_status:
2513 if task_action == "CREATE":
2514 task_status_create = new_status
2515 db_ro_task_update[task_path] = new_status
tierno70eeb182020-10-19 16:38:00 +00002516
sousaedu80135b92021-02-17 15:05:18 +01002517 if target_update or db_vim_update:
tierno70eeb182020-10-19 16:38:00 +00002518 if target_update == "DELETE":
2519 self._update_target(task, None)
2520 elif target_update == "COPY_VIM_INFO":
2521 self._update_target(task, ro_task["vim_info"])
2522 else:
2523 self._update_target(task, db_vim_update)
2524
2525 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002526 if (
2527 isinstance(e, DbException)
2528 and e.http_code == HTTPStatus.NOT_FOUND
2529 ):
tierno70eeb182020-10-19 16:38:00 +00002530 # if the vnfrs or nsrs has been removed from database, this task must be removed
sousaedu80135b92021-02-17 15:05:18 +01002531 self.logger.debug(
2532 "marking to delete task={}".format(task["task_id"])
2533 )
tierno70eeb182020-10-19 16:38:00 +00002534 self.tasks_to_delete.append(task)
2535 else:
sousaedu80135b92021-02-17 15:05:18 +01002536 self.logger.error(
2537 "Unexpected exception at _update_target task={}: {}".format(
2538 task["task_id"], e
2539 ),
2540 exc_info=True,
2541 )
tierno70eeb182020-10-19 16:38:00 +00002542
tiernof1b640f2020-12-09 15:06:01 +00002543 locked_at = ro_task["locked_at"]
sousaedu80135b92021-02-17 15:05:18 +01002544
tiernof1b640f2020-12-09 15:06:01 +00002545 if lock_object:
sousaedu80135b92021-02-17 15:05:18 +01002546 locked_at = [
2547 lock_object["locked_at"],
2548 lock_object["locked_at"] + self.task_locked_time,
2549 ]
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03002550 # locked_at contains two times to avoid race condition. In case the lock has been renewed, it will
tiernof1b640f2020-12-09 15:06:01 +00002551 # contain exactly locked_at + self.task_locked_time
2552 LockRenew.remove_lock_object(lock_object)
sousaedu80135b92021-02-17 15:05:18 +01002553
2554 q_filter = {
2555 "_id": ro_task["_id"],
2556 "to_check_at": ro_task["to_check_at"],
2557 "locked_at": locked_at,
2558 }
tierno70eeb182020-10-19 16:38:00 +00002559 # modify own task. Try filtering by to_next_check. For race condition if to_check_at has been modified,
2560 # outside this task (by ro_nbi) do not update it
2561 db_ro_task_update["locked_by"] = None
Gulsum Atici1f58a1a2023-06-19 13:59:46 +03002562 # locked_at converted to int only for debugging. When it is not decimals it means it has been unlocked
tiernof1b640f2020-12-09 15:06:01 +00002563 db_ro_task_update["locked_at"] = int(now - self.task_locked_time)
2564 db_ro_task_update["modified_at"] = now
tierno70eeb182020-10-19 16:38:00 +00002565 db_ro_task_update["to_check_at"] = next_check_at
sousaedu80135b92021-02-17 15:05:18 +01002566
gallardo2f4aaaa2022-01-31 16:50:48 +00002567 """
2568 # Log RO tasks only when loglevel is DEBUG
2569 if self.logger.getEffectiveLevel() == logging.DEBUG:
2570 db_ro_task_update_log = db_ro_task_update.copy()
2571 db_ro_task_update_log["_id"] = q_filter["_id"]
2572 self._log_ro_task(None, db_ro_task_update_log, None, "TASK_WF", "SET_TASK")
2573 """
2574
sousaedu80135b92021-02-17 15:05:18 +01002575 if not self.db.set_one(
2576 "ro_tasks",
2577 update_dict=db_ro_task_update,
2578 q_filter=q_filter,
2579 fail_on_empty=False,
2580 ):
tierno70eeb182020-10-19 16:38:00 +00002581 del db_ro_task_update["to_check_at"]
2582 del q_filter["to_check_at"]
gallardo2f4aaaa2022-01-31 16:50:48 +00002583 """
2584 # Log RO tasks only when loglevel is DEBUG
2585 if self.logger.getEffectiveLevel() == logging.DEBUG:
2586 self._log_ro_task(
2587 None,
2588 db_ro_task_update_log,
2589 None,
2590 "TASK_WF",
2591 "SET_TASK " + str(q_filter),
2592 )
2593 """
sousaedu80135b92021-02-17 15:05:18 +01002594 self.db.set_one(
2595 "ro_tasks",
2596 q_filter=q_filter,
2597 update_dict=db_ro_task_update,
2598 fail_on_empty=True,
2599 )
tierno70eeb182020-10-19 16:38:00 +00002600 except DbException as e:
sousaedu80135b92021-02-17 15:05:18 +01002601 self.logger.error(
2602 "ro_task={} Error updating database {}".format(ro_task_id, e)
2603 )
tierno70eeb182020-10-19 16:38:00 +00002604 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002605 self.logger.error(
2606 "Error executing ro_task={}: {}".format(ro_task_id, e), exc_info=True
2607 )
tierno70eeb182020-10-19 16:38:00 +00002608
2609 def _update_target(self, task, ro_vim_item_update):
2610 table, _, temp = task["target_record"].partition(":")
2611 _id, _, path_vim_status = temp.partition(":")
sousaedu80135b92021-02-17 15:05:18 +01002612 path_item = path_vim_status[: path_vim_status.rfind(".")]
2613 path_item = path_item[: path_item.rfind(".")]
tierno70eeb182020-10-19 16:38:00 +00002614 # path_vim_status: dot separated list targeting vim information, e.g. "vdur.10.vim_info.vim:id"
2615 # path_item: dot separated list targeting record information, e.g. "vdur.10"
sousaedu80135b92021-02-17 15:05:18 +01002616
tierno70eeb182020-10-19 16:38:00 +00002617 if ro_vim_item_update:
sousaedu80135b92021-02-17 15:05:18 +01002618 update_dict = {
2619 path_vim_status + "." + k: v
2620 for k, v in ro_vim_item_update.items()
2621 if k
aticig79ac6df2022-05-06 16:09:52 +03002622 in (
2623 "vim_id",
2624 "vim_details",
2625 "vim_message",
2626 "vim_name",
2627 "vim_status",
2628 "interfaces",
aticig37ecec02022-05-25 03:12:36 +03002629 "interfaces_backup",
aticig79ac6df2022-05-06 16:09:52 +03002630 )
sousaedu80135b92021-02-17 15:05:18 +01002631 }
2632
tierno70eeb182020-10-19 16:38:00 +00002633 if path_vim_status.startswith("vdur."):
2634 # for backward compatibility, add vdur.name apart from vdur.vim_name
2635 if ro_vim_item_update.get("vim_name"):
2636 update_dict[path_item + ".name"] = ro_vim_item_update["vim_name"]
sousaedu80135b92021-02-17 15:05:18 +01002637
tierno70eeb182020-10-19 16:38:00 +00002638 # for backward compatibility, add vdur.vim-id apart from vdur.vim_id
2639 if ro_vim_item_update.get("vim_id"):
2640 update_dict[path_item + ".vim-id"] = ro_vim_item_update["vim_id"]
sousaedu80135b92021-02-17 15:05:18 +01002641
tierno70eeb182020-10-19 16:38:00 +00002642 # update general status
2643 if ro_vim_item_update.get("vim_status"):
sousaedu80135b92021-02-17 15:05:18 +01002644 update_dict[path_item + ".status"] = ro_vim_item_update[
2645 "vim_status"
2646 ]
2647
tierno70eeb182020-10-19 16:38:00 +00002648 if ro_vim_item_update.get("interfaces"):
2649 path_interfaces = path_item + ".interfaces"
sousaedu80135b92021-02-17 15:05:18 +01002650
tierno70eeb182020-10-19 16:38:00 +00002651 for i, iface in enumerate(ro_vim_item_update.get("interfaces")):
2652 if iface:
sousaedu80135b92021-02-17 15:05:18 +01002653 update_dict.update(
2654 {
2655 path_interfaces + ".{}.".format(i) + k: v
2656 for k, v in iface.items()
2657 if k in ("vlan", "compute_node", "pci")
2658 }
2659 )
2660
tierno70eeb182020-10-19 16:38:00 +00002661 # put ip_address and mac_address with ip-address and mac-address
sousaedu80135b92021-02-17 15:05:18 +01002662 if iface.get("ip_address"):
2663 update_dict[
2664 path_interfaces + ".{}.".format(i) + "ip-address"
2665 ] = iface["ip_address"]
2666
2667 if iface.get("mac_address"):
2668 update_dict[
2669 path_interfaces + ".{}.".format(i) + "mac-address"
2670 ] = iface["mac_address"]
2671
tierno70eeb182020-10-19 16:38:00 +00002672 if iface.get("mgmt_vnf_interface") and iface.get("ip_address"):
sousaedu80135b92021-02-17 15:05:18 +01002673 update_dict["ip-address"] = iface.get("ip_address").split(
2674 ";"
2675 )[0]
2676
tierno70eeb182020-10-19 16:38:00 +00002677 if iface.get("mgmt_vdu_interface") and iface.get("ip_address"):
sousaedu80135b92021-02-17 15:05:18 +01002678 update_dict[path_item + ".ip-address"] = iface.get(
2679 "ip_address"
2680 ).split(";")[0]
tierno70eeb182020-10-19 16:38:00 +00002681
2682 self.db.set_one(table, q_filter={"_id": _id}, update_dict=update_dict)
aticig37ecec02022-05-25 03:12:36 +03002683
2684 # If interfaces exists, it backups VDU interfaces in the DB for healing operations
2685 if ro_vim_item_update.get("interfaces"):
2686 search_key = path_vim_status + ".interfaces"
2687 if update_dict.get(search_key):
2688 interfaces_backup_update = {
2689 path_vim_status + ".interfaces_backup": update_dict[search_key]
2690 }
2691
2692 self.db.set_one(
2693 table,
2694 q_filter={"_id": _id},
2695 update_dict=interfaces_backup_update,
2696 )
2697
tierno70eeb182020-10-19 16:38:00 +00002698 else:
2699 update_dict = {path_item + ".status": "DELETED"}
sousaedu80135b92021-02-17 15:05:18 +01002700 self.db.set_one(
2701 table,
2702 q_filter={"_id": _id},
2703 update_dict=update_dict,
2704 unset={path_vim_status: None},
2705 )
tierno70eeb182020-10-19 16:38:00 +00002706
2707 def _process_delete_db_tasks(self):
2708 """
2709 Delete task from database because vnfrs or nsrs or both have been deleted
2710 :return: None. Uses and modify self.tasks_to_delete
2711 """
2712 while self.tasks_to_delete:
2713 task = self.tasks_to_delete[0]
2714 vnfrs_deleted = None
2715 nsr_id = task["nsr_id"]
sousaedu80135b92021-02-17 15:05:18 +01002716
tierno70eeb182020-10-19 16:38:00 +00002717 if task["target_record"].startswith("vnfrs:"):
2718 # check if nsrs is present
2719 if self.db.get_one("nsrs", {"_id": nsr_id}, fail_on_empty=False):
2720 vnfrs_deleted = task["target_record"].split(":")[1]
sousaedu80135b92021-02-17 15:05:18 +01002721
tierno70eeb182020-10-19 16:38:00 +00002722 try:
2723 self.delete_db_tasks(self.db, nsr_id, vnfrs_deleted)
2724 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002725 self.logger.error(
2726 "Error deleting task={}: {}".format(task["task_id"], e)
2727 )
tierno70eeb182020-10-19 16:38:00 +00002728 self.tasks_to_delete.pop(0)
2729
2730 @staticmethod
2731 def delete_db_tasks(db, nsr_id, vnfrs_deleted):
2732 """
2733 Static method because it is called from osm_ng_ro.ns
2734 :param db: instance of database to use
2735 :param nsr_id: affected nsrs id
2736 :param vnfrs_deleted: only tasks with this vnfr id. If None, all affected by nsr_id
2737 :return: None, exception is fails
2738 """
2739 retries = 5
2740 for retry in range(retries):
2741 ro_tasks = db.get_list("ro_tasks", {"tasks.nsr_id": nsr_id})
2742 now = time.time()
2743 conflict = False
sousaedu80135b92021-02-17 15:05:18 +01002744
tierno70eeb182020-10-19 16:38:00 +00002745 for ro_task in ro_tasks:
2746 db_update = {}
2747 to_delete_ro_task = True
sousaedu80135b92021-02-17 15:05:18 +01002748
tierno70eeb182020-10-19 16:38:00 +00002749 for index, task in enumerate(ro_task["tasks"]):
2750 if not task:
2751 pass
sousaedu80135b92021-02-17 15:05:18 +01002752 elif (not vnfrs_deleted and task["nsr_id"] == nsr_id) or (
2753 vnfrs_deleted
2754 and task["target_record"].startswith("vnfrs:" + vnfrs_deleted)
2755 ):
tierno70eeb182020-10-19 16:38:00 +00002756 db_update["tasks.{}".format(index)] = None
2757 else:
sousaedu80135b92021-02-17 15:05:18 +01002758 # used by other nsr, ro_task cannot be deleted
2759 to_delete_ro_task = False
2760
tierno70eeb182020-10-19 16:38:00 +00002761 # delete or update if nobody has changed ro_task meanwhile. Used modified_at for known if changed
2762 if to_delete_ro_task:
sousaedu80135b92021-02-17 15:05:18 +01002763 if not db.del_one(
2764 "ro_tasks",
2765 q_filter={
2766 "_id": ro_task["_id"],
2767 "modified_at": ro_task["modified_at"],
2768 },
2769 fail_on_empty=False,
2770 ):
tierno70eeb182020-10-19 16:38:00 +00002771 conflict = True
2772 elif db_update:
2773 db_update["modified_at"] = now
sousaedu80135b92021-02-17 15:05:18 +01002774 if not db.set_one(
2775 "ro_tasks",
2776 q_filter={
2777 "_id": ro_task["_id"],
2778 "modified_at": ro_task["modified_at"],
2779 },
2780 update_dict=db_update,
2781 fail_on_empty=False,
2782 ):
tierno70eeb182020-10-19 16:38:00 +00002783 conflict = True
2784 if not conflict:
2785 return
2786 else:
2787 raise NsWorkerException("Exceeded {} retries".format(retries))
2788
tierno1d213f42020-04-24 14:02:51 +00002789 def run(self):
2790 # load database
tierno86153522020-12-06 18:27:16 +00002791 self.logger.info("Starting")
tierno1d213f42020-04-24 14:02:51 +00002792 while True:
tierno70eeb182020-10-19 16:38:00 +00002793 # step 1: get commands from queue
tierno1d213f42020-04-24 14:02:51 +00002794 try:
tierno86153522020-12-06 18:27:16 +00002795 if self.vim_targets:
2796 task = self.task_queue.get(block=False)
2797 else:
2798 if not self.idle:
2799 self.logger.debug("enters in idle state")
2800 self.idle = True
2801 task = self.task_queue.get(block=True)
2802 self.idle = False
2803
tierno1d213f42020-04-24 14:02:51 +00002804 if task[0] == "terminate":
2805 break
tierno70eeb182020-10-19 16:38:00 +00002806 elif task[0] == "load_vim":
tierno86153522020-12-06 18:27:16 +00002807 self.logger.info("order to load vim {}".format(task[1]))
tierno1d213f42020-04-24 14:02:51 +00002808 self._load_vim(task[1])
tierno70eeb182020-10-19 16:38:00 +00002809 elif task[0] == "unload_vim":
tierno86153522020-12-06 18:27:16 +00002810 self.logger.info("order to unload vim {}".format(task[1]))
tierno70eeb182020-10-19 16:38:00 +00002811 self._unload_vim(task[1])
2812 elif task[0] == "reload_vim":
2813 self._reload_vim(task[1])
2814 elif task[0] == "check_vim":
tierno86153522020-12-06 18:27:16 +00002815 self.logger.info("order to check vim {}".format(task[1]))
tierno70eeb182020-10-19 16:38:00 +00002816 self._check_vim(task[1])
tierno1d213f42020-04-24 14:02:51 +00002817 continue
tierno70eeb182020-10-19 16:38:00 +00002818 except Exception as e:
2819 if isinstance(e, queue.Empty):
2820 pass
2821 else:
sousaedu80135b92021-02-17 15:05:18 +01002822 self.logger.critical(
2823 "Error processing task: {}".format(e), exc_info=True
2824 )
tierno1d213f42020-04-24 14:02:51 +00002825
tierno70eeb182020-10-19 16:38:00 +00002826 # step 2: process pending_tasks, delete not needed tasks
tierno1d213f42020-04-24 14:02:51 +00002827 try:
tierno70eeb182020-10-19 16:38:00 +00002828 if self.tasks_to_delete:
2829 self._process_delete_db_tasks()
tierno1d213f42020-04-24 14:02:51 +00002830 busy = False
gallardo2f4aaaa2022-01-31 16:50:48 +00002831 """
2832 # Log RO tasks only when loglevel is DEBUG
2833 if self.logger.getEffectiveLevel() == logging.DEBUG:
2834 _ = self._get_db_all_tasks()
2835 """
tierno1d213f42020-04-24 14:02:51 +00002836 ro_task = self._get_db_task()
2837 if ro_task:
aticigddff2b02022-09-03 23:06:37 +03002838 self.logger.debug("Task to process: {}".format(ro_task))
palaciosj8f2060b2022-02-24 12:05:59 +00002839 time.sleep(1)
tierno70eeb182020-10-19 16:38:00 +00002840 self._process_pending_tasks(ro_task)
tierno1d213f42020-04-24 14:02:51 +00002841 busy = True
2842 if not busy:
2843 time.sleep(5)
2844 except Exception as e:
sousaedu80135b92021-02-17 15:05:18 +01002845 self.logger.critical(
2846 "Unexpected exception at run: " + str(e), exc_info=True
2847 )
tierno1d213f42020-04-24 14:02:51 +00002848
tierno86153522020-12-06 18:27:16 +00002849 self.logger.info("Finishing")