cookbooks/postgresql/recipes/default.rb

   1 #
   2 # Cookbook:: postgresql
   3 # Recipe:: default
   4 #
   5 # Copyright:: 2012, OpenStreetMap Foundation
   6 #
   7 # Licensed under the Apache License, Version 2.0 (the "License");
   8 # you may not use this file except in compliance with the License.
   9 # You may obtain a copy of the License at
  10 #
  11 #     https://www.apache.org/licenses/LICENSE-2.0
  12 #
  13 # Unless required by applicable law or agreed to in writing, software
  14 # distributed under the License is distributed on an "AS IS" BASIS,
  15 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 # See the License for the specific language governing permissions and
  17 # limitations under the License.
  18 #
  19
  20 include_recipe "apt::postgresql"
  21 include_recipe "prometheus"
  22
  23 package %w[
  24   locales-all
  25   pgbackrest
  26   postgresql-common
  27 ]
  28
  29 if node[:postgresql][:pgbackrest][:credentials_bag]
  30   pgbackrest_credentials = data_bag_item(node[:postgresql][:pgbackrest][:credentials_bag],
  31                                          node[:postgresql][:pgbackrest][:credentials_item])
  32 end
  33
  34 template "/etc/pgbackrest.conf" do
  35   source "pgbackrest.conf.erb"
  36   owner "postgres"
  37   group "postgres"
  38   mode "640"
  39   variables :credentials => pgbackrest_credentials
  40 end
  41
  42 prometheus_exporter "pgbackrest" do
  43   port 9854
  44   user "postgres"
  45   remove_ipc false
  46 end
  47
  48 node[:postgresql][:versions].each do |version|
  49   package %W[
  50     postgresql-#{version}
  51     postgresql-client-#{version}
  52     postgresql-contrib-#{version}
  53     postgresql-server-dev-#{version}
  54   ]
  55
  56   defaults = node[:postgresql][:settings][:defaults] || {}
  57   settings = node[:postgresql][:settings][version] || {}
  58
  59   standby_mode = settings[:standby_mode] || defaults[:standby_mode]
  60   primary_conninfo = settings[:primary_conninfo] || defaults[:primary_conninfo]
  61
  62   passwords = if primary_conninfo
  63                 data_bag_item(primary_conninfo[:passwords][:bag],
  64                               primary_conninfo[:passwords][:item])
  65               end
  66
  67   template "/etc/postgresql/#{version}/main/postgresql.conf" do
  68     source "postgresql.conf.erb"
  69     owner "postgres"
  70     group "postgres"
  71     mode "644"
  72     variables :version => version,
  73               :defaults => defaults,
  74               :settings => settings,
  75               :primary_conninfo => primary_conninfo,
  76               :passwords => passwords
  77     notifies :reload, "service[postgresql]"
  78     only_if { ::Dir.exist?("/etc/postgresql/#{version}/main") }
  79   end
  80
  81   template "/etc/postgresql/#{version}/main/pg_hba.conf" do
  82     source "pg_hba.conf.erb"
  83     owner "postgres"
  84     group "postgres"
  85     mode "640"
  86     variables :early_rules => settings[:early_authentication_rules] || defaults[:early_authentication_rules],
  87               :late_rules => settings[:late_authentication_rules] || defaults[:late_authentication_rules]
  88     notifies :reload, "service[postgresql]"
  89     only_if { ::Dir.exist?("/etc/postgresql/#{version}/main") }
  90   end
  91
  92   template "/etc/postgresql/#{version}/main/pg_ident.conf" do
  93     source "pg_ident.conf.erb"
  94     owner "postgres"
  95     group "postgres"
  96     mode "640"
  97     variables :maps => settings[:user_name_maps] || defaults[:user_name_maps]
  98     notifies :reload, "service[postgresql]"
  99     only_if { ::Dir.exist?("/etc/postgresql/#{version}/main") }
 100   end
 101
 102   link "/var/lib/postgresql/#{version}/main/server.crt" do
 103     to "/etc/ssl/certs/ssl-cert-snakeoil.pem"
 104     only_if { ::Dir.exist?("/var/lib/postgresql/#{version}/main") }
 105   end
 106
 107   link "/var/lib/postgresql/#{version}/main/server.key" do
 108     to "/etc/ssl/private/ssl-cert-snakeoil.key"
 109     only_if { ::Dir.exist?("/var/lib/postgresql/#{version}/main") }
 110   end
 111
 112   if standby_mode == "on"
 113     file "/var/lib/postgresql/#{version}/main/standby.signal" do
 114       owner "postgres"
 115       group "postgres"
 116       mode "640"
 117       only_if { ::Dir.exist?("/var/lib/postgresql/#{version}/main") }
 118     end
 119   else
 120     file "/var/lib/postgresql/#{version}/main/standby.signal" do
 121       action :delete
 122       only_if { ::Dir.exist?("/var/lib/postgresql/#{version}/main") }
 123     end
 124   end
 125 end
 126
 127 service "postgresql" do
 128   action [:enable, :start]
 129   supports :status => true, :restart => true, :reload => true
 130 end
 131
 132 ohai_plugin "postgresql" do
 133   template "ohai.rb.erb"
 134 end
 135
 136 package "pgtop"
 137 package "libdbd-pg-perl"
 138
 139 clusters = node[:postgresql][:clusters] || []
 140 passwords = data_bag_item("postgresql", "passwords")
 141
 142 clusters.each do |name, details|
 143   prometheus_suffix = name.tr("/", "-")
 144   prometheus_database = node[:postgresql][:monitor_database]
 145
 146   postgresql_user "prometheus" do
 147     cluster name
 148     password passwords["prometheus"]
 149     roles "pg_monitor"
 150     not_if { ::File.exist?("/var/lib/postgresql/#{name}/standby.signal") }
 151   end
 152
 153   prometheus_exporter "postgres" do
 154     port 10000 + details[:port].to_i
 155     service "postgres-#{prometheus_suffix}"
 156     labels "cluster" => name
 157     scrape_interval "1m"
 158     scrape_timeout "1m"
 159     options %w[
 160       --collector.database_wraparound
 161       --collector.long_running_transactions
 162       --collector.process_idle
 163       --collector.stat_activity_autovacuum
 164       --collector.stat_wal_receiver
 165       --collector.statio_user_indexes
 166     ]
 167     environment "DATA_SOURCE_NAME" => "postgres:///#{prometheus_database}?host=/run/postgresql&port=#{details[:port]}&user=prometheus&password=#{passwords['prometheus']}"
 168     restrict_address_families "AF_UNIX"
 169     subscribes :restart, "template[/etc/prometheus/exporters/postgres_queries.yml]"
 170   end
 171
 172   if node[:postgresql][:monitor_queries]
 173     template "/etc/prometheus/exporters/sql_exporter.yml" do
 174       source "sql_exporter.yml.erb"
 175       owner "root"
 176       group "root"
 177       mode "644"
 178     end
 179
 180     prometheus_exporter "sql" do
 181       port 20000 + details[:port].to_i
 182       service "sql-#{prometheus_suffix}"
 183       labels "cluster" => name
 184       scrape_interval "1m"
 185       scrape_timeout "1m"
 186       options "--config.file=/etc/prometheus/exporters/sql_exporter.yml"
 187       environment "SQLEXPORTER_TARGET_DSN" => "postgres://prometheus:#{passwords['prometheus']}@/run/postgresql:#{details[:port]}/#{prometheus_database}"
 188       restrict_address_families "AF_UNIX"
 189       memory_deny_write_execute false
 190       subscribes :restart, "template[/etc/prometheus/exporters/sql_exporter.yml]"
 191     end
 192   else
 193     prometheus_exporter "sql" do
 194       action :delete
 195       service "sql-#{prometheus_suffix}"
 196     end
 197
 198     file "/etc/prometheus/exporters/sql_exporter.yml" do
 199       action :delete
 200     end
 201   end
 202 end