From ef7b1d362683b9d03aca03888d862b811fac721e Mon Sep 17 00:00:00 2001
From: Takashi Kajinami <kajinamit@oss.nttdata.com>
Date: Wed, 18 Sep 2024 23:15:38 +0900
Subject: [PATCH] Expose rabbit_qos_prefetch_count

The option was supported by puppet-oslo but was not configurable via
this module.

Change-Id: I2d3dd1249832b4fc528004ef1268a4fc1c5e1878
---
 manifests/init.pp                                           | 6 ++++++
 .../notes/rabbit_qos_prefetch_count-c8a97c06fc25d7e4.yaml   | 4 ++++
 spec/classes/mistral_init_spec.rb                           | 1 +
 3 files changed, 11 insertions(+)
 create mode 100644 releasenotes/notes/rabbit_qos_prefetch_count-c8a97c06fc25d7e4.yaml

diff --git a/manifests/init.pp b/manifests/init.pp
index c82f6a2..43b11a0 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -63,6 +63,10 @@
 #   will be run through a green thread.
 #   Defaults to $facts['os_service_default']
 #
+# [*rabbit_qos_prefetch_count*]
+#   (Optional) Specifies the number of messages to prefetch.
+#   Defaults to $facts['os_service_default']
+#
 # [*rabbit_quorum_queue*]
 #   (Optional) Use quorum queues in RabbitMQ.
 #   Defaults to $facts['os_service_default']
@@ -187,6 +191,7 @@ class mistral(
   $rabbit_heartbeat_timeout_threshold = $facts['os_service_default'],
   $rabbit_heartbeat_rate              = $facts['os_service_default'],
   $rabbit_heartbeat_in_pthread        = $facts['os_service_default'],
+  $rabbit_qos_prefetch_count          = $facts['os_service_default'],
   $rabbit_quorum_queue                = $facts['os_service_default'],
   $rabbit_transient_quorum_queue      = $facts['os_service_default'],
   $rabbit_quorum_delivery_limit       = $facts['os_service_default'],
@@ -260,6 +265,7 @@ class mistral(
     heartbeat_timeout_threshold     => $rabbit_heartbeat_timeout_threshold,
     heartbeat_rate                  => $rabbit_heartbeat_rate,
     heartbeat_in_pthread            => $rabbit_heartbeat_in_pthread,
+    rabbit_qos_prefetch_count       => $rabbit_qos_prefetch_count,
     amqp_durable_queues             => $amqp_durable_queues,
     rabbit_quorum_queue             => $rabbit_quorum_queue,
     rabbit_transient_quorum_queue   => $rabbit_transient_quorum_queue,
diff --git a/releasenotes/notes/rabbit_qos_prefetch_count-c8a97c06fc25d7e4.yaml b/releasenotes/notes/rabbit_qos_prefetch_count-c8a97c06fc25d7e4.yaml
new file mode 100644
index 0000000..cd26926
--- /dev/null
+++ b/releasenotes/notes/rabbit_qos_prefetch_count-c8a97c06fc25d7e4.yaml
@@ -0,0 +1,4 @@
+---
+features:
+  - |
+    The new ``mistral::rabbit_qos_prefetch_count`` parameter has been added.
diff --git a/spec/classes/mistral_init_spec.rb b/spec/classes/mistral_init_spec.rb
index dbf0d93..75118ef 100644
--- a/spec/classes/mistral_init_spec.rb
+++ b/spec/classes/mistral_init_spec.rb
@@ -42,6 +42,7 @@ describe 'mistral' do
           :heartbeat_timeout_threshold        => '<SERVICE DEFAULT>',
           :heartbeat_rate                     => '<SERVICE DEFAULT>',
           :heartbeat_in_pthread               => '<SERVICE DEFAULT>',
+          :rabbit_qos_prefetch_count          => '<SERVICE DEFAULT>',
           :kombu_reconnect_delay              => '<SERVICE DEFAULT>',
           :kombu_failover_strategy            => '<SERVICE DEFAULT>',
           :amqp_durable_queues                => '<SERVICE DEFAULT>',