122 lines
4.7 KiB
Python
122 lines
4.7 KiB
Python
# Copyright (c) 2015 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.
|
|
|
|
import six
|
|
|
|
from oslo_log import log as logging
|
|
|
|
from nova.i18n import _
|
|
from nova_solverscheduler.scheduler.solvers import constraints
|
|
from nova_solverscheduler.scheduler.solvers import utils as solver_utils
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class SameRackConstraint(constraints.BaseLinearConstraint):
|
|
"""Place instances in the same racks as those of specified instances.
|
|
If the specified instances are in hosts without rack config, then place
|
|
instances in the same hosts as those of specified instances.
|
|
"""
|
|
|
|
def get_constraint_matrix(self, hosts, filter_properties):
|
|
num_hosts = len(hosts)
|
|
num_instances = filter_properties.get('num_instances')
|
|
|
|
constraint_matrix = [[True for j in xrange(num_instances)]
|
|
for i in xrange(num_hosts)]
|
|
|
|
scheduler_hints = filter_properties.get('scheduler_hints') or {}
|
|
affinity_uuids = scheduler_hints.get('same_rack', [])
|
|
|
|
if not affinity_uuids:
|
|
return constraint_matrix
|
|
|
|
if isinstance(affinity_uuids, six.string_types):
|
|
affinity_uuids = [affinity_uuids]
|
|
|
|
host_racks_map = solver_utils.get_host_racks_map(hosts)
|
|
|
|
affinity_racks = set([])
|
|
affinity_hosts = set([])
|
|
|
|
for i in xrange(num_hosts):
|
|
host_name = hosts[i].host
|
|
host_racks = host_racks_map.get(host_name, set([]))
|
|
if solver_utils.instance_uuids_overlap(hosts[i], affinity_uuids):
|
|
affinity_racks = affinity_racks.union(host_racks)
|
|
affinity_hosts.add(host_name)
|
|
|
|
for i in xrange(num_hosts):
|
|
host_name = hosts[i].host
|
|
host_racks = host_racks_map.get(host_name, set([]))
|
|
if host_name in affinity_hosts:
|
|
LOG.debug(_("%(host)s passed same-rack check."),
|
|
{'host': host_name})
|
|
continue
|
|
elif (len(host_racks) == 0) or any([rack not in affinity_racks
|
|
for rack in host_racks]):
|
|
constraint_matrix[i] = [False for j in xrange(num_instances)]
|
|
else:
|
|
LOG.debug(_("%(host)s passed same-rack check."),
|
|
{'host': host_name})
|
|
|
|
return constraint_matrix
|
|
|
|
|
|
class DifferentRackConstraint(constraints.BaseLinearConstraint):
|
|
"""Place instances in different racks as those of specified instances.
|
|
If the specified instances are in hosts without rack config, then place
|
|
instances in different hosts as those of specified instances.
|
|
"""
|
|
|
|
def get_constraint_matrix(self, hosts, filter_properties):
|
|
num_hosts = len(hosts)
|
|
num_instances = filter_properties.get('num_instances')
|
|
|
|
constraint_matrix = [[True for j in xrange(num_instances)]
|
|
for i in xrange(num_hosts)]
|
|
|
|
scheduler_hints = filter_properties.get('scheduler_hints') or {}
|
|
affinity_uuids = scheduler_hints.get('different_rack', [])
|
|
|
|
if not affinity_uuids:
|
|
return constraint_matrix
|
|
|
|
if isinstance(affinity_uuids, six.string_types):
|
|
affinity_uuids = [affinity_uuids]
|
|
|
|
host_racks_map = solver_utils.get_host_racks_map(hosts)
|
|
|
|
affinity_racks = set([])
|
|
affinity_hosts = set([])
|
|
|
|
for i in xrange(num_hosts):
|
|
host_name = hosts[i].host
|
|
host_racks = host_racks_map.get(host_name, set([]))
|
|
if solver_utils.instance_uuids_overlap(hosts[i], affinity_uuids):
|
|
affinity_racks = affinity_racks.union(host_racks)
|
|
affinity_hosts.add(host_name)
|
|
|
|
for i in xrange(num_hosts):
|
|
host_name = hosts[i].host
|
|
host_racks = host_racks_map.get(host_name, set([]))
|
|
if any([rack in affinity_racks for rack in host_racks]) or (
|
|
host_name in affinity_hosts):
|
|
constraint_matrix[i] = [False for j in xrange(num_instances)]
|
|
LOG.debug(_("%(host)s didnot pass different-rack check."),
|
|
{'host': host_name})
|
|
|
|
return constraint_matrix
|