98 lines
3.8 KiB
Python
98 lines
3.8 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
#
|
|
# Copyright 2011 Cisco Systems, 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.
|
|
#
|
|
# @author: Sumit Naiksatam, Cisco Systems, Inc.
|
|
#
|
|
|
|
from nova import exception as excp
|
|
from nova import flags
|
|
from nova import log as logging
|
|
from nova.scheduler import driver
|
|
from quantum.client import Client
|
|
from quantum.common.wsgi import Serializer
|
|
|
|
LOG = logging.getLogger('quantum.plugins.cisco.nova.quantum_aware_scheduler')
|
|
|
|
FLAGS = flags.FLAGS
|
|
flags.DEFINE_string('quantum_host', "127.0.0.1",
|
|
'IP address of the quantum network service.')
|
|
flags.DEFINE_integer('quantum_port', 9696,
|
|
'Listening port for Quantum network service')
|
|
|
|
HOST = FLAGS.quantum_host
|
|
PORT = FLAGS.quantum_port
|
|
USE_SSL = False
|
|
ACTION_PREFIX_EXT = '/v1.0'
|
|
ACTION_PREFIX_CSCO = ACTION_PREFIX_EXT + \
|
|
'/extensions/csco/tenants/{tenant_id}'
|
|
TENANT_ID = 'nova'
|
|
CSCO_EXT_NAME = 'Cisco Nova Tenant'
|
|
ACTION = '/schedule_host'
|
|
|
|
|
|
class QuantumPortAwareScheduler(driver.Scheduler):
|
|
"""
|
|
Quantum network service dependent scheduler
|
|
Obtains the hostname from Quantum using an extension API
|
|
"""
|
|
def __init__(self):
|
|
# We have to send a dummy tenant name here since the client
|
|
# needs some tenant name, but the tenant name will not be used
|
|
# since the extensions URL does not require it
|
|
client = Client(HOST, PORT, USE_SSL, format='json',
|
|
action_prefix=ACTION_PREFIX_EXT, tenant="dummy")
|
|
request_url = "/extensions"
|
|
data = client.do_request('GET', request_url)
|
|
LOG.debug("Obtained supported extensions from Quantum: %s" % data)
|
|
for ext in data['extensions']:
|
|
name = ext['name']
|
|
if name == CSCO_EXT_NAME:
|
|
LOG.debug("Quantum plugin supports required \"%s\" extension"
|
|
"for the scheduler." % name)
|
|
return
|
|
LOG.error("Quantum plugin does not support required \"%s\" extension"
|
|
" for the scheduler. Scheduler will quit." % CSCO_EXT_NAME)
|
|
raise excp.ServiceUnavailable()
|
|
|
|
def schedule(self, context, topic, *args, **kwargs):
|
|
"""Gets the host name from the Quantum service"""
|
|
instance_id = kwargs['instance_id']
|
|
user_id = \
|
|
kwargs['request_spec']['instance_properties']['user_id']
|
|
project_id = \
|
|
kwargs['request_spec']['instance_properties']['project_id']
|
|
|
|
instance_data_dict = \
|
|
{'novatenant': \
|
|
{'instance_id': instance_id,
|
|
'instance_desc': \
|
|
{'user_id': user_id,
|
|
'project_id': project_id}}}
|
|
|
|
client = Client(HOST, PORT, USE_SSL, format='json', tenant=TENANT_ID,
|
|
action_prefix=ACTION_PREFIX_CSCO)
|
|
request_url = "/novatenants/" + project_id + ACTION
|
|
data = client.do_request('PUT', request_url, body=instance_data_dict)
|
|
|
|
hostname = data["host_list"]["host_1"]
|
|
if not hostname:
|
|
raise driver.NoValidHost(_("Scheduler was unable to locate a host"
|
|
" for this request. Is the appropriate"
|
|
" service running?"))
|
|
|
|
LOG.debug(_("Quantum service returned host: %s") % hostname)
|
|
return hostname
|