diff options
Diffstat (limited to 'moonv4/moon_orchestrator/moon_orchestrator/drivers.py')
-rw-r--r-- | moonv4/moon_orchestrator/moon_orchestrator/drivers.py | 175 |
1 files changed, 0 insertions, 175 deletions
diff --git a/moonv4/moon_orchestrator/moon_orchestrator/drivers.py b/moonv4/moon_orchestrator/moon_orchestrator/drivers.py deleted file mode 100644 index 08c53be3..00000000 --- a/moonv4/moon_orchestrator/moon_orchestrator/drivers.py +++ /dev/null @@ -1,175 +0,0 @@ -# Copyright 2015 Open Platform for NFV Project, Inc. and its contributors -# This software is distributed under the terms and conditions of the 'Apache-2.0' -# license which can be found in the file 'LICENSE' in this package distribution -# or at 'http://www.apache.org/licenses/LICENSE-2.0'. - -from kubernetes import client, config -import logging -import urllib3.exceptions -from python_moonutilities import configuration - -LOG = logging.getLogger("moon.orchestrator.drivers") - - -def get_driver(): - try: - return K8S() - except urllib3.exceptions.MaxRetryError as e: - LOG.exception(e) - return Docker() - - -class Driver: - - def __init__(self): - self.cache = {} - # example of cache: - # { - # "uuid_of_pod": { - # "ip": "", - # "hostname": "", - # "port": 30001, - # "pdp": "", - # "keystone_project_id": "", - # "plugin_name": "", - # "namespace": "" - # } - # } - - def get_pods(self, namespace=None): - raise NotImplementedError - - def load_pod(self, data, api_client=None, ext_client=None): - raise NotImplementedError - - def delete_pod(self, uuid=None, name=None): - raise NotImplementedError - - def get_slaves(self): - raise NotImplementedError - - -class K8S(Driver): - - def __init__(self): - super(K8S, self).__init__() - config.load_kube_config() - self.client = client.CoreV1Api() - - def get_pods(self, name=None): - if name: - pods = self.client.list_pod_for_all_namespaces(watch=False) - for pod in pods.items: - LOG.info("get_pods {}".format(pod.metadata.name)) - if name in pod.metadata.name: - return pod - else: - return None - LOG.info("get_pods cache={}".format(self.cache)) - return self.cache - - @staticmethod - def __create_pod(client, data): - pod_manifest = { - 'apiVersion': 'extensions/v1beta1', - 'kind': 'Deployment', - 'metadata': { - 'name': data[0].get('name') - }, - 'spec': { - 'replicas': 1, - 'template': { - 'metadata': {'labels': {'app': data[0].get('name')}}, - 'hostname': data[0].get('name'), - 'spec': { - 'containers': [] - } - }, - } - } - for _data in data: - pod_manifest['spec']['template']['spec']['containers'].append( - { - 'image': _data.get('container', "busybox"), - 'name': _data.get('name'), - 'hostname': _data.get('name'), - 'ports': [ - {"containerPort": _data.get('port', 80)}, - ], - 'env': [ - {'name': "UUID", "value": _data.get('name', "None")}, - {'name': "TYPE", "value": _data.get('genre', "None")}, - {'name': "PORT", "value": str(_data.get('port', 80))}, - {'name': "PDP_ID", "value": _data.get('pdp_id', "None")}, - {'name': "META_RULE_ID", "value": _data.get('meta_rule_id', "None")}, - {'name': "KEYSTONE_PROJECT_ID", - "value": _data.get('keystone_project_id', "None")}, - ] - } - ) - resp = client.create_namespaced_deployment(body=pod_manifest, - namespace='moon') - LOG.info("Pod {} created!".format(data[0].get('name'))) - # logger.info(yaml.dump(pod_manifest, sys.stdout)) - # logger.info(resp) - return resp - - @staticmethod - def __create_service(client, data, expose=False): - service_manifest = { - 'apiVersion': 'v1', - 'kind': 'Service', - 'metadata': { - 'name': data.get('name'), - 'namespace': 'moon' - }, - 'spec': { - 'ports': [{ - 'port': data.get('port', 80), - 'targetPort': data.get('port', 80) - }], - 'selector': { - 'app': data.get('name') - }, - # 'type': 'NodePort', - 'endpoints': [{ - 'port': data.get('port', 80), - 'protocol': 'TCP', - }], - } - } - if expose: - service_manifest['spec']['ports'][0]['nodePort'] = \ - configuration.increment_port() - service_manifest['spec']['type'] = "NodePort" - resp = client.create_namespaced_service(namespace="moon", - body=service_manifest) - LOG.info("Service {} created!".format(data.get('name'))) - return resp - - def load_pod(self, data, api_client=None, ext_client=None, expose=False): - _client = api_client if api_client else self.client - pod = self.__create_pod(client=ext_client, data=data) - service = self.__create_service(client=_client, data=data[0], - expose=expose) - self.cache[pod.metadata.uid] = data - - def delete_pod(self, uuid=None, name=None): - LOG.info("Deleting pod {}".format(uuid)) - # TODO: delete_namespaced_deployment - # https://github.com/kubernetes-incubator/client-python/blob/master/kubernetes/client/apis/extensions_v1beta1_api.py - - def get_slaves(self): - contexts, active_context = config.list_kube_config_contexts() - return contexts, active_context - - -class Docker(Driver): - - def load_pod(self, data, api_client=None, ext_client=None): - LOG.info("Creating pod {}".format(data[0].get('name'))) - raise NotImplementedError - - def delete_pod(self, uuid=None, name=None): - LOG.info("Deleting pod {}".format(uuid)) - raise NotImplementedError |