blob: 5bbeade55f7c65ea7c99522224da6736446f437f [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
22import os
23import pdb
24
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
tierno59d22d22018-09-25 18:10:19 +020031
Patricia Reinoso15ce47b2022-10-26 08:58:39 +000032from osm_lcm import ns, paas, vim_sdn, netslice
tierno69f0d382020-05-07 13:08:09 +000033from osm_lcm.ng_ro import NgRoException, NgRoClient
34from osm_lcm.ROclient import ROClient, ROClientException
quilesj7e13aeb2019-10-08 13:34:55 +020035
tierno94f06112020-02-11 12:38:19 +000036from time import time
tierno8069ce52019-08-28 15:34:33 +000037from osm_lcm.lcm_utils import versiontuple, LcmException, TaskRegistry, LcmExceptionExit
tiernoa4dea5a2020-01-05 16:29:30 +000038from osm_lcm import version as lcm_version, version_date as lcm_version_date
tierno8069ce52019-08-28 15:34:33 +000039
bravof922c4172020-11-24 21:21:43 -030040from osm_common import msglocal, msgkafka
tierno98768132018-09-11 12:07:21 +020041from osm_common import version as common_version
tierno59d22d22018-09-25 18:10:19 +020042from osm_common.dbbase import DbException
tiernoc0e42e22018-05-11 11:36:10 +020043from osm_common.fsbase import FsException
44from osm_common.msgbase import MsgException
bravof922c4172020-11-24 21:21:43 -030045from osm_lcm.data_utils.database.database import Database
46from osm_lcm.data_utils.filesystem.filesystem import Filesystem
aticig56b86c22022-06-29 10:43:05 +030047from osm_lcm.lcm_hc import get_health_check_file
tierno275411e2018-05-16 14:33:32 +020048from os import environ, path
tierno16427352019-04-22 11:37:36 +000049from random import choice as random_choice
tierno59d22d22018-09-25 18:10:19 +020050from n2vc import version as n2vc_version
bravof922c4172020-11-24 21:21:43 -030051import traceback
tiernoc0e42e22018-05-11 11:36:10 +020052
garciadeblas5697b8b2021-03-24 09:17:02 +010053if os.getenv("OSMLCM_PDB_DEBUG", None) is not None:
quilesj7e13aeb2019-10-08 13:34:55 +020054 pdb.set_trace()
55
tiernoc0e42e22018-05-11 11:36:10 +020056
tierno275411e2018-05-16 14:33:32 +020057__author__ = "Alfonso Tierno"
tiernoe64f7fb2019-09-11 08:55:52 +000058min_RO_version = "6.0.2"
tierno6e9d2eb2018-09-12 17:47:18 +020059min_n2vc_version = "0.0.2"
quilesj7e13aeb2019-10-08 13:34:55 +020060
tierno16427352019-04-22 11:37:36 +000061min_common_version = "0.1.19"
tierno275411e2018-05-16 14:33:32 +020062
63
tiernoc0e42e22018-05-11 11:36:10 +020064class Lcm:
65
garciadeblas5697b8b2021-03-24 09:17:02 +010066 ping_interval_pace = (
67 120 # how many time ping is send once is confirmed all is running
68 )
69 ping_interval_boot = 5 # how many time ping is sent when booting
70 cfg_logger_name = {
71 "message": "lcm.msg",
72 "database": "lcm.db",
73 "storage": "lcm.fs",
74 "tsdb": "lcm.prometheus",
75 }
tierno991e95d2020-07-21 12:41:25 +000076 # ^ contains for each section at lcm.cfg the used logger name
tiernoa9843d82018-10-24 10:44:20 +020077
tierno59d22d22018-09-25 18:10:19 +020078 def __init__(self, config_file, loop=None):
tiernoc0e42e22018-05-11 11:36:10 +020079 """
80 Init, Connect to database, filesystem storage, and messaging
81 :param config: two level dictionary with configuration. Top level should contain 'database', 'storage',
82 :return: None
83 """
tiernoc0e42e22018-05-11 11:36:10 +020084 self.db = None
85 self.msg = None
tierno16427352019-04-22 11:37:36 +000086 self.msg_admin = None
tiernoc0e42e22018-05-11 11:36:10 +020087 self.fs = None
88 self.pings_not_received = 1
tiernoc2564fe2019-01-28 16:18:56 +000089 self.consecutive_errors = 0
90 self.first_start = False
tiernoc0e42e22018-05-11 11:36:10 +020091
tiernoc0e42e22018-05-11 11:36:10 +020092 # logging
garciadeblas5697b8b2021-03-24 09:17:02 +010093 self.logger = logging.getLogger("lcm")
tierno16427352019-04-22 11:37:36 +000094 # get id
95 self.worker_id = self.get_process_id()
tiernoc0e42e22018-05-11 11:36:10 +020096 # load configuration
97 config = self.read_config_file(config_file)
98 self.config = config
aticig56b86c22022-06-29 10:43:05 +030099 self.health_check_file = get_health_check_file(self.config)
tierno744303e2020-01-13 16:46:31 +0000100 self.config["ro_config"] = {
tierno69f0d382020-05-07 13:08:09 +0000101 "ng": config["RO"].get("ng", False),
102 "uri": config["RO"].get("uri"),
tierno750b2452018-05-17 16:39:29 +0200103 "tenant": config.get("tenant", "osm"),
tierno69f0d382020-05-07 13:08:09 +0000104 "logger_name": "lcm.roclient",
105 "loglevel": config["RO"].get("loglevel", "ERROR"),
tiernoc0e42e22018-05-11 11:36:10 +0200106 }
tierno69f0d382020-05-07 13:08:09 +0000107 if not self.config["ro_config"]["uri"]:
garciadeblas5697b8b2021-03-24 09:17:02 +0100108 self.config["ro_config"]["uri"] = "http://{}:{}/".format(
109 config["RO"]["host"], config["RO"]["port"]
110 )
111 elif (
112 "/ro" in self.config["ro_config"]["uri"][-4:]
113 or "/openmano" in self.config["ro_config"]["uri"][-10:]
114 ):
tierno2357f4e2020-10-19 16:38:59 +0000115 # uri ends with '/ro', '/ro/', '/openmano', '/openmano/'
116 index = self.config["ro_config"]["uri"][-1].rfind("/")
garciadeblas5697b8b2021-03-24 09:17:02 +0100117 self.config["ro_config"]["uri"] = self.config["ro_config"]["uri"][index + 1]
tiernoc0e42e22018-05-11 11:36:10 +0200118
tierno59d22d22018-09-25 18:10:19 +0200119 self.loop = loop or asyncio.get_event_loop()
garciadeblas5697b8b2021-03-24 09:17:02 +0100120 self.ns = (
121 self.netslice
122 ) = (
123 self.vim
Patricia Reinoso15ce47b2022-10-26 08:58:39 +0000124 ) = (
125 self.wim
126 ) = self.sdn = self.k8scluster = self.vca = self.k8srepo = self.paas = None
tiernoc0e42e22018-05-11 11:36:10 +0200127
128 # logging
garciadeblas5697b8b2021-03-24 09:17:02 +0100129 log_format_simple = (
130 "%(asctime)s %(levelname)s %(name)s %(filename)s:%(lineno)s %(message)s"
131 )
132 log_formatter_simple = logging.Formatter(
133 log_format_simple, datefmt="%Y-%m-%dT%H:%M:%S"
134 )
tiernoc0e42e22018-05-11 11:36:10 +0200135 config["database"]["logger_name"] = "lcm.db"
136 config["storage"]["logger_name"] = "lcm.fs"
137 config["message"]["logger_name"] = "lcm.msg"
tierno86aa62f2018-08-20 11:57:04 +0000138 if config["global"].get("logfile"):
garciadeblas5697b8b2021-03-24 09:17:02 +0100139 file_handler = logging.handlers.RotatingFileHandler(
140 config["global"]["logfile"], maxBytes=100e6, backupCount=9, delay=0
141 )
tiernoc0e42e22018-05-11 11:36:10 +0200142 file_handler.setFormatter(log_formatter_simple)
143 self.logger.addHandler(file_handler)
tierno86aa62f2018-08-20 11:57:04 +0000144 if not config["global"].get("nologging"):
tiernoc0e42e22018-05-11 11:36:10 +0200145 str_handler = logging.StreamHandler()
146 str_handler.setFormatter(log_formatter_simple)
147 self.logger.addHandler(str_handler)
148
149 if config["global"].get("loglevel"):
150 self.logger.setLevel(config["global"]["loglevel"])
151
152 # logging other modules
tierno991e95d2020-07-21 12:41:25 +0000153 for k1, logname in self.cfg_logger_name.items():
tiernoc0e42e22018-05-11 11:36:10 +0200154 config[k1]["logger_name"] = logname
155 logger_module = logging.getLogger(logname)
tierno86aa62f2018-08-20 11:57:04 +0000156 if config[k1].get("logfile"):
garciadeblas5697b8b2021-03-24 09:17:02 +0100157 file_handler = logging.handlers.RotatingFileHandler(
158 config[k1]["logfile"], maxBytes=100e6, backupCount=9, delay=0
159 )
tiernoc0e42e22018-05-11 11:36:10 +0200160 file_handler.setFormatter(log_formatter_simple)
161 logger_module.addHandler(file_handler)
tierno86aa62f2018-08-20 11:57:04 +0000162 if config[k1].get("loglevel"):
tiernoc0e42e22018-05-11 11:36:10 +0200163 logger_module.setLevel(config[k1]["loglevel"])
garciadeblas5697b8b2021-03-24 09:17:02 +0100164 self.logger.critical(
165 "starting osm/lcm version {} {}".format(lcm_version, lcm_version_date)
166 )
tierno59d22d22018-09-25 18:10:19 +0200167
tiernoc0e42e22018-05-11 11:36:10 +0200168 # check version of N2VC
169 # TODO enhance with int conversion or from distutils.version import LooseVersion
170 # or with list(map(int, version.split(".")))
tierno59d22d22018-09-25 18:10:19 +0200171 if versiontuple(n2vc_version) < versiontuple(min_n2vc_version):
garciadeblas5697b8b2021-03-24 09:17:02 +0100172 raise LcmException(
173 "Not compatible osm/N2VC version '{}'. Needed '{}' or higher".format(
174 n2vc_version, min_n2vc_version
175 )
176 )
tierno59d22d22018-09-25 18:10:19 +0200177 # check version of common
tierno27246d82018-09-27 15:59:09 +0200178 if versiontuple(common_version) < versiontuple(min_common_version):
garciadeblas5697b8b2021-03-24 09:17:02 +0100179 raise LcmException(
180 "Not compatible osm/common version '{}'. Needed '{}' or higher".format(
181 common_version, min_common_version
182 )
183 )
tierno22f4f9c2018-06-11 18:53:39 +0200184
tiernoc0e42e22018-05-11 11:36:10 +0200185 try:
bravof922c4172020-11-24 21:21:43 -0300186 self.db = Database(config).instance.db
tiernoc0e42e22018-05-11 11:36:10 +0200187
bravof922c4172020-11-24 21:21:43 -0300188 self.fs = Filesystem(config).instance.fs
sousaedu40365e82021-07-26 15:24:21 +0200189 self.fs.sync()
tiernoc0e42e22018-05-11 11:36:10 +0200190
quilesj7e13aeb2019-10-08 13:34:55 +0200191 # copy message configuration in order to remove 'group_id' for msg_admin
tiernoc2564fe2019-01-28 16:18:56 +0000192 config_message = config["message"].copy()
193 config_message["loop"] = self.loop
194 if config_message["driver"] == "local":
tiernoc0e42e22018-05-11 11:36:10 +0200195 self.msg = msglocal.MsgLocal()
tiernoc2564fe2019-01-28 16:18:56 +0000196 self.msg.connect(config_message)
tierno16427352019-04-22 11:37:36 +0000197 self.msg_admin = msglocal.MsgLocal()
198 config_message.pop("group_id", None)
199 self.msg_admin.connect(config_message)
tiernoc2564fe2019-01-28 16:18:56 +0000200 elif config_message["driver"] == "kafka":
tiernoc0e42e22018-05-11 11:36:10 +0200201 self.msg = msgkafka.MsgKafka()
tiernoc2564fe2019-01-28 16:18:56 +0000202 self.msg.connect(config_message)
tierno16427352019-04-22 11:37:36 +0000203 self.msg_admin = msgkafka.MsgKafka()
204 config_message.pop("group_id", None)
205 self.msg_admin.connect(config_message)
tiernoc0e42e22018-05-11 11:36:10 +0200206 else:
garciadeblas5697b8b2021-03-24 09:17:02 +0100207 raise LcmException(
208 "Invalid configuration param '{}' at '[message]':'driver'".format(
209 config["message"]["driver"]
210 )
211 )
tiernoc0e42e22018-05-11 11:36:10 +0200212 except (DbException, FsException, MsgException) as e:
213 self.logger.critical(str(e), exc_info=True)
214 raise LcmException(str(e))
215
kuused124bfe2019-06-18 12:09:24 +0200216 # contains created tasks/futures to be able to cancel
bravof922c4172020-11-24 21:21:43 -0300217 self.lcm_tasks = TaskRegistry(self.worker_id, self.logger)
kuused124bfe2019-06-18 12:09:24 +0200218
tierno22f4f9c2018-06-11 18:53:39 +0200219 async def check_RO_version(self):
tiernoe64f7fb2019-09-11 08:55:52 +0000220 tries = 14
221 last_error = None
222 while True:
tierno2357f4e2020-10-19 16:38:59 +0000223 ro_uri = self.config["ro_config"]["uri"]
tiernoe64f7fb2019-09-11 08:55:52 +0000224 try:
tierno2357f4e2020-10-19 16:38:59 +0000225 # try new RO, if fail old RO
226 try:
227 self.config["ro_config"]["uri"] = ro_uri + "ro"
tierno69f0d382020-05-07 13:08:09 +0000228 ro_server = NgRoClient(self.loop, **self.config["ro_config"])
tierno2357f4e2020-10-19 16:38:59 +0000229 ro_version = await ro_server.get_version()
230 self.config["ro_config"]["ng"] = True
231 except Exception:
232 self.config["ro_config"]["uri"] = ro_uri + "openmano"
tierno69f0d382020-05-07 13:08:09 +0000233 ro_server = ROClient(self.loop, **self.config["ro_config"])
tierno2357f4e2020-10-19 16:38:59 +0000234 ro_version = await ro_server.get_version()
235 self.config["ro_config"]["ng"] = False
tiernoe64f7fb2019-09-11 08:55:52 +0000236 if versiontuple(ro_version) < versiontuple(min_RO_version):
garciadeblas5697b8b2021-03-24 09:17:02 +0100237 raise LcmException(
238 "Not compatible osm/RO version '{}'. Needed '{}' or higher".format(
239 ro_version, min_RO_version
240 )
241 )
242 self.logger.info(
243 "Connected to RO version {} new-generation version {}".format(
244 ro_version, self.config["ro_config"]["ng"]
245 )
246 )
tiernoe64f7fb2019-09-11 08:55:52 +0000247 return
tierno69f0d382020-05-07 13:08:09 +0000248 except (ROClientException, NgRoException) as e:
tierno2357f4e2020-10-19 16:38:59 +0000249 self.config["ro_config"]["uri"] = ro_uri
tiernoe64f7fb2019-09-11 08:55:52 +0000250 tries -= 1
bravof922c4172020-11-24 21:21:43 -0300251 traceback.print_tb(e.__traceback__)
garciadeblas5697b8b2021-03-24 09:17:02 +0100252 error_text = "Error while connecting to RO on {}: {}".format(
253 self.config["ro_config"]["uri"], e
254 )
tiernoe64f7fb2019-09-11 08:55:52 +0000255 if tries <= 0:
256 self.logger.critical(error_text)
257 raise LcmException(error_text)
258 if last_error != error_text:
259 last_error = error_text
garciadeblas5697b8b2021-03-24 09:17:02 +0100260 self.logger.error(
261 error_text + ". Waiting until {} seconds".format(5 * tries)
262 )
tiernoe64f7fb2019-09-11 08:55:52 +0000263 await asyncio.sleep(5)
tierno22f4f9c2018-06-11 18:53:39 +0200264
tiernoc0e42e22018-05-11 11:36:10 +0200265 async def test(self, param=None):
266 self.logger.debug("Starting/Ending test task: {}".format(param))
267
tiernoc0e42e22018-05-11 11:36:10 +0200268 async def kafka_ping(self):
269 self.logger.debug("Task kafka_ping Enter")
270 consecutive_errors = 0
271 first_start = True
272 kafka_has_received = False
273 self.pings_not_received = 1
274 while True:
275 try:
tierno16427352019-04-22 11:37:36 +0000276 await self.msg_admin.aiowrite(
garciadeblas5697b8b2021-03-24 09:17:02 +0100277 "admin",
278 "ping",
279 {
280 "from": "lcm",
281 "to": "lcm",
282 "worker_id": self.worker_id,
283 "version": lcm_version,
284 },
285 self.loop,
286 )
tiernoc0e42e22018-05-11 11:36:10 +0200287 # time between pings are low when it is not received and at starting
garciadeblas5697b8b2021-03-24 09:17:02 +0100288 wait_time = (
289 self.ping_interval_boot
290 if not kafka_has_received
291 else self.ping_interval_pace
292 )
tiernoc0e42e22018-05-11 11:36:10 +0200293 if not self.pings_not_received:
294 kafka_has_received = True
295 self.pings_not_received += 1
296 await asyncio.sleep(wait_time, loop=self.loop)
297 if self.pings_not_received > 10:
298 raise LcmException("It is not receiving pings from Kafka bus")
299 consecutive_errors = 0
300 first_start = False
301 except LcmException:
302 raise
303 except Exception as e:
304 # if not first_start is the first time after starting. So leave more time and wait
305 # to allow kafka starts
306 if consecutive_errors == 8 if not first_start else 30:
garciadeblas5697b8b2021-03-24 09:17:02 +0100307 self.logger.error(
308 "Task kafka_read task exit error too many errors. Exception: {}".format(
309 e
310 )
311 )
tiernoc0e42e22018-05-11 11:36:10 +0200312 raise
313 consecutive_errors += 1
garciadeblas5697b8b2021-03-24 09:17:02 +0100314 self.logger.error(
315 "Task kafka_read retrying after Exception {}".format(e)
316 )
tierno16427352019-04-22 11:37:36 +0000317 wait_time = 2 if not first_start else 5
tiernoc0e42e22018-05-11 11:36:10 +0200318 await asyncio.sleep(wait_time, loop=self.loop)
319
Patricia Reinoso15ce47b2022-10-26 08:58:39 +0000320 def _kafka_read_paas(self, command, params, order_id):
321 paas_id = params.get("_id")
322
323 if command == "created":
324 task = asyncio.ensure_future(self.paas.create(params, order_id))
325 self.lcm_tasks.register("paas", paas_id, order_id, "paas_create", task)
326 elif command == "edited":
327 task = asyncio.ensure_future(self.paas.edit(params, order_id))
328 self.lcm_tasks.register("paas", paas_id, order_id, "paas_edit", task)
329 elif command == "delete":
330 task = asyncio.ensure_future(self.paas.delete(params, order_id))
331 self.lcm_tasks.register("paas", paas_id, order_id, "paas_delete", task)
332 elif command == "deleted":
333 self.logger.debug("PaaS {} already deleted from DB".format(paas_id))
334 else:
335 self.logger.error("Invalid command {} for PaaS topic".format(command))
336
gcalvinoed7f6d42018-12-14 14:44:56 +0100337 def kafka_read_callback(self, topic, command, params):
338 order_id = 1
339
340 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
347 if command == "exit":
348 raise LcmExceptionExit
349 elif command.startswith("#"):
350 return
351 elif command == "echo":
352 # just for test
353 print(params)
354 sys.stdout.flush()
355 return
356 elif command == "test":
357 asyncio.Task(self.test(params), loop=self.loop)
358 return
359
360 if topic == "admin":
361 if command == "ping" and params["to"] == "lcm" and params["from"] == "lcm":
tierno16427352019-04-22 11:37:36 +0000362 if params.get("worker_id") != self.worker_id:
363 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100364 self.pings_not_received = 0
tierno3e359b12019-02-03 02:29:13 +0100365 try:
aticig56b86c22022-06-29 10:43:05 +0300366 with open(self.health_check_file, "w") as f:
tierno3e359b12019-02-03 02:29:13 +0100367 f.write(str(time()))
368 except Exception as e:
garciadeblas5697b8b2021-03-24 09:17:02 +0100369 self.logger.error(
370 "Cannot write into '{}' for healthcheck: {}".format(
aticig56b86c22022-06-29 10:43:05 +0300371 self.health_check_file, e
garciadeblas5697b8b2021-03-24 09:17:02 +0100372 )
373 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100374 return
magnussonle9198bb2020-01-21 13:00:51 +0100375 elif topic == "pla":
376 if command == "placement":
377 self.ns.update_nsrs_with_pla_result(params)
378 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100379 elif topic == "k8scluster":
380 if command == "create" or command == "created":
381 k8scluster_id = params.get("_id")
382 task = asyncio.ensure_future(self.k8scluster.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100383 self.lcm_tasks.register(
384 "k8scluster", k8scluster_id, order_id, "k8scluster_create", task
385 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100386 return
387 elif command == "delete" or command == "deleted":
388 k8scluster_id = params.get("_id")
389 task = asyncio.ensure_future(self.k8scluster.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100390 self.lcm_tasks.register(
391 "k8scluster", k8scluster_id, order_id, "k8scluster_delete", task
392 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100393 return
David Garciac1fe90a2021-03-31 19:12:02 +0200394 elif topic == "vca":
395 if command == "create" or command == "created":
396 vca_id = params.get("_id")
397 task = asyncio.ensure_future(self.vca.create(params, order_id))
398 self.lcm_tasks.register("vca", vca_id, order_id, "vca_create", task)
399 return
400 elif command == "delete" or command == "deleted":
401 vca_id = params.get("_id")
402 task = asyncio.ensure_future(self.vca.delete(params, order_id))
403 self.lcm_tasks.register("vca", vca_id, order_id, "vca_delete", task)
404 return
Patricia Reinoso15ce47b2022-10-26 08:58:39 +0000405 elif topic == "paas":
406 self._kafka_read_paas(command, params, order_id)
407 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100408 elif topic == "k8srepo":
409 if command == "create" or command == "created":
410 k8srepo_id = params.get("_id")
411 self.logger.debug("k8srepo_id = {}".format(k8srepo_id))
412 task = asyncio.ensure_future(self.k8srepo.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100413 self.lcm_tasks.register(
414 "k8srepo", k8srepo_id, order_id, "k8srepo_create", task
415 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100416 return
417 elif command == "delete" or command == "deleted":
418 k8srepo_id = params.get("_id")
419 task = asyncio.ensure_future(self.k8srepo.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100420 self.lcm_tasks.register(
421 "k8srepo", k8srepo_id, order_id, "k8srepo_delete", task
422 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100423 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100424 elif topic == "ns":
tierno307425f2020-01-26 23:35:59 +0000425 if command == "instantiate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100426 # self.logger.debug("Deploying NS {}".format(nsr_id))
427 nslcmop = params
428 nslcmop_id = nslcmop["_id"]
429 nsr_id = nslcmop["nsInstanceId"]
430 task = asyncio.ensure_future(self.ns.instantiate(nsr_id, nslcmop_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100431 self.lcm_tasks.register(
432 "ns", nsr_id, nslcmop_id, "ns_instantiate", task
433 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100434 return
tierno307425f2020-01-26 23:35:59 +0000435 elif command == "terminate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100436 # self.logger.debug("Deleting NS {}".format(nsr_id))
437 nslcmop = params
438 nslcmop_id = nslcmop["_id"]
439 nsr_id = nslcmop["nsInstanceId"]
440 self.lcm_tasks.cancel(topic, nsr_id)
441 task = asyncio.ensure_future(self.ns.terminate(nsr_id, nslcmop_id))
442 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_terminate", task)
443 return
ksaikiranr3fde2c72021-03-15 10:39:06 +0530444 elif command == "vca_status_refresh":
445 nslcmop = params
446 nslcmop_id = nslcmop["_id"]
447 nsr_id = nslcmop["nsInstanceId"]
garciadeblas5697b8b2021-03-24 09:17:02 +0100448 task = asyncio.ensure_future(
449 self.ns.vca_status_refresh(nsr_id, nslcmop_id)
450 )
451 self.lcm_tasks.register(
452 "ns", nsr_id, nslcmop_id, "ns_vca_status_refresh", task
453 )
ksaikiranr3fde2c72021-03-15 10:39:06 +0530454 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100455 elif command == "action":
456 # self.logger.debug("Update NS {}".format(nsr_id))
457 nslcmop = params
458 nslcmop_id = nslcmop["_id"]
459 nsr_id = nslcmop["nsInstanceId"]
460 task = asyncio.ensure_future(self.ns.action(nsr_id, nslcmop_id))
461 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_action", task)
462 return
aticigdffa6212022-04-12 15:27:53 +0300463 elif command == "update":
464 # self.logger.debug("Update NS {}".format(nsr_id))
465 nslcmop = params
466 nslcmop_id = nslcmop["_id"]
467 nsr_id = nslcmop["nsInstanceId"]
468 task = asyncio.ensure_future(self.ns.update(nsr_id, nslcmop_id))
469 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_update", task)
470 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100471 elif command == "scale":
472 # self.logger.debug("Update NS {}".format(nsr_id))
473 nslcmop = params
474 nslcmop_id = nslcmop["_id"]
475 nsr_id = nslcmop["nsInstanceId"]
476 task = asyncio.ensure_future(self.ns.scale(nsr_id, nslcmop_id))
477 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_scale", task)
478 return
garciadeblas07f4e4c2022-06-09 09:42:58 +0200479 elif command == "heal":
480 # self.logger.debug("Healing NS {}".format(nsr_id))
481 nslcmop = params
482 nslcmop_id = nslcmop["_id"]
483 nsr_id = nslcmop["nsInstanceId"]
484 task = asyncio.ensure_future(self.ns.heal(nsr_id, nslcmop_id))
preethika.p28b0bf82022-09-23 07:36:28 +0000485 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_heal", task)
garciadeblas07f4e4c2022-06-09 09:42:58 +0200486 return
elumalai80bcf1c2022-04-28 18:05:01 +0530487 elif command == "migrate":
488 nslcmop = params
489 nslcmop_id = nslcmop["_id"]
490 nsr_id = nslcmop["nsInstanceId"]
491 task = asyncio.ensure_future(self.ns.migrate(nsr_id, nslcmop_id))
492 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_migrate", task)
493 return
govindarajul4ff4b512022-05-02 20:02:41 +0530494 elif command == "verticalscale":
495 nslcmop = params
496 nslcmop_id = nslcmop["_id"]
497 nsr_id = nslcmop["nsInstanceId"]
498 task = asyncio.ensure_future(self.ns.vertical_scale(nsr_id, nslcmop_id))
preethika.p28b0bf82022-09-23 07:36:28 +0000499 self.logger.debug(
500 "nsr_id,nslcmop_id,task {},{},{}".format(nsr_id, nslcmop_id, task)
501 )
502 self.lcm_tasks.register(
503 "ns", nsr_id, nslcmop_id, "ns_verticalscale", task
504 )
505 self.logger.debug(
506 "LCM task registered {},{},{} ".format(nsr_id, nslcmop_id, task)
507 )
govindarajul4ff4b512022-05-02 20:02:41 +0530508 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100509 elif command == "show":
tiernoc2564fe2019-01-28 16:18:56 +0000510 nsr_id = params
gcalvinoed7f6d42018-12-14 14:44:56 +0100511 try:
512 db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
garciadeblas5697b8b2021-03-24 09:17:02 +0100513 print(
514 "nsr:\n _id={}\n operational-status: {}\n config-status: {}"
515 "\n detailed-status: {}\n deploy: {}\n tasks: {}"
516 "".format(
517 nsr_id,
518 db_nsr["operational-status"],
519 db_nsr["config-status"],
520 db_nsr["detailed-status"],
521 db_nsr["_admin"]["deployed"],
522 self.lcm_ns_tasks.get(nsr_id),
523 )
524 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100525 except Exception as e:
526 print("nsr {} not found: {}".format(nsr_id, e))
527 sys.stdout.flush()
528 return
529 elif command == "deleted":
530 return # TODO cleaning of task just in case should be done
garciadeblas5697b8b2021-03-24 09:17:02 +0100531 elif command in (
elumalaica7ece02022-04-12 12:47:32 +0530532 "vnf_terminated",
elumalaib9e357c2022-04-27 09:58:38 +0530533 "policy_updated",
garciadeblas5697b8b2021-03-24 09:17:02 +0100534 "terminated",
535 "instantiated",
536 "scaled",
garciadeblas07f4e4c2022-06-09 09:42:58 +0200537 "healed",
garciadeblas5697b8b2021-03-24 09:17:02 +0100538 "actioned",
aticigdffa6212022-04-12 15:27:53 +0300539 "updated",
elumalai80bcf1c2022-04-28 18:05:01 +0530540 "migrated",
govindarajul4ff4b512022-05-02 20:02:41 +0530541 "verticalscaled",
garciadeblas5697b8b2021-03-24 09:17:02 +0100542 ): # "scaled-cooldown-time"
gcalvinoed7f6d42018-12-14 14:44:56 +0100543 return
elumalaica7ece02022-04-12 12:47:32 +0530544
gcalvinoed7f6d42018-12-14 14:44:56 +0100545 elif topic == "nsi": # netslice LCM processes (instantiate, terminate, etc)
tierno307425f2020-01-26 23:35:59 +0000546 if command == "instantiate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100547 # self.logger.debug("Instantiating Network Slice {}".format(nsilcmop["netsliceInstanceId"]))
548 nsilcmop = params
549 nsilcmop_id = nsilcmop["_id"] # slice operation id
550 nsir_id = nsilcmop["netsliceInstanceId"] # slice record id
garciadeblas5697b8b2021-03-24 09:17:02 +0100551 task = asyncio.ensure_future(
552 self.netslice.instantiate(nsir_id, nsilcmop_id)
553 )
554 self.lcm_tasks.register(
555 "nsi", nsir_id, nsilcmop_id, "nsi_instantiate", task
556 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100557 return
tierno307425f2020-01-26 23:35:59 +0000558 elif command == "terminate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100559 # self.logger.debug("Terminating Network Slice NS {}".format(nsilcmop["netsliceInstanceId"]))
560 nsilcmop = params
561 nsilcmop_id = nsilcmop["_id"] # slice operation id
562 nsir_id = nsilcmop["netsliceInstanceId"] # slice record id
563 self.lcm_tasks.cancel(topic, nsir_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100564 task = asyncio.ensure_future(
565 self.netslice.terminate(nsir_id, nsilcmop_id)
566 )
567 self.lcm_tasks.register(
568 "nsi", nsir_id, nsilcmop_id, "nsi_terminate", task
569 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100570 return
571 elif command == "show":
tiernoc2564fe2019-01-28 16:18:56 +0000572 nsir_id = params
gcalvinoed7f6d42018-12-14 14:44:56 +0100573 try:
574 db_nsir = self.db.get_one("nsirs", {"_id": nsir_id})
garciadeblas5697b8b2021-03-24 09:17:02 +0100575 print(
576 "nsir:\n _id={}\n operational-status: {}\n config-status: {}"
577 "\n detailed-status: {}\n deploy: {}\n tasks: {}"
578 "".format(
579 nsir_id,
580 db_nsir["operational-status"],
581 db_nsir["config-status"],
582 db_nsir["detailed-status"],
583 db_nsir["_admin"]["deployed"],
584 self.lcm_netslice_tasks.get(nsir_id),
585 )
586 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100587 except Exception as e:
588 print("nsir {} not found: {}".format(nsir_id, e))
589 sys.stdout.flush()
590 return
591 elif command == "deleted":
592 return # TODO cleaning of task just in case should be done
garciadeblas5697b8b2021-03-24 09:17:02 +0100593 elif command in (
594 "terminated",
595 "instantiated",
596 "scaled",
garciadeblas07f4e4c2022-06-09 09:42:58 +0200597 "healed",
garciadeblas5697b8b2021-03-24 09:17:02 +0100598 "actioned",
599 ): # "scaled-cooldown-time"
gcalvinoed7f6d42018-12-14 14:44:56 +0100600 return
601 elif topic == "vim_account":
602 vim_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000603 if command in ("create", "created"):
tierno2357f4e2020-10-19 16:38:59 +0000604 if not self.config["ro_config"].get("ng"):
605 task = asyncio.ensure_future(self.vim.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100606 self.lcm_tasks.register(
607 "vim_account", vim_id, order_id, "vim_create", task
608 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100609 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100610 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100611 self.lcm_tasks.cancel(topic, vim_id)
kuuse6a470c62019-07-10 13:52:45 +0200612 task = asyncio.ensure_future(self.vim.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100613 self.lcm_tasks.register(
614 "vim_account", vim_id, order_id, "vim_delete", task
615 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100616 return
617 elif command == "show":
618 print("not implemented show with vim_account")
619 sys.stdout.flush()
620 return
tiernof210c1c2019-10-16 09:09:58 +0000621 elif command in ("edit", "edited"):
tierno2357f4e2020-10-19 16:38:59 +0000622 if not self.config["ro_config"].get("ng"):
623 task = asyncio.ensure_future(self.vim.edit(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100624 self.lcm_tasks.register(
625 "vim_account", vim_id, order_id, "vim_edit", task
626 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100627 return
tiernof210c1c2019-10-16 09:09:58 +0000628 elif command == "deleted":
629 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100630 elif topic == "wim_account":
631 wim_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000632 if command in ("create", "created"):
tierno2357f4e2020-10-19 16:38:59 +0000633 if not self.config["ro_config"].get("ng"):
634 task = asyncio.ensure_future(self.wim.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100635 self.lcm_tasks.register(
636 "wim_account", wim_id, order_id, "wim_create", task
637 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100638 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100639 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100640 self.lcm_tasks.cancel(topic, wim_id)
kuuse6a470c62019-07-10 13:52:45 +0200641 task = asyncio.ensure_future(self.wim.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100642 self.lcm_tasks.register(
643 "wim_account", wim_id, order_id, "wim_delete", task
644 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100645 return
646 elif command == "show":
647 print("not implemented show with wim_account")
648 sys.stdout.flush()
649 return
tiernof210c1c2019-10-16 09:09:58 +0000650 elif command in ("edit", "edited"):
gcalvinoed7f6d42018-12-14 14:44:56 +0100651 task = asyncio.ensure_future(self.wim.edit(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100652 self.lcm_tasks.register(
653 "wim_account", wim_id, order_id, "wim_edit", task
654 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100655 return
tiernof210c1c2019-10-16 09:09:58 +0000656 elif command == "deleted":
657 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100658 elif topic == "sdn":
659 _sdn_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000660 if command in ("create", "created"):
tierno2357f4e2020-10-19 16:38:59 +0000661 if not self.config["ro_config"].get("ng"):
662 task = asyncio.ensure_future(self.sdn.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100663 self.lcm_tasks.register(
664 "sdn", _sdn_id, order_id, "sdn_create", task
665 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100666 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100667 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100668 self.lcm_tasks.cancel(topic, _sdn_id)
kuuse6a470c62019-07-10 13:52:45 +0200669 task = asyncio.ensure_future(self.sdn.delete(params, order_id))
gcalvinoed7f6d42018-12-14 14:44:56 +0100670 self.lcm_tasks.register("sdn", _sdn_id, order_id, "sdn_delete", task)
671 return
tiernof210c1c2019-10-16 09:09:58 +0000672 elif command in ("edit", "edited"):
gcalvinoed7f6d42018-12-14 14:44:56 +0100673 task = asyncio.ensure_future(self.sdn.edit(params, order_id))
674 self.lcm_tasks.register("sdn", _sdn_id, order_id, "sdn_edit", task)
675 return
tiernof210c1c2019-10-16 09:09:58 +0000676 elif command == "deleted":
677 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100678 self.logger.critical("unknown topic {} and command '{}'".format(topic, command))
679
tiernoc0e42e22018-05-11 11:36:10 +0200680 async def kafka_read(self):
garciadeblas5697b8b2021-03-24 09:17:02 +0100681 self.logger.debug(
682 "Task kafka_read Enter with worker_id={}".format(self.worker_id)
683 )
tiernoc0e42e22018-05-11 11:36:10 +0200684 # future = asyncio.Future()
gcalvinoed7f6d42018-12-14 14:44:56 +0100685 self.consecutive_errors = 0
686 self.first_start = True
687 while self.consecutive_errors < 10:
tiernoc0e42e22018-05-11 11:36:10 +0200688 try:
garciadeblas5697b8b2021-03-24 09:17:02 +0100689 topics = (
690 "ns",
691 "vim_account",
692 "wim_account",
693 "sdn",
694 "nsi",
695 "k8scluster",
696 "vca",
Patricia Reinoso15ce47b2022-10-26 08:58:39 +0000697 "paas",
garciadeblas5697b8b2021-03-24 09:17:02 +0100698 "k8srepo",
699 "pla",
700 )
701 topics_admin = ("admin",)
tierno16427352019-04-22 11:37:36 +0000702 await asyncio.gather(
garciadeblas5697b8b2021-03-24 09:17:02 +0100703 self.msg.aioread(
704 topics, self.loop, self.kafka_read_callback, from_beginning=True
705 ),
706 self.msg_admin.aioread(
707 topics_admin,
708 self.loop,
709 self.kafka_read_callback,
710 group_id=False,
711 ),
tierno16427352019-04-22 11:37:36 +0000712 )
tiernoc0e42e22018-05-11 11:36:10 +0200713
gcalvinoed7f6d42018-12-14 14:44:56 +0100714 except LcmExceptionExit:
715 self.logger.debug("Bye!")
716 break
tiernoc0e42e22018-05-11 11:36:10 +0200717 except Exception as e:
718 # if not first_start is the first time after starting. So leave more time and wait
719 # to allow kafka starts
gcalvinoed7f6d42018-12-14 14:44:56 +0100720 if self.consecutive_errors == 8 if not self.first_start else 30:
garciadeblas5697b8b2021-03-24 09:17:02 +0100721 self.logger.error(
722 "Task kafka_read task exit error too many errors. Exception: {}".format(
723 e
724 )
725 )
tiernoc0e42e22018-05-11 11:36:10 +0200726 raise
gcalvinoed7f6d42018-12-14 14:44:56 +0100727 self.consecutive_errors += 1
garciadeblas5697b8b2021-03-24 09:17:02 +0100728 self.logger.error(
729 "Task kafka_read retrying after Exception {}".format(e)
730 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100731 wait_time = 2 if not self.first_start else 5
tiernoc0e42e22018-05-11 11:36:10 +0200732 await asyncio.sleep(wait_time, loop=self.loop)
733
734 # self.logger.debug("Task kafka_read terminating")
735 self.logger.debug("Task kafka_read exit")
736
737 def start(self):
tierno22f4f9c2018-06-11 18:53:39 +0200738
739 # check RO version
740 self.loop.run_until_complete(self.check_RO_version())
741
aticig15db6142022-01-24 12:51:26 +0300742 self.ns = ns.NsLcm(self.msg, self.lcm_tasks, self.config, self.loop)
garciadeblas5697b8b2021-03-24 09:17:02 +0100743 self.netslice = netslice.NetsliceLcm(
744 self.msg, self.lcm_tasks, self.config, self.loop, self.ns
745 )
bravof922c4172020-11-24 21:21:43 -0300746 self.vim = vim_sdn.VimLcm(self.msg, self.lcm_tasks, self.config, self.loop)
747 self.wim = vim_sdn.WimLcm(self.msg, self.lcm_tasks, self.config, self.loop)
748 self.sdn = vim_sdn.SdnLcm(self.msg, self.lcm_tasks, self.config, self.loop)
garciadeblas5697b8b2021-03-24 09:17:02 +0100749 self.k8scluster = vim_sdn.K8sClusterLcm(
750 self.msg, self.lcm_tasks, self.config, self.loop
751 )
David Garciac1fe90a2021-03-31 19:12:02 +0200752 self.vca = vim_sdn.VcaLcm(self.msg, self.lcm_tasks, self.config, self.loop)
Patricia Reinoso15ce47b2022-10-26 08:58:39 +0000753 self.paas = paas.PaasLcm(self.msg, self.lcm_tasks, self.config, self.loop)
garciadeblas5697b8b2021-03-24 09:17:02 +0100754 self.k8srepo = vim_sdn.K8sRepoLcm(
755 self.msg, self.lcm_tasks, self.config, self.loop
756 )
tierno2357f4e2020-10-19 16:38:59 +0000757
garciadeblas5697b8b2021-03-24 09:17:02 +0100758 self.loop.run_until_complete(
759 asyncio.gather(self.kafka_read(), self.kafka_ping())
760 )
bravof73bac502021-05-11 07:38:47 -0400761
tiernoc0e42e22018-05-11 11:36:10 +0200762 # TODO
763 # self.logger.debug("Terminating cancelling creation tasks")
tiernoca2e16a2018-06-29 15:25:24 +0200764 # self.lcm_tasks.cancel("ALL", "create")
tiernoc0e42e22018-05-11 11:36:10 +0200765 # timeout = 200
766 # while self.is_pending_tasks():
767 # self.logger.debug("Task kafka_read terminating. Waiting for tasks termination")
768 # await asyncio.sleep(2, loop=self.loop)
769 # timeout -= 2
770 # if not timeout:
tiernoca2e16a2018-06-29 15:25:24 +0200771 # self.lcm_tasks.cancel("ALL", "ALL")
tiernoc0e42e22018-05-11 11:36:10 +0200772 self.loop.close()
773 self.loop = None
774 if self.db:
775 self.db.db_disconnect()
776 if self.msg:
777 self.msg.disconnect()
tierno16427352019-04-22 11:37:36 +0000778 if self.msg_admin:
779 self.msg_admin.disconnect()
tiernoc0e42e22018-05-11 11:36:10 +0200780 if self.fs:
781 self.fs.fs_disconnect()
782
tiernoc0e42e22018-05-11 11:36:10 +0200783 def read_config_file(self, config_file):
784 # TODO make a [ini] + yaml inside parser
785 # the configparser library is not suitable, because it does not admit comments at the end of line,
786 # and not parse integer or boolean
787 try:
tierno744303e2020-01-13 16:46:31 +0000788 # read file as yaml format
tiernoc0e42e22018-05-11 11:36:10 +0200789 with open(config_file) as f:
Luisccdc2162022-07-01 14:35:49 +0000790 conf = yaml.safe_load(f)
tierno744303e2020-01-13 16:46:31 +0000791 # Ensure all sections are not empty
garciadeblas5697b8b2021-03-24 09:17:02 +0100792 for k in (
793 "global",
794 "timeout",
795 "RO",
796 "VCA",
797 "database",
798 "storage",
799 "message",
800 ):
tierno744303e2020-01-13 16:46:31 +0000801 if not conf.get(k):
802 conf[k] = {}
803
804 # read all environ that starts with OSMLCM_
tiernoc0e42e22018-05-11 11:36:10 +0200805 for k, v in environ.items():
806 if not k.startswith("OSMLCM_"):
807 continue
tierno744303e2020-01-13 16:46:31 +0000808 subject, _, item = k[7:].lower().partition("_")
809 if not item:
tierno17a612f2018-10-23 11:30:42 +0200810 continue
tierno744303e2020-01-13 16:46:31 +0000811 if subject in ("ro", "vca"):
tierno17a612f2018-10-23 11:30:42 +0200812 # put in capital letter
tierno744303e2020-01-13 16:46:31 +0000813 subject = subject.upper()
tiernoc0e42e22018-05-11 11:36:10 +0200814 try:
tierno744303e2020-01-13 16:46:31 +0000815 if item == "port" or subject == "timeout":
816 conf[subject][item] = int(v)
tiernoc0e42e22018-05-11 11:36:10 +0200817 else:
tierno744303e2020-01-13 16:46:31 +0000818 conf[subject][item] = v
tiernoc0e42e22018-05-11 11:36:10 +0200819 except Exception as e:
garciadeblas5697b8b2021-03-24 09:17:02 +0100820 self.logger.warning(
821 "skipping environ '{}' on exception '{}'".format(k, e)
822 )
tierno744303e2020-01-13 16:46:31 +0000823
824 # backward compatibility of VCA parameters
825
garciadeblas5697b8b2021-03-24 09:17:02 +0100826 if "pubkey" in conf["VCA"]:
827 conf["VCA"]["public_key"] = conf["VCA"].pop("pubkey")
828 if "cacert" in conf["VCA"]:
829 conf["VCA"]["ca_cert"] = conf["VCA"].pop("cacert")
830 if "apiproxy" in conf["VCA"]:
831 conf["VCA"]["api_proxy"] = conf["VCA"].pop("apiproxy")
tierno744303e2020-01-13 16:46:31 +0000832
garciadeblas5697b8b2021-03-24 09:17:02 +0100833 if "enableosupgrade" in conf["VCA"]:
834 conf["VCA"]["enable_os_upgrade"] = conf["VCA"].pop("enableosupgrade")
835 if isinstance(conf["VCA"].get("enable_os_upgrade"), str):
836 if conf["VCA"]["enable_os_upgrade"].lower() == "false":
837 conf["VCA"]["enable_os_upgrade"] = False
838 elif conf["VCA"]["enable_os_upgrade"].lower() == "true":
839 conf["VCA"]["enable_os_upgrade"] = True
tierno744303e2020-01-13 16:46:31 +0000840
garciadeblas5697b8b2021-03-24 09:17:02 +0100841 if "aptmirror" in conf["VCA"]:
842 conf["VCA"]["apt_mirror"] = conf["VCA"].pop("aptmirror")
tiernoc0e42e22018-05-11 11:36:10 +0200843
844 return conf
845 except Exception as e:
846 self.logger.critical("At config file '{}': {}".format(config_file, e))
847 exit(1)
848
tierno16427352019-04-22 11:37:36 +0000849 @staticmethod
850 def get_process_id():
851 """
852 Obtain a unique ID for this process. If running from inside docker, it will get docker ID. If not it
853 will provide a random one
854 :return: Obtained ID
855 """
856 # Try getting docker id. If fails, get pid
857 try:
858 with open("/proc/self/cgroup", "r") as f:
859 text_id_ = f.readline()
860 _, _, text_id = text_id_.rpartition("/")
garciadeblas5697b8b2021-03-24 09:17:02 +0100861 text_id = text_id.replace("\n", "")[:12]
tierno16427352019-04-22 11:37:36 +0000862 if text_id:
863 return text_id
864 except Exception:
865 pass
866 # Return a random id
garciadeblas5697b8b2021-03-24 09:17:02 +0100867 return "".join(random_choice("0123456789abcdef") for _ in range(12))
tierno16427352019-04-22 11:37:36 +0000868
tiernoc0e42e22018-05-11 11:36:10 +0200869
tierno275411e2018-05-16 14:33:32 +0200870def usage():
garciadeblas5697b8b2021-03-24 09:17:02 +0100871 print(
872 """Usage: {} [options]
quilesj7e13aeb2019-10-08 13:34:55 +0200873 -c|--config [configuration_file]: loads the configuration file (default: ./lcm.cfg)
tiernoa9843d82018-10-24 10:44:20 +0200874 --health-check: do not run lcm, but inspect kafka bus to determine if lcm is healthy
tierno275411e2018-05-16 14:33:32 +0200875 -h|--help: shows this help
garciadeblas5697b8b2021-03-24 09:17:02 +0100876 """.format(
877 sys.argv[0]
878 )
879 )
tierno750b2452018-05-17 16:39:29 +0200880 # --log-socket-host HOST: send logs to this host")
881 # --log-socket-port PORT: send logs using this port (default: 9022)")
tierno275411e2018-05-16 14:33:32 +0200882
883
garciadeblas5697b8b2021-03-24 09:17:02 +0100884if __name__ == "__main__":
quilesj7e13aeb2019-10-08 13:34:55 +0200885
tierno275411e2018-05-16 14:33:32 +0200886 try:
tierno8c16b052020-02-05 15:08:32 +0000887 # print("SYS.PATH='{}'".format(sys.path))
tierno275411e2018-05-16 14:33:32 +0200888 # load parameters and configuration
quilesj7e13aeb2019-10-08 13:34:55 +0200889 # -h
890 # -c value
891 # --config value
892 # --help
893 # --health-check
garciadeblas5697b8b2021-03-24 09:17:02 +0100894 opts, args = getopt.getopt(
895 sys.argv[1:], "hc:", ["config=", "help", "health-check"]
896 )
tierno275411e2018-05-16 14:33:32 +0200897 # TODO add "log-socket-host=", "log-socket-port=", "log-file="
898 config_file = None
899 for o, a in opts:
900 if o in ("-h", "--help"):
901 usage()
902 sys.exit()
903 elif o in ("-c", "--config"):
904 config_file = a
tiernoa9843d82018-10-24 10:44:20 +0200905 elif o == "--health-check":
tierno94f06112020-02-11 12:38:19 +0000906 from osm_lcm.lcm_hc import health_check
garciadeblas5697b8b2021-03-24 09:17:02 +0100907
aticig56b86c22022-06-29 10:43:05 +0300908 health_check(config_file, Lcm.ping_interval_pace)
tierno275411e2018-05-16 14:33:32 +0200909 # elif o == "--log-socket-port":
910 # log_socket_port = a
911 # elif o == "--log-socket-host":
912 # log_socket_host = a
913 # elif o == "--log-file":
914 # log_file = a
915 else:
916 assert False, "Unhandled option"
quilesj7e13aeb2019-10-08 13:34:55 +0200917
tierno275411e2018-05-16 14:33:32 +0200918 if config_file:
919 if not path.isfile(config_file):
garciadeblas5697b8b2021-03-24 09:17:02 +0100920 print(
921 "configuration file '{}' does not exist".format(config_file),
922 file=sys.stderr,
923 )
tierno275411e2018-05-16 14:33:32 +0200924 exit(1)
925 else:
garciadeblas5697b8b2021-03-24 09:17:02 +0100926 for config_file in (
927 __file__[: __file__.rfind(".")] + ".cfg",
928 "./lcm.cfg",
929 "/etc/osm/lcm.cfg",
930 ):
tierno275411e2018-05-16 14:33:32 +0200931 if path.isfile(config_file):
932 break
933 else:
garciadeblas5697b8b2021-03-24 09:17:02 +0100934 print(
935 "No configuration file 'lcm.cfg' found neither at local folder nor at /etc/osm/",
936 file=sys.stderr,
937 )
tierno275411e2018-05-16 14:33:32 +0200938 exit(1)
939 lcm = Lcm(config_file)
tierno3e359b12019-02-03 02:29:13 +0100940 lcm.start()
tierno22f4f9c2018-06-11 18:53:39 +0200941 except (LcmException, getopt.GetoptError) as e:
tierno275411e2018-05-16 14:33:32 +0200942 print(str(e), file=sys.stderr)
943 # usage()
944 exit(1)