62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
# Copyright (c) 2014 Red Hat, 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.
|
|
|
|
import functools
|
|
|
|
from oslo_utils import importutils
|
|
|
|
from mogan.scheduler import utils
|
|
|
|
|
|
class LazyLoader(object):
|
|
|
|
def __init__(self, klass, *args, **kwargs):
|
|
self.klass = klass
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
self.instance = None
|
|
|
|
def __getattr__(self, name):
|
|
return functools.partial(self.__run_method, name)
|
|
|
|
def __run_method(self, __name, *args, **kwargs):
|
|
if self.instance is None:
|
|
self.instance = self.klass(*self.args, **self.kwargs)
|
|
return getattr(self.instance, __name)(*args, **kwargs)
|
|
|
|
|
|
class SchedulerClient(object):
|
|
"""Client library for placing calls to the scheduler."""
|
|
|
|
def __init__(self):
|
|
self.queryclient = LazyLoader(importutils.import_class(
|
|
'mogan.scheduler.client.query.SchedulerQueryClient'))
|
|
self.reportclient = LazyLoader(importutils.import_class(
|
|
'mogan.scheduler.client.report.SchedulerReportClient'))
|
|
|
|
@utils.retry_select_destinations
|
|
def select_destinations(self, context, spec_obj, filter_properties):
|
|
return self.queryclient.select_destinations(context, spec_obj,
|
|
filter_properties)
|
|
|
|
def set_inventory_for_provider(self, rp_uuid, rp_name, inv_data,
|
|
res_class):
|
|
self.reportclient.set_inventory_for_provider(
|
|
rp_uuid,
|
|
rp_name,
|
|
inv_data,
|
|
res_class
|
|
)
|