From 8ffc47e48ea42bb2c967e844a2ce316d592ab0ee Mon Sep 17 00:00:00 2001
From: Rajesh Tailor <ratailor@redhat.com>
Date: Fri, 15 May 2020 11:37:10 +0530
Subject: [PATCH] Add parameter for `query_placement_for_availability_zone`

Add new parameter `scheduler/query_placement_for_availability_zone`
that allows the scheduler to look up a host aggregate with metadata
key of availability zone set to the value provided by incoming request,
and request result from placement be limited to that aggregate.

Change-Id: Ie02c732d9d75e1d5783f2aa841f244c9bee695ed
---
 manifests/scheduler.pp                                   | 9 +++++++++
 ...placement_for_availability_zone-701dcf6b0662351e.yaml | 8 ++++++++
 spec/classes/nova_scheduler_spec.rb                      | 9 +++++++++
 3 files changed, 26 insertions(+)
 create mode 100644 releasenotes/notes/nova_scheduler_query_placement_for_availability_zone-701dcf6b0662351e.yaml

diff --git a/manifests/scheduler.pp b/manifests/scheduler.pp
index 6ea482f09..c36b73af8 100644
--- a/manifests/scheduler.pp
+++ b/manifests/scheduler.pp
@@ -61,6 +61,13 @@
 #   flavor/image.
 #   Defaults to $::os_service_default
 #
+# [*query_placement_for_availability_zone*]
+#   (Optional) This setting allows the scheduler to look up a host aggregate
+#   with metadata key of availability zone set to the value provided by
+#   incoming request, and request result from placement be limited to that
+#   aggregate.
+#   Defaults to $::os_service_default
+#
 class nova::scheduler(
   $enabled                                  = true,
   $manage_service                           = true,
@@ -73,6 +80,7 @@ class nova::scheduler(
   $placement_aggregate_required_for_tenants = $::os_service_default,
   $max_placement_results                    = $::os_service_default,
   $enable_isolated_aggregate_filtering      = $::os_service_default,
+  $query_placement_for_availability_zone    = $::os_service_default,
 ) {
 
   include nova::deps
@@ -97,5 +105,6 @@ class nova::scheduler(
     'scheduler/placement_aggregate_required_for_tenants': value => $placement_aggregate_required_for_tenants;
     'scheduler/max_placement_results':                    value => $max_placement_results;
     'scheduler/enable_isolated_aggregate_filtering':      value => $enable_isolated_aggregate_filtering;
+    'scheduler/query_placement_for_availability_zone':    value => $query_placement_for_availability_zone;
   }
 }
diff --git a/releasenotes/notes/nova_scheduler_query_placement_for_availability_zone-701dcf6b0662351e.yaml b/releasenotes/notes/nova_scheduler_query_placement_for_availability_zone-701dcf6b0662351e.yaml
new file mode 100644
index 000000000..17d221ef2
--- /dev/null
+++ b/releasenotes/notes/nova_scheduler_query_placement_for_availability_zone-701dcf6b0662351e.yaml
@@ -0,0 +1,8 @@
+---
+features:
+  - |
+    Add parameter `query_placement_for_availability_zone` that allows the
+    scheduler to look up a host aggregate with metadata key of availability
+    zone set to the value provided by incoming request, and request result
+    from placement be limited to that aggregate.
+    Default value for this parameter is false.
diff --git a/spec/classes/nova_scheduler_spec.rb b/spec/classes/nova_scheduler_spec.rb
index 705a45e4b..a754e4b69 100644
--- a/spec/classes/nova_scheduler_spec.rb
+++ b/spec/classes/nova_scheduler_spec.rb
@@ -25,6 +25,7 @@ describe 'nova::scheduler' do
     it { is_expected.to contain_nova_config('scheduler/placement_aggregate_required_for_tenants').with_value('<SERVICE DEFAULT>') }
     it { is_expected.to contain_nova_config('scheduler/max_placement_results').with_value('<SERVICE DEFAULT>') }
     it { is_expected.to contain_nova_config('scheduler/enable_isolated_aggregate_filtering').with_value('<SERVICE DEFAULT>') }
+    it { is_expected.to contain_nova_config('scheduler/query_placement_for_availability_zone').with_value('<SERVICE DEFAULT>') }
 
     it { is_expected.to contain_class('nova::availability_zone') }
 
@@ -111,6 +112,14 @@ describe 'nova::scheduler' do
       it { is_expected.to contain_nova_config('scheduler/enable_isolated_aggregate_filtering').with_value(true) }
     end
 
+    context 'with query_placement_for_availability_zone' do
+      let :params do
+        { :query_placement_for_availability_zone => true }
+      end
+
+      it { is_expected.to contain_nova_config('scheduler/query_placement_for_availability_zone').with_value(true) }
+    end
+
     context 'with default database parameters' do
       let :pre_condition do
         "include nova"