Loading src/service/service/service_handlers/__init__.py +2 −2 Original line number Diff line number Diff line Loading @@ -47,8 +47,8 @@ SERVICE_HANDLERS = [ ]), (MicrowaveServiceHandler, [ { FilterFieldEnum.SERVICE_TYPE : ORM_ServiceTypeEnum.L2NM, FilterFieldEnum.DEVICE_DRIVER : ORM_DeviceDriverEnum.IETF_NETWORK_TOPOLOGY, FilterFieldEnum.SERVICE_TYPE : ServiceTypeEnum.SERVICETYPE_L2NM, FilterFieldEnum.DEVICE_DRIVER : DeviceDriverEnum.DEVICEDRIVER_IETF_NETWORK_TOPOLOGY, } ]), ] src/service/service/service_handlers/microwave/MicrowaveServiceHandler.py +48 −55 Original line number Diff line number Diff line Loading @@ -14,57 +14,51 @@ import anytree, json, logging from typing import Any, Dict, List, Optional, Tuple, Union from common.orm.Database import Database from common.orm.HighLevel import get_object from common.orm.backend.Tools import key_to_str from common.proto.context_pb2 import ConfigActionEnum, ConfigRule, DeviceId, Service from common.tools.object_factory.ConfigRule import json_config_rule_delete, json_config_rule_set from common.tools.object_factory.Device import json_device_id from common.type_checkers.Checkers import chk_type from context.client.ContextClient import ContextClient from device.client.DeviceClient import DeviceClient from device.proto.context_pb2 import Device from service.service.database.ConfigModel import ORM_ConfigActionEnum, get_config_rules from service.service.database.ContextModel import ContextModel from service.service.database.DeviceModel import DeviceModel from service.service.database.ServiceModel import ServiceModel from service.service.service_handler_api._ServiceHandler import _ServiceHandler from service.service.service_handler_api.AnyTreeTools import TreeNode, delete_subnode, get_subnode, set_subnode_value from service.service.service_handlers.Tools import config_rule_set, config_rule_delete from service.service.task_scheduler.TaskExecutor import TaskExecutor LOGGER = logging.getLogger(__name__) class MicrowaveServiceHandler(_ServiceHandler): def __init__( # pylint: disable=super-init-not-called self, db_service : ServiceModel, database : Database, context_client : ContextClient, device_client : DeviceClient, **settings self, service : Service, task_executor : TaskExecutor, **settings ) -> None: self.__db_service = db_service self.__database = database self.__context_client = context_client # pylint: disable=unused-private-member self.__device_client = device_client self.__db_context : ContextModel = get_object(self.__database, ContextModel, self.__db_service.context_fk) str_service_key = key_to_str([self.__db_context.context_uuid, self.__db_service.service_uuid]) db_config = get_config_rules(self.__database, str_service_key, 'running') self.__service = service self.__task_executor = task_executor # pylint: disable=unused-private-member self.__resolver = anytree.Resolver(pathattr='name') self.__config = TreeNode('.') for action, resource_key, resource_value in db_config: if action == ORM_ConfigActionEnum.SET: for config_rule in service.service_config.config_rules: action = config_rule.action if config_rule.WhichOneof('config_rule') != 'custom': continue resource_key = config_rule.custom.resource_key resource_value = config_rule.custom.resource_value if action == ConfigActionEnum.CONFIGACTION_SET: try: resource_value = json.loads(resource_value) except: # pylint: disable=bare-except pass set_subnode_value(self.__resolver, self.__config, resource_key, resource_value) elif action == ORM_ConfigActionEnum.DELETE: elif action == ConfigActionEnum.CONFIGACTION_DELETE: delete_subnode(self.__resolver, self.__config, resource_key) def SetEndpoint(self, endpoints : List[Tuple[str, str, Optional[str]]]) -> List[Union[bool, Exception]]: def SetEndpoint( self, endpoints : List[Tuple[str, str, Optional[str]]], connection_uuid : Optional[str] = None ) -> List[Union[bool, Exception]]: LOGGER.info('[SetEndpoint] endpoints={:s}'.format(str(endpoints))) LOGGER.info('[SetEndpoint] connection_uuid={:s}'.format(str(connection_uuid))) chk_type('endpoints', endpoints, list) if len(endpoints) != 2: return [] service_uuid = self.__db_service.service_uuid service_settings : TreeNode = get_subnode(self.__resolver, self.__config, 'settings', None) if service_settings is None: raise Exception('Unable to settings for Service({:s})'.format(str(service_uuid))) service_uuid = self.__service.service_id.service_uuid.uuid settings : TreeNode = get_subnode(self.__resolver, self.__config, '/settings', None) if settings is None: raise Exception('Unable to retrieve settings for Service({:s})'.format(str(service_uuid))) json_settings : Dict = service_settings.value json_settings : Dict = settings.value vlan_id = json_settings.get('vlan_id', 121) # endpoints are retrieved in the following format --> '/endpoints/endpoint[172.26.60.243:9]' try: Loading @@ -81,44 +75,43 @@ class MicrowaveServiceHandler(_ServiceHandler): results = [] try: device_uuid = endpoints[0][0] db_device : DeviceModel = get_object(self.__database, DeviceModel, device_uuid, raise_if_not_found=True) json_device = db_device.dump(include_config_rules=False, include_drivers=True, include_endpoints=True) json_device_config : Dict = json_device.setdefault('device_config', {}) json_device_config_rules : List = json_device_config.setdefault('config_rules', []) json_device_config_rules.extend([ config_rule_set('/service[{:s}]'.format(service_uuid), { device = self.__task_executor.get_device(DeviceId(**json_device_id(device_uuid))) json_config_rule = json_config_rule_set('/service[{:s}]'.format(service_uuid), { 'uuid' : service_uuid, 'node_id_src': node_id_src, 'tp_id_src' : tp_id_src, 'node_id_dst': node_id_dst, 'tp_id_dst' : tp_id_dst, 'vlan_id' : vlan_id, }), ]) self.__device_client.ConfigureDevice(Device(**json_device)) }) del device.device_config.config_rules[:] device.device_config.config_rules.append(ConfigRule(**json_config_rule)) self.__task_executor.configure_device(device) results.append(True) except Exception as e: # pylint: disable=broad-except LOGGER.exception('Unable to SetEndpoint for Service({:s})'.format(str(service_uuid))) LOGGER.exception('Unable to configure Service({:s})'.format(str(service_uuid))) results.append(e) return results def DeleteEndpoint(self, endpoints : List[Tuple[str, str, Optional[str]]]) -> List[Union[bool, Exception]]: def DeleteEndpoint( self, endpoints : List[Tuple[str, str, Optional[str]]], connection_uuid : Optional[str] = None ) -> List[Union[bool, Exception]]: LOGGER.info('[DeleteEndpoint] endpoints={:s}'.format(str(endpoints))) LOGGER.info('[DeleteEndpoint] connection_uuid={:s}'.format(str(connection_uuid))) chk_type('endpoints', endpoints, list) if len(endpoints) != 2: return [] service_uuid = self.__db_service.service_uuid service_uuid = self.__service.service_id.service_uuid.uuid results = [] try: device_uuid = endpoints[0][0] db_device : DeviceModel = get_object(self.__database, DeviceModel, device_uuid, raise_if_not_found=True) json_device = db_device.dump(include_config_rules=False, include_drivers=True, include_endpoints=True) json_device_config : Dict = json_device.setdefault('device_config', {}) json_device_config_rules : List = json_device_config.setdefault('config_rules', []) json_device_config_rules.extend([ config_rule_delete('/service[{:s}]'.format(service_uuid), {'uuid': service_uuid}) ]) self.__device_client.ConfigureDevice(Device(**json_device)) device = self.__task_executor.get_device(DeviceId(**json_device_id(device_uuid))) json_config_rule = json_config_rule_delete('/service[{:s}]'.format(service_uuid), {'uuid': service_uuid}) del device.device_config.config_rules[:] device.device_config.config_rules.append(ConfigRule(**json_config_rule)) self.__task_executor.configure_device(device) results.append(True) except Exception as e: # pylint: disable=broad-except LOGGER.exception('Unable to DeleteEndpoint for Service({:s})'.format(str(service_uuid))) Loading src/service/service/service_handlers/microwave/__init__.py 0 → 100644 +14 −0 Original line number Diff line number Diff line # Copyright 2021-2023 H2020 TeraFlow (https://www.teraflow-h2020.eu/) # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. Loading
src/service/service/service_handlers/__init__.py +2 −2 Original line number Diff line number Diff line Loading @@ -47,8 +47,8 @@ SERVICE_HANDLERS = [ ]), (MicrowaveServiceHandler, [ { FilterFieldEnum.SERVICE_TYPE : ORM_ServiceTypeEnum.L2NM, FilterFieldEnum.DEVICE_DRIVER : ORM_DeviceDriverEnum.IETF_NETWORK_TOPOLOGY, FilterFieldEnum.SERVICE_TYPE : ServiceTypeEnum.SERVICETYPE_L2NM, FilterFieldEnum.DEVICE_DRIVER : DeviceDriverEnum.DEVICEDRIVER_IETF_NETWORK_TOPOLOGY, } ]), ]
src/service/service/service_handlers/microwave/MicrowaveServiceHandler.py +48 −55 Original line number Diff line number Diff line Loading @@ -14,57 +14,51 @@ import anytree, json, logging from typing import Any, Dict, List, Optional, Tuple, Union from common.orm.Database import Database from common.orm.HighLevel import get_object from common.orm.backend.Tools import key_to_str from common.proto.context_pb2 import ConfigActionEnum, ConfigRule, DeviceId, Service from common.tools.object_factory.ConfigRule import json_config_rule_delete, json_config_rule_set from common.tools.object_factory.Device import json_device_id from common.type_checkers.Checkers import chk_type from context.client.ContextClient import ContextClient from device.client.DeviceClient import DeviceClient from device.proto.context_pb2 import Device from service.service.database.ConfigModel import ORM_ConfigActionEnum, get_config_rules from service.service.database.ContextModel import ContextModel from service.service.database.DeviceModel import DeviceModel from service.service.database.ServiceModel import ServiceModel from service.service.service_handler_api._ServiceHandler import _ServiceHandler from service.service.service_handler_api.AnyTreeTools import TreeNode, delete_subnode, get_subnode, set_subnode_value from service.service.service_handlers.Tools import config_rule_set, config_rule_delete from service.service.task_scheduler.TaskExecutor import TaskExecutor LOGGER = logging.getLogger(__name__) class MicrowaveServiceHandler(_ServiceHandler): def __init__( # pylint: disable=super-init-not-called self, db_service : ServiceModel, database : Database, context_client : ContextClient, device_client : DeviceClient, **settings self, service : Service, task_executor : TaskExecutor, **settings ) -> None: self.__db_service = db_service self.__database = database self.__context_client = context_client # pylint: disable=unused-private-member self.__device_client = device_client self.__db_context : ContextModel = get_object(self.__database, ContextModel, self.__db_service.context_fk) str_service_key = key_to_str([self.__db_context.context_uuid, self.__db_service.service_uuid]) db_config = get_config_rules(self.__database, str_service_key, 'running') self.__service = service self.__task_executor = task_executor # pylint: disable=unused-private-member self.__resolver = anytree.Resolver(pathattr='name') self.__config = TreeNode('.') for action, resource_key, resource_value in db_config: if action == ORM_ConfigActionEnum.SET: for config_rule in service.service_config.config_rules: action = config_rule.action if config_rule.WhichOneof('config_rule') != 'custom': continue resource_key = config_rule.custom.resource_key resource_value = config_rule.custom.resource_value if action == ConfigActionEnum.CONFIGACTION_SET: try: resource_value = json.loads(resource_value) except: # pylint: disable=bare-except pass set_subnode_value(self.__resolver, self.__config, resource_key, resource_value) elif action == ORM_ConfigActionEnum.DELETE: elif action == ConfigActionEnum.CONFIGACTION_DELETE: delete_subnode(self.__resolver, self.__config, resource_key) def SetEndpoint(self, endpoints : List[Tuple[str, str, Optional[str]]]) -> List[Union[bool, Exception]]: def SetEndpoint( self, endpoints : List[Tuple[str, str, Optional[str]]], connection_uuid : Optional[str] = None ) -> List[Union[bool, Exception]]: LOGGER.info('[SetEndpoint] endpoints={:s}'.format(str(endpoints))) LOGGER.info('[SetEndpoint] connection_uuid={:s}'.format(str(connection_uuid))) chk_type('endpoints', endpoints, list) if len(endpoints) != 2: return [] service_uuid = self.__db_service.service_uuid service_settings : TreeNode = get_subnode(self.__resolver, self.__config, 'settings', None) if service_settings is None: raise Exception('Unable to settings for Service({:s})'.format(str(service_uuid))) service_uuid = self.__service.service_id.service_uuid.uuid settings : TreeNode = get_subnode(self.__resolver, self.__config, '/settings', None) if settings is None: raise Exception('Unable to retrieve settings for Service({:s})'.format(str(service_uuid))) json_settings : Dict = service_settings.value json_settings : Dict = settings.value vlan_id = json_settings.get('vlan_id', 121) # endpoints are retrieved in the following format --> '/endpoints/endpoint[172.26.60.243:9]' try: Loading @@ -81,44 +75,43 @@ class MicrowaveServiceHandler(_ServiceHandler): results = [] try: device_uuid = endpoints[0][0] db_device : DeviceModel = get_object(self.__database, DeviceModel, device_uuid, raise_if_not_found=True) json_device = db_device.dump(include_config_rules=False, include_drivers=True, include_endpoints=True) json_device_config : Dict = json_device.setdefault('device_config', {}) json_device_config_rules : List = json_device_config.setdefault('config_rules', []) json_device_config_rules.extend([ config_rule_set('/service[{:s}]'.format(service_uuid), { device = self.__task_executor.get_device(DeviceId(**json_device_id(device_uuid))) json_config_rule = json_config_rule_set('/service[{:s}]'.format(service_uuid), { 'uuid' : service_uuid, 'node_id_src': node_id_src, 'tp_id_src' : tp_id_src, 'node_id_dst': node_id_dst, 'tp_id_dst' : tp_id_dst, 'vlan_id' : vlan_id, }), ]) self.__device_client.ConfigureDevice(Device(**json_device)) }) del device.device_config.config_rules[:] device.device_config.config_rules.append(ConfigRule(**json_config_rule)) self.__task_executor.configure_device(device) results.append(True) except Exception as e: # pylint: disable=broad-except LOGGER.exception('Unable to SetEndpoint for Service({:s})'.format(str(service_uuid))) LOGGER.exception('Unable to configure Service({:s})'.format(str(service_uuid))) results.append(e) return results def DeleteEndpoint(self, endpoints : List[Tuple[str, str, Optional[str]]]) -> List[Union[bool, Exception]]: def DeleteEndpoint( self, endpoints : List[Tuple[str, str, Optional[str]]], connection_uuid : Optional[str] = None ) -> List[Union[bool, Exception]]: LOGGER.info('[DeleteEndpoint] endpoints={:s}'.format(str(endpoints))) LOGGER.info('[DeleteEndpoint] connection_uuid={:s}'.format(str(connection_uuid))) chk_type('endpoints', endpoints, list) if len(endpoints) != 2: return [] service_uuid = self.__db_service.service_uuid service_uuid = self.__service.service_id.service_uuid.uuid results = [] try: device_uuid = endpoints[0][0] db_device : DeviceModel = get_object(self.__database, DeviceModel, device_uuid, raise_if_not_found=True) json_device = db_device.dump(include_config_rules=False, include_drivers=True, include_endpoints=True) json_device_config : Dict = json_device.setdefault('device_config', {}) json_device_config_rules : List = json_device_config.setdefault('config_rules', []) json_device_config_rules.extend([ config_rule_delete('/service[{:s}]'.format(service_uuid), {'uuid': service_uuid}) ]) self.__device_client.ConfigureDevice(Device(**json_device)) device = self.__task_executor.get_device(DeviceId(**json_device_id(device_uuid))) json_config_rule = json_config_rule_delete('/service[{:s}]'.format(service_uuid), {'uuid': service_uuid}) del device.device_config.config_rules[:] device.device_config.config_rules.append(ConfigRule(**json_config_rule)) self.__task_executor.configure_device(device) results.append(True) except Exception as e: # pylint: disable=broad-except LOGGER.exception('Unable to DeleteEndpoint for Service({:s})'.format(str(service_uuid))) Loading
src/service/service/service_handlers/microwave/__init__.py 0 → 100644 +14 −0 Original line number Diff line number Diff line # Copyright 2021-2023 H2020 TeraFlow (https://www.teraflow-h2020.eu/) # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License.
