o2app/entrypoints/resource_watcher.py

   1 # Copyright (C) 2021 Wind River Systems, Inc.
   2 #
   3 #  Licensed under the Apache License, Version 2.0 (the "License");
   4 #  you may not use this file except in compliance with the License.
   5 #  You may obtain a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 #  Unless required by applicable law or agreed to in writing, software
  10 #  distributed under the License is distributed on an "AS IS" BASIS,
  11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 #  See the License for the specific language governing permissions and
  13 #  limitations under the License.
  14
  15 import cotyledon
  16
  17 from o2app import bootstrap
  18 from o2common.service.watcher.base import WatcherTree
  19 from o2common.service.watcher.worker import PollWorker
  20
  21 from o2ims.service.watcher.ocloud_watcher import OcloudWatcher
  22 from o2ims.service.watcher.ocloud_watcher import DmsWatcher
  23 from o2ims.service.watcher.resourcepool_watcher import ResourcePoolWatcher
  24 from o2ims.service.watcher.alarm_watcher import AlarmWatcher
  25
  26 from o2ims.adapter.clients.ocloud_client import StxDmsClient
  27 from o2ims.adapter.clients.ocloud_client import StxOcloudClient
  28 from o2ims.adapter.clients.ocloud_client import StxResourcePoolClient
  29 from o2ims.adapter.clients.fault_client import StxAlarmClient
  30
  31 from o2ims.service.watcher.pserver_watcher import PServerWatcher
  32 from o2ims.adapter.clients.ocloud_client import StxPserverClient
  33
  34 from o2ims.service.watcher.pserver_cpu_watcher import PServerCpuWatcher
  35 from o2ims.adapter.clients.ocloud_client import StxCpuClient
  36
  37 from o2ims.service.watcher.pserver_mem_watcher import PServerMemWatcher
  38 from o2ims.adapter.clients.ocloud_client import StxMemClient
  39
  40 from o2ims.service.watcher.pserver_if_watcher import PServerIfWatcher
  41 from o2ims.adapter.clients.ocloud_client import StxIfClient
  42
  43 # from o2ims.service.watcher.pserver_port_watcher import PServerIfPortWatcher
  44 # from o2ims.adapter.clients.ocloud_client import StxIfPortClient
  45
  46 from o2ims.service.watcher.pserver_eth_watcher import PServerEthWatcher
  47 from o2ims.adapter.clients.ocloud_client import StxEthClient
  48
  49 # from o2ims.service.watcher.pserver_dev_watcher import PServerDevWatcher
  50 # from o2ims.adapter.clients.ocloud_client import StxDevClient
  51 from o2ims.service.watcher.pserver_acc_watcher import PServerAccWatcher
  52 from o2ims.adapter.clients.ocloud_client import StxAccClient
  53
  54 from o2ims.adapter.clients.alarm_dict_client import load_alarm_definition,\
  55     load_alarm_dictionary_from_conf_file
  56 from o2ims.service.watcher.agg_compute_watcher import ComputeAggWatcher
  57 from o2ims.service.watcher.agg_network_watcher import NetworkAggWatcher
  58 from o2ims.service.watcher.agg_storage_watcher import StorageAggWatcher
  59 from o2ims.service.watcher.agg_undefined_watcher import UndefinedAggWatcher
  60 from o2ims.adapter.clients.aggregate_client import ComputeAggClient, \
  61     NetworkAggClient, StorageAggClient, UndefinedAggClient
  62
  63 from o2common.helper import o2logging
  64 logger = o2logging.get_logger(__name__)
  65
  66 # r = redis.Redis(**config.get_redis_host_and_port())
  67
  68
  69 class WatcherService(cotyledon.Service):
  70     def __init__(self, worker_id, args=None) -> None:
  71         super().__init__(worker_id)
  72         self.args = args
  73         self.bus = bootstrap.bootstrap()
  74         self.worker = PollWorker(bus=self.bus)
  75         load_alarm_definition(self.bus.uow)
  76         load_alarm_dictionary_from_conf_file(self.bus.uow)
  77
  78     def run(self):
  79         try:
  80             root = WatcherTree(OcloudWatcher(
  81                 StxOcloudClient(), self.bus))
  82             root.addchild(
  83                 DmsWatcher(StxDmsClient(), self.bus))
  84             # root.addchild(
  85             #     AlarmWatcher(StxFaultClient(), self.bus))
  86
  87             child_respool = root.addchild(
  88                 ResourcePoolWatcher(StxResourcePoolClient(),
  89                                     self.bus))
  90
  91             # Add Aggregate watch
  92             child_respool.addchild(
  93                 ComputeAggWatcher(ComputeAggClient(), self.bus))
  94             child_respool.addchild(
  95                 NetworkAggWatcher(NetworkAggClient(), self.bus))
  96             child_respool.addchild(
  97                 StorageAggWatcher(StorageAggClient(), self.bus))
  98             child_respool.addchild(
  99                 UndefinedAggWatcher(UndefinedAggClient(), self.bus))
 100
 101             # Add Resource watch
 102             child_pserver = child_respool.addchild(
 103                 PServerWatcher(StxPserverClient(), self.bus))
 104             child_pserver.addchild(
 105                 PServerCpuWatcher(StxCpuClient(), self.bus))
 106             child_pserver.addchild(
 107                 PServerMemWatcher(StxMemClient(), self.bus))
 108             child_pserver.addchild(
 109                 PServerEthWatcher(StxEthClient(), self.bus))
 110             child_pserver.addchild(
 111                 PServerIfWatcher(StxIfClient(), self.bus))
 112             # child_if.addchild(
 113             #     PServerIfPortWatcher(StxIfPortClient(), self.bus))
 114             # child_pserver.addchild(
 115             #     PServerDevWatcher(StxDevClient(), self.bus))
 116             child_pserver.addchild(
 117                 PServerAccWatcher(StxAccClient(), self.bus))
 118
 119             # Add Alarm watch
 120             child_respool.addchild(
 121                 AlarmWatcher(StxAlarmClient(self.bus.uow), self.bus))
 122
 123             self.worker.add_watcher(root)
 124
 125             self.worker.start()
 126         except Exception as ex:
 127             logger.warning("WorkerService Exception:" + str(ex))
 128         finally:
 129             self.worker.stop()
 130
 131
 132 def start_watchers(sm: cotyledon.ServiceManager = None):
 133     watchersm = sm if sm else cotyledon.ServiceManager()
 134     watchersm.add(WatcherService, workers=1, args=())
 135     watchersm.run()
 136
 137
 138 def main():
 139     logger.info("Resource watcher starting")
 140     start_watchers()
 141
 142
 143 if __name__ == "__main__":
 144     main()