blob: c685ed14a2ab9d551cd4b3bde4a42982c5f705db [file] [log] [blame]
tiernoc0e42e22018-05-11 11:36:10 +02001#!/usr/bin/python3
2# -*- coding: utf-8 -*-
3
tierno2e215512018-11-28 09:37:52 +00004##
5# Copyright 2018 Telefonica S.A.
6#
7# Licensed under the Apache License, Version 2.0 (the "License"); you may
8# not use this file except in compliance with the License. You may obtain
9# a copy of the License at
10#
11# http://www.apache.org/licenses/LICENSE-2.0
12#
13# Unless required by applicable law or agreed to in writing, software
14# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
15# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
16# License for the specific language governing permissions and limitations
17# under the License.
18##
19
quilesj7e13aeb2019-10-08 13:34:55 +020020
21# DEBUG WITH PDB
quilesj7e13aeb2019-10-08 13:34:55 +020022import pdb
23
k4.rahul44306072023-05-05 14:24:31 +053024import os
tiernoc0e42e22018-05-11 11:36:10 +020025import asyncio
26import yaml
tierno275411e2018-05-16 14:33:32 +020027import logging
28import logging.handlers
29import getopt
tierno275411e2018-05-16 14:33:32 +020030import sys
Gabriel Cuba4c0e6802023-10-09 13:22:38 -050031from random import SystemRandom
tierno59d22d22018-09-25 18:10:19 +020032
rshri932105f2024-07-05 15:11:55 +000033from osm_lcm import ns, vim_sdn, netslice, k8s
tierno69f0d382020-05-07 13:08:09 +000034from osm_lcm.ng_ro import NgRoException, NgRoClient
35from osm_lcm.ROclient import ROClient, ROClientException
quilesj7e13aeb2019-10-08 13:34:55 +020036
tierno94f06112020-02-11 12:38:19 +000037from time import time
tierno8069ce52019-08-28 15:34:33 +000038from osm_lcm.lcm_utils import versiontuple, LcmException, TaskRegistry, LcmExceptionExit
tiernoa4dea5a2020-01-05 16:29:30 +000039from osm_lcm import version as lcm_version, version_date as lcm_version_date
tierno8069ce52019-08-28 15:34:33 +000040
bravof922c4172020-11-24 21:21:43 -030041from osm_common import msglocal, msgkafka
tierno98768132018-09-11 12:07:21 +020042from osm_common import version as common_version
tierno59d22d22018-09-25 18:10:19 +020043from osm_common.dbbase import DbException
tiernoc0e42e22018-05-11 11:36:10 +020044from osm_common.fsbase import FsException
45from osm_common.msgbase import MsgException
bravof922c4172020-11-24 21:21:43 -030046from osm_lcm.data_utils.database.database import Database
47from osm_lcm.data_utils.filesystem.filesystem import Filesystem
Luis Vegaa27dc532022-11-11 20:10:49 +000048from osm_lcm.data_utils.lcm_config import LcmCfg
garciadeblas96b94f52024-07-08 16:18:21 +020049from osm_lcm.data_utils.list_utils import find_in_list
aticig56b86c22022-06-29 10:43:05 +030050from osm_lcm.lcm_hc import get_health_check_file
Gabriel Cuba4c0e6802023-10-09 13:22:38 -050051from os import path, getenv
tierno59d22d22018-09-25 18:10:19 +020052from n2vc import version as n2vc_version
bravof922c4172020-11-24 21:21:43 -030053import traceback
tiernoc0e42e22018-05-11 11:36:10 +020054
Gabriel Cuba4c0e6802023-10-09 13:22:38 -050055if getenv("OSMLCM_PDB_DEBUG", None) is not None:
quilesj7e13aeb2019-10-08 13:34:55 +020056 pdb.set_trace()
57
tiernoc0e42e22018-05-11 11:36:10 +020058
tierno275411e2018-05-16 14:33:32 +020059__author__ = "Alfonso Tierno"
tiernoe64f7fb2019-09-11 08:55:52 +000060min_RO_version = "6.0.2"
tierno6e9d2eb2018-09-12 17:47:18 +020061min_n2vc_version = "0.0.2"
quilesj7e13aeb2019-10-08 13:34:55 +020062
tierno16427352019-04-22 11:37:36 +000063min_common_version = "0.1.19"
tierno275411e2018-05-16 14:33:32 +020064
65
tiernoc0e42e22018-05-11 11:36:10 +020066class Lcm:
garciadeblas96b94f52024-07-08 16:18:21 +020067 profile_collection_mapping = {
68 "infra_controller_profiles": "k8sinfra_controller",
69 "infra_config_profiles": "k8sinfra_config",
70 "resource_profiles": "k8sresource",
71 "app_profiles": "k8sapp",
72 }
73
garciadeblas5697b8b2021-03-24 09:17:02 +010074 ping_interval_pace = (
75 120 # how many time ping is send once is confirmed all is running
76 )
77 ping_interval_boot = 5 # how many time ping is sent when booting
Luis Vegaa27dc532022-11-11 20:10:49 +000078
79 main_config = LcmCfg()
tiernoa9843d82018-10-24 10:44:20 +020080
Gabriel Cubae7898982023-05-11 01:57:21 -050081 def __init__(self, config_file):
tiernoc0e42e22018-05-11 11:36:10 +020082 """
83 Init, Connect to database, filesystem storage, and messaging
84 :param config: two level dictionary with configuration. Top level should contain 'database', 'storage',
85 :return: None
86 """
tiernoc0e42e22018-05-11 11:36:10 +020087 self.db = None
88 self.msg = None
tierno16427352019-04-22 11:37:36 +000089 self.msg_admin = None
tiernoc0e42e22018-05-11 11:36:10 +020090 self.fs = None
91 self.pings_not_received = 1
tiernoc2564fe2019-01-28 16:18:56 +000092 self.consecutive_errors = 0
93 self.first_start = False
tiernoc0e42e22018-05-11 11:36:10 +020094
tiernoc0e42e22018-05-11 11:36:10 +020095 # logging
garciadeblas5697b8b2021-03-24 09:17:02 +010096 self.logger = logging.getLogger("lcm")
tierno16427352019-04-22 11:37:36 +000097 # get id
98 self.worker_id = self.get_process_id()
tiernoc0e42e22018-05-11 11:36:10 +020099 # load configuration
100 config = self.read_config_file(config_file)
Luis Vegaa27dc532022-11-11 20:10:49 +0000101 self.main_config.set_from_dict(config)
102 self.main_config.transform()
103 self.main_config.load_from_env()
104 self.logger.critical("Loaded configuration:" + str(self.main_config.to_dict()))
105 # TODO: check if lcm_hc.py is necessary
106 self.health_check_file = get_health_check_file(self.main_config.to_dict())
garciadeblas5697b8b2021-03-24 09:17:02 +0100107 self.ns = (
108 self.netslice
109 ) = (
110 self.vim
rshri932105f2024-07-05 15:11:55 +0000111 ) = (
112 self.wim
113 ) = (
114 self.sdn
115 ) = (
116 self.k8scluster
117 ) = (
118 self.vca
119 ) = (
120 self.k8srepo
121 ) = (
122 self.cluster
123 ) = (
124 self.k8s_app
125 ) = self.k8s_resource = self.k8s_infra_controller = self.k8s_infra_config = None
tiernoc0e42e22018-05-11 11:36:10 +0200126
127 # logging
garciadeblas5697b8b2021-03-24 09:17:02 +0100128 log_format_simple = (
129 "%(asctime)s %(levelname)s %(name)s %(filename)s:%(lineno)s %(message)s"
130 )
131 log_formatter_simple = logging.Formatter(
132 log_format_simple, datefmt="%Y-%m-%dT%H:%M:%S"
133 )
Luis Vegaa27dc532022-11-11 20:10:49 +0000134 if self.main_config.globalConfig.logfile:
garciadeblas5697b8b2021-03-24 09:17:02 +0100135 file_handler = logging.handlers.RotatingFileHandler(
Luis Vegaa27dc532022-11-11 20:10:49 +0000136 self.main_config.globalConfig.logfile,
137 maxBytes=100e6,
138 backupCount=9,
139 delay=0,
garciadeblas5697b8b2021-03-24 09:17:02 +0100140 )
tiernoc0e42e22018-05-11 11:36:10 +0200141 file_handler.setFormatter(log_formatter_simple)
142 self.logger.addHandler(file_handler)
Luis Vegaa27dc532022-11-11 20:10:49 +0000143 if not self.main_config.globalConfig.to_dict()["nologging"]:
tiernoc0e42e22018-05-11 11:36:10 +0200144 str_handler = logging.StreamHandler()
145 str_handler.setFormatter(log_formatter_simple)
146 self.logger.addHandler(str_handler)
147
Luis Vegaa27dc532022-11-11 20:10:49 +0000148 if self.main_config.globalConfig.to_dict()["loglevel"]:
149 self.logger.setLevel(self.main_config.globalConfig.loglevel)
tiernoc0e42e22018-05-11 11:36:10 +0200150
151 # logging other modules
garciadeblas4623e982024-09-11 14:28:38 +0200152 for logger in ("message", "database", "storage", "tsdb", "gitops"):
Luis Vegaa27dc532022-11-11 20:10:49 +0000153 logger_config = self.main_config.to_dict()[logger]
154 logger_module = logging.getLogger(logger_config["logger_name"])
155 if logger_config["logfile"]:
garciadeblas5697b8b2021-03-24 09:17:02 +0100156 file_handler = logging.handlers.RotatingFileHandler(
Luis Vegaa27dc532022-11-11 20:10:49 +0000157 logger_config["logfile"], maxBytes=100e6, backupCount=9, delay=0
garciadeblas5697b8b2021-03-24 09:17:02 +0100158 )
tiernoc0e42e22018-05-11 11:36:10 +0200159 file_handler.setFormatter(log_formatter_simple)
160 logger_module.addHandler(file_handler)
Luis Vegaa27dc532022-11-11 20:10:49 +0000161 if logger_config["loglevel"]:
162 logger_module.setLevel(logger_config["loglevel"])
garciadeblas5697b8b2021-03-24 09:17:02 +0100163 self.logger.critical(
164 "starting osm/lcm version {} {}".format(lcm_version, lcm_version_date)
165 )
tierno59d22d22018-09-25 18:10:19 +0200166
tiernoc0e42e22018-05-11 11:36:10 +0200167 # check version of N2VC
168 # TODO enhance with int conversion or from distutils.version import LooseVersion
169 # or with list(map(int, version.split(".")))
tierno59d22d22018-09-25 18:10:19 +0200170 if versiontuple(n2vc_version) < versiontuple(min_n2vc_version):
garciadeblas5697b8b2021-03-24 09:17:02 +0100171 raise LcmException(
172 "Not compatible osm/N2VC version '{}'. Needed '{}' or higher".format(
173 n2vc_version, min_n2vc_version
174 )
175 )
tierno59d22d22018-09-25 18:10:19 +0200176 # check version of common
tierno27246d82018-09-27 15:59:09 +0200177 if versiontuple(common_version) < versiontuple(min_common_version):
garciadeblas5697b8b2021-03-24 09:17:02 +0100178 raise LcmException(
179 "Not compatible osm/common version '{}'. Needed '{}' or higher".format(
180 common_version, min_common_version
181 )
182 )
tierno22f4f9c2018-06-11 18:53:39 +0200183
tiernoc0e42e22018-05-11 11:36:10 +0200184 try:
Luis Vegaa27dc532022-11-11 20:10:49 +0000185 self.db = Database(self.main_config.to_dict()).instance.db
tiernoc0e42e22018-05-11 11:36:10 +0200186
Luis Vegaa27dc532022-11-11 20:10:49 +0000187 self.fs = Filesystem(self.main_config.to_dict()).instance.fs
sousaedu40365e82021-07-26 15:24:21 +0200188 self.fs.sync()
tiernoc0e42e22018-05-11 11:36:10 +0200189
quilesj7e13aeb2019-10-08 13:34:55 +0200190 # copy message configuration in order to remove 'group_id' for msg_admin
Luis Vegaa27dc532022-11-11 20:10:49 +0000191 config_message = self.main_config.message.to_dict()
Gabriel Cubae7898982023-05-11 01:57:21 -0500192 config_message["loop"] = asyncio.get_event_loop()
tiernoc2564fe2019-01-28 16:18:56 +0000193 if config_message["driver"] == "local":
tiernoc0e42e22018-05-11 11:36:10 +0200194 self.msg = msglocal.MsgLocal()
tiernoc2564fe2019-01-28 16:18:56 +0000195 self.msg.connect(config_message)
tierno16427352019-04-22 11:37:36 +0000196 self.msg_admin = msglocal.MsgLocal()
197 config_message.pop("group_id", None)
198 self.msg_admin.connect(config_message)
tiernoc2564fe2019-01-28 16:18:56 +0000199 elif config_message["driver"] == "kafka":
tiernoc0e42e22018-05-11 11:36:10 +0200200 self.msg = msgkafka.MsgKafka()
tiernoc2564fe2019-01-28 16:18:56 +0000201 self.msg.connect(config_message)
tierno16427352019-04-22 11:37:36 +0000202 self.msg_admin = msgkafka.MsgKafka()
203 config_message.pop("group_id", None)
204 self.msg_admin.connect(config_message)
tiernoc0e42e22018-05-11 11:36:10 +0200205 else:
garciadeblas5697b8b2021-03-24 09:17:02 +0100206 raise LcmException(
207 "Invalid configuration param '{}' at '[message]':'driver'".format(
Luis Vegaa27dc532022-11-11 20:10:49 +0000208 self.main_config.message.driver
garciadeblas5697b8b2021-03-24 09:17:02 +0100209 )
210 )
tiernoc0e42e22018-05-11 11:36:10 +0200211 except (DbException, FsException, MsgException) as e:
212 self.logger.critical(str(e), exc_info=True)
213 raise LcmException(str(e))
214
kuused124bfe2019-06-18 12:09:24 +0200215 # contains created tasks/futures to be able to cancel
bravof922c4172020-11-24 21:21:43 -0300216 self.lcm_tasks = TaskRegistry(self.worker_id, self.logger)
kuused124bfe2019-06-18 12:09:24 +0200217
rshri932105f2024-07-05 15:11:55 +0000218 self.logger.info(
219 "Worker_id: {} main_config: {} lcm tasks: {}".format(
220 self.worker_id, self.main_config, self.lcm_tasks
221 )
222 )
223
tierno22f4f9c2018-06-11 18:53:39 +0200224 async def check_RO_version(self):
tiernoe64f7fb2019-09-11 08:55:52 +0000225 tries = 14
226 last_error = None
227 while True:
Luis Vegaa27dc532022-11-11 20:10:49 +0000228 ro_uri = self.main_config.RO.uri
229 if not ro_uri:
230 ro_uri = ""
tiernoe64f7fb2019-09-11 08:55:52 +0000231 try:
tierno2357f4e2020-10-19 16:38:59 +0000232 # try new RO, if fail old RO
233 try:
Luis Vegaa27dc532022-11-11 20:10:49 +0000234 self.main_config.RO.uri = ro_uri + "ro"
Gabriel Cubae7898982023-05-11 01:57:21 -0500235 ro_server = NgRoClient(**self.main_config.RO.to_dict())
tierno2357f4e2020-10-19 16:38:59 +0000236 ro_version = await ro_server.get_version()
Luis Vegaa27dc532022-11-11 20:10:49 +0000237 self.main_config.RO.ng = True
tierno2357f4e2020-10-19 16:38:59 +0000238 except Exception:
Luis Vegaa27dc532022-11-11 20:10:49 +0000239 self.main_config.RO.uri = ro_uri + "openmano"
Gabriel Cubae7898982023-05-11 01:57:21 -0500240 ro_server = ROClient(**self.main_config.RO.to_dict())
tierno2357f4e2020-10-19 16:38:59 +0000241 ro_version = await ro_server.get_version()
Luis Vegaa27dc532022-11-11 20:10:49 +0000242 self.main_config.RO.ng = False
tiernoe64f7fb2019-09-11 08:55:52 +0000243 if versiontuple(ro_version) < versiontuple(min_RO_version):
garciadeblas5697b8b2021-03-24 09:17:02 +0100244 raise LcmException(
245 "Not compatible osm/RO version '{}'. Needed '{}' or higher".format(
246 ro_version, min_RO_version
247 )
248 )
249 self.logger.info(
250 "Connected to RO version {} new-generation version {}".format(
Luis Vegaa27dc532022-11-11 20:10:49 +0000251 ro_version, self.main_config.RO.ng
garciadeblas5697b8b2021-03-24 09:17:02 +0100252 )
253 )
tiernoe64f7fb2019-09-11 08:55:52 +0000254 return
tierno69f0d382020-05-07 13:08:09 +0000255 except (ROClientException, NgRoException) as e:
Luis Vegaa27dc532022-11-11 20:10:49 +0000256 self.main_config.RO.uri = ro_uri
tiernoe64f7fb2019-09-11 08:55:52 +0000257 tries -= 1
bravof922c4172020-11-24 21:21:43 -0300258 traceback.print_tb(e.__traceback__)
garciadeblas5697b8b2021-03-24 09:17:02 +0100259 error_text = "Error while connecting to RO on {}: {}".format(
Luis Vegaa27dc532022-11-11 20:10:49 +0000260 self.main_config.RO.uri, e
garciadeblas5697b8b2021-03-24 09:17:02 +0100261 )
tiernoe64f7fb2019-09-11 08:55:52 +0000262 if tries <= 0:
263 self.logger.critical(error_text)
264 raise LcmException(error_text)
265 if last_error != error_text:
266 last_error = error_text
garciadeblas5697b8b2021-03-24 09:17:02 +0100267 self.logger.error(
268 error_text + ". Waiting until {} seconds".format(5 * tries)
269 )
tiernoe64f7fb2019-09-11 08:55:52 +0000270 await asyncio.sleep(5)
tierno22f4f9c2018-06-11 18:53:39 +0200271
tiernoc0e42e22018-05-11 11:36:10 +0200272 async def test(self, param=None):
273 self.logger.debug("Starting/Ending test task: {}".format(param))
274
tiernoc0e42e22018-05-11 11:36:10 +0200275 async def kafka_ping(self):
276 self.logger.debug("Task kafka_ping Enter")
277 consecutive_errors = 0
278 first_start = True
279 kafka_has_received = False
280 self.pings_not_received = 1
281 while True:
282 try:
tierno16427352019-04-22 11:37:36 +0000283 await self.msg_admin.aiowrite(
garciadeblas5697b8b2021-03-24 09:17:02 +0100284 "admin",
285 "ping",
286 {
287 "from": "lcm",
288 "to": "lcm",
289 "worker_id": self.worker_id,
290 "version": lcm_version,
291 },
garciadeblas5697b8b2021-03-24 09:17:02 +0100292 )
tiernoc0e42e22018-05-11 11:36:10 +0200293 # time between pings are low when it is not received and at starting
garciadeblas5697b8b2021-03-24 09:17:02 +0100294 wait_time = (
295 self.ping_interval_boot
296 if not kafka_has_received
297 else self.ping_interval_pace
298 )
tiernoc0e42e22018-05-11 11:36:10 +0200299 if not self.pings_not_received:
300 kafka_has_received = True
301 self.pings_not_received += 1
Gabriel Cubae7898982023-05-11 01:57:21 -0500302 await asyncio.sleep(wait_time)
tiernoc0e42e22018-05-11 11:36:10 +0200303 if self.pings_not_received > 10:
304 raise LcmException("It is not receiving pings from Kafka bus")
305 consecutive_errors = 0
306 first_start = False
307 except LcmException:
308 raise
309 except Exception as e:
310 # if not first_start is the first time after starting. So leave more time and wait
311 # to allow kafka starts
312 if consecutive_errors == 8 if not first_start else 30:
garciadeblas5697b8b2021-03-24 09:17:02 +0100313 self.logger.error(
314 "Task kafka_read task exit error too many errors. Exception: {}".format(
315 e
316 )
317 )
tiernoc0e42e22018-05-11 11:36:10 +0200318 raise
319 consecutive_errors += 1
garciadeblas5697b8b2021-03-24 09:17:02 +0100320 self.logger.error(
321 "Task kafka_read retrying after Exception {}".format(e)
322 )
tierno16427352019-04-22 11:37:36 +0000323 wait_time = 2 if not first_start else 5
Gabriel Cubae7898982023-05-11 01:57:21 -0500324 await asyncio.sleep(wait_time)
tiernoc0e42e22018-05-11 11:36:10 +0200325
garciadeblas96b94f52024-07-08 16:18:21 +0200326 def get_operation_params(self, item, operation_id):
327 operation_history = item.get("operationHistory", [])
328 operation = find_in_list(
329 operation_history, lambda op: op["op_id"] == operation_id
330 )
331 return operation.get("operationParams", {})
332
Gabriel Cubab6049d32023-10-30 13:44:49 -0500333 async def kafka_read_callback(self, topic, command, params):
gcalvinoed7f6d42018-12-14 14:44:56 +0100334 order_id = 1
rshri932105f2024-07-05 15:11:55 +0000335 self.logger.info(
336 "Topic: {} command: {} params: {} order ID: {}".format(
337 topic, command, params, order_id
338 )
339 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100340 if topic != "admin" and command != "ping":
garciadeblas5697b8b2021-03-24 09:17:02 +0100341 self.logger.debug(
342 "Task kafka_read receives {} {}: {}".format(topic, command, params)
343 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100344 self.consecutive_errors = 0
345 self.first_start = False
346 order_id += 1
rshri932105f2024-07-05 15:11:55 +0000347 self.logger.info(
348 "Consecutive error: {} First start: {}".format(
349 self.consecutive_errors, self.first_start
350 )
351 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100352 if command == "exit":
353 raise LcmExceptionExit
354 elif command.startswith("#"):
355 return
356 elif command == "echo":
357 # just for test
358 print(params)
359 sys.stdout.flush()
360 return
361 elif command == "test":
Gabriel Cubae7898982023-05-11 01:57:21 -0500362 asyncio.Task(self.test(params))
gcalvinoed7f6d42018-12-14 14:44:56 +0100363 return
364
365 if topic == "admin":
366 if command == "ping" and params["to"] == "lcm" and params["from"] == "lcm":
tierno16427352019-04-22 11:37:36 +0000367 if params.get("worker_id") != self.worker_id:
368 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100369 self.pings_not_received = 0
tierno3e359b12019-02-03 02:29:13 +0100370 try:
aticig56b86c22022-06-29 10:43:05 +0300371 with open(self.health_check_file, "w") as f:
tierno3e359b12019-02-03 02:29:13 +0100372 f.write(str(time()))
373 except Exception as e:
garciadeblas5697b8b2021-03-24 09:17:02 +0100374 self.logger.error(
375 "Cannot write into '{}' for healthcheck: {}".format(
aticig56b86c22022-06-29 10:43:05 +0300376 self.health_check_file, e
garciadeblas5697b8b2021-03-24 09:17:02 +0100377 )
378 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100379 return
Gabriel Cubab6049d32023-10-30 13:44:49 -0500380 elif topic == "nslcmops":
381 if command == "cancel":
382 nslcmop_id = params["_id"]
383 self.logger.debug("Cancelling nslcmop {}".format(nslcmop_id))
384 nsr_id = params["nsInstanceId"]
385 # cancel the tasks and wait
386 for task in self.lcm_tasks.cancel("ns", nsr_id, nslcmop_id):
387 try:
388 await task
389 self.logger.debug(
390 "Cancelled task ended {},{},{}".format(
391 nsr_id, nslcmop_id, task
392 )
393 )
394 except asyncio.CancelledError:
395 self.logger.debug(
396 "Task already cancelled and finished {},{},{}".format(
397 nsr_id, nslcmop_id, task
398 )
399 )
400 # update DB
401 q_filter = {"_id": nslcmop_id}
402 update_dict = {
403 "operationState": "FAILED_TEMP",
404 "isCancelPending": False,
405 }
406 unset_dict = {
407 "cancelMode": None,
408 }
409 self.db.set_one(
410 "nslcmops",
411 q_filter=q_filter,
412 update_dict=update_dict,
413 fail_on_empty=False,
414 unset=unset_dict,
415 )
416 self.logger.debug("LCM task cancelled {},{}".format(nsr_id, nslcmop_id))
417 return
magnussonle9198bb2020-01-21 13:00:51 +0100418 elif topic == "pla":
419 if command == "placement":
420 self.ns.update_nsrs_with_pla_result(params)
421 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100422 elif topic == "k8scluster":
423 if command == "create" or command == "created":
424 k8scluster_id = params.get("_id")
425 task = asyncio.ensure_future(self.k8scluster.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100426 self.lcm_tasks.register(
427 "k8scluster", k8scluster_id, order_id, "k8scluster_create", task
428 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100429 return
dariofaccin8bbeeb02023-01-23 18:13:27 +0100430 elif command == "edit" or command == "edited":
431 k8scluster_id = params.get("_id")
432 task = asyncio.ensure_future(self.k8scluster.edit(params, order_id))
433 self.lcm_tasks.register(
434 "k8scluster", k8scluster_id, order_id, "k8scluster_edit", task
435 )
436 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100437 elif command == "delete" or command == "deleted":
438 k8scluster_id = params.get("_id")
439 task = asyncio.ensure_future(self.k8scluster.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100440 self.lcm_tasks.register(
441 "k8scluster", k8scluster_id, order_id, "k8scluster_delete", task
442 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100443 return
David Garciac1fe90a2021-03-31 19:12:02 +0200444 elif topic == "vca":
445 if command == "create" or command == "created":
446 vca_id = params.get("_id")
447 task = asyncio.ensure_future(self.vca.create(params, order_id))
448 self.lcm_tasks.register("vca", vca_id, order_id, "vca_create", task)
449 return
Dario Faccin8e53c6d2023-01-10 10:38:41 +0000450 elif command == "edit" or command == "edited":
451 vca_id = params.get("_id")
452 task = asyncio.ensure_future(self.vca.edit(params, order_id))
453 self.lcm_tasks.register("vca", vca_id, order_id, "vca_edit", task)
454 return
David Garciac1fe90a2021-03-31 19:12:02 +0200455 elif command == "delete" or command == "deleted":
456 vca_id = params.get("_id")
457 task = asyncio.ensure_future(self.vca.delete(params, order_id))
458 self.lcm_tasks.register("vca", vca_id, order_id, "vca_delete", task)
459 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100460 elif topic == "k8srepo":
461 if command == "create" or command == "created":
462 k8srepo_id = params.get("_id")
463 self.logger.debug("k8srepo_id = {}".format(k8srepo_id))
464 task = asyncio.ensure_future(self.k8srepo.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100465 self.lcm_tasks.register(
466 "k8srepo", k8srepo_id, order_id, "k8srepo_create", task
467 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100468 return
469 elif command == "delete" or command == "deleted":
470 k8srepo_id = params.get("_id")
471 task = asyncio.ensure_future(self.k8srepo.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100472 self.lcm_tasks.register(
473 "k8srepo", k8srepo_id, order_id, "k8srepo_delete", task
474 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100475 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100476 elif topic == "ns":
tierno307425f2020-01-26 23:35:59 +0000477 if command == "instantiate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100478 # self.logger.debug("Deploying NS {}".format(nsr_id))
rshri932105f2024-07-05 15:11:55 +0000479 self.logger.info("NS instantiate")
gcalvinoed7f6d42018-12-14 14:44:56 +0100480 nslcmop = params
481 nslcmop_id = nslcmop["_id"]
482 nsr_id = nslcmop["nsInstanceId"]
rshri932105f2024-07-05 15:11:55 +0000483 self.logger.info(
484 "NsLCMOP: {} NsLCMOP_ID:{} nsr_id: {}".format(
485 nslcmop, nslcmop_id, nsr_id
486 )
487 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100488 task = asyncio.ensure_future(self.ns.instantiate(nsr_id, nslcmop_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100489 self.lcm_tasks.register(
490 "ns", nsr_id, nslcmop_id, "ns_instantiate", task
491 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100492 return
tierno307425f2020-01-26 23:35:59 +0000493 elif command == "terminate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100494 # self.logger.debug("Deleting NS {}".format(nsr_id))
495 nslcmop = params
496 nslcmop_id = nslcmop["_id"]
497 nsr_id = nslcmop["nsInstanceId"]
498 self.lcm_tasks.cancel(topic, nsr_id)
499 task = asyncio.ensure_future(self.ns.terminate(nsr_id, nslcmop_id))
500 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_terminate", task)
501 return
ksaikiranr3fde2c72021-03-15 10:39:06 +0530502 elif command == "vca_status_refresh":
503 nslcmop = params
504 nslcmop_id = nslcmop["_id"]
505 nsr_id = nslcmop["nsInstanceId"]
garciadeblas5697b8b2021-03-24 09:17:02 +0100506 task = asyncio.ensure_future(
507 self.ns.vca_status_refresh(nsr_id, nslcmop_id)
508 )
509 self.lcm_tasks.register(
510 "ns", nsr_id, nslcmop_id, "ns_vca_status_refresh", task
511 )
ksaikiranr3fde2c72021-03-15 10:39:06 +0530512 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100513 elif command == "action":
514 # self.logger.debug("Update NS {}".format(nsr_id))
515 nslcmop = params
516 nslcmop_id = nslcmop["_id"]
517 nsr_id = nslcmop["nsInstanceId"]
518 task = asyncio.ensure_future(self.ns.action(nsr_id, nslcmop_id))
519 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_action", task)
520 return
aticigdffa6212022-04-12 15:27:53 +0300521 elif command == "update":
522 # self.logger.debug("Update NS {}".format(nsr_id))
523 nslcmop = params
524 nslcmop_id = nslcmop["_id"]
525 nsr_id = nslcmop["nsInstanceId"]
526 task = asyncio.ensure_future(self.ns.update(nsr_id, nslcmop_id))
527 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_update", task)
528 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100529 elif command == "scale":
530 # self.logger.debug("Update NS {}".format(nsr_id))
531 nslcmop = params
532 nslcmop_id = nslcmop["_id"]
533 nsr_id = nslcmop["nsInstanceId"]
534 task = asyncio.ensure_future(self.ns.scale(nsr_id, nslcmop_id))
535 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_scale", task)
536 return
garciadeblas07f4e4c2022-06-09 09:42:58 +0200537 elif command == "heal":
538 # self.logger.debug("Healing NS {}".format(nsr_id))
539 nslcmop = params
540 nslcmop_id = nslcmop["_id"]
541 nsr_id = nslcmop["nsInstanceId"]
542 task = asyncio.ensure_future(self.ns.heal(nsr_id, nslcmop_id))
preethika.p28b0bf82022-09-23 07:36:28 +0000543 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_heal", task)
garciadeblas07f4e4c2022-06-09 09:42:58 +0200544 return
elumalai80bcf1c2022-04-28 18:05:01 +0530545 elif command == "migrate":
546 nslcmop = params
547 nslcmop_id = nslcmop["_id"]
548 nsr_id = nslcmop["nsInstanceId"]
549 task = asyncio.ensure_future(self.ns.migrate(nsr_id, nslcmop_id))
550 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_migrate", task)
551 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100552 elif command == "show":
tiernoc2564fe2019-01-28 16:18:56 +0000553 nsr_id = params
gcalvinoed7f6d42018-12-14 14:44:56 +0100554 try:
555 db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
garciadeblas5697b8b2021-03-24 09:17:02 +0100556 print(
557 "nsr:\n _id={}\n operational-status: {}\n config-status: {}"
558 "\n detailed-status: {}\n deploy: {}\n tasks: {}"
559 "".format(
560 nsr_id,
561 db_nsr["operational-status"],
562 db_nsr["config-status"],
563 db_nsr["detailed-status"],
564 db_nsr["_admin"]["deployed"],
Gabriel Cuba411af2e2023-01-06 17:23:22 -0500565 self.lcm_tasks.task_registry["ns"].get(nsr_id, ""),
garciadeblas5697b8b2021-03-24 09:17:02 +0100566 )
567 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100568 except Exception as e:
569 print("nsr {} not found: {}".format(nsr_id, e))
570 sys.stdout.flush()
571 return
572 elif command == "deleted":
573 return # TODO cleaning of task just in case should be done
garciadeblas5697b8b2021-03-24 09:17:02 +0100574 elif command in (
elumalaica7ece02022-04-12 12:47:32 +0530575 "vnf_terminated",
elumalaib9e357c2022-04-27 09:58:38 +0530576 "policy_updated",
garciadeblas5697b8b2021-03-24 09:17:02 +0100577 "terminated",
578 "instantiated",
579 "scaled",
garciadeblas07f4e4c2022-06-09 09:42:58 +0200580 "healed",
garciadeblas5697b8b2021-03-24 09:17:02 +0100581 "actioned",
aticigdffa6212022-04-12 15:27:53 +0300582 "updated",
elumalai80bcf1c2022-04-28 18:05:01 +0530583 "migrated",
rshri932105f2024-07-05 15:11:55 +0000584 "verticalscaled",
garciadeblas5697b8b2021-03-24 09:17:02 +0100585 ): # "scaled-cooldown-time"
gcalvinoed7f6d42018-12-14 14:44:56 +0100586 return
elumalaica7ece02022-04-12 12:47:32 +0530587
gcalvinoed7f6d42018-12-14 14:44:56 +0100588 elif topic == "nsi": # netslice LCM processes (instantiate, terminate, etc)
tierno307425f2020-01-26 23:35:59 +0000589 if command == "instantiate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100590 # self.logger.debug("Instantiating Network Slice {}".format(nsilcmop["netsliceInstanceId"]))
591 nsilcmop = params
592 nsilcmop_id = nsilcmop["_id"] # slice operation id
593 nsir_id = nsilcmop["netsliceInstanceId"] # slice record id
garciadeblas5697b8b2021-03-24 09:17:02 +0100594 task = asyncio.ensure_future(
595 self.netslice.instantiate(nsir_id, nsilcmop_id)
596 )
597 self.lcm_tasks.register(
598 "nsi", nsir_id, nsilcmop_id, "nsi_instantiate", task
599 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100600 return
tierno307425f2020-01-26 23:35:59 +0000601 elif command == "terminate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100602 # self.logger.debug("Terminating Network Slice NS {}".format(nsilcmop["netsliceInstanceId"]))
603 nsilcmop = params
604 nsilcmop_id = nsilcmop["_id"] # slice operation id
605 nsir_id = nsilcmop["netsliceInstanceId"] # slice record id
606 self.lcm_tasks.cancel(topic, nsir_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100607 task = asyncio.ensure_future(
608 self.netslice.terminate(nsir_id, nsilcmop_id)
609 )
610 self.lcm_tasks.register(
611 "nsi", nsir_id, nsilcmop_id, "nsi_terminate", task
612 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100613 return
614 elif command == "show":
tiernoc2564fe2019-01-28 16:18:56 +0000615 nsir_id = params
gcalvinoed7f6d42018-12-14 14:44:56 +0100616 try:
617 db_nsir = self.db.get_one("nsirs", {"_id": nsir_id})
garciadeblas5697b8b2021-03-24 09:17:02 +0100618 print(
619 "nsir:\n _id={}\n operational-status: {}\n config-status: {}"
620 "\n detailed-status: {}\n deploy: {}\n tasks: {}"
621 "".format(
622 nsir_id,
623 db_nsir["operational-status"],
624 db_nsir["config-status"],
625 db_nsir["detailed-status"],
626 db_nsir["_admin"]["deployed"],
Gabriel Cuba411af2e2023-01-06 17:23:22 -0500627 self.lcm_tasks.task_registry["nsi"].get(nsir_id, ""),
garciadeblas5697b8b2021-03-24 09:17:02 +0100628 )
629 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100630 except Exception as e:
631 print("nsir {} not found: {}".format(nsir_id, e))
632 sys.stdout.flush()
633 return
634 elif command == "deleted":
635 return # TODO cleaning of task just in case should be done
garciadeblas5697b8b2021-03-24 09:17:02 +0100636 elif command in (
637 "terminated",
638 "instantiated",
639 "scaled",
garciadeblas07f4e4c2022-06-09 09:42:58 +0200640 "healed",
garciadeblas5697b8b2021-03-24 09:17:02 +0100641 "actioned",
642 ): # "scaled-cooldown-time"
gcalvinoed7f6d42018-12-14 14:44:56 +0100643 return
644 elif topic == "vim_account":
645 vim_id = params["_id"]
garciadeblas96b94f52024-07-08 16:18:21 +0200646 op_id = vim_id
647 op_params = params
yshah771dea82024-07-05 15:11:49 +0000648 db_vim = self.db.get_one("vim_accounts", {"_id": vim_id})
garciadeblas96b94f52024-07-08 16:18:21 +0200649 vim_config = db_vim.get("config", {})
650 self.logger.debug("Db Vim: {}".format(db_vim))
tiernof210c1c2019-10-16 09:09:58 +0000651 if command in ("create", "created"):
garciadeblas96b94f52024-07-08 16:18:21 +0200652 self.logger.debug("Main config: {}".format(self.main_config.to_dict()))
653 if "credentials" in vim_config:
654 self.logger.info("Vim add cloud credentials")
655 task = asyncio.ensure_future(
656 self.cloud_credentials.add(op_id, op_params, db_vim)
657 )
658 self.lcm_tasks.register(
659 "vim_account", vim_id, op_id, "cloud_credentials_add", task
660 )
Luis Vegaa27dc532022-11-11 20:10:49 +0000661 if not self.main_config.RO.ng:
garciadeblas96b94f52024-07-08 16:18:21 +0200662 self.logger.info("Calling RO to create VIM (no NG-RO)")
tierno2357f4e2020-10-19 16:38:59 +0000663 task = asyncio.ensure_future(self.vim.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100664 self.lcm_tasks.register(
665 "vim_account", vim_id, order_id, "vim_create", task
666 )
garciadeblas96b94f52024-07-08 16:18:21 +0200667 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100668 elif command == "delete" or command == "deleted":
garciadeblas96b94f52024-07-08 16:18:21 +0200669 self.lcm_tasks.cancel(topic, vim_id)
670 if "credentials" in vim_config:
671 self.logger.info("Vim remove cloud credentials")
yshah771dea82024-07-05 15:11:49 +0000672 task = asyncio.ensure_future(
garciadeblas96b94f52024-07-08 16:18:21 +0200673 self.cloud_credentials.remove(op_id, op_params, db_vim)
yshah771dea82024-07-05 15:11:49 +0000674 )
675 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200676 "vim_account", vim_id, op_id, "cloud_credentials_remove", task
yshah771dea82024-07-05 15:11:49 +0000677 )
garciadeblas96b94f52024-07-08 16:18:21 +0200678 task = asyncio.ensure_future(self.vim.delete(params, order_id))
679 self.lcm_tasks.register(
680 "vim_account", vim_id, order_id, "vim_delete", task
681 )
682 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100683 elif command == "show":
684 print("not implemented show with vim_account")
685 sys.stdout.flush()
686 return
tiernof210c1c2019-10-16 09:09:58 +0000687 elif command in ("edit", "edited"):
garciadeblas96b94f52024-07-08 16:18:21 +0200688 if "credentials" in vim_config:
689 self.logger.info("Vim update cloud credentials")
yshah771dea82024-07-05 15:11:49 +0000690 task = asyncio.ensure_future(
garciadeblas96b94f52024-07-08 16:18:21 +0200691 self.cloud_credentials.edit(op_id, op_params, db_vim)
yshah771dea82024-07-05 15:11:49 +0000692 )
693 self.lcm_tasks.register(
694 "vim_account", vim_id, op_id, "cloud_credentials_update", task
695 )
Luis Vegaa27dc532022-11-11 20:10:49 +0000696 if not self.main_config.RO.ng:
tierno2357f4e2020-10-19 16:38:59 +0000697 task = asyncio.ensure_future(self.vim.edit(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100698 self.lcm_tasks.register(
699 "vim_account", vim_id, order_id, "vim_edit", task
700 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100701 return
tiernof210c1c2019-10-16 09:09:58 +0000702 elif command == "deleted":
703 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100704 elif topic == "wim_account":
705 wim_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000706 if command in ("create", "created"):
Luis Vegaa27dc532022-11-11 20:10:49 +0000707 if not self.main_config.RO.ng:
tierno2357f4e2020-10-19 16:38:59 +0000708 task = asyncio.ensure_future(self.wim.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100709 self.lcm_tasks.register(
710 "wim_account", wim_id, order_id, "wim_create", task
711 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100712 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100713 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100714 self.lcm_tasks.cancel(topic, wim_id)
kuuse6a470c62019-07-10 13:52:45 +0200715 task = asyncio.ensure_future(self.wim.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100716 self.lcm_tasks.register(
717 "wim_account", wim_id, order_id, "wim_delete", task
718 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100719 return
720 elif command == "show":
721 print("not implemented show with wim_account")
722 sys.stdout.flush()
723 return
tiernof210c1c2019-10-16 09:09:58 +0000724 elif command in ("edit", "edited"):
gcalvinoed7f6d42018-12-14 14:44:56 +0100725 task = asyncio.ensure_future(self.wim.edit(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100726 self.lcm_tasks.register(
727 "wim_account", wim_id, order_id, "wim_edit", task
728 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100729 return
tiernof210c1c2019-10-16 09:09:58 +0000730 elif command == "deleted":
731 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100732 elif topic == "sdn":
733 _sdn_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000734 if command in ("create", "created"):
Luis Vegaa27dc532022-11-11 20:10:49 +0000735 if not self.main_config.RO.ng:
tierno2357f4e2020-10-19 16:38:59 +0000736 task = asyncio.ensure_future(self.sdn.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100737 self.lcm_tasks.register(
738 "sdn", _sdn_id, order_id, "sdn_create", task
739 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100740 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100741 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100742 self.lcm_tasks.cancel(topic, _sdn_id)
kuuse6a470c62019-07-10 13:52:45 +0200743 task = asyncio.ensure_future(self.sdn.delete(params, order_id))
gcalvinoed7f6d42018-12-14 14:44:56 +0100744 self.lcm_tasks.register("sdn", _sdn_id, order_id, "sdn_delete", task)
745 return
tiernof210c1c2019-10-16 09:09:58 +0000746 elif command in ("edit", "edited"):
gcalvinoed7f6d42018-12-14 14:44:56 +0100747 task = asyncio.ensure_future(self.sdn.edit(params, order_id))
748 self.lcm_tasks.register("sdn", _sdn_id, order_id, "sdn_edit", task)
749 return
tiernof210c1c2019-10-16 09:09:58 +0000750 elif command == "deleted":
751 return # TODO cleaning of task just in case should be done
rshri932105f2024-07-05 15:11:55 +0000752 elif topic == "cluster":
garciadeblas96b94f52024-07-08 16:18:21 +0200753 if command != "get_creds":
754 op_id = params["operation_id"]
755 cluster_id = params["cluster_id"]
756 db_cluster = self.db.get_one("clusters", {"_id": cluster_id})
757 op_params = self.get_operation_params(db_cluster, op_id)
758 db_content = {
759 "cluster": db_cluster,
760 }
rshri932105f2024-07-05 15:11:55 +0000761 if command == "create" or command == "created":
rshri932105f2024-07-05 15:11:55 +0000762 self.logger.debug("cluster_id = {}".format(cluster_id))
garciadeblas96b94f52024-07-08 16:18:21 +0200763 # db_vim = self.db.get_one("vim_accounts", {"_id": db_cluster["vim_account"]})
764 db_vim = self.db.get_one(
765 "vim_accounts", {"name": db_cluster["vim_account"]}
766 )
767 db_content["vim_account"] = db_vim
768 task = asyncio.ensure_future(
769 self.cluster.create(op_id, op_params, db_content)
770 )
rshri932105f2024-07-05 15:11:55 +0000771 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200772 "cluster", cluster_id, op_id, "cluster_create", task
rshri932105f2024-07-05 15:11:55 +0000773 )
774 return
775 elif command == "delete" or command == "deleted":
garciadeblas96b94f52024-07-08 16:18:21 +0200776 task = asyncio.ensure_future(
777 self.cluster.delete(op_id, op_params, db_content)
778 )
rshri932105f2024-07-05 15:11:55 +0000779 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200780 "cluster", cluster_id, op_id, "cluster_delete", task
rshri932105f2024-07-05 15:11:55 +0000781 )
782 return
783 elif command == "add" or command == "added":
garciadeblas96b94f52024-07-08 16:18:21 +0200784 profile_type = params["profile_type"]
785 profile_collection = self.profile_collection_mapping[profile_type]
786 db_profile = self.db.get_one(
787 profile_collection, {"_id": params["profile_id"]}
788 )
789 db_profile["profile_type"] = profile_type
790 db_content["profile"] = db_profile
791 task = asyncio.ensure_future(
792 self.cluster.attach_profile(op_id, op_params, db_content)
793 )
rshri932105f2024-07-05 15:11:55 +0000794 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200795 "cluster", cluster_id, op_id, "profile_add", task
rshri932105f2024-07-05 15:11:55 +0000796 )
797 return
798 elif command == "remove" or command == "removed":
garciadeblas96b94f52024-07-08 16:18:21 +0200799 profile_type = params["profile_type"]
800 profile_collection = self.profile_collection_mapping[profile_type]
801 db_profile = self.db.get_one(
802 profile_collection, {"_id": params["profile_id"]}
803 )
804 db_profile["profile_type"] = profile_type
805 db_content["profile"] = db_profile
806 task = asyncio.ensure_future(
807 self.cluster.detach_profile(op_id, op_params, db_content)
808 )
rshri932105f2024-07-05 15:11:55 +0000809 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200810 "cluster", cluster_id, op_id, "profile_remove", task
rshri932105f2024-07-05 15:11:55 +0000811 )
812 return
813 elif command == "register" or command == "registered":
garciadeblas96b94f52024-07-08 16:18:21 +0200814 task = asyncio.ensure_future(
815 self.cluster.register(op_id, op_params, db_content)
816 )
rshri932105f2024-07-05 15:11:55 +0000817 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200818 "cluster", cluster_id, op_id, "cluster_register", task
rshri932105f2024-07-05 15:11:55 +0000819 )
820 return
821 elif command == "deregister" or command == "deregistered":
garciadeblas96b94f52024-07-08 16:18:21 +0200822 task = asyncio.ensure_future(
823 self.cluster.deregister(op_id, op_params, db_content)
824 )
rshri932105f2024-07-05 15:11:55 +0000825 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200826 "cluster", cluster_id, op_id, "cluster_deregister", task
rshri932105f2024-07-05 15:11:55 +0000827 )
828 return
yshah771dea82024-07-05 15:11:49 +0000829 elif command == "get_creds":
garciadeblas96b94f52024-07-08 16:18:21 +0200830 cluster_id = params["_id"]
831 db_cluster = self.db.get_one("clusters", {"_id": cluster_id})
832 task = asyncio.ensure_future(self.cluster.get_creds(db_cluster))
yshah771dea82024-07-05 15:11:49 +0000833 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200834 "cluster", cluster_id, cluster_id, "cluster_get_credentials", task
yshah771dea82024-07-05 15:11:49 +0000835 )
836 return
837 elif command == "upgrade" or command == "scale":
garciadeblas96b94f52024-07-08 16:18:21 +0200838 # db_vim = self.db.get_one("vim_accounts", {"_id": db_cluster["vim_account"]})
839 db_vim = self.db.get_one(
840 "vim_accounts", {"name": db_cluster["vim_account"]}
841 )
842 db_content["vim_account"] = db_vim
843 task = asyncio.ensure_future(
844 self.cluster.update(op_id, op_params, db_content)
845 )
846 self.lcm_tasks.register(
847 "cluster", cluster_id, op_id, "cluster_update", task
848 )
yshah771dea82024-07-05 15:11:49 +0000849 return
rshri932105f2024-07-05 15:11:55 +0000850 elif topic == "k8s_app":
garciadeblas96b94f52024-07-08 16:18:21 +0200851 op_id = params["operation_id"]
852 profile_id = params["profile_id"]
853 db_profile = self.db.get_one("k8sapp", {"_id": profile_id})
854 db_profile["profile_type"] = "applications"
855 op_params = self.get_operation_params(db_profile, op_id)
rshri932105f2024-07-05 15:11:55 +0000856 if command == "profile_create" or command == "profile_created":
garciadeblas96b94f52024-07-08 16:18:21 +0200857 self.logger.debug("Create k8s_app_id = {}".format(profile_id))
858 task = asyncio.ensure_future(
859 self.k8s_app.create(op_id, op_params, db_profile)
860 )
rshri932105f2024-07-05 15:11:55 +0000861 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200862 "k8s_app", profile_id, op_id, "k8s_app_create", task
rshri932105f2024-07-05 15:11:55 +0000863 )
864 return
865 elif command == "delete" or command == "deleted":
garciadeblas96b94f52024-07-08 16:18:21 +0200866 self.logger.debug("Delete k8s_app_id = {}".format(profile_id))
867 task = asyncio.ensure_future(
868 self.k8s_app.delete(op_id, op_params, db_profile)
869 )
rshri932105f2024-07-05 15:11:55 +0000870 self.lcm_tasks.register(
garciadeblas96b94f52024-07-08 16:18:21 +0200871 "k8s_app", profile_id, op_id, "k8s_app_delete", task
rshri932105f2024-07-05 15:11:55 +0000872 )
873 return
874 elif topic == "k8s_resource":
garciadeblas96b94f52024-07-08 16:18:21 +0200875 op_id = params["operation_id"]
876 profile_id = params["profile_id"]
877 db_profile = self.db.get_one("k8sresource", {"_id": profile_id})
878 db_profile["profile_type"] = "managed-resources"
879 op_params = self.get_operation_params(db_profile, op_id)
rshri932105f2024-07-05 15:11:55 +0000880 if command == "profile_create" or command == "profile_created":
garciadeblas96b94f52024-07-08 16:18:21 +0200881 self.logger.debug("Create k8s_resource_id = {}".format(profile_id))
882 task = asyncio.ensure_future(
883 self.k8s_resource.create(op_id, op_params, db_profile)
884 )
rshri932105f2024-07-05 15:11:55 +0000885 self.lcm_tasks.register(
886 "k8s_resource",
garciadeblas96b94f52024-07-08 16:18:21 +0200887 profile_id,
888 op_id,
rshri932105f2024-07-05 15:11:55 +0000889 "k8s_resource_create",
890 task,
891 )
892 return
893 elif command == "delete" or command == "deleted":
garciadeblas96b94f52024-07-08 16:18:21 +0200894 self.logger.debug("Delete k8s_resource_id = {}".format(profile_id))
895 task = asyncio.ensure_future(
896 self.k8s_resource.delete(op_id, op_params, db_profile)
897 )
rshri932105f2024-07-05 15:11:55 +0000898 self.lcm_tasks.register(
899 "k8s_resource",
garciadeblas96b94f52024-07-08 16:18:21 +0200900 profile_id,
901 op_id,
rshri932105f2024-07-05 15:11:55 +0000902 "k8s_resource_delete",
903 task,
904 )
905 return
906
907 elif topic == "k8s_infra_controller":
garciadeblas96b94f52024-07-08 16:18:21 +0200908 op_id = params["operation_id"]
909 profile_id = params["profile_id"]
910 db_profile = self.db.get_one("k8sinfra_controller", {"_id": profile_id})
911 db_profile["profile_type"] = "infra-controllers"
912 op_params = self.get_operation_params(db_profile, op_id)
rshri932105f2024-07-05 15:11:55 +0000913 if command == "profile_create" or command == "profile_created":
rshri932105f2024-07-05 15:11:55 +0000914 self.logger.debug(
garciadeblas96b94f52024-07-08 16:18:21 +0200915 "Create k8s_infra_controller_id = {}".format(profile_id)
rshri932105f2024-07-05 15:11:55 +0000916 )
917 task = asyncio.ensure_future(
garciadeblas96b94f52024-07-08 16:18:21 +0200918 self.k8s_infra_controller.create(op_id, op_params, db_profile)
rshri932105f2024-07-05 15:11:55 +0000919 )
920 self.lcm_tasks.register(
921 "k8s_infra_controller",
garciadeblas96b94f52024-07-08 16:18:21 +0200922 profile_id,
923 op_id,
rshri932105f2024-07-05 15:11:55 +0000924 "k8s_infra_controller_create",
925 task,
926 )
927 return
928 elif command == "delete" or command == "deleted":
garciadeblas96b94f52024-07-08 16:18:21 +0200929 self.logger.debug(
930 "Delete k8s_infra_controller_id = {}".format(profile_id)
931 )
rshri932105f2024-07-05 15:11:55 +0000932 task = asyncio.ensure_future(
garciadeblas96b94f52024-07-08 16:18:21 +0200933 self.k8s_infra_controller.delete(op_id, op_params, db_profile)
rshri932105f2024-07-05 15:11:55 +0000934 )
935 self.lcm_tasks.register(
936 "k8s_infra_controller",
garciadeblas96b94f52024-07-08 16:18:21 +0200937 profile_id,
938 op_id,
rshri932105f2024-07-05 15:11:55 +0000939 "k8s_infra_controller_delete",
940 task,
941 )
942 return
943
944 elif topic == "k8s_infra_config":
garciadeblas96b94f52024-07-08 16:18:21 +0200945 op_id = params["operation_id"]
946 profile_id = params["profile_id"]
947 db_profile = self.db.get_one("k8sinfra_config", {"_id": profile_id})
948 db_profile["profile_type"] = "infra-configs"
949 op_params = self.get_operation_params(db_profile, op_id)
rshri932105f2024-07-05 15:11:55 +0000950 if command == "profile_create" or command == "profile_created":
garciadeblas96b94f52024-07-08 16:18:21 +0200951 self.logger.debug("Create k8s_infra_config_id = {}".format(profile_id))
rshri932105f2024-07-05 15:11:55 +0000952 task = asyncio.ensure_future(
garciadeblas96b94f52024-07-08 16:18:21 +0200953 self.k8s_infra_config.create(op_id, op_params, db_profile)
rshri932105f2024-07-05 15:11:55 +0000954 )
955 self.lcm_tasks.register(
956 "k8s_infra_config",
garciadeblas96b94f52024-07-08 16:18:21 +0200957 profile_id,
958 op_id,
rshri932105f2024-07-05 15:11:55 +0000959 "k8s_infra_config_create",
960 task,
961 )
962 return
963 elif command == "delete" or command == "deleted":
garciadeblas96b94f52024-07-08 16:18:21 +0200964 self.logger.debug("Delete k8s_infra_config_id = {}".format(profile_id))
rshri932105f2024-07-05 15:11:55 +0000965 task = asyncio.ensure_future(
garciadeblas96b94f52024-07-08 16:18:21 +0200966 self.k8s_infra_config.delete(op_id, op_params, db_profile)
rshri932105f2024-07-05 15:11:55 +0000967 )
968 self.lcm_tasks.register(
969 "k8s_infra_config",
garciadeblas96b94f52024-07-08 16:18:21 +0200970 profile_id,
971 op_id,
rshri932105f2024-07-05 15:11:55 +0000972 "k8s_infra_config_delete",
973 task,
974 )
975 return
yshah771dea82024-07-05 15:11:49 +0000976 elif topic == "oka":
garciadeblas96b94f52024-07-08 16:18:21 +0200977 op_id = params["operation_id"]
978 oka_id = params["oka_id"]
979 db_oka = self.db.get_one("okas", {"_id": oka_id})
980 op_params = self.get_operation_params(db_oka, op_id)
yshah771dea82024-07-05 15:11:49 +0000981 if command == "create":
garciadeblas96b94f52024-07-08 16:18:21 +0200982 task = asyncio.ensure_future(self.oka.create(op_id, op_params, db_oka))
983 self.lcm_tasks.register("oka", oka_id, op_id, "oka_create", task)
yshah771dea82024-07-05 15:11:49 +0000984 return
985 elif command == "edit":
garciadeblas96b94f52024-07-08 16:18:21 +0200986 task = asyncio.ensure_future(self.oka.edit(op_id, op_params, db_oka))
987 self.lcm_tasks.register("oka", oka_id, op_id, "oka_edit", task)
yshah771dea82024-07-05 15:11:49 +0000988 return
989 elif command == "delete":
garciadeblas96b94f52024-07-08 16:18:21 +0200990 task = asyncio.ensure_future(self.oka.delete(op_id, op_params, db_oka))
991 self.lcm_tasks.register("oka", oka_id, op_id, "oka_delete", task)
yshah771dea82024-07-05 15:11:49 +0000992 return
993 elif topic == "ksu":
garciadeblas96b94f52024-07-08 16:18:21 +0200994 op_id = params["operation_id"]
995 op_params = None
996 db_content = None
997 if not (command == "clone" or command == "move"):
998 # op_params is a list
999 # db_content is a list of KSU
1000 db_content = []
1001 op_params = []
1002 for ksu_id in params["ksus_list"]:
1003 db_ksu = self.db.get_one("ksus", {"_id": ksu_id})
1004 db_content.append(db_ksu)
1005 ksu_params = {}
1006 if command == "delete":
1007 ksu_params["profile"] = {}
1008 ksu_params["profile"]["profile_type"] = db_ksu["profile"][
1009 "profile_type"
1010 ]
1011 ksu_params["profile"]["_id"] = db_ksu["profile"]["_id"]
1012 else:
1013 ksu_params = self.get_operation_params(db_ksu, op_id)
1014 # Update ksu_params["profile"] with profile name and age-pubkey
1015 profile_type = ksu_params["profile"]["profile_type"]
1016 profile_id = ksu_params["profile"]["_id"]
1017 profile_collection = self.profile_collection_mapping[profile_type]
1018 db_profile = self.db.get_one(
1019 profile_collection, {"_id": profile_id}
1020 )
1021 ksu_params["profile"]["name"] = db_profile["name"]
1022 ksu_params["profile"]["age_pubkey"] = db_profile.get(
1023 "age_pubkey", ""
1024 )
1025 if command == "create" or command == "edit" or command == "edited":
1026 # Update ksu_params["oka"] with sw_catalog_path (when missing)
1027 for oka in ksu_params["oka"]:
1028 if "sw_catalog_path" not in oka:
1029 oka_id = oka["_id"]
1030 db_oka = self.db.get_one("okas", {"_id": oka_id})
1031 oka[
1032 "sw_catalog_path"
1033 ] = f"infra-controllers/{db_oka['git_name']}"
1034 op_params.append(ksu_params)
1035 else:
1036 # db_content and op_params are single items
1037 db_content = self.db.get_one("ksus", {"_id": params["_id"]})
1038 db_content = db_ksu
1039 op_params = self.get_operation_params(db_ksu, op_id)
yshah771dea82024-07-05 15:11:49 +00001040 if command == "create":
garciadeblas96b94f52024-07-08 16:18:21 +02001041 task = asyncio.ensure_future(
1042 self.ksu.create(op_id, op_params, db_content)
1043 )
1044 self.lcm_tasks.register("ksu", ksu_id, op_id, "ksu_create", task)
yshah771dea82024-07-05 15:11:49 +00001045 return
1046 elif command == "edit" or command == "edited":
garciadeblas96b94f52024-07-08 16:18:21 +02001047 task = asyncio.ensure_future(
1048 self.ksu.edit(op_id, op_params, db_content)
1049 )
1050 self.lcm_tasks.register("ksu", ksu_id, op_id, "ksu_edit", task)
yshah771dea82024-07-05 15:11:49 +00001051 return
1052 elif command == "delete":
garciadeblas96b94f52024-07-08 16:18:21 +02001053 task = asyncio.ensure_future(
1054 self.ksu.delete(op_id, op_params, db_content)
1055 )
1056 self.lcm_tasks.register("ksu", ksu_id, op_id, "ksu_delete", task)
yshah771dea82024-07-05 15:11:49 +00001057 return
1058 elif command == "clone":
garciadeblas96b94f52024-07-08 16:18:21 +02001059 task = asyncio.ensure_future(
1060 self.ksu.clone(op_id, op_params, db_content)
1061 )
1062 self.lcm_tasks.register("ksu", ksu_id, op_id, "ksu_clone", task)
yshah771dea82024-07-05 15:11:49 +00001063 return
1064 elif command == "move":
garciadeblas96b94f52024-07-08 16:18:21 +02001065 task = asyncio.ensure_future(
1066 self.ksu.move(op_id, op_params, db_content)
1067 )
1068 self.lcm_tasks.register("ksu", ksu_id, op_id, "ksu_move", task)
yshah771dea82024-07-05 15:11:49 +00001069 return
rshri932105f2024-07-05 15:11:55 +00001070
gcalvinoed7f6d42018-12-14 14:44:56 +01001071 self.logger.critical("unknown topic {} and command '{}'".format(topic, command))
1072
tiernoc0e42e22018-05-11 11:36:10 +02001073 async def kafka_read(self):
garciadeblas5697b8b2021-03-24 09:17:02 +01001074 self.logger.debug(
1075 "Task kafka_read Enter with worker_id={}".format(self.worker_id)
1076 )
gcalvinoed7f6d42018-12-14 14:44:56 +01001077 self.consecutive_errors = 0
1078 self.first_start = True
rshri932105f2024-07-05 15:11:55 +00001079 self.logger.info(
1080 "Consecutive errors: {} first start: {}".format(
1081 self.consecutive_errors, self.first_start
1082 )
1083 )
gcalvinoed7f6d42018-12-14 14:44:56 +01001084 while self.consecutive_errors < 10:
tiernoc0e42e22018-05-11 11:36:10 +02001085 try:
garciadeblas5697b8b2021-03-24 09:17:02 +01001086 topics = (
1087 "ns",
1088 "vim_account",
1089 "wim_account",
1090 "sdn",
1091 "nsi",
1092 "k8scluster",
1093 "vca",
1094 "k8srepo",
1095 "pla",
Gabriel Cubab6049d32023-10-30 13:44:49 -05001096 "nslcmops",
rshri932105f2024-07-05 15:11:55 +00001097 "cluster",
1098 "k8s_app",
1099 "k8s_resource",
1100 "k8s_infra_controller",
1101 "k8s_infra_config",
yshah771dea82024-07-05 15:11:49 +00001102 "oka",
1103 "ksu",
rshri932105f2024-07-05 15:11:55 +00001104 )
1105 self.logger.info(
1106 "Consecutive errors: {} first start: {}".format(
1107 self.consecutive_errors, self.first_start
1108 )
garciadeblas5697b8b2021-03-24 09:17:02 +01001109 )
1110 topics_admin = ("admin",)
tierno16427352019-04-22 11:37:36 +00001111 await asyncio.gather(
garciadeblas5697b8b2021-03-24 09:17:02 +01001112 self.msg.aioread(
Gabriel Cubab6049d32023-10-30 13:44:49 -05001113 topics,
1114 aiocallback=self.kafka_read_callback,
1115 from_beginning=True,
garciadeblas5697b8b2021-03-24 09:17:02 +01001116 ),
1117 self.msg_admin.aioread(
1118 topics_admin,
Gabriel Cubab6049d32023-10-30 13:44:49 -05001119 aiocallback=self.kafka_read_callback,
garciadeblas5697b8b2021-03-24 09:17:02 +01001120 group_id=False,
1121 ),
tierno16427352019-04-22 11:37:36 +00001122 )
tiernoc0e42e22018-05-11 11:36:10 +02001123
gcalvinoed7f6d42018-12-14 14:44:56 +01001124 except LcmExceptionExit:
1125 self.logger.debug("Bye!")
1126 break
tiernoc0e42e22018-05-11 11:36:10 +02001127 except Exception as e:
1128 # if not first_start is the first time after starting. So leave more time and wait
1129 # to allow kafka starts
gcalvinoed7f6d42018-12-14 14:44:56 +01001130 if self.consecutive_errors == 8 if not self.first_start else 30:
garciadeblas5697b8b2021-03-24 09:17:02 +01001131 self.logger.error(
1132 "Task kafka_read task exit error too many errors. Exception: {}".format(
1133 e
1134 )
1135 )
tiernoc0e42e22018-05-11 11:36:10 +02001136 raise
gcalvinoed7f6d42018-12-14 14:44:56 +01001137 self.consecutive_errors += 1
garciadeblas5697b8b2021-03-24 09:17:02 +01001138 self.logger.error(
1139 "Task kafka_read retrying after Exception {}".format(e)
1140 )
gcalvinoed7f6d42018-12-14 14:44:56 +01001141 wait_time = 2 if not self.first_start else 5
Gabriel Cubae7898982023-05-11 01:57:21 -05001142 await asyncio.sleep(wait_time)
tiernoc0e42e22018-05-11 11:36:10 +02001143
tiernoc0e42e22018-05-11 11:36:10 +02001144 self.logger.debug("Task kafka_read exit")
1145
Gabriel Cubae7898982023-05-11 01:57:21 -05001146 async def kafka_read_ping(self):
1147 await asyncio.gather(self.kafka_read(), self.kafka_ping())
1148
Mark Beierl1addc932023-05-18 15:11:34 -04001149 async def start(self):
rshri932105f2024-07-05 15:11:55 +00001150 self.logger.info("Start LCM")
tierno22f4f9c2018-06-11 18:53:39 +02001151 # check RO version
Mark Beierl1addc932023-05-18 15:11:34 -04001152 await self.check_RO_version()
tierno22f4f9c2018-06-11 18:53:39 +02001153
Gabriel Cubae7898982023-05-11 01:57:21 -05001154 self.ns = ns.NsLcm(self.msg, self.lcm_tasks, self.main_config)
Luis Vegaa27dc532022-11-11 20:10:49 +00001155 # TODO: modify the rest of classes to use the LcmCfg object instead of dicts
garciadeblas5697b8b2021-03-24 09:17:02 +01001156 self.netslice = netslice.NetsliceLcm(
Gabriel Cubae7898982023-05-11 01:57:21 -05001157 self.msg, self.lcm_tasks, self.main_config.to_dict(), self.ns
garciadeblas5697b8b2021-03-24 09:17:02 +01001158 )
Gabriel Cubae7898982023-05-11 01:57:21 -05001159 self.vim = vim_sdn.VimLcm(self.msg, self.lcm_tasks, self.main_config.to_dict())
1160 self.wim = vim_sdn.WimLcm(self.msg, self.lcm_tasks, self.main_config.to_dict())
1161 self.sdn = vim_sdn.SdnLcm(self.msg, self.lcm_tasks, self.main_config.to_dict())
garciadeblas5697b8b2021-03-24 09:17:02 +01001162 self.k8scluster = vim_sdn.K8sClusterLcm(
Gabriel Cubae7898982023-05-11 01:57:21 -05001163 self.msg, self.lcm_tasks, self.main_config.to_dict()
garciadeblas5697b8b2021-03-24 09:17:02 +01001164 )
Gabriel Cubae7898982023-05-11 01:57:21 -05001165 self.vca = vim_sdn.VcaLcm(self.msg, self.lcm_tasks, self.main_config.to_dict())
garciadeblas5697b8b2021-03-24 09:17:02 +01001166 self.k8srepo = vim_sdn.K8sRepoLcm(
Gabriel Cubae7898982023-05-11 01:57:21 -05001167 self.msg, self.lcm_tasks, self.main_config.to_dict()
garciadeblas5697b8b2021-03-24 09:17:02 +01001168 )
rshri932105f2024-07-05 15:11:55 +00001169 self.cluster = k8s.ClusterLcm(
1170 self.msg, self.lcm_tasks, self.main_config.to_dict()
1171 )
1172 self.k8s_app = k8s.K8sAppLcm(
1173 self.msg, self.lcm_tasks, self.main_config.to_dict()
1174 )
1175 self.k8s_resource = k8s.K8sResourceLcm(
1176 self.msg, self.lcm_tasks, self.main_config.to_dict()
1177 )
1178 self.k8s_infra_controller = k8s.K8sInfraControllerLcm(
1179 self.msg, self.lcm_tasks, self.main_config.to_dict()
1180 )
1181 self.k8s_infra_config = k8s.K8sInfraConfigLcm(
1182 self.msg, self.lcm_tasks, self.main_config.to_dict()
1183 )
yshah771dea82024-07-05 15:11:49 +00001184 self.cloud_credentials = k8s.CloudCredentialsLcm(
1185 self.msg, self.lcm_tasks, self.main_config.to_dict()
1186 )
1187 self.oka = k8s.OkaLcm(self.msg, self.lcm_tasks, self.main_config.to_dict())
1188 self.ksu = k8s.KsuLcm(self.msg, self.lcm_tasks, self.main_config.to_dict())
rshri932105f2024-07-05 15:11:55 +00001189
1190 self.logger.info(
1191 "Msg: {} lcm tasks: {} main config: {}".format(
1192 self.msg, self.lcm_tasks, self.main_config
1193 )
1194 )
tierno2357f4e2020-10-19 16:38:59 +00001195
Mark Beierl1addc932023-05-18 15:11:34 -04001196 await self.kafka_read_ping()
bravof73bac502021-05-11 07:38:47 -04001197
tiernoc0e42e22018-05-11 11:36:10 +02001198 # TODO
1199 # self.logger.debug("Terminating cancelling creation tasks")
tiernoca2e16a2018-06-29 15:25:24 +02001200 # self.lcm_tasks.cancel("ALL", "create")
tiernoc0e42e22018-05-11 11:36:10 +02001201 # timeout = 200
1202 # while self.is_pending_tasks():
1203 # self.logger.debug("Task kafka_read terminating. Waiting for tasks termination")
Gabriel Cubae7898982023-05-11 01:57:21 -05001204 # await asyncio.sleep(2)
tiernoc0e42e22018-05-11 11:36:10 +02001205 # timeout -= 2
1206 # if not timeout:
tiernoca2e16a2018-06-29 15:25:24 +02001207 # self.lcm_tasks.cancel("ALL", "ALL")
tiernoc0e42e22018-05-11 11:36:10 +02001208 if self.db:
1209 self.db.db_disconnect()
1210 if self.msg:
1211 self.msg.disconnect()
tierno16427352019-04-22 11:37:36 +00001212 if self.msg_admin:
1213 self.msg_admin.disconnect()
tiernoc0e42e22018-05-11 11:36:10 +02001214 if self.fs:
1215 self.fs.fs_disconnect()
1216
tiernoc0e42e22018-05-11 11:36:10 +02001217 def read_config_file(self, config_file):
tiernoc0e42e22018-05-11 11:36:10 +02001218 try:
Gabriel Cubaa89a5a72022-11-26 18:55:15 -05001219 with open(config_file) as f:
1220 return yaml.safe_load(f)
tiernoc0e42e22018-05-11 11:36:10 +02001221 except Exception as e:
1222 self.logger.critical("At config file '{}': {}".format(config_file, e))
Gabriel Cubaa89a5a72022-11-26 18:55:15 -05001223 exit(1)
tiernoc0e42e22018-05-11 11:36:10 +02001224
tierno16427352019-04-22 11:37:36 +00001225 @staticmethod
1226 def get_process_id():
1227 """
1228 Obtain a unique ID for this process. If running from inside docker, it will get docker ID. If not it
1229 will provide a random one
1230 :return: Obtained ID
1231 """
Gabriel Cuba4c0e6802023-10-09 13:22:38 -05001232
1233 def get_docker_id():
1234 try:
1235 with open("/proc/self/cgroup", "r") as f:
1236 text_id_ = f.readline()
1237 _, _, text_id = text_id_.rpartition("/")
1238 return text_id.replace("\n", "")[:12]
1239 except Exception:
1240 return None
1241
1242 def generate_random_id():
1243 return "".join(SystemRandom().choice("0123456789abcdef") for _ in range(12))
1244
1245 # Try getting docker id. If it fails, generate a random id
1246 docker_id = get_docker_id()
1247 return docker_id if docker_id else generate_random_id()
tierno16427352019-04-22 11:37:36 +00001248
tiernoc0e42e22018-05-11 11:36:10 +02001249
tierno275411e2018-05-16 14:33:32 +02001250def usage():
garciadeblas5697b8b2021-03-24 09:17:02 +01001251 print(
1252 """Usage: {} [options]
quilesj7e13aeb2019-10-08 13:34:55 +02001253 -c|--config [configuration_file]: loads the configuration file (default: ./lcm.cfg)
tiernoa9843d82018-10-24 10:44:20 +02001254 --health-check: do not run lcm, but inspect kafka bus to determine if lcm is healthy
tierno275411e2018-05-16 14:33:32 +02001255 -h|--help: shows this help
garciadeblas5697b8b2021-03-24 09:17:02 +01001256 """.format(
1257 sys.argv[0]
1258 )
1259 )
tierno750b2452018-05-17 16:39:29 +02001260 # --log-socket-host HOST: send logs to this host")
1261 # --log-socket-port PORT: send logs using this port (default: 9022)")
tierno275411e2018-05-16 14:33:32 +02001262
1263
garciadeblas5697b8b2021-03-24 09:17:02 +01001264if __name__ == "__main__":
tierno275411e2018-05-16 14:33:32 +02001265 try:
tierno8c16b052020-02-05 15:08:32 +00001266 # print("SYS.PATH='{}'".format(sys.path))
tierno275411e2018-05-16 14:33:32 +02001267 # load parameters and configuration
quilesj7e13aeb2019-10-08 13:34:55 +02001268 # -h
1269 # -c value
1270 # --config value
1271 # --help
1272 # --health-check
garciadeblas5697b8b2021-03-24 09:17:02 +01001273 opts, args = getopt.getopt(
1274 sys.argv[1:], "hc:", ["config=", "help", "health-check"]
1275 )
tierno275411e2018-05-16 14:33:32 +02001276 # TODO add "log-socket-host=", "log-socket-port=", "log-file="
1277 config_file = None
1278 for o, a in opts:
1279 if o in ("-h", "--help"):
1280 usage()
1281 sys.exit()
1282 elif o in ("-c", "--config"):
1283 config_file = a
tiernoa9843d82018-10-24 10:44:20 +02001284 elif o == "--health-check":
tierno94f06112020-02-11 12:38:19 +00001285 from osm_lcm.lcm_hc import health_check
garciadeblas5697b8b2021-03-24 09:17:02 +01001286
aticig56b86c22022-06-29 10:43:05 +03001287 health_check(config_file, Lcm.ping_interval_pace)
tierno275411e2018-05-16 14:33:32 +02001288 else:
Gabriel Cuba4c0e6802023-10-09 13:22:38 -05001289 print(f"Unhandled option: {o}")
1290 exit(1)
quilesj7e13aeb2019-10-08 13:34:55 +02001291
tierno275411e2018-05-16 14:33:32 +02001292 if config_file:
1293 if not path.isfile(config_file):
garciadeblas5697b8b2021-03-24 09:17:02 +01001294 print(
1295 "configuration file '{}' does not exist".format(config_file),
1296 file=sys.stderr,
1297 )
tierno275411e2018-05-16 14:33:32 +02001298 exit(1)
1299 else:
garciadeblas5697b8b2021-03-24 09:17:02 +01001300 for config_file in (
1301 __file__[: __file__.rfind(".")] + ".cfg",
1302 "./lcm.cfg",
1303 "/etc/osm/lcm.cfg",
1304 ):
tierno275411e2018-05-16 14:33:32 +02001305 if path.isfile(config_file):
1306 break
1307 else:
garciadeblas5697b8b2021-03-24 09:17:02 +01001308 print(
1309 "No configuration file 'lcm.cfg' found neither at local folder nor at /etc/osm/",
1310 file=sys.stderr,
1311 )
tierno275411e2018-05-16 14:33:32 +02001312 exit(1)
k4.rahul44306072023-05-05 14:24:31 +05301313 config_file = os.path.realpath(os.path.normpath(os.path.abspath(config_file)))
tierno275411e2018-05-16 14:33:32 +02001314 lcm = Lcm(config_file)
Mark Beierl1addc932023-05-18 15:11:34 -04001315 asyncio.run(lcm.start())
tierno22f4f9c2018-06-11 18:53:39 +02001316 except (LcmException, getopt.GetoptError) as e:
tierno275411e2018-05-16 14:33:32 +02001317 print(str(e), file=sys.stderr)
1318 # usage()
1319 exit(1)