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 o2ims.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 o2ims import config
\r
24 # from dcmanagerclient.api import client
\r
25 from cgtsclient.client import get_client
\r
27 logger = logging.getLogger(__name__)
\r
30 class StxSaOcloudClient(BaseClient):
\r
31 def __init__(self, driver=None):
\r
33 self.driver = driver if driver else StxSaClientImp()
\r
35 # def list(self) -> List[ocloudModel.StxGenericModel]:
\r
36 # return self._list()
\r
38 # def get(self, id) -> ocloudModel.StxGenericModel:
\r
39 # return self._get(id)
\r
41 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
42 return self.driver.getInstanceInfo()
\r
45 return [self.driver.getInstanceInfo()]
\r
48 class StxSaResourcePoolClient(BaseClient):
\r
51 self.driver = StxSaClientImp()
\r
53 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
54 return self.driver.getInstanceInfo()
\r
57 return [self.driver.getInstanceInfo()]
\r
60 class StxSaDmsClient(BaseClient):
\r
63 self.driver = StxSaClientImp()
\r
65 def _get(self, name) -> ocloudModel.StxGenericModel:
\r
66 return self.driver.getK8sDetail(name)
\r
69 return self.driver.getK8sList()
\r
72 class StxPserverClient(BaseClient):
\r
75 self.driver = StxSaClientImp()
\r
77 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
78 return self.driver.getPserver(id)
\r
80 def _list(self) -> List[ocloudModel.StxGenericModel]:
\r
81 return self.driver.getPserverList()
\r
84 class StxCpuClient(BaseClient):
\r
85 def __init__(self, pserver_id):
\r
87 self._pserver_id = pserver_id
\r
88 self.driver = StxSaClientImp()
\r
90 def _get(self, id) -> ocloudModel.StxGenericModel:
\r
91 return self.driver.getCpu(id)
\r
93 def _list(self) -> List[ocloudModel.StxGenericModel]:
\r
94 return self.driver.getCpuList(self._pserver_id)
\r
96 # internal driver which implement client call to Stx Standalone instance
\r
99 class StxSaClientImp(object):
\r
100 def __init__(self, stx_client=None):
\r
102 self.stxclient = stx_client if stx_client else self.getStxClient()
\r
104 def getStxClient():
\r
105 os_client_args = config.get_stx_access_info()
\r
106 config_client = get_client(**os_client_args)
\r
107 return config_client
\r
109 def getInstanceInfo(self) -> ocloudModel.StxGenericModel:
\r
110 systems = self.stxclient.isystem.list()
\r
111 logger.debug("systems:" + str(systems[0].to_dict()))
\r
112 return ocloudModel.StxGenericModel(systems[0]) if systems else None
\r
114 def getPserverList(self) -> List[ocloudModel.StxGenericModel]:
\r
115 hosts = self.stxclient.ihost.list()
\r
116 logger.debug("host 1:" + str(hosts[0].to_dict()))
\r
117 return [ocloudModel.StxGenericModel(self._hostconverter(host))
\r
118 for host in hosts if host]
\r
120 def getPserver(self, id) -> ocloudModel.StxGenericModel:
\r
121 host = self.stxclient.ihost.get(id)
\r
122 logger.debug("host:" + str(host.to_dict()))
\r
123 return ocloudModel.StxGenericModel(self._hostconverter(host))
\r
125 def getK8sList(self) -> List[ocloudModel.StxGenericModel]:
\r
126 k8sclusters = self.stxclient.kube_cluster.list()
\r
127 logger.debug("k8sresources:" + str(k8sclusters[0].to_dict()))
\r
129 return [ocloudModel.StxGenericModel(self._k8sconverter(k8sres))
\r
130 for k8sres in k8sclusters if k8sres]
\r
132 def getK8sDetail(self, name) -> ocloudModel.StxGenericModel:
\r
133 k8scluster = self.stxclient.kube_cluster.get(name)
\r
134 logger.debug("k8sresource:" + str(k8scluster.to_dict()))
\r
135 return ocloudModel.StxGenericModel(self._k8sconverter(k8scluster))
\r
137 def getCpuList(self, hostid) -> List[ocloudModel.StxGenericModel]:
\r
138 cpulist = self.stxclient.icpu.list(hostid)
\r
139 return [ocloudModel.StxGenericModel(self._cpuconverter(cpures))
\r
140 for cpures in cpulist if cpures]
\r
142 def getCpu(self, id) -> ocloudModel.StxGenericModel:
\r
143 cpuinfo = self.stxclient.icpu.get(id)
\r
144 return ocloudModel.StxGenericModel(self._cpuconverter(cpuinfo))
\r
146 def _getIsystems(self):
\r
147 return self.stxclient.isystem.list()
\r
149 def _getIsystem(self, id=None):
\r
151 return self.stxclient.isystem.get(id)
\r
153 isystems = self.stxclient.isystem.list()
\r
154 if len(isystems) != 1 and not id:
\r
155 raise Exception('No system uuid was provided and '
\r
156 'more than one system exists in the account.')
\r
160 def _hostconverter(host):
\r
161 setattr(host, "name", host.hostname)
\r
165 def _cpuconverter(cpu):
\r
166 setattr(cpu, "name", "core-"+str(cpu.core))
\r
170 def _k8sconverter(host):
\r
171 setattr(host, "name", host.cluster_name)
\r
172 setattr(host, "uuid",
\r
173 uuid.uuid3(uuid.NAMESPACE_URL, host.cluster_name))
\r
174 setattr(host, 'updated_at', None)
\r
175 setattr(host, 'created_at', None)
\r
176 logger.debug("k8s cluster name/uuid:" +
\r
177 host.name + "/" + str(host.uuid))
\r