1 # Copyright (C) 2021 Wind River Systems, Inc.
\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
7 # http://www.apache.org/licenses/LICENSE-2.0
\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
15 # client talking to Stx standalone
\r
18 from o2common.service.client.base_client import BaseClient
\r
19 from typing import List
\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
25 # from dcmanagerclient.api import client
\r
26 from cgtsclient.client import get_client
\r
28 from o2common.helper import o2logging
\r
29 logger = o2logging.get_logger(__name__)
\r
32 class StxSaOcloudClient(BaseClient):
\r
33 def __init__(self, driver=None):
\r
35 self.driver = driver if driver else StxSaClientImp()
\r
37 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
38 return self.driver.getInstanceInfo()
\r
40 def _list(self, **filters):
\r
41 return [self.driver.getInstanceInfo()]
\r
44 class StxSaResourcePoolClient(BaseClient):
\r
47 self.driver = StxSaClientImp()
\r
49 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
50 return self.driver.getInstanceInfo()
\r
52 def _list(self, **filters):
\r
53 return [self.driver.getInstanceInfo()]
\r
56 class StxSaDmsClient(BaseClient):
\r
59 self.driver = StxSaClientImp()
\r
61 def _get(self, name) -> ocloudModel.StxGenericModel:
\r
62 return self.driver.getK8sDetail(name)
\r
64 def _list(self, **filters):
\r
65 return self.driver.getK8sList(**filters)
\r
68 class StxPserverClient(BaseClient):
\r
71 self.driver = StxSaClientImp()
\r
73 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
74 return self.driver.getPserver(id)
\r
76 def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
77 return self.driver.getPserverList(**filters)
\r
80 class StxCpuClient(BaseClient):
\r
83 # self._pserver_id = pserver_id
\r
84 self.driver = StxSaClientImp()
\r
86 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
87 return self.driver.getCpu(id)
\r
89 def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
90 return self.driver.getCpuList(**filters)
\r
93 class StxMemClient(BaseClient):
\r
96 # self._pserver_id = pserver_id
\r
97 self.driver = StxSaClientImp()
\r
99 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
100 return self.driver.getMem(id)
\r
102 def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
103 return self.driver.getMemList(**filters)
\r
106 class StxPortClient(BaseClient):
\r
107 def __init__(self):
\r
109 # self._pserver_id = pserver_id
\r
110 self.driver = StxSaClientImp()
\r
112 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
113 return self.driver.getPort(id)
\r
115 def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
116 return self.driver.getPortList(**filters)
\r
119 class StxIfClient(BaseClient):
\r
120 def __init__(self):
\r
122 # self._pserver_id = pserver_id
\r
123 self.driver = StxSaClientImp()
\r
125 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
126 return self.driver.getIf(id)
\r
128 def _list(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
129 return self.driver.getIfList(**filters)
\r
132 # internal driver which implement client call to Stx Standalone instance
\r
133 class StxSaClientImp(object):
\r
134 def __init__(self, stx_client=None):
\r
136 self.stxclient = stx_client if stx_client else self.getStxClient()
\r
138 def getStxClient(self):
\r
139 os_client_args = config.get_stx_access_info()
\r
140 config_client = get_client(**os_client_args)
\r
141 return config_client
\r
143 def getInstanceInfo(self) -> ocloudModel.StxGenericModel:
\r
144 systems = self.stxclient.isystem.list()
\r
145 logger.debug("systems:" + str(systems[0].to_dict()))
\r
146 return ocloudModel.StxGenericModel(
\r
147 ResourceTypeEnum.OCLOUD, systems[0]) if systems else None
\r
149 def getPserverList(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
150 # resourcepoolid = filters.get("resourcepoolid", None)
\r
151 hosts = self.stxclient.ihost.list()
\r
152 logger.debug("host 1:" + str(hosts[0].to_dict()))
\r
153 return [ocloudModel.StxGenericModel(
\r
154 ResourceTypeEnum.PSERVER, self._hostconverter(host))
\r
155 for host in hosts if host]
\r
157 def getPserver(self, id) -> ocloudModel.StxGenericModel:
\r
158 host = self.stxclient.ihost.get(id)
\r
159 logger.debug("host:" + str(host.to_dict()))
\r
160 return ocloudModel.StxGenericModel(
\r
161 ResourceTypeEnum.PSERVER, self._hostconverter(host))
\r
163 def getK8sList(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
164 k8sclusters = self.stxclient.kube_cluster.list()
\r
165 logger.debug("k8sresources[0]:" + str(k8sclusters[0].to_dict()))
\r
166 return [ocloudModel.StxGenericModel(
\r
167 ResourceTypeEnum.DMS,
\r
168 self._k8sconverter(k8sres), self._k8shasher(k8sres))
\r
169 for k8sres in k8sclusters if k8sres]
\r
171 def getK8sDetail(self, name) -> ocloudModel.StxGenericModel:
\r
173 k8sclusters = self.stxclient.kube_cluster.list()
\r
174 # logger.debug("k8sresources[0]:" + str(k8sclusters[0].to_dict()))
\r
175 k8scluster = k8sclusters.pop()
\r
177 k8scluster = self.stxclient.kube_cluster.get(name)
\r
181 logger.debug("k8sresource:" + str(k8scluster.to_dict()))
\r
182 return ocloudModel.StxGenericModel(
\r
183 ResourceTypeEnum.DMS,
\r
184 self._k8sconverter(k8scluster), self._k8shasher(k8scluster))
\r
186 def getCpuList(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
187 hostid = filters.get("hostid", None)
\r
188 assert (hostid is not None), "missing hostid to query icpu list"
\r
189 cpulist = self.stxclient.icpu.list(hostid)
\r
190 return [ocloudModel.StxGenericModel(
\r
191 ResourceTypeEnum.PSERVER_CPU,
\r
192 self._cpuconverter(cpures)) for cpures in cpulist if cpures]
\r
194 def getCpu(self, id) -> ocloudModel.StxGenericModel:
\r
195 cpuinfo = self.stxclient.icpu.get(id)
\r
196 return ocloudModel.StxGenericModel(
\r
197 ResourceTypeEnum.PSERVER_CPU, self._cpuconverter(cpuinfo))
\r
199 def getMemList(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
200 hostid = filters.get("hostid", None)
\r
201 assert (hostid is not None), "missing hostid to query imem list"
\r
202 memlist = self.stxclient.imemory.list(hostid)
\r
203 return [ocloudModel.StxGenericModel(
\r
204 ResourceTypeEnum.PSERVER_RAM,
\r
205 self._memconverter(memories)) for memories in memlist if memories]
\r
207 def getMem(self, id) -> ocloudModel.StxGenericModel:
\r
208 meminfo = self.stxclient.imemory.get(id)
\r
209 return ocloudModel.StxGenericModel(
\r
210 ResourceTypeEnum.PSERVER_RAM, self._memconverter(meminfo))
\r
212 def getPortList(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
213 hostid = filters.get("hostid", None)
\r
214 assert (hostid is not None), "missing hostid to query port list"
\r
215 portlist = self.stxclient.port.list(hostid)
\r
216 return [ocloudModel.StxGenericModel(
\r
217 ResourceTypeEnum.PSERVER_PORT,
\r
218 port) for port in portlist if port]
\r
220 def getPort(self, id) -> ocloudModel.StxGenericModel:
\r
221 portinfo = self.stxclient.port.get(id)
\r
222 return ocloudModel.StxGenericModel(
\r
223 ResourceTypeEnum.PSERVER_PORT, portinfo)
\r
225 def getIfList(self, **filters) -> List[ocloudModel.StxGenericModel]:
\r
226 hostid = filters.get("hostid", None)
\r
227 assert (hostid is not None), "missing hostid to query iinterface list"
\r
228 iflist = self.stxclient.iinterface.list(hostid)
\r
229 return [ocloudModel.StxGenericModel(
\r
230 ResourceTypeEnum.PSERVER_IF,
\r
231 self._ifconverter(ifs)) for ifs in iflist if ifs]
\r
233 def getIf(self, id) -> ocloudModel.StxGenericModel:
\r
234 ifinfo = self.stxclient.iinterface.get(id)
\r
235 return ocloudModel.StxGenericModel(
\r
236 ResourceTypeEnum.PSERVER_IF, self._ifconverter(ifinfo))
\r
238 def _getIsystems(self):
\r
239 return self.stxclient.isystem.list()
\r
241 def _getIsystem(self, id=None):
\r
243 return self.stxclient.isystem.get(id)
\r
245 isystems = self.stxclient.isystem.list()
\r
246 if len(isystems) != 1 and not id:
\r
247 raise Exception('No system uuid was provided and '
\r
248 'more than one system exists in the account.')
\r
252 def _hostconverter(host):
\r
253 setattr(host, "name", host.hostname)
\r
257 def _cpuconverter(cpu):
\r
258 setattr(cpu, "name", "cpu-"+str(cpu.cpu))
\r
262 def _memconverter(mem):
\r
263 setattr(mem, "name", "mem-node-"+str(mem.numa_node))
\r
267 def _ifconverter(ifs):
\r
268 setattr(ifs, "name", ifs.ifname)
\r
269 setattr(ifs, 'updated_at', None)
\r
270 setattr(ifs, 'created_at', None)
\r
274 def _k8sconverter(cluster):
\r
275 setattr(cluster, "name", cluster.cluster_name)
\r
276 setattr(cluster, "uuid",
\r
277 uuid.uuid3(uuid.NAMESPACE_URL, cluster.cluster_name))
\r
278 setattr(cluster, 'updated_at', None)
\r
279 setattr(cluster, 'created_at', None)
\r
280 setattr(cluster, 'events', [])
\r
281 logger.debug("k8s cluster name/uuid:" +
\r
282 cluster.name + "/" + str(cluster.uuid))
\r
286 def _k8shasher(cluster):
\r
287 return str(hash((cluster.cluster_name,
\r
288 cluster.cluster_api_endpoint, cluster.admin_user)))
\r