blob: 731488cf92119bedcd0bd77d06a4bedfec478fd8 [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
bravof73bac502021-05-11 07:38:47 -040032from osm_lcm import ns, 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
tierno275411e2018-05-16 14:33:32 +020047from os import environ, path
tierno16427352019-04-22 11:37:36 +000048from random import choice as random_choice
tierno59d22d22018-09-25 18:10:19 +020049from n2vc import version as n2vc_version
bravof922c4172020-11-24 21:21:43 -030050import traceback
tiernoc0e42e22018-05-11 11:36:10 +020051
garciadeblas5697b8b2021-03-24 09:17:02 +010052if os.getenv("OSMLCM_PDB_DEBUG", None) is not None:
quilesj7e13aeb2019-10-08 13:34:55 +020053 pdb.set_trace()
54
tiernoc0e42e22018-05-11 11:36:10 +020055
tierno275411e2018-05-16 14:33:32 +020056__author__ = "Alfonso Tierno"
tiernoe64f7fb2019-09-11 08:55:52 +000057min_RO_version = "6.0.2"
tierno6e9d2eb2018-09-12 17:47:18 +020058min_n2vc_version = "0.0.2"
quilesj7e13aeb2019-10-08 13:34:55 +020059
tierno16427352019-04-22 11:37:36 +000060min_common_version = "0.1.19"
garciadeblas5697b8b2021-03-24 09:17:02 +010061health_check_file = (
62 path.expanduser("~") + "/time_last_ping"
63) # TODO find better location for this file
tierno275411e2018-05-16 14:33:32 +020064
65
tiernoc0e42e22018-05-11 11:36:10 +020066class Lcm:
67
garciadeblas5697b8b2021-03-24 09:17:02 +010068 ping_interval_pace = (
69 120 # how many time ping is send once is confirmed all is running
70 )
71 ping_interval_boot = 5 # how many time ping is sent when booting
72 cfg_logger_name = {
73 "message": "lcm.msg",
74 "database": "lcm.db",
75 "storage": "lcm.fs",
76 "tsdb": "lcm.prometheus",
77 }
tierno991e95d2020-07-21 12:41:25 +000078 # ^ contains for each section at lcm.cfg the used logger name
tiernoa9843d82018-10-24 10:44:20 +020079
tierno59d22d22018-09-25 18:10:19 +020080 def __init__(self, config_file, loop=None):
tiernoc0e42e22018-05-11 11:36:10 +020081 """
82 Init, Connect to database, filesystem storage, and messaging
83 :param config: two level dictionary with configuration. Top level should contain 'database', 'storage',
84 :return: None
85 """
tiernoc0e42e22018-05-11 11:36:10 +020086 self.db = None
87 self.msg = None
tierno16427352019-04-22 11:37:36 +000088 self.msg_admin = None
tiernoc0e42e22018-05-11 11:36:10 +020089 self.fs = None
90 self.pings_not_received = 1
tiernoc2564fe2019-01-28 16:18:56 +000091 self.consecutive_errors = 0
92 self.first_start = False
tiernoc0e42e22018-05-11 11:36:10 +020093
tiernoc0e42e22018-05-11 11:36:10 +020094 # logging
garciadeblas5697b8b2021-03-24 09:17:02 +010095 self.logger = logging.getLogger("lcm")
tierno16427352019-04-22 11:37:36 +000096 # get id
97 self.worker_id = self.get_process_id()
tiernoc0e42e22018-05-11 11:36:10 +020098 # load configuration
99 config = self.read_config_file(config_file)
100 self.config = config
tierno744303e2020-01-13 16:46:31 +0000101 self.config["ro_config"] = {
tierno69f0d382020-05-07 13:08:09 +0000102 "ng": config["RO"].get("ng", False),
103 "uri": config["RO"].get("uri"),
tierno750b2452018-05-17 16:39:29 +0200104 "tenant": config.get("tenant", "osm"),
tierno69f0d382020-05-07 13:08:09 +0000105 "logger_name": "lcm.roclient",
106 "loglevel": config["RO"].get("loglevel", "ERROR"),
tiernoc0e42e22018-05-11 11:36:10 +0200107 }
tierno69f0d382020-05-07 13:08:09 +0000108 if not self.config["ro_config"]["uri"]:
garciadeblas5697b8b2021-03-24 09:17:02 +0100109 self.config["ro_config"]["uri"] = "http://{}:{}/".format(
110 config["RO"]["host"], config["RO"]["port"]
111 )
112 elif (
113 "/ro" in self.config["ro_config"]["uri"][-4:]
114 or "/openmano" in self.config["ro_config"]["uri"][-10:]
115 ):
tierno2357f4e2020-10-19 16:38:59 +0000116 # uri ends with '/ro', '/ro/', '/openmano', '/openmano/'
117 index = self.config["ro_config"]["uri"][-1].rfind("/")
garciadeblas5697b8b2021-03-24 09:17:02 +0100118 self.config["ro_config"]["uri"] = self.config["ro_config"]["uri"][index + 1]
tiernoc0e42e22018-05-11 11:36:10 +0200119
tierno59d22d22018-09-25 18:10:19 +0200120 self.loop = loop or asyncio.get_event_loop()
garciadeblas5697b8b2021-03-24 09:17:02 +0100121 self.ns = (
122 self.netslice
123 ) = (
124 self.vim
125 ) = self.wim = self.sdn = self.k8scluster = self.vca = self.k8srepo = 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 )
tiernoc0e42e22018-05-11 11:36:10 +0200134 config["database"]["logger_name"] = "lcm.db"
135 config["storage"]["logger_name"] = "lcm.fs"
136 config["message"]["logger_name"] = "lcm.msg"
tierno86aa62f2018-08-20 11:57:04 +0000137 if config["global"].get("logfile"):
garciadeblas5697b8b2021-03-24 09:17:02 +0100138 file_handler = logging.handlers.RotatingFileHandler(
139 config["global"]["logfile"], maxBytes=100e6, backupCount=9, delay=0
140 )
tiernoc0e42e22018-05-11 11:36:10 +0200141 file_handler.setFormatter(log_formatter_simple)
142 self.logger.addHandler(file_handler)
tierno86aa62f2018-08-20 11:57:04 +0000143 if not config["global"].get("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
148 if config["global"].get("loglevel"):
149 self.logger.setLevel(config["global"]["loglevel"])
150
151 # logging other modules
tierno991e95d2020-07-21 12:41:25 +0000152 for k1, logname in self.cfg_logger_name.items():
tiernoc0e42e22018-05-11 11:36:10 +0200153 config[k1]["logger_name"] = logname
154 logger_module = logging.getLogger(logname)
tierno86aa62f2018-08-20 11:57:04 +0000155 if config[k1].get("logfile"):
garciadeblas5697b8b2021-03-24 09:17:02 +0100156 file_handler = logging.handlers.RotatingFileHandler(
157 config[k1]["logfile"], maxBytes=100e6, backupCount=9, delay=0
158 )
tiernoc0e42e22018-05-11 11:36:10 +0200159 file_handler.setFormatter(log_formatter_simple)
160 logger_module.addHandler(file_handler)
tierno86aa62f2018-08-20 11:57:04 +0000161 if config[k1].get("loglevel"):
tiernoc0e42e22018-05-11 11:36:10 +0200162 logger_module.setLevel(config[k1]["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:
bravof922c4172020-11-24 21:21:43 -0300185 self.db = Database(config).instance.db
tiernoc0e42e22018-05-11 11:36:10 +0200186
bravof922c4172020-11-24 21:21:43 -0300187 self.fs = Filesystem(config).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
tiernoc2564fe2019-01-28 16:18:56 +0000191 config_message = config["message"].copy()
192 config_message["loop"] = self.loop
193 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(
208 config["message"]["driver"]
209 )
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
tierno22f4f9c2018-06-11 18:53:39 +0200218 async def check_RO_version(self):
tiernoe64f7fb2019-09-11 08:55:52 +0000219 tries = 14
220 last_error = None
221 while True:
tierno2357f4e2020-10-19 16:38:59 +0000222 ro_uri = self.config["ro_config"]["uri"]
tiernoe64f7fb2019-09-11 08:55:52 +0000223 try:
tierno2357f4e2020-10-19 16:38:59 +0000224 # try new RO, if fail old RO
225 try:
226 self.config["ro_config"]["uri"] = ro_uri + "ro"
tierno69f0d382020-05-07 13:08:09 +0000227 ro_server = NgRoClient(self.loop, **self.config["ro_config"])
tierno2357f4e2020-10-19 16:38:59 +0000228 ro_version = await ro_server.get_version()
229 self.config["ro_config"]["ng"] = True
230 except Exception:
231 self.config["ro_config"]["uri"] = ro_uri + "openmano"
tierno69f0d382020-05-07 13:08:09 +0000232 ro_server = ROClient(self.loop, **self.config["ro_config"])
tierno2357f4e2020-10-19 16:38:59 +0000233 ro_version = await ro_server.get_version()
234 self.config["ro_config"]["ng"] = False
tiernoe64f7fb2019-09-11 08:55:52 +0000235 if versiontuple(ro_version) < versiontuple(min_RO_version):
garciadeblas5697b8b2021-03-24 09:17:02 +0100236 raise LcmException(
237 "Not compatible osm/RO version '{}'. Needed '{}' or higher".format(
238 ro_version, min_RO_version
239 )
240 )
241 self.logger.info(
242 "Connected to RO version {} new-generation version {}".format(
243 ro_version, self.config["ro_config"]["ng"]
244 )
245 )
tiernoe64f7fb2019-09-11 08:55:52 +0000246 return
tierno69f0d382020-05-07 13:08:09 +0000247 except (ROClientException, NgRoException) as e:
tierno2357f4e2020-10-19 16:38:59 +0000248 self.config["ro_config"]["uri"] = ro_uri
tiernoe64f7fb2019-09-11 08:55:52 +0000249 tries -= 1
bravof922c4172020-11-24 21:21:43 -0300250 traceback.print_tb(e.__traceback__)
garciadeblas5697b8b2021-03-24 09:17:02 +0100251 error_text = "Error while connecting to RO on {}: {}".format(
252 self.config["ro_config"]["uri"], e
253 )
tiernoe64f7fb2019-09-11 08:55:52 +0000254 if tries <= 0:
255 self.logger.critical(error_text)
256 raise LcmException(error_text)
257 if last_error != error_text:
258 last_error = error_text
garciadeblas5697b8b2021-03-24 09:17:02 +0100259 self.logger.error(
260 error_text + ". Waiting until {} seconds".format(5 * tries)
261 )
tiernoe64f7fb2019-09-11 08:55:52 +0000262 await asyncio.sleep(5)
tierno22f4f9c2018-06-11 18:53:39 +0200263
tiernoc0e42e22018-05-11 11:36:10 +0200264 async def test(self, param=None):
265 self.logger.debug("Starting/Ending test task: {}".format(param))
266
tiernoc0e42e22018-05-11 11:36:10 +0200267 async def kafka_ping(self):
268 self.logger.debug("Task kafka_ping Enter")
269 consecutive_errors = 0
270 first_start = True
271 kafka_has_received = False
272 self.pings_not_received = 1
273 while True:
274 try:
tierno16427352019-04-22 11:37:36 +0000275 await self.msg_admin.aiowrite(
garciadeblas5697b8b2021-03-24 09:17:02 +0100276 "admin",
277 "ping",
278 {
279 "from": "lcm",
280 "to": "lcm",
281 "worker_id": self.worker_id,
282 "version": lcm_version,
283 },
284 self.loop,
285 )
tiernoc0e42e22018-05-11 11:36:10 +0200286 # time between pings are low when it is not received and at starting
garciadeblas5697b8b2021-03-24 09:17:02 +0100287 wait_time = (
288 self.ping_interval_boot
289 if not kafka_has_received
290 else self.ping_interval_pace
291 )
tiernoc0e42e22018-05-11 11:36:10 +0200292 if not self.pings_not_received:
293 kafka_has_received = True
294 self.pings_not_received += 1
295 await asyncio.sleep(wait_time, loop=self.loop)
296 if self.pings_not_received > 10:
297 raise LcmException("It is not receiving pings from Kafka bus")
298 consecutive_errors = 0
299 first_start = False
300 except LcmException:
301 raise
302 except Exception as e:
303 # if not first_start is the first time after starting. So leave more time and wait
304 # to allow kafka starts
305 if consecutive_errors == 8 if not first_start else 30:
garciadeblas5697b8b2021-03-24 09:17:02 +0100306 self.logger.error(
307 "Task kafka_read task exit error too many errors. Exception: {}".format(
308 e
309 )
310 )
tiernoc0e42e22018-05-11 11:36:10 +0200311 raise
312 consecutive_errors += 1
garciadeblas5697b8b2021-03-24 09:17:02 +0100313 self.logger.error(
314 "Task kafka_read retrying after Exception {}".format(e)
315 )
tierno16427352019-04-22 11:37:36 +0000316 wait_time = 2 if not first_start else 5
tiernoc0e42e22018-05-11 11:36:10 +0200317 await asyncio.sleep(wait_time, loop=self.loop)
318
gcalvinoed7f6d42018-12-14 14:44:56 +0100319 def kafka_read_callback(self, topic, command, params):
320 order_id = 1
321
322 if topic != "admin" and command != "ping":
garciadeblas5697b8b2021-03-24 09:17:02 +0100323 self.logger.debug(
324 "Task kafka_read receives {} {}: {}".format(topic, command, params)
325 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100326 self.consecutive_errors = 0
327 self.first_start = False
328 order_id += 1
329 if command == "exit":
330 raise LcmExceptionExit
331 elif command.startswith("#"):
332 return
333 elif command == "echo":
334 # just for test
335 print(params)
336 sys.stdout.flush()
337 return
338 elif command == "test":
339 asyncio.Task(self.test(params), loop=self.loop)
340 return
341
342 if topic == "admin":
343 if command == "ping" and params["to"] == "lcm" and params["from"] == "lcm":
tierno16427352019-04-22 11:37:36 +0000344 if params.get("worker_id") != self.worker_id:
345 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100346 self.pings_not_received = 0
tierno3e359b12019-02-03 02:29:13 +0100347 try:
348 with open(health_check_file, "w") as f:
349 f.write(str(time()))
350 except Exception as e:
garciadeblas5697b8b2021-03-24 09:17:02 +0100351 self.logger.error(
352 "Cannot write into '{}' for healthcheck: {}".format(
353 health_check_file, e
354 )
355 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100356 return
magnussonle9198bb2020-01-21 13:00:51 +0100357 elif topic == "pla":
358 if command == "placement":
359 self.ns.update_nsrs_with_pla_result(params)
360 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100361 elif topic == "k8scluster":
362 if command == "create" or command == "created":
363 k8scluster_id = params.get("_id")
364 task = asyncio.ensure_future(self.k8scluster.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100365 self.lcm_tasks.register(
366 "k8scluster", k8scluster_id, order_id, "k8scluster_create", task
367 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100368 return
369 elif command == "delete" or command == "deleted":
370 k8scluster_id = params.get("_id")
371 task = asyncio.ensure_future(self.k8scluster.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100372 self.lcm_tasks.register(
373 "k8scluster", k8scluster_id, order_id, "k8scluster_delete", task
374 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100375 return
David Garciac1fe90a2021-03-31 19:12:02 +0200376 elif topic == "vca":
377 if command == "create" or command == "created":
378 vca_id = params.get("_id")
379 task = asyncio.ensure_future(self.vca.create(params, order_id))
380 self.lcm_tasks.register("vca", vca_id, order_id, "vca_create", task)
381 return
382 elif command == "delete" or command == "deleted":
383 vca_id = params.get("_id")
384 task = asyncio.ensure_future(self.vca.delete(params, order_id))
385 self.lcm_tasks.register("vca", vca_id, order_id, "vca_delete", task)
386 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100387 elif topic == "k8srepo":
388 if command == "create" or command == "created":
389 k8srepo_id = params.get("_id")
390 self.logger.debug("k8srepo_id = {}".format(k8srepo_id))
391 task = asyncio.ensure_future(self.k8srepo.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100392 self.lcm_tasks.register(
393 "k8srepo", k8srepo_id, order_id, "k8srepo_create", task
394 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100395 return
396 elif command == "delete" or command == "deleted":
397 k8srepo_id = params.get("_id")
398 task = asyncio.ensure_future(self.k8srepo.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100399 self.lcm_tasks.register(
400 "k8srepo", k8srepo_id, order_id, "k8srepo_delete", task
401 )
calvinosanch9f9c6f22019-11-04 13:37:39 +0100402 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100403 elif topic == "ns":
tierno307425f2020-01-26 23:35:59 +0000404 if command == "instantiate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100405 # self.logger.debug("Deploying NS {}".format(nsr_id))
406 nslcmop = params
407 nslcmop_id = nslcmop["_id"]
408 nsr_id = nslcmop["nsInstanceId"]
409 task = asyncio.ensure_future(self.ns.instantiate(nsr_id, nslcmop_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100410 self.lcm_tasks.register(
411 "ns", nsr_id, nslcmop_id, "ns_instantiate", task
412 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100413 return
tierno307425f2020-01-26 23:35:59 +0000414 elif command == "terminate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100415 # self.logger.debug("Deleting NS {}".format(nsr_id))
416 nslcmop = params
417 nslcmop_id = nslcmop["_id"]
418 nsr_id = nslcmop["nsInstanceId"]
419 self.lcm_tasks.cancel(topic, nsr_id)
420 task = asyncio.ensure_future(self.ns.terminate(nsr_id, nslcmop_id))
421 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_terminate", task)
422 return
ksaikiranr3fde2c72021-03-15 10:39:06 +0530423 elif command == "vca_status_refresh":
424 nslcmop = params
425 nslcmop_id = nslcmop["_id"]
426 nsr_id = nslcmop["nsInstanceId"]
garciadeblas5697b8b2021-03-24 09:17:02 +0100427 task = asyncio.ensure_future(
428 self.ns.vca_status_refresh(nsr_id, nslcmop_id)
429 )
430 self.lcm_tasks.register(
431 "ns", nsr_id, nslcmop_id, "ns_vca_status_refresh", task
432 )
ksaikiranr3fde2c72021-03-15 10:39:06 +0530433 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100434 elif command == "action":
435 # self.logger.debug("Update NS {}".format(nsr_id))
436 nslcmop = params
437 nslcmop_id = nslcmop["_id"]
438 nsr_id = nslcmop["nsInstanceId"]
439 task = asyncio.ensure_future(self.ns.action(nsr_id, nslcmop_id))
440 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_action", task)
441 return
aticigdffa6212022-04-12 15:27:53 +0300442 elif command == "update":
443 # self.logger.debug("Update NS {}".format(nsr_id))
444 nslcmop = params
445 nslcmop_id = nslcmop["_id"]
446 nsr_id = nslcmop["nsInstanceId"]
447 task = asyncio.ensure_future(self.ns.update(nsr_id, nslcmop_id))
448 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_update", task)
449 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100450 elif command == "scale":
451 # self.logger.debug("Update NS {}".format(nsr_id))
452 nslcmop = params
453 nslcmop_id = nslcmop["_id"]
454 nsr_id = nslcmop["nsInstanceId"]
455 task = asyncio.ensure_future(self.ns.scale(nsr_id, nslcmop_id))
456 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_scale", task)
457 return
elumalai80bcf1c2022-04-28 18:05:01 +0530458 elif command == "migrate":
459 nslcmop = params
460 nslcmop_id = nslcmop["_id"]
461 nsr_id = nslcmop["nsInstanceId"]
462 task = asyncio.ensure_future(self.ns.migrate(nsr_id, nslcmop_id))
463 self.lcm_tasks.register("ns", nsr_id, nslcmop_id, "ns_migrate", task)
464 return
gcalvinoed7f6d42018-12-14 14:44:56 +0100465 elif command == "show":
tiernoc2564fe2019-01-28 16:18:56 +0000466 nsr_id = params
gcalvinoed7f6d42018-12-14 14:44:56 +0100467 try:
468 db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
garciadeblas5697b8b2021-03-24 09:17:02 +0100469 print(
470 "nsr:\n _id={}\n operational-status: {}\n config-status: {}"
471 "\n detailed-status: {}\n deploy: {}\n tasks: {}"
472 "".format(
473 nsr_id,
474 db_nsr["operational-status"],
475 db_nsr["config-status"],
476 db_nsr["detailed-status"],
477 db_nsr["_admin"]["deployed"],
478 self.lcm_ns_tasks.get(nsr_id),
479 )
480 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100481 except Exception as e:
482 print("nsr {} not found: {}".format(nsr_id, e))
483 sys.stdout.flush()
484 return
485 elif command == "deleted":
486 return # TODO cleaning of task just in case should be done
garciadeblas5697b8b2021-03-24 09:17:02 +0100487 elif command in (
488 "terminated",
489 "instantiated",
490 "scaled",
491 "actioned",
aticigdffa6212022-04-12 15:27:53 +0300492 "updated",
elumalai80bcf1c2022-04-28 18:05:01 +0530493 "migrated",
garciadeblas5697b8b2021-03-24 09:17:02 +0100494 ): # "scaled-cooldown-time"
gcalvinoed7f6d42018-12-14 14:44:56 +0100495 return
496 elif topic == "nsi": # netslice LCM processes (instantiate, terminate, etc)
tierno307425f2020-01-26 23:35:59 +0000497 if command == "instantiate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100498 # self.logger.debug("Instantiating Network Slice {}".format(nsilcmop["netsliceInstanceId"]))
499 nsilcmop = params
500 nsilcmop_id = nsilcmop["_id"] # slice operation id
501 nsir_id = nsilcmop["netsliceInstanceId"] # slice record id
garciadeblas5697b8b2021-03-24 09:17:02 +0100502 task = asyncio.ensure_future(
503 self.netslice.instantiate(nsir_id, nsilcmop_id)
504 )
505 self.lcm_tasks.register(
506 "nsi", nsir_id, nsilcmop_id, "nsi_instantiate", task
507 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100508 return
tierno307425f2020-01-26 23:35:59 +0000509 elif command == "terminate":
gcalvinoed7f6d42018-12-14 14:44:56 +0100510 # self.logger.debug("Terminating Network Slice NS {}".format(nsilcmop["netsliceInstanceId"]))
511 nsilcmop = params
512 nsilcmop_id = nsilcmop["_id"] # slice operation id
513 nsir_id = nsilcmop["netsliceInstanceId"] # slice record id
514 self.lcm_tasks.cancel(topic, nsir_id)
garciadeblas5697b8b2021-03-24 09:17:02 +0100515 task = asyncio.ensure_future(
516 self.netslice.terminate(nsir_id, nsilcmop_id)
517 )
518 self.lcm_tasks.register(
519 "nsi", nsir_id, nsilcmop_id, "nsi_terminate", task
520 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100521 return
522 elif command == "show":
tiernoc2564fe2019-01-28 16:18:56 +0000523 nsir_id = params
gcalvinoed7f6d42018-12-14 14:44:56 +0100524 try:
525 db_nsir = self.db.get_one("nsirs", {"_id": nsir_id})
garciadeblas5697b8b2021-03-24 09:17:02 +0100526 print(
527 "nsir:\n _id={}\n operational-status: {}\n config-status: {}"
528 "\n detailed-status: {}\n deploy: {}\n tasks: {}"
529 "".format(
530 nsir_id,
531 db_nsir["operational-status"],
532 db_nsir["config-status"],
533 db_nsir["detailed-status"],
534 db_nsir["_admin"]["deployed"],
535 self.lcm_netslice_tasks.get(nsir_id),
536 )
537 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100538 except Exception as e:
539 print("nsir {} not found: {}".format(nsir_id, e))
540 sys.stdout.flush()
541 return
542 elif command == "deleted":
543 return # TODO cleaning of task just in case should be done
garciadeblas5697b8b2021-03-24 09:17:02 +0100544 elif command in (
545 "terminated",
546 "instantiated",
547 "scaled",
548 "actioned",
549 ): # "scaled-cooldown-time"
gcalvinoed7f6d42018-12-14 14:44:56 +0100550 return
551 elif topic == "vim_account":
552 vim_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000553 if command in ("create", "created"):
tierno2357f4e2020-10-19 16:38:59 +0000554 if not self.config["ro_config"].get("ng"):
555 task = asyncio.ensure_future(self.vim.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100556 self.lcm_tasks.register(
557 "vim_account", vim_id, order_id, "vim_create", task
558 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100559 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100560 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100561 self.lcm_tasks.cancel(topic, vim_id)
kuuse6a470c62019-07-10 13:52:45 +0200562 task = asyncio.ensure_future(self.vim.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100563 self.lcm_tasks.register(
564 "vim_account", vim_id, order_id, "vim_delete", task
565 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100566 return
567 elif command == "show":
568 print("not implemented show with vim_account")
569 sys.stdout.flush()
570 return
tiernof210c1c2019-10-16 09:09:58 +0000571 elif command in ("edit", "edited"):
tierno2357f4e2020-10-19 16:38:59 +0000572 if not self.config["ro_config"].get("ng"):
573 task = asyncio.ensure_future(self.vim.edit(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100574 self.lcm_tasks.register(
575 "vim_account", vim_id, order_id, "vim_edit", task
576 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100577 return
tiernof210c1c2019-10-16 09:09:58 +0000578 elif command == "deleted":
579 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100580 elif topic == "wim_account":
581 wim_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000582 if command in ("create", "created"):
tierno2357f4e2020-10-19 16:38:59 +0000583 if not self.config["ro_config"].get("ng"):
584 task = asyncio.ensure_future(self.wim.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100585 self.lcm_tasks.register(
586 "wim_account", wim_id, order_id, "wim_create", task
587 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100588 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100589 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100590 self.lcm_tasks.cancel(topic, wim_id)
kuuse6a470c62019-07-10 13:52:45 +0200591 task = asyncio.ensure_future(self.wim.delete(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100592 self.lcm_tasks.register(
593 "wim_account", wim_id, order_id, "wim_delete", task
594 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100595 return
596 elif command == "show":
597 print("not implemented show with wim_account")
598 sys.stdout.flush()
599 return
tiernof210c1c2019-10-16 09:09:58 +0000600 elif command in ("edit", "edited"):
gcalvinoed7f6d42018-12-14 14:44:56 +0100601 task = asyncio.ensure_future(self.wim.edit(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100602 self.lcm_tasks.register(
603 "wim_account", wim_id, order_id, "wim_edit", task
604 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100605 return
tiernof210c1c2019-10-16 09:09:58 +0000606 elif command == "deleted":
607 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100608 elif topic == "sdn":
609 _sdn_id = params["_id"]
tiernof210c1c2019-10-16 09:09:58 +0000610 if command in ("create", "created"):
tierno2357f4e2020-10-19 16:38:59 +0000611 if not self.config["ro_config"].get("ng"):
612 task = asyncio.ensure_future(self.sdn.create(params, order_id))
garciadeblas5697b8b2021-03-24 09:17:02 +0100613 self.lcm_tasks.register(
614 "sdn", _sdn_id, order_id, "sdn_create", task
615 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100616 return
calvinosanch9f9c6f22019-11-04 13:37:39 +0100617 elif command == "delete" or command == "deleted":
gcalvinoed7f6d42018-12-14 14:44:56 +0100618 self.lcm_tasks.cancel(topic, _sdn_id)
kuuse6a470c62019-07-10 13:52:45 +0200619 task = asyncio.ensure_future(self.sdn.delete(params, order_id))
gcalvinoed7f6d42018-12-14 14:44:56 +0100620 self.lcm_tasks.register("sdn", _sdn_id, order_id, "sdn_delete", task)
621 return
tiernof210c1c2019-10-16 09:09:58 +0000622 elif command in ("edit", "edited"):
gcalvinoed7f6d42018-12-14 14:44:56 +0100623 task = asyncio.ensure_future(self.sdn.edit(params, order_id))
624 self.lcm_tasks.register("sdn", _sdn_id, order_id, "sdn_edit", task)
625 return
tiernof210c1c2019-10-16 09:09:58 +0000626 elif command == "deleted":
627 return # TODO cleaning of task just in case should be done
gcalvinoed7f6d42018-12-14 14:44:56 +0100628 self.logger.critical("unknown topic {} and command '{}'".format(topic, command))
629
tiernoc0e42e22018-05-11 11:36:10 +0200630 async def kafka_read(self):
garciadeblas5697b8b2021-03-24 09:17:02 +0100631 self.logger.debug(
632 "Task kafka_read Enter with worker_id={}".format(self.worker_id)
633 )
tiernoc0e42e22018-05-11 11:36:10 +0200634 # future = asyncio.Future()
gcalvinoed7f6d42018-12-14 14:44:56 +0100635 self.consecutive_errors = 0
636 self.first_start = True
637 while self.consecutive_errors < 10:
tiernoc0e42e22018-05-11 11:36:10 +0200638 try:
garciadeblas5697b8b2021-03-24 09:17:02 +0100639 topics = (
640 "ns",
641 "vim_account",
642 "wim_account",
643 "sdn",
644 "nsi",
645 "k8scluster",
646 "vca",
647 "k8srepo",
648 "pla",
649 )
650 topics_admin = ("admin",)
tierno16427352019-04-22 11:37:36 +0000651 await asyncio.gather(
garciadeblas5697b8b2021-03-24 09:17:02 +0100652 self.msg.aioread(
653 topics, self.loop, self.kafka_read_callback, from_beginning=True
654 ),
655 self.msg_admin.aioread(
656 topics_admin,
657 self.loop,
658 self.kafka_read_callback,
659 group_id=False,
660 ),
tierno16427352019-04-22 11:37:36 +0000661 )
tiernoc0e42e22018-05-11 11:36:10 +0200662
gcalvinoed7f6d42018-12-14 14:44:56 +0100663 except LcmExceptionExit:
664 self.logger.debug("Bye!")
665 break
tiernoc0e42e22018-05-11 11:36:10 +0200666 except Exception as e:
667 # if not first_start is the first time after starting. So leave more time and wait
668 # to allow kafka starts
gcalvinoed7f6d42018-12-14 14:44:56 +0100669 if self.consecutive_errors == 8 if not self.first_start else 30:
garciadeblas5697b8b2021-03-24 09:17:02 +0100670 self.logger.error(
671 "Task kafka_read task exit error too many errors. Exception: {}".format(
672 e
673 )
674 )
tiernoc0e42e22018-05-11 11:36:10 +0200675 raise
gcalvinoed7f6d42018-12-14 14:44:56 +0100676 self.consecutive_errors += 1
garciadeblas5697b8b2021-03-24 09:17:02 +0100677 self.logger.error(
678 "Task kafka_read retrying after Exception {}".format(e)
679 )
gcalvinoed7f6d42018-12-14 14:44:56 +0100680 wait_time = 2 if not self.first_start else 5
tiernoc0e42e22018-05-11 11:36:10 +0200681 await asyncio.sleep(wait_time, loop=self.loop)
682
683 # self.logger.debug("Task kafka_read terminating")
684 self.logger.debug("Task kafka_read exit")
685
686 def start(self):
tierno22f4f9c2018-06-11 18:53:39 +0200687
688 # check RO version
689 self.loop.run_until_complete(self.check_RO_version())
690
aticig15db6142022-01-24 12:51:26 +0300691 self.ns = ns.NsLcm(self.msg, self.lcm_tasks, self.config, self.loop)
garciadeblas5697b8b2021-03-24 09:17:02 +0100692 self.netslice = netslice.NetsliceLcm(
693 self.msg, self.lcm_tasks, self.config, self.loop, self.ns
694 )
bravof922c4172020-11-24 21:21:43 -0300695 self.vim = vim_sdn.VimLcm(self.msg, self.lcm_tasks, self.config, self.loop)
696 self.wim = vim_sdn.WimLcm(self.msg, self.lcm_tasks, self.config, self.loop)
697 self.sdn = vim_sdn.SdnLcm(self.msg, self.lcm_tasks, self.config, self.loop)
garciadeblas5697b8b2021-03-24 09:17:02 +0100698 self.k8scluster = vim_sdn.K8sClusterLcm(
699 self.msg, self.lcm_tasks, self.config, self.loop
700 )
David Garciac1fe90a2021-03-31 19:12:02 +0200701 self.vca = vim_sdn.VcaLcm(self.msg, self.lcm_tasks, self.config, self.loop)
garciadeblas5697b8b2021-03-24 09:17:02 +0100702 self.k8srepo = vim_sdn.K8sRepoLcm(
703 self.msg, self.lcm_tasks, self.config, self.loop
704 )
tierno2357f4e2020-10-19 16:38:59 +0000705
garciadeblas5697b8b2021-03-24 09:17:02 +0100706 self.loop.run_until_complete(
707 asyncio.gather(self.kafka_read(), self.kafka_ping())
708 )
bravof73bac502021-05-11 07:38:47 -0400709
tiernoc0e42e22018-05-11 11:36:10 +0200710 # TODO
711 # self.logger.debug("Terminating cancelling creation tasks")
tiernoca2e16a2018-06-29 15:25:24 +0200712 # self.lcm_tasks.cancel("ALL", "create")
tiernoc0e42e22018-05-11 11:36:10 +0200713 # timeout = 200
714 # while self.is_pending_tasks():
715 # self.logger.debug("Task kafka_read terminating. Waiting for tasks termination")
716 # await asyncio.sleep(2, loop=self.loop)
717 # timeout -= 2
718 # if not timeout:
tiernoca2e16a2018-06-29 15:25:24 +0200719 # self.lcm_tasks.cancel("ALL", "ALL")
tiernoc0e42e22018-05-11 11:36:10 +0200720 self.loop.close()
721 self.loop = None
722 if self.db:
723 self.db.db_disconnect()
724 if self.msg:
725 self.msg.disconnect()
tierno16427352019-04-22 11:37:36 +0000726 if self.msg_admin:
727 self.msg_admin.disconnect()
tiernoc0e42e22018-05-11 11:36:10 +0200728 if self.fs:
729 self.fs.fs_disconnect()
730
tiernoc0e42e22018-05-11 11:36:10 +0200731 def read_config_file(self, config_file):
732 # TODO make a [ini] + yaml inside parser
733 # the configparser library is not suitable, because it does not admit comments at the end of line,
734 # and not parse integer or boolean
735 try:
tierno744303e2020-01-13 16:46:31 +0000736 # read file as yaml format
tiernoc0e42e22018-05-11 11:36:10 +0200737 with open(config_file) as f:
tiernoda6fb102019-11-23 00:36:52 +0000738 conf = yaml.load(f, Loader=yaml.Loader)
tierno744303e2020-01-13 16:46:31 +0000739 # Ensure all sections are not empty
garciadeblas5697b8b2021-03-24 09:17:02 +0100740 for k in (
741 "global",
742 "timeout",
743 "RO",
744 "VCA",
745 "database",
746 "storage",
747 "message",
748 ):
tierno744303e2020-01-13 16:46:31 +0000749 if not conf.get(k):
750 conf[k] = {}
751
752 # read all environ that starts with OSMLCM_
tiernoc0e42e22018-05-11 11:36:10 +0200753 for k, v in environ.items():
754 if not k.startswith("OSMLCM_"):
755 continue
tierno744303e2020-01-13 16:46:31 +0000756 subject, _, item = k[7:].lower().partition("_")
757 if not item:
tierno17a612f2018-10-23 11:30:42 +0200758 continue
tierno744303e2020-01-13 16:46:31 +0000759 if subject in ("ro", "vca"):
tierno17a612f2018-10-23 11:30:42 +0200760 # put in capital letter
tierno744303e2020-01-13 16:46:31 +0000761 subject = subject.upper()
tiernoc0e42e22018-05-11 11:36:10 +0200762 try:
tierno744303e2020-01-13 16:46:31 +0000763 if item == "port" or subject == "timeout":
764 conf[subject][item] = int(v)
tiernoc0e42e22018-05-11 11:36:10 +0200765 else:
tierno744303e2020-01-13 16:46:31 +0000766 conf[subject][item] = v
tiernoc0e42e22018-05-11 11:36:10 +0200767 except Exception as e:
garciadeblas5697b8b2021-03-24 09:17:02 +0100768 self.logger.warning(
769 "skipping environ '{}' on exception '{}'".format(k, e)
770 )
tierno744303e2020-01-13 16:46:31 +0000771
772 # backward compatibility of VCA parameters
773
garciadeblas5697b8b2021-03-24 09:17:02 +0100774 if "pubkey" in conf["VCA"]:
775 conf["VCA"]["public_key"] = conf["VCA"].pop("pubkey")
776 if "cacert" in conf["VCA"]:
777 conf["VCA"]["ca_cert"] = conf["VCA"].pop("cacert")
778 if "apiproxy" in conf["VCA"]:
779 conf["VCA"]["api_proxy"] = conf["VCA"].pop("apiproxy")
tierno744303e2020-01-13 16:46:31 +0000780
garciadeblas5697b8b2021-03-24 09:17:02 +0100781 if "enableosupgrade" in conf["VCA"]:
782 conf["VCA"]["enable_os_upgrade"] = conf["VCA"].pop("enableosupgrade")
783 if isinstance(conf["VCA"].get("enable_os_upgrade"), str):
784 if conf["VCA"]["enable_os_upgrade"].lower() == "false":
785 conf["VCA"]["enable_os_upgrade"] = False
786 elif conf["VCA"]["enable_os_upgrade"].lower() == "true":
787 conf["VCA"]["enable_os_upgrade"] = True
tierno744303e2020-01-13 16:46:31 +0000788
garciadeblas5697b8b2021-03-24 09:17:02 +0100789 if "aptmirror" in conf["VCA"]:
790 conf["VCA"]["apt_mirror"] = conf["VCA"].pop("aptmirror")
tiernoc0e42e22018-05-11 11:36:10 +0200791
792 return conf
793 except Exception as e:
794 self.logger.critical("At config file '{}': {}".format(config_file, e))
795 exit(1)
796
tierno16427352019-04-22 11:37:36 +0000797 @staticmethod
798 def get_process_id():
799 """
800 Obtain a unique ID for this process. If running from inside docker, it will get docker ID. If not it
801 will provide a random one
802 :return: Obtained ID
803 """
804 # Try getting docker id. If fails, get pid
805 try:
806 with open("/proc/self/cgroup", "r") as f:
807 text_id_ = f.readline()
808 _, _, text_id = text_id_.rpartition("/")
garciadeblas5697b8b2021-03-24 09:17:02 +0100809 text_id = text_id.replace("\n", "")[:12]
tierno16427352019-04-22 11:37:36 +0000810 if text_id:
811 return text_id
812 except Exception:
813 pass
814 # Return a random id
garciadeblas5697b8b2021-03-24 09:17:02 +0100815 return "".join(random_choice("0123456789abcdef") for _ in range(12))
tierno16427352019-04-22 11:37:36 +0000816
tiernoc0e42e22018-05-11 11:36:10 +0200817
tierno275411e2018-05-16 14:33:32 +0200818def usage():
garciadeblas5697b8b2021-03-24 09:17:02 +0100819 print(
820 """Usage: {} [options]
quilesj7e13aeb2019-10-08 13:34:55 +0200821 -c|--config [configuration_file]: loads the configuration file (default: ./lcm.cfg)
tiernoa9843d82018-10-24 10:44:20 +0200822 --health-check: do not run lcm, but inspect kafka bus to determine if lcm is healthy
tierno275411e2018-05-16 14:33:32 +0200823 -h|--help: shows this help
garciadeblas5697b8b2021-03-24 09:17:02 +0100824 """.format(
825 sys.argv[0]
826 )
827 )
tierno750b2452018-05-17 16:39:29 +0200828 # --log-socket-host HOST: send logs to this host")
829 # --log-socket-port PORT: send logs using this port (default: 9022)")
tierno275411e2018-05-16 14:33:32 +0200830
831
garciadeblas5697b8b2021-03-24 09:17:02 +0100832if __name__ == "__main__":
quilesj7e13aeb2019-10-08 13:34:55 +0200833
tierno275411e2018-05-16 14:33:32 +0200834 try:
tierno8c16b052020-02-05 15:08:32 +0000835 # print("SYS.PATH='{}'".format(sys.path))
tierno275411e2018-05-16 14:33:32 +0200836 # load parameters and configuration
quilesj7e13aeb2019-10-08 13:34:55 +0200837 # -h
838 # -c value
839 # --config value
840 # --help
841 # --health-check
garciadeblas5697b8b2021-03-24 09:17:02 +0100842 opts, args = getopt.getopt(
843 sys.argv[1:], "hc:", ["config=", "help", "health-check"]
844 )
tierno275411e2018-05-16 14:33:32 +0200845 # TODO add "log-socket-host=", "log-socket-port=", "log-file="
846 config_file = None
847 for o, a in opts:
848 if o in ("-h", "--help"):
849 usage()
850 sys.exit()
851 elif o in ("-c", "--config"):
852 config_file = a
tiernoa9843d82018-10-24 10:44:20 +0200853 elif o == "--health-check":
tierno94f06112020-02-11 12:38:19 +0000854 from osm_lcm.lcm_hc import health_check
garciadeblas5697b8b2021-03-24 09:17:02 +0100855
tierno94f06112020-02-11 12:38:19 +0000856 health_check(health_check_file, Lcm.ping_interval_pace)
tierno275411e2018-05-16 14:33:32 +0200857 # elif o == "--log-socket-port":
858 # log_socket_port = a
859 # elif o == "--log-socket-host":
860 # log_socket_host = a
861 # elif o == "--log-file":
862 # log_file = a
863 else:
864 assert False, "Unhandled option"
quilesj7e13aeb2019-10-08 13:34:55 +0200865
tierno275411e2018-05-16 14:33:32 +0200866 if config_file:
867 if not path.isfile(config_file):
garciadeblas5697b8b2021-03-24 09:17:02 +0100868 print(
869 "configuration file '{}' does not exist".format(config_file),
870 file=sys.stderr,
871 )
tierno275411e2018-05-16 14:33:32 +0200872 exit(1)
873 else:
garciadeblas5697b8b2021-03-24 09:17:02 +0100874 for config_file in (
875 __file__[: __file__.rfind(".")] + ".cfg",
876 "./lcm.cfg",
877 "/etc/osm/lcm.cfg",
878 ):
tierno275411e2018-05-16 14:33:32 +0200879 if path.isfile(config_file):
880 break
881 else:
garciadeblas5697b8b2021-03-24 09:17:02 +0100882 print(
883 "No configuration file 'lcm.cfg' found neither at local folder nor at /etc/osm/",
884 file=sys.stderr,
885 )
tierno275411e2018-05-16 14:33:32 +0200886 exit(1)
887 lcm = Lcm(config_file)
tierno3e359b12019-02-03 02:29:13 +0100888 lcm.start()
tierno22f4f9c2018-06-11 18:53:39 +0200889 except (LcmException, getopt.GetoptError) as e:
tierno275411e2018-05-16 14:33:32 +0200890 print(str(e), file=sys.stderr)
891 # usage()
892 exit(1)