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 o2ims.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 o2ims 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 import logging\r
  28 logger = logging.getLogger(__name__)\r
  29 \r
  30 \r
  31 class StxSaOcloudClient(BaseClient):\r
  32     def __init__(self, driver=None):\r
  33         super().__init__()\r
  34         self.driver = driver if driver else StxSaClientImp()\r
  35 \r
  36     # def list(self) -> List[ocloudModel.StxGenericModel]:\r
  37     #     return self._list()\r
  38 \r
  39     # def get(self, id) -> ocloudModel.StxGenericModel:\r
  40     #     return self._get(id)\r
  41 \r
  42     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  43         return self.driver.getInstanceInfo()\r
  44 \r
  45     def _list(self):\r
  46         return [self.driver.getInstanceInfo()]\r
  47 \r
  48 \r
  49 class StxSaResourcePoolClient(BaseClient):\r
  50     def __init__(self):\r
  51         super().__init__()\r
  52         self.driver = StxSaClientImp()\r
  53 \r
  54     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  55         return self.driver.getInstanceInfo()\r
  56 \r
  57     def _list(self):\r
  58         return [self.driver.getInstanceInfo()]\r
  59 \r
  60 \r
  61 class StxSaDmsClient(BaseClient):\r
  62     def __init__(self):\r
  63         super().__init__()\r
  64         self.driver = StxSaClientImp()\r
  65 \r
  66     def _get(self, name) -> ocloudModel.StxGenericModel:\r
  67         return self.driver.getK8sDetail(name)\r
  68 \r
  69     def _list(self):\r
  70         return self.driver.getK8sList()\r
  71 \r
  72 \r
  73 class StxPserverClient(BaseClient):\r
  74     def __init__(self):\r
  75         super().__init__()\r
  76         self.driver = StxSaClientImp()\r
  77 \r
  78     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  79         return self.driver.getPserver(id)\r
  80 \r
  81     def _list(self) -> List[ocloudModel.StxGenericModel]:\r
  82         return self.driver.getPserverList()\r
  83 \r
  84 \r
  85 class StxCpuClient(BaseClient):\r
  86     def __init__(self, pserver_id):\r
  87         super().__init__()\r
  88         self._pserver_id = pserver_id\r
  89         self.driver = StxSaClientImp()\r
  90 \r
  91     def _get(self, id) -> ocloudModel.StxGenericModel:\r
  92         return self.driver.getCpu(id)\r
  93 \r
  94     def _list(self) -> List[ocloudModel.StxGenericModel]:\r
  95         return self.driver.getCpuList(self._pserver_id)\r
  96 \r
  97 # internal driver which implement client call to Stx Standalone instance\r
  98 \r
  99 \r
 100 class StxSaClientImp(object):\r
 101     def __init__(self, stx_client=None):\r
 102         super().__init__()\r
 103         self.stxclient = stx_client if stx_client else self.getStxClient()\r
 104 \r
 105     def getStxClient(self):\r
 106         os_client_args = config.get_stx_access_info()\r
 107         config_client = get_client(**os_client_args)\r
 108         return config_client\r
 109 \r
 110     def getInstanceInfo(self) -> ocloudModel.StxGenericModel:\r
 111         systems = self.stxclient.isystem.list()\r
 112         logger.debug("systems:" + str(systems[0].to_dict()))\r
 113         return ocloudModel.StxGenericModel(\r
 114             ResourceTypeEnum.OCLOUD, systems[0]) if systems else None\r
 115 \r
 116     def getPserverList(self) -> List[ocloudModel.StxGenericModel]:\r
 117         hosts = self.stxclient.ihost.list()\r
 118         logger.debug("host 1:" + str(hosts[0].to_dict()))\r
 119         return [ocloudModel.StxGenericModel(\r
 120             ResourceTypeEnum.PSERVER, self._hostconverter(host))\r
 121                 for host in hosts if host]\r
 122 \r
 123     def getPserver(self, id) -> ocloudModel.StxGenericModel:\r
 124         host = self.stxclient.ihost.get(id)\r
 125         logger.debug("host:" + str(host.to_dict()))\r
 126         return ocloudModel.StxGenericModel(\r
 127             ResourceTypeEnum.PSERVER, self._hostconverter(host))\r
 128 \r
 129     def getK8sList(self) -> List[ocloudModel.StxGenericModel]:\r
 130         k8sclusters = self.stxclient.kube_cluster.list()\r
 131         logger.debug("k8sresources[0]:" + str(k8sclusters[0].to_dict()))\r
 132         return [ocloudModel.StxGenericModel(\r
 133             ResourceTypeEnum.DMS,\r
 134             self._k8sconverter(k8sres), self._k8shasher(k8sres))\r
 135             for k8sres in k8sclusters if k8sres]\r
 136 \r
 137     def getK8sDetail(self, name) -> ocloudModel.StxGenericModel:\r
 138         if not name:\r
 139             k8sclusters = self.stxclient.kube_cluster.list()\r
 140             # logger.debug("k8sresources[0]:" + str(k8sclusters[0].to_dict()))\r
 141             k8scluster = k8sclusters.pop()\r
 142         else:\r
 143             k8scluster = self.stxclient.kube_cluster.get(name)\r
 144 \r
 145         if not k8scluster:\r
 146             return None\r
 147         logger.debug("k8sresource:" + str(k8scluster.to_dict()))\r
 148         return ocloudModel.StxGenericModel(\r
 149             ResourceTypeEnum.DMS,\r
 150             self._k8sconverter(k8scluster), self._k8shasher(k8scluster))\r
 151 \r
 152     def getCpuList(self, hostid) -> List[ocloudModel.StxGenericModel]:\r
 153         cpulist = self.stxclient.icpu.list(hostid)\r
 154         return [ocloudModel.StxGenericModel(\r
 155             ResourceTypeEnum.OCLOUD,\r
 156             self._cpuconverter(cpures)) for cpures in cpulist if cpures]\r
 157 \r
 158     def getCpu(self, id) -> ocloudModel.StxGenericModel:\r
 159         cpuinfo = self.stxclient.icpu.get(id)\r
 160         return ocloudModel.StxGenericModel(\r
 161             ResourceTypeEnum.OCLOUD, self._cpuconverter(cpuinfo))\r
 162 \r
 163     def _getIsystems(self):\r
 164         return self.stxclient.isystem.list()\r
 165 \r
 166     def _getIsystem(self, id=None):\r
 167         if id:\r
 168             return self.stxclient.isystem.get(id)\r
 169         else:\r
 170             isystems = self.stxclient.isystem.list()\r
 171             if len(isystems) != 1 and not id:\r
 172                 raise Exception('No system uuid was provided and '\r
 173                                 'more than one system exists in the account.')\r
 174             return isystems[0]\r
 175 \r
 176     @staticmethod\r
 177     def _hostconverter(host):\r
 178         setattr(host, "name", host.hostname)\r
 179         return host\r
 180 \r
 181     @staticmethod\r
 182     def _cpuconverter(cpu):\r
 183         setattr(cpu, "name", "core-"+str(cpu.core))\r
 184         return cpu\r
 185 \r
 186     @staticmethod\r
 187     def _k8sconverter(cluster):\r
 188         setattr(cluster, "name", cluster.cluster_name)\r
 189         setattr(cluster, "uuid",\r
 190                 uuid.uuid3(uuid.NAMESPACE_URL, cluster.cluster_name))\r
 191         setattr(cluster, 'updated_at', None)\r
 192         setattr(cluster, 'created_at', None)\r
 193         logger.debug("k8s cluster name/uuid:" +\r
 194                      cluster.name + "/" + str(cluster.uuid))\r
 195         return cluster\r
 196 \r
 197     @staticmethod\r
 198     def _k8shasher(cluster):\r
 199         return str(hash((cluster.cluster_name,\r
 200                          cluster.cluster_api_endpoint, cluster.admin_user)))\r