94 lines
3.5 KiB
Python
94 lines
3.5 KiB
Python
# Copyright 2014 VMware, Inc.
|
|
# All Rights Reserved
|
|
#
|
|
# 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.
|
|
|
|
from oslo_config import cfg
|
|
import stevedore
|
|
|
|
from oslo_log import log
|
|
|
|
from neutron.i18n import _LE, _LI
|
|
from vmware_nsx.common import exceptions as nsx_exc
|
|
|
|
LOG = log.getLogger(__name__)
|
|
ROUTER_TYPE_DRIVERS = ["distributed", "exclusive", "shared"]
|
|
|
|
|
|
class RouterTypeManager(stevedore.named.NamedExtensionManager):
|
|
"""Manage router segment types using drivers."""
|
|
|
|
def __init__(self, plugin):
|
|
# Mapping from type name to DriverManager
|
|
self.drivers = {}
|
|
|
|
LOG.info(_LI("Configured router type driver names: %s"),
|
|
ROUTER_TYPE_DRIVERS)
|
|
super(RouterTypeManager, self).__init__(
|
|
'vmware_nsx.neutron.nsxv.router_type_drivers',
|
|
ROUTER_TYPE_DRIVERS,
|
|
invoke_on_load=True,
|
|
invoke_args=(plugin,))
|
|
LOG.info(_LI("Loaded type driver names: %s"), self.names())
|
|
self._register_types()
|
|
self._check_tenant_router_types(cfg.CONF.nsxv.tenant_router_types)
|
|
|
|
def _register_types(self):
|
|
for ext in self:
|
|
router_type = ext.obj.get_type()
|
|
if router_type in self.drivers:
|
|
LOG.error(_LE("Type driver '%(new_driver)s' ignored because "
|
|
"type driver '%(old_driver)s' is already "
|
|
"registered for type '%(type)s'"),
|
|
{'new_driver': ext.name,
|
|
'old_driver': self.drivers[router_type].name,
|
|
'type': router_type})
|
|
else:
|
|
self.drivers[router_type] = ext
|
|
LOG.info(_LI("Registered types: %s"), self.drivers.keys())
|
|
|
|
def _check_tenant_router_types(self, types):
|
|
self.tenant_router_types = []
|
|
for router_type in types:
|
|
if router_type in self.drivers:
|
|
self.tenant_router_types.append(router_type)
|
|
else:
|
|
msg = _("No type driver for tenant router_type: %s. "
|
|
"Service terminated!") % router_type
|
|
LOG.error(msg)
|
|
raise SystemExit(msg)
|
|
LOG.info(_LI("Tenant router_types: %s"), self.tenant_router_types)
|
|
|
|
def get_tenant_router_driver(self, context, router_type):
|
|
driver = self.drivers.get(router_type)
|
|
if driver:
|
|
return driver.obj
|
|
raise nsx_exc.NoRouterAvailable()
|
|
|
|
def decide_tenant_router_type(self, context, router_type=None):
|
|
if router_type is None:
|
|
for rt in self.tenant_router_types:
|
|
driver = self.drivers.get(rt)
|
|
if driver:
|
|
return rt
|
|
raise nsx_exc.NoRouterAvailable()
|
|
elif context.is_admin:
|
|
driver = self.drivers.get(router_type)
|
|
if driver:
|
|
return router_type
|
|
elif router_type in self.tenant_router_types:
|
|
driver = self.drivers.get(router_type)
|
|
if driver:
|
|
return router_type
|
|
raise nsx_exc.NoRouterAvailable()
|