From 2e6fa9d0a37eb696efc95c75279d35e338dcf898 Mon Sep 17 00:00:00 2001
From: Peter Palfrader <peter@palfrader.org>
Date: Sun, 29 Sep 2019 16:30:28 +0200
Subject: [PATCH] Allow us to gradually move a server with multiple clusters to
 move to managed hba

---
 modules/roles/manifests/postgresql/server.pp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/modules/roles/manifests/postgresql/server.pp b/modules/roles/manifests/postgresql/server.pp
index 99a82376b..fd8f9164b 100644
--- a/modules/roles/manifests/postgresql/server.pp
+++ b/modules/roles/manifests/postgresql/server.pp
@@ -3,18 +3,24 @@
 #
 # @param manage_clusters_hba manage clusters' pg_hba.conf using postgres::cluster.  Eventually should should be true for every host and we can drop the param
 class roles::postgresql::server(
-  Boolean $manage_clusters_hba = false,
+  Variant[Boolean,Array[Integer]] $manage_clusters_hba = false,
 ) {
   $clusters = $facts['postgresql_clusters']
   $clusters.each |$cluster| {
     $version      = $cluster['version']
     $cluster_name = $cluster['cluster']
     $port         = $cluster['port']
+
+    if $manage_clusters_hba =~ Boolean {
+      $hba = $manage_clusters_hba
+    } else {
+      $hba = $port in $manage_clusters_hba
+    }
     postgres::cluster { "${version}::${cluster_name}":
       pg_version => $version,
       pg_cluster => $cluster_name,
       pg_port    => $port,
-      manage_hba => $manage_clusters_hba,
+      manage_hba => $hba,
     }
   }
 }
-- 
2.20.1