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