56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
# Copyright (c) 2012 The Cloudscaling Group, 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 nova import db
|
|
from nova.scheduler import filters
|
|
|
|
|
|
class TypeAffinityFilter(filters.BaseHostFilter):
|
|
"""TypeAffinityFilter doesn't allow more then one VM type per host.
|
|
|
|
Note: this works best with ram_weight_multiplier
|
|
(spread) set to 1 (default).
|
|
"""
|
|
|
|
def host_passes(self, host_state, filter_properties):
|
|
"""Dynamically limits hosts to one instance type
|
|
|
|
Return False if host has any instance types other then the requested
|
|
type. Return True if all instance types match or if host is empty.
|
|
"""
|
|
|
|
instance_type = filter_properties.get('instance_type')
|
|
context = filter_properties['context'].elevated()
|
|
instances_other_type = db.instance_get_all_by_host_and_not_type(
|
|
context, host_state.host, instance_type['id'])
|
|
return len(instances_other_type) == 0
|
|
|
|
|
|
class AggregateTypeAffinityFilter(filters.BaseHostFilter):
|
|
"""AggregateTypeAffinityFilter limits instance_type by aggregate
|
|
|
|
return True if no instance_type key is set or if the aggregate metadata
|
|
key 'instance_type' has the instance_type name as a value
|
|
"""
|
|
|
|
def host_passes(self, host_state, filter_properties):
|
|
instance_type = filter_properties.get('instance_type')
|
|
context = filter_properties['context'].elevated()
|
|
metadata = db.aggregate_metadata_get_by_host(
|
|
context, host_state.host, key='instance_type')
|
|
return (len(metadata) == 0 or
|
|
instance_type['name'] in metadata['instance_type'])
|