diff --git a/nova/scheduler/filters/affinity_filter.py b/nova/scheduler/filters/affinity_filter.py index e2a38b1e63e..03bf0dd6eee 100644 --- a/nova/scheduler/filters/affinity_filter.py +++ b/nova/scheduler/filters/affinity_filter.py @@ -25,8 +25,11 @@ class AffinityFilter(filters.BaseHostFilter): def __init__(self): self.compute_api = compute.API() - def _affinity_host(self, context, instance_id): - return self.compute_api.get(context, instance_id)['host'] + def _all_hosts(self, context): + all_hosts = {} + for instance in self.compute_api.get_all(context): + all_hosts[instance['uuid']] = instance['host'] + return all_hosts class DifferentHostFilter(AffinityFilter): @@ -41,8 +44,9 @@ def host_passes(self, host_state, filter_properties): if isinstance(affinity_uuids, basestring): affinity_uuids = [affinity_uuids] if affinity_uuids: + all_hosts = self._all_hosts(context) return not any([i for i in affinity_uuids - if self._affinity_host(context, i) == me]) + if all_hosts.get(i) == me]) # With no different_host key return True @@ -61,9 +65,10 @@ def host_passes(self, host_state, filter_properties): if isinstance(affinity_uuids, basestring): affinity_uuids = [affinity_uuids] if affinity_uuids: + all_hosts = self._all_hosts(context) return any([i for i in affinity_uuids - if self._affinity_host(context, i) == me]) + if all_hosts.get(i) == me]) # With no same_host key return True