Since the function _get_subnet is widely used by different components, I move it to kuryr_kubernetes.utils as a part of common utilities. Change-Id: I9a80fb55f5c02274fb50c4c92eb3514ccb42830e
		
			
				
	
	
		
			55 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			55 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
# Copyright (c) 2016 Mirantis, 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
 | 
						|
 | 
						|
from kuryr_kubernetes import config
 | 
						|
from kuryr_kubernetes.controller.drivers import base
 | 
						|
from kuryr_kubernetes import utils
 | 
						|
 | 
						|
 | 
						|
class DefaultPodSubnetDriver(base.PodSubnetsDriver):
 | 
						|
    """Provides subnet for Pod port based on a configuration option."""
 | 
						|
 | 
						|
    def get_subnets(self, pod, project_id):
 | 
						|
        subnet_id = config.CONF.neutron_defaults.pod_subnet
 | 
						|
 | 
						|
        if not subnet_id:
 | 
						|
            # NOTE(ivc): this option is only required for
 | 
						|
            # DefaultPodSubnetDriver and its subclasses, but it may be
 | 
						|
            # optional for other drivers (e.g. when each namespace has own
 | 
						|
            # subnet)
 | 
						|
            raise cfg.RequiredOptError('pod_subnet',
 | 
						|
                                       cfg.OptGroup('neutron_defaults'))
 | 
						|
 | 
						|
        return {subnet_id: utils.get_subnet(subnet_id)}
 | 
						|
 | 
						|
 | 
						|
class DefaultServiceSubnetDriver(base.ServiceSubnetsDriver):
 | 
						|
    """Provides subnet for Service's LBaaS based on a configuration option."""
 | 
						|
 | 
						|
    def get_subnets(self, service, project_id):
 | 
						|
        subnet_id = config.CONF.neutron_defaults.service_subnet
 | 
						|
 | 
						|
        if not subnet_id:
 | 
						|
            # NOTE(ivc): this option is only required for
 | 
						|
            # DefaultServiceSubnetDriver and its subclasses, but it may be
 | 
						|
            # optional for other drivers (e.g. when each namespace has own
 | 
						|
            # subnet)
 | 
						|
            raise cfg.RequiredOptError('service_subnet',
 | 
						|
                                       cfg.OptGroup('neutron_defaults'))
 | 
						|
 | 
						|
        return {subnet_id: utils.get_subnet(subnet_id)}
 |