o2ims/adapter/clients/ocloud_sa_client.py

   1 # Copyright (C) 2021 Wind River Systems, Inc.\r
   2 #\r
   3 #  Licensed under the Apache License, Version 2.0 (the "License");\r
   4 #  you may not use this file except in compliance with the License.\r
   5 #  You may obtain a copy of the License at\r
   6 #\r
   7 #      http://www.apache.org/licenses/LICENSE-2.0\r
   8 #\r
   9 #  Unless required by applicable law or agreed to in writing, software\r
  10 #  distributed under the License is distributed on an "AS IS" BASIS,\r
  11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  12 #  See the License for the specific language governing permissions and\r
  13 #  limitations under the License.\r
  14 \r
  15 # client talking to Stx standalone\r
  16 \r
  17 import uuid\r
  18 from o2common.service.client.base_client import BaseClient\r
  19 from typing import List\r
  20 # Optional,  Set\r
  21 from o2ims.domain import stx_object as ocloudModel\r
  22 from o2common.config import config\r
  23 from o2ims.domain.resource_type import ResourceTypeEnum\r
  24 \r
  25 # from dcmanagerclient.api import client\r
  26 from cgtsclient.client import get_client\r
  27 \r
  28 from o2common.helper import o2logging\r
  29 logger = o2logging.get_logger(__name__)\r
  30 \r
  31 \r
  32 class StxSaOcloudClient(BaseClient):\r
  33     def __init__(self, driver=None):\r
  34         super().__init__()\r
  35         self.driver = driver if driver else StxSaClientImp()\r
  36 \r
  37     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  38         return self.driver.getInstanceInfo()\r
  39 \r
  40     def _list(self, **filters):\r
  41         return [self.driver.getInstanceInfo()]\r
  42 \r
  43 \r
  44 class StxSaResourcePoolClient(BaseClient):\r
  45     def __init__(self):\r
  46         super().__init__()\r
  47         self.driver = StxSaClientImp()\r
  48 \r
  49     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  50         return self.driver.getInstanceInfo()\r
  51 \r
  52     def _list(self, **filters):\r
  53         return [self.driver.getInstanceInfo()]\r
  54 \r
  55 \r
  56 class StxSaDmsClient(BaseClient):\r
  57     def __init__(self):\r
  58         super().__init__()\r
  59         self.driver = StxSaClientImp()\r
  60 \r
  61     def _get(self, name) -> ocloudModel.StxGenericModel:\r
  62         return self.driver.getK8sDetail(name)\r
  63 \r
  64     def _list(self, **filters):\r
  65         return self.driver.getK8sList(**filters)\r
  66 \r
  67 \r
  68 class StxPserverClient(BaseClient):\r
  69     def __init__(self):\r
  70         super().__init__()\r
  71         self.driver = StxSaClientImp()\r
  72 \r
  73     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  74         return self.driver.getPserver(id)\r
  75 \r
  76     def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
  77         return self.driver.getPserverList(**filters)\r
  78 \r
  79 \r
  80 class StxCpuClient(BaseClient):\r
  81     def __init__(self):\r
  82         super().__init__()\r
  83         # self._pserver_id = pserver_id\r
  84         self.driver = StxSaClientImp()\r
  85 \r
  86     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  87         return self.driver.getCpu(id)\r
  88 \r
  89     def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
  90         return self.driver.getCpuList(**filters)\r
  91 \r
  92 \r
  93 class StxMemClient(BaseClient):\r
  94     def __init__(self):\r
  95         super().__init__()\r
  96         self.driver = StxSaClientImp()\r
  97 \r
  98     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  99         return self.driver.getMem(id)\r
 100 \r
 101     def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 102         return self.driver.getMemList(**filters)\r
 103 \r
 104 \r
 105 class StxEthClient(BaseClient):\r
 106     def __init__(self):\r
 107         super().__init__()\r
 108         self.driver = StxSaClientImp()\r
 109 \r
 110     def _get(self, id) -> ocloudModel.StxGenericModel:\r
 111         return self.driver.getEthernet(id)\r
 112 \r
 113     def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 114         return self.driver.getEthernetList(**filters)\r
 115 \r
 116 \r
 117 class StxIfClient(BaseClient):\r
 118     def __init__(self):\r
 119         super().__init__()\r
 120         self.driver = StxSaClientImp()\r
 121 \r
 122     def _get(self, id) -> ocloudModel.StxGenericModel:\r
 123         return self.driver.getIf(id)\r
 124 \r
 125     def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 126         return self.driver.getIfList(**filters)\r
 127 \r
 128 \r
 129 class StxIfPortClient(BaseClient):\r
 130     def __init__(self):\r
 131         super().__init__()\r
 132         self.driver = StxSaClientImp()\r
 133 \r
 134     def _get(self, id) -> ocloudModel.StxGenericModel:\r
 135         return self.driver.getPort(id)\r
 136 \r
 137     def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 138         return self.driver.getPortList(**filters)\r
 139 \r
 140 \r
 141 # internal driver which implement client call to Stx Standalone instance\r
 142 class StxSaClientImp(object):\r
 143     def __init__(self, stx_client=None):\r
 144         super().__init__()\r
 145         self.stxclient = stx_client if stx_client else self.getStxClient()\r
 146 \r
 147     def getStxClient(self):\r
 148         os_client_args = config.get_stx_access_info()\r
 149         config_client = get_client(**os_client_args)\r
 150         return config_client\r
 151 \r
 152     def getInstanceInfo(self) -> ocloudModel.StxGenericModel:\r
 153         systems = self.stxclient.isystem.list()\r
 154         logger.debug('systems:' + str(systems[0].to_dict()))\r
 155         return ocloudModel.StxGenericModel(\r
 156             ResourceTypeEnum.OCLOUD, systems[0]) if systems else None\r
 157 \r
 158     def getPserverList(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 159         # resourcepoolid = filters.get("resourcepoolid", None)\r
 160         hosts = self.stxclient.ihost.list()\r
 161         logger.debug('host 1:' + str(hosts[0].to_dict()))\r
 162         return [ocloudModel.StxGenericModel(\r
 163             ResourceTypeEnum.PSERVER, self._hostconverter(host))\r
 164             for host in hosts if host]\r
 165 \r
 166     def getPserver(self, id) -> ocloudModel.StxGenericModel:\r
 167         host = self.stxclient.ihost.get(id)\r
 168         logger.debug('host:' + str(host.to_dict()))\r
 169         return ocloudModel.StxGenericModel(\r
 170             ResourceTypeEnum.PSERVER, self._hostconverter(host))\r
 171 \r
 172     def getK8sList(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 173         k8sclusters = self.stxclient.kube_cluster.list()\r
 174         logger.debug('k8sresources[0]:' + str(k8sclusters[0].to_dict()))\r
 175         return [ocloudModel.StxGenericModel(\r
 176             ResourceTypeEnum.DMS,\r
 177             self._k8sconverter(k8sres), self._k8shasher(k8sres))\r
 178             for k8sres in k8sclusters if k8sres]\r
 179 \r
 180     def getK8sDetail(self, name) -> ocloudModel.StxGenericModel:\r
 181         if not name:\r
 182             k8sclusters = self.stxclient.kube_cluster.list()\r
 183             # logger.debug("k8sresources[0]:" + str(k8sclusters[0].to_dict()))\r
 184             k8scluster = k8sclusters.pop()\r
 185         else:\r
 186             k8scluster = self.stxclient.kube_cluster.get(name)\r
 187 \r
 188         if not k8scluster:\r
 189             return None\r
 190         logger.debug('k8sresource:' + str(k8scluster.to_dict()))\r
 191         return ocloudModel.StxGenericModel(\r
 192             ResourceTypeEnum.DMS,\r
 193             self._k8sconverter(k8scluster), self._k8shasher(k8scluster))\r
 194 \r
 195     def getCpuList(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 196         hostid = filters.get('hostid', None)\r
 197         assert (hostid is not None), 'missing hostid to query icpu list'\r
 198         cpulist = self.stxclient.icpu.list(hostid)\r
 199         return [ocloudModel.StxGenericModel(\r
 200             ResourceTypeEnum.PSERVER_CPU,\r
 201             self._cpuconverter(cpures)) for cpures in cpulist if cpures]\r
 202 \r
 203     def getCpu(self, id) -> ocloudModel.StxGenericModel:\r
 204         cpuinfo = self.stxclient.icpu.get(id)\r
 205         return ocloudModel.StxGenericModel(\r
 206             ResourceTypeEnum.PSERVER_CPU, self._cpuconverter(cpuinfo))\r
 207 \r
 208     def getMemList(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 209         hostid = filters.get('hostid', None)\r
 210         assert (hostid is not None), 'missing hostid to query imem list'\r
 211         memlist = self.stxclient.imemory.list(hostid)\r
 212         return [ocloudModel.StxGenericModel(\r
 213             ResourceTypeEnum.PSERVER_RAM,\r
 214             self._memconverter(memories)) for memories in memlist if memories]\r
 215 \r
 216     def getMem(self, id) -> ocloudModel.StxGenericModel:\r
 217         meminfo = self.stxclient.imemory.get(id)\r
 218         return ocloudModel.StxGenericModel(\r
 219             ResourceTypeEnum.PSERVER_RAM, self._memconverter(meminfo))\r
 220 \r
 221     def getEthernetList(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 222         hostid = filters.get('hostid', None)\r
 223         assert (hostid is not None), 'missing hostid to query port list'\r
 224         ethlist = self.stxclient.ethernet_port.list(hostid)\r
 225         return [ocloudModel.StxGenericModel(\r
 226             ResourceTypeEnum.PSERVER_ETH,\r
 227             self._ethconverter(eth)) for eth in ethlist if eth]\r
 228 \r
 229     def getEthernet(self, id) -> ocloudModel.StxGenericModel:\r
 230         ethinfo = self.stxclient.ethernet_port.get(id)\r
 231         return ocloudModel.StxGenericModel(\r
 232             ResourceTypeEnum.PSERVER_ETH, self._ethconverter(ethinfo))\r
 233 \r
 234     def getIfList(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 235         hostid = filters.get('hostid', None)\r
 236         assert (hostid is not None), 'missing hostid to query iinterface list'\r
 237         iflist = self.stxclient.iinterface.list(hostid)\r
 238         return [ocloudModel.StxGenericModel(\r
 239             ResourceTypeEnum.PSERVER_IF,\r
 240             self._ifconverter(ifs)) for ifs in iflist if ifs]\r
 241 \r
 242     def getIf(self, id) -> ocloudModel.StxGenericModel:\r
 243         ifinfo = self.stxclient.iinterface.get(id)\r
 244         return ocloudModel.StxGenericModel(\r
 245             ResourceTypeEnum.PSERVER_IF, self._ifconverter(ifinfo))\r
 246 \r
 247     def getPortList(self, **filters) -> List[ocloudModel.StxGenericModel]:\r
 248         ifid = filters.get('interfaceid', None)\r
 249         assert (ifid is not None), 'missing interface id to query port list'\r
 250         portlist = self.stxclient.iinterface.list_ports(ifid)\r
 251         return [ocloudModel.StxGenericModel(\r
 252             ResourceTypeEnum.PSERVER_IF_PORT,\r
 253             port) for port in portlist if port]\r
 254 \r
 255     def getPort(self, id) -> ocloudModel.StxGenericModel:\r
 256         portinfo = self.stxclient.port.get(id)\r
 257         return ocloudModel.StxGenericModel(\r
 258             ResourceTypeEnum.PSERVER_IF_PORT, portinfo)\r
 259 \r
 260     def _getIsystems(self):\r
 261         return self.stxclient.isystem.list()\r
 262 \r
 263     def _getIsystem(self, id=None):\r
 264         if id:\r
 265             return self.stxclient.isystem.get(id)\r
 266         else:\r
 267             isystems = self.stxclient.isystem.list()\r
 268             if len(isystems) != 1 and not id:\r
 269                 raise Exception('No system uuid was provided and '\r
 270                                 'more than one system exists in the account.')\r
 271             return isystems[0]\r
 272 \r
 273     @ staticmethod\r
 274     def _hostconverter(host):\r
 275         setattr(host, 'name', host.hostname)\r
 276         return host\r
 277 \r
 278     @ staticmethod\r
 279     def _cpuconverter(cpu):\r
 280         setattr(cpu, 'name', cpu.ihost_uuid.split(\r
 281             '-', 1)[0] + '-cpu-'+str(cpu.cpu))\r
 282         return cpu\r
 283 \r
 284     @ staticmethod\r
 285     def _memconverter(mem):\r
 286         setattr(mem, 'name', mem.ihost_uuid.split('-', 1)[0] +\r
 287                 '-mem-node-'+str(mem.numa_node))\r
 288         return mem\r
 289 \r
 290     @ staticmethod\r
 291     def _ethconverter(eth):\r
 292         setattr(eth, 'name', eth.host_uuid.split('-', 1)[0] + '-'+eth.name)\r
 293         setattr(eth, 'updated_at', None)\r
 294         setattr(eth, 'created_at', None)\r
 295         return eth\r
 296 \r
 297     @ staticmethod\r
 298     def _ifconverter(ifs):\r
 299         setattr(ifs, 'name', ifs.ihost_uuid.split('-', 1)[0] + '-'+ifs.ifname)\r
 300         setattr(ifs, 'updated_at', None)\r
 301         setattr(ifs, 'created_at', None)\r
 302         return ifs\r
 303 \r
 304     @ staticmethod\r
 305     def _k8sconverter(cluster):\r
 306         setattr(cluster, 'name', cluster.cluster_name)\r
 307         setattr(cluster, 'uuid',\r
 308                 uuid.uuid3(uuid.NAMESPACE_URL, cluster.cluster_name))\r
 309         setattr(cluster, 'updated_at', None)\r
 310         setattr(cluster, 'created_at', None)\r
 311         setattr(cluster, 'events', [])\r
 312         logger.debug('k8s cluster name/uuid:' +\r
 313                      cluster.name + '/' + str(cluster.uuid))\r
 314         return cluster\r
 315 \r
 316     @ staticmethod\r
 317     def _k8shasher(cluster):\r
 318         return str(hash((cluster.cluster_name,\r
 319                          cluster.cluster_api_endpoint, cluster.admin_user)))\r