83 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			83 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
#
 | 
						|
# Cookbook:: kosmos-postgresql
 | 
						|
# Recipe:: replica
 | 
						|
#
 | 
						|
# The MIT License (MIT)
 | 
						|
#
 | 
						|
# Copyright:: 2019, Kosmos Developers
 | 
						|
#
 | 
						|
# Permission is hereby granted, free of charge, to any person obtaining a copy
 | 
						|
# of this software and associated documentation files (the "Software"), to deal
 | 
						|
# in the Software without restriction, including without limitation the rights
 | 
						|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 | 
						|
# copies of the Software, and to permit persons to whom the Software is
 | 
						|
# furnished to do so, subject to the following conditions:
 | 
						|
#
 | 
						|
# The above copyright notice and this permission notice shall be included in
 | 
						|
# all copies or substantial portions of the Software.
 | 
						|
#
 | 
						|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 | 
						|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 | 
						|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 | 
						|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 | 
						|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 | 
						|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 | 
						|
# THE SOFTWARE.
 | 
						|
 | 
						|
postgresql_version = "12"
 | 
						|
postgresql_service = "postgresql@#{postgresql_version}-main"
 | 
						|
 | 
						|
postgresql_custom_server postgresql_version do
 | 
						|
  role "replica"
 | 
						|
end
 | 
						|
 | 
						|
service postgresql_service do
 | 
						|
  supports restart: true, status: true, reload: true
 | 
						|
end
 | 
						|
 | 
						|
postgresql_data_bag_item = data_bag_item('credentials', 'postgresql')
 | 
						|
 | 
						|
primary = postgresql_primary
 | 
						|
 | 
						|
unless primary.nil?
 | 
						|
  postgresql_data_dir = "#{node["kosmos_encfs"]["data_directory"]}/postgresql/#{postgresql_version}/main"
 | 
						|
 | 
						|
  if node['kosmos-postgresql']['ready_to_set_up_replica']
 | 
						|
    execute "set up replication" do
 | 
						|
      command <<-EOF
 | 
						|
systemctl stop #{postgresql_service}
 | 
						|
mv #{postgresql_data_dir} #{postgresql_data_dir}.old
 | 
						|
pg_basebackup -h #{primary[:ipaddress]} -U replication -D #{postgresql_data_dir} -R
 | 
						|
chown -R postgres:postgres #{postgresql_data_dir}
 | 
						|
systemctl start #{postgresql_service}
 | 
						|
      EOF
 | 
						|
      environment 'PGPASSWORD' => postgresql_data_bag_item['replication_password']
 | 
						|
      sensitive true
 | 
						|
      not_if { ::File.exist? "#{postgresql_data_dir}/standby.signal" }
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  postgresql_access "replication" do
 | 
						|
    access_type "host"
 | 
						|
    access_db "replication"
 | 
						|
    access_user "replication"
 | 
						|
    access_addr "#{primary[:ipaddress]}/32"
 | 
						|
    access_method "md5"
 | 
						|
    notifies :reload, "service[#{postgresql_service}]", :immediately
 | 
						|
  end
 | 
						|
 | 
						|
  # On the next Chef run the replica will be set up
 | 
						|
  node.normal['kosmos-postgresql']['ready_to_set_up_replica'] = true
 | 
						|
 | 
						|
  unless node.chef_environment == "development"
 | 
						|
    include_recipe "firewall"
 | 
						|
 | 
						|
    firewall_rule "postgresql primary #{primary[:hostname]}" do
 | 
						|
      port        5432
 | 
						|
      protocol    :tcp
 | 
						|
      command     :allow
 | 
						|
      source primary[:ipaddress]
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 |