731488cf92119bedcd0bd77d06a4bedfec478fd8
[osm/LCM.git] / osm_lcm / lcm.py
1 #!/usr/bin/python3
2 # -*- coding: utf-8 -*-
3
4 ##
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
20
21 # DEBUG WITH PDB
22 import os
23 import pdb
24
25 import asyncio
26 import yaml
27 import logging
28 import logging.handlers
29 import getopt
30 import sys
31
32 from osm_lcm import ns, vim_sdn, netslice
33 from osm_lcm.ng_ro import NgRoException, NgRoClient
34 from osm_lcm.ROclient import ROClient, ROClientException
35
36 from time import time
37 from osm_lcm.lcm_utils import versiontuple, LcmException, TaskRegistry, LcmExceptionExit
38 from osm_lcm import version as lcm_version, version_date as lcm_version_date
39
40 from osm_common import msglocal, msgkafka
41 from osm_common import version as common_version
42 from osm_common.dbbase import DbException
43 from osm_common.fsbase import FsException
44 from osm_common.msgbase import MsgException
45 from osm_lcm.data_utils.database.database import Database
46 from osm_lcm.data_utils.filesystem.filesystem import Filesystem
47 from os import environ, path
48 from random import choice as random_choice
49 from n2vc import version as n2vc_version
50 import traceback
51
52 if os.getenv("OSMLCM_PDB_DEBUG", None) is not None:
53 pdb.set_trace()
54
55
56 __author__ = "Alfonso Tierno"
57 min_RO_version = "6.0.2"
58 min_n2vc_version = "0.0.2"
59
60 min_common_version = "0.1.19"
61 health_check_file = (
62 path.expanduser("~") + "/time_last_ping"
63 ) # TODO find better location for this file
64
65
66 class Lcm:
67
68 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 }
78 # ^ contains for each section at lcm.cfg the used logger name
79
80 def __init__(self, config_file, loop=None):
81 """
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 """
86 self.db = None
87 self.msg = None
88 self.msg_admin = None
89 self.fs = None
90 self.pings_not_received = 1
91 self.consecutive_errors = 0
92 self.first_start = False
93
94 # logging
95 self.logger = logging.getLogger("lcm")
96 # get id
97 self.worker_id = self.get_process_id()
98 # load configuration
99 config = self.read_config_file(config_file)
100 self.config = config
101 self.config["ro_config"] = {
102 "ng": config["RO"].get("ng", False),
103 "uri": config["RO"].get("uri"),
104 "tenant": config.get("tenant", "osm"),
105 "logger_name": "lcm.roclient",
106 "loglevel": config["RO"].get("loglevel", "ERROR"),
107 }
108 if not self.config["ro_config"]["uri"]:
109 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 ):
116 # uri ends with '/ro', '/ro/', '/openmano', '/openmano/'
117 index = self.config["ro_config"]["uri"][-1].rfind("/")
118 self.config["ro_config"]["uri"] = self.config["ro_config"]["uri"][index + 1]
119
120 self.loop = loop or asyncio.get_event_loop()
121 self.ns = (
122 self.netslice
123 ) = (
124 self.vim
125 ) = self.wim = self.sdn = self.k8scluster = self.vca = self.k8srepo = None
126
127 # logging
128 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 )
134 config["database"]["logger_name"] = "lcm.db"
135 config["storage"]["logger_name"] = "lcm.fs"
136 config["message"]["logger_name"] = "lcm.msg"
137 if config["global"].get("logfile"):
138 file_handler = logging.handlers.RotatingFileHandler(
139 config["global"]["logfile"], maxBytes=100e6, backupCount=9, delay=0
140 )
141 file_handler.setFormatter(log_formatter_simple)
142 self.logger.addHandler(file_handler)
143 if not config["global"].get("nologging"):
144 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
152 for k1, logname in self.cfg_logger_name.items():
153 config[k1]["logger_name"] = logname
154 logger_module = logging.getLogger(logname)
155 if config[k1].get("logfile"):
156 file_handler = logging.handlers.RotatingFileHandler(
157 config[k1]["logfile"], maxBytes=100e6, backupCount=9, delay=0
158 )
159 file_handler.setFormatter(log_formatter_simple)
160 logger_module.addHandler(file_handler)
161 if config[k1].get("loglevel"):
162 logger_module.setLevel(config[k1]["loglevel"])
163 self.logger.critical(
164 "starting osm/lcm version {} {}".format(lcm_version, lcm_version_date)
165 )
166
167 # check version of N2VC
168 # TODO enhance with int conversion or from distutils.version import LooseVersion
169 # or with list(map(int, version.split(".")))
170 if versiontuple(n2vc_version) < versiontuple(min_n2vc_version):
171 raise LcmException(
172 "Not compatible osm/N2VC version '{}'. Needed '{}' or higher".format(
173 n2vc_version, min_n2vc_version
174 )
175 )
176 # check version of common
177 if versiontuple(common_version) < versiontuple(min_common_version):
178 raise LcmException(
179 "Not compatible osm/common version '{}'. Needed '{}' or higher".format(
180 common_version, min_common_version
181 )
182 )
183
184 try:
185 self.db = Database(config).instance.db
186
187 self.fs = Filesystem(config).instance.fs
188 self.fs.sync()
189
190 # copy message configuration in order to remove 'group_id' for msg_admin
191 config_message = config["message"].copy()
192 config_message["loop"] = self.loop
193 if config_message["driver"] == "local":
194 self.msg = msglocal.MsgLocal()
195 self.msg.connect(config_message)
196 self.msg_admin = msglocal.MsgLocal()
197 config_message.pop("group_id", None)
198 self.msg_admin.connect(config_message)
199 elif config_message["driver"] == "kafka":
200 self.msg = msgkafka.MsgKafka()
201 self.msg.connect(config_message)
202 self.msg_admin = msgkafka.MsgKafka()
203 config_message.pop("group_id", None)
204 self.msg_admin.connect(config_message)
205 else:
206 raise LcmException(
207 "Invalid configuration param '{}' at '[message]':'driver'".format(
208 config["message"]["driver"]
209 )
210 )
211 except (DbException, FsException, MsgException) as e:
212 self.logger.critical(str(e), exc_info=True)
213 raise LcmException(str(e))
214
215 # contains created tasks/futures to be able to cancel
216 self.lcm_tasks = TaskRegistry(self.worker_id, self.logger)
217
218 async def check_RO_version(self):
219 tries = 14
220 last_error = None
221 while True:
222 ro_uri = self.config["ro_config"]["uri"]
223 try:
224 # try new RO, if fail old RO
225 try:
226 self.config["ro_config"]["uri"] = ro_uri + "ro"
227 ro_server = NgRoClient(self.loop, **self.config["ro_config"])
228 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"
232 ro_server = ROClient(self.loop, **self.config["ro_config"])
233 ro_version = await ro_server.get_version()
234 self.config["ro_config"]["ng"] = False
235 if versiontuple(ro_version) < versiontuple(min_RO_version):
236 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 )
246 return
247 except (ROClientException, NgRoException) as e:
248 self.config["ro_config"]["uri"] = ro_uri
249 tries -= 1
250 traceback.print_tb(e.__traceback__)
251 error_text = "Error while connecting to RO on {}: {}".format(
252 self.config["ro_config"]["uri"], e
253 )
254 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
259 self.logger.error(
260 error_text + ". Waiting until {} seconds".format(5 * tries)
261 )
262 await asyncio.sleep(5)
263
264 async def test(self, param=None):
265 self.logger.debug("Starting/Ending test task: {}".format(param))
266
267 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:
275 await self.msg_admin.aiowrite(
276 "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 )
286 # time between pings are low when it is not received and at starting
287 wait_time = (
288 self.ping_interval_boot
289 if not kafka_has_received
290 else self.ping_interval_pace
291 )
292 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:
306 self.logger.error(
307 "Task kafka_read task exit error too many errors. Exception: {}".format(
308 e
309 )
310 )
311 raise
312 consecutive_errors += 1
313 self.logger.error(
314 "Task kafka_read retrying after Exception {}".format(e)
315 )
316 wait_time = 2 if not first_start else 5
317 await asyncio.sleep(wait_time, loop=self.loop)
318
319 def kafka_read_callback(self, topic, command, params):
320 order_id = 1
321
322 if topic != "admin" and command != "ping":
323 self.logger.debug(
324 "Task kafka_read receives {} {}: {}".format(topic, command, params)
325 )
326 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":
344 if params.get("worker_id") != self.worker_id:
345 return
346 self.pings_not_received = 0
347 try:
348 with open(health_check_file, "w") as f:
349 f.write(str(time()))
350 except Exception as e:
351 self.logger.error(
352 "Cannot write into '{}' for healthcheck: {}".format(
353 health_check_file, e
354 )
355 )
356 return
357 elif topic == "pla":
358 if command == "placement":
359 self.ns.update_nsrs_with_pla_result(params)
360 return
361 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))
365 self.lcm_tasks.register(
366 "k8scluster", k8scluster_id, order_id, "k8scluster_create", task
367 )
368 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))
372 self.lcm_tasks.register(
373 "k8scluster", k8scluster_id, order_id, "k8scluster_delete", task
374 )
375 return
376 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
387 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))
392 self.lcm_tasks.register(
393 "k8srepo", k8srepo_id, order_id, "k8srepo_create", task
394 )
395 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))
399 self.lcm_tasks.register(
400 "k8srepo", k8srepo_id, order_id, "k8srepo_delete", task
401 )
402 return
403 elif topic == "ns":
404 if command == "instantiate":
405 # 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))
410 self.lcm_tasks.register(
411 "ns", nsr_id, nslcmop_id, "ns_instantiate", task
412 )
413 return
414 elif command == "terminate":
415 # 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
423 elif command == "vca_status_refresh":
424 nslcmop = params
425 nslcmop_id = nslcmop["_id"]
426 nsr_id = nslcmop["nsInstanceId"]
427 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 )
433 return
434 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
442 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
450 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
458 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
465 elif command == "show":
466 nsr_id = params
467 try:
468 db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
469 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 )
481 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
487 elif command in (
488 "terminated",
489 "instantiated",
490 "scaled",
491 "actioned",
492 "updated",
493 "migrated",
494 ): # "scaled-cooldown-time"
495 return
496 elif topic == "nsi": # netslice LCM processes (instantiate, terminate, etc)
497 if command == "instantiate":
498 # 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
502 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 )
508 return
509 elif command == "terminate":
510 # 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)
515 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 )
521 return
522 elif command == "show":
523 nsir_id = params
524 try:
525 db_nsir = self.db.get_one("nsirs", {"_id": nsir_id})
526 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 )
538 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
544 elif command in (
545 "terminated",
546 "instantiated",
547 "scaled",
548 "actioned",
549 ): # "scaled-cooldown-time"
550 return
551 elif topic == "vim_account":
552 vim_id = params["_id"]
553 if command in ("create", "created"):
554 if not self.config["ro_config"].get("ng"):
555 task = asyncio.ensure_future(self.vim.create(params, order_id))
556 self.lcm_tasks.register(
557 "vim_account", vim_id, order_id, "vim_create", task
558 )
559 return
560 elif command == "delete" or command == "deleted":
561 self.lcm_tasks.cancel(topic, vim_id)
562 task = asyncio.ensure_future(self.vim.delete(params, order_id))
563 self.lcm_tasks.register(
564 "vim_account", vim_id, order_id, "vim_delete", task
565 )
566 return
567 elif command == "show":
568 print("not implemented show with vim_account")
569 sys.stdout.flush()
570 return
571 elif command in ("edit", "edited"):
572 if not self.config["ro_config"].get("ng"):
573 task = asyncio.ensure_future(self.vim.edit(params, order_id))
574 self.lcm_tasks.register(
575 "vim_account", vim_id, order_id, "vim_edit", task
576 )
577 return
578 elif command == "deleted":
579 return # TODO cleaning of task just in case should be done
580 elif topic == "wim_account":
581 wim_id = params["_id"]
582 if command in ("create", "created"):
583 if not self.config["ro_config"].get("ng"):
584 task = asyncio.ensure_future(self.wim.create(params, order_id))
585 self.lcm_tasks.register(
586 "wim_account", wim_id, order_id, "wim_create", task
587 )
588 return
589 elif command == "delete" or command == "deleted":
590 self.lcm_tasks.cancel(topic, wim_id)
591 task = asyncio.ensure_future(self.wim.delete(params, order_id))
592 self.lcm_tasks.register(
593 "wim_account", wim_id, order_id, "wim_delete", task
594 )
595 return
596 elif command == "show":
597 print("not implemented show with wim_account")
598 sys.stdout.flush()
599 return
600 elif command in ("edit", "edited"):
601 task = asyncio.ensure_future(self.wim.edit(params, order_id))
602 self.lcm_tasks.register(
603 "wim_account", wim_id, order_id, "wim_edit", task
604 )
605 return
606 elif command == "deleted":
607 return # TODO cleaning of task just in case should be done
608 elif topic == "sdn":
609 _sdn_id = params["_id"]
610 if command in ("create", "created"):
611 if not self.config["ro_config"].get("ng"):
612 task = asyncio.ensure_future(self.sdn.create(params, order_id))
613 self.lcm_tasks.register(
614 "sdn", _sdn_id, order_id, "sdn_create", task
615 )
616 return
617 elif command == "delete" or command == "deleted":
618 self.lcm_tasks.cancel(topic, _sdn_id)
619 task = asyncio.ensure_future(self.sdn.delete(params, order_id))
620 self.lcm_tasks.register("sdn", _sdn_id, order_id, "sdn_delete", task)
621 return
622 elif command in ("edit", "edited"):
623 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
626 elif command == "deleted":
627 return # TODO cleaning of task just in case should be done
628 self.logger.critical("unknown topic {} and command '{}'".format(topic, command))
629
630 async def kafka_read(self):
631 self.logger.debug(
632 "Task kafka_read Enter with worker_id={}".format(self.worker_id)
633 )
634 # future = asyncio.Future()
635 self.consecutive_errors = 0
636 self.first_start = True
637 while self.consecutive_errors < 10:
638 try:
639 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",)
651 await asyncio.gather(
652 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 ),
661 )
662
663 except LcmExceptionExit:
664 self.logger.debug("Bye!")
665 break
666 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
669 if self.consecutive_errors == 8 if not self.first_start else 30:
670 self.logger.error(
671 "Task kafka_read task exit error too many errors. Exception: {}".format(
672 e
673 )
674 )
675 raise
676 self.consecutive_errors += 1
677 self.logger.error(
678 "Task kafka_read retrying after Exception {}".format(e)
679 )
680 wait_time = 2 if not self.first_start else 5
681 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):
687
688 # check RO version
689 self.loop.run_until_complete(self.check_RO_version())
690
691 self.ns = ns.NsLcm(self.msg, self.lcm_tasks, self.config, self.loop)
692 self.netslice = netslice.NetsliceLcm(
693 self.msg, self.lcm_tasks, self.config, self.loop, self.ns
694 )
695 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)
698 self.k8scluster = vim_sdn.K8sClusterLcm(
699 self.msg, self.lcm_tasks, self.config, self.loop
700 )
701 self.vca = vim_sdn.VcaLcm(self.msg, self.lcm_tasks, self.config, self.loop)
702 self.k8srepo = vim_sdn.K8sRepoLcm(
703 self.msg, self.lcm_tasks, self.config, self.loop
704 )
705
706 self.loop.run_until_complete(
707 asyncio.gather(self.kafka_read(), self.kafka_ping())
708 )
709
710 # TODO
711 # self.logger.debug("Terminating cancelling creation tasks")
712 # self.lcm_tasks.cancel("ALL", "create")
713 # 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:
719 # self.lcm_tasks.cancel("ALL", "ALL")
720 self.loop.close()
721 self.loop = None
722 if self.db:
723 self.db.db_disconnect()
724 if self.msg:
725 self.msg.disconnect()
726 if self.msg_admin:
727 self.msg_admin.disconnect()
728 if self.fs:
729 self.fs.fs_disconnect()
730
731 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:
736 # read file as yaml format
737 with open(config_file) as f:
738 conf = yaml.load(f, Loader=yaml.Loader)
739 # Ensure all sections are not empty
740 for k in (
741 "global",
742 "timeout",
743 "RO",
744 "VCA",
745 "database",
746 "storage",
747 "message",
748 ):
749 if not conf.get(k):
750 conf[k] = {}
751
752 # read all environ that starts with OSMLCM_
753 for k, v in environ.items():
754 if not k.startswith("OSMLCM_"):
755 continue
756 subject, _, item = k[7:].lower().partition("_")
757 if not item:
758 continue
759 if subject in ("ro", "vca"):
760 # put in capital letter
761 subject = subject.upper()
762 try:
763 if item == "port" or subject == "timeout":
764 conf[subject][item] = int(v)
765 else:
766 conf[subject][item] = v
767 except Exception as e:
768 self.logger.warning(
769 "skipping environ '{}' on exception '{}'".format(k, e)
770 )
771
772 # backward compatibility of VCA parameters
773
774 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")
780
781 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
788
789 if "aptmirror" in conf["VCA"]:
790 conf["VCA"]["apt_mirror"] = conf["VCA"].pop("aptmirror")
791
792 return conf
793 except Exception as e:
794 self.logger.critical("At config file '{}': {}".format(config_file, e))
795 exit(1)
796
797 @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("/")
809 text_id = text_id.replace("\n", "")[:12]
810 if text_id:
811 return text_id
812 except Exception:
813 pass
814 # Return a random id
815 return "".join(random_choice("0123456789abcdef") for _ in range(12))
816
817
818 def usage():
819 print(
820 """Usage: {} [options]
821 -c|--config [configuration_file]: loads the configuration file (default: ./lcm.cfg)
822 --health-check: do not run lcm, but inspect kafka bus to determine if lcm is healthy
823 -h|--help: shows this help
824 """.format(
825 sys.argv[0]
826 )
827 )
828 # --log-socket-host HOST: send logs to this host")
829 # --log-socket-port PORT: send logs using this port (default: 9022)")
830
831
832 if __name__ == "__main__":
833
834 try:
835 # print("SYS.PATH='{}'".format(sys.path))
836 # load parameters and configuration
837 # -h
838 # -c value
839 # --config value
840 # --help
841 # --health-check
842 opts, args = getopt.getopt(
843 sys.argv[1:], "hc:", ["config=", "help", "health-check"]
844 )
845 # 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
853 elif o == "--health-check":
854 from osm_lcm.lcm_hc import health_check
855
856 health_check(health_check_file, Lcm.ping_interval_pace)
857 # 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"
865
866 if config_file:
867 if not path.isfile(config_file):
868 print(
869 "configuration file '{}' does not exist".format(config_file),
870 file=sys.stderr,
871 )
872 exit(1)
873 else:
874 for config_file in (
875 __file__[: __file__.rfind(".")] + ".cfg",
876 "./lcm.cfg",
877 "/etc/osm/lcm.cfg",
878 ):
879 if path.isfile(config_file):
880 break
881 else:
882 print(
883 "No configuration file 'lcm.cfg' found neither at local folder nor at /etc/osm/",
884 file=sys.stderr,
885 )
886 exit(1)
887 lcm = Lcm(config_file)
888 lcm.start()
889 except (LcmException, getopt.GetoptError) as e:
890 print(str(e), file=sys.stderr)
891 # usage()
892 exit(1)