Merge pull request #31 from plmayekar/standby-cluster

adding standby cluster config

manifests/init.pp

@@ -242,6 +242,14 @@
 #   Patroni service enable property
 # @param custom_pip_provider
 #   Use custom pip path when installing pip packages
+# @param is_standby
+#   Boolean to use Standby cluster
+# @param standby_cluster_host
+#   Refer to Standby configuration `host` setting
+# @param standby_cluster_port
+#   Refer to Standby configuration `port` setting
+# @param standby_cluster_primary_slot_name
+#   Refer to Standby configuration `slot` setting
 class patroni (
 
   # Global Settings
@@ -272,6 +280,10 @@
   Hash $bootstrap_users = {},
   Variant[Undef,String] $bootstrap_post_bootstrap = undef,
   Variant[Undef,String] $bootstrap_post_init = undef,
+  Boolean $is_standby = false,
+  String $standby_cluster_host = '127.0.0.1',
+  Stdlib::Port $standby_cluster_port = 5432,
+  Optional[String] $standby_cluster_primary_slot_name = 'patroni',
 
   # PostgreSQL Settings
   String $superuser_username = 'postgres',

spec/classes/patroni_spec.rb

@@ -285,6 +285,23 @@
         )
       end
 
+      context 'is_standby => true' do
+        let(:params) { { 'scope' => 'testscope', 'is_standby' => true } }
+
+        it 'has valid config' do
+          content = catalogue.resource('file', 'patroni_config').send(:parameters)[:content]
+          config = YAML.safe_load(content)
+          expected = {
+            'standby_cluster' => {
+              'host' => '127.0.0.1',
+              'port' => '5432',
+              'primary_slot_name' => 'patroni',
+            },
+          }
+          expect(config).to include(expected)
+        end
+      end
+
       context 'use_etcd => true' do
         let(:params) { { 'scope' => 'testscope', 'use_etcd' => true } }
 

templates/postgresql.yml.erb

@@ -4,6 +4,12 @@ name: <%= @hostname %>
 
 bootstrap:
   dcs:
+    <%- if @is_standby -%>
+    standby_cluster:
+      host: <%= @standby_cluster_host %>
+      port: <%= @standby_cluster_port %>
+      primary_slot_name: <%= @standby_cluster_primary_slot_name %>
+    <%- end -%>
     ttl: <%= @dcs_ttl %>
     loop_wait: <%= @dcs_loop_wait %>
     retry_timeout: <%= @dcs_retry_timeout %>