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 "munin"
  22 include_recipe "prometheus"
  23
  24 package "locales-all"
  25 package "postgresql-common"
  26
  27 node[:postgresql][:versions].each do |version|
  28   package "postgresql-#{version}"
  29   package "postgresql-client-#{version}"
  30   package "postgresql-contrib-#{version}"
  31   package "postgresql-server-dev-#{version}"
  32
  33   defaults = node[:postgresql][:settings][:defaults] || {}
  34   settings = node[:postgresql][:settings][version] || {}
  35
  36   standby_mode = settings[:standby_mode] || defaults[:standby_mode]
  37   primary_conninfo = settings[:primary_conninfo] || defaults[:primary_conninfo]
  38
  39   passwords = if primary_conninfo
  40                 data_bag_item(primary_conninfo[:passwords][:bag],
  41                               primary_conninfo[:passwords][:item])
  42               end
  43
  44   template "/etc/postgresql/#{version}/main/postgresql.conf" do
  45     source "postgresql.conf.erb"
  46     owner "postgres"
  47     group "postgres"
  48     mode "644"
  49     variables :version => version,
  50               :defaults => defaults,
  51               :settings => settings,
  52               :primary_conninfo => primary_conninfo,
  53               :passwords => passwords
  54     notifies :reload, "service[postgresql]"
  55     only_if { ::Dir.exist?("/etc/postgresql/#{version}/main") }
  56   end
  57
  58   template "/etc/postgresql/#{version}/main/pg_hba.conf" do
  59     source "pg_hba.conf.erb"
  60     owner "postgres"
  61     group "postgres"
  62     mode "640"
  63     variables :early_rules => settings[:early_authentication_rules] || defaults[:early_authentication_rules],
  64               :late_rules => settings[:late_authentication_rules] || defaults[:late_authentication_rules]
  65     notifies :reload, "service[postgresql]"
  66     only_if { ::Dir.exist?("/etc/postgresql/#{version}/main") }
  67   end
  68
  69   template "/etc/postgresql/#{version}/main/pg_ident.conf" do
  70     source "pg_ident.conf.erb"
  71     owner "postgres"
  72     group "postgres"
  73     mode "640"
  74     variables :maps => settings[:user_name_maps] || defaults[:user_name_maps]
  75     notifies :reload, "service[postgresql]"
  76     only_if { ::Dir.exist?("/etc/postgresql/#{version}/main") }
  77   end
  78
  79   link "/var/lib/postgresql/#{version}/main/server.crt" do
  80     to "/etc/ssl/certs/ssl-cert-snakeoil.pem"
  81     only_if { ::Dir.exist?("/var/lib/postgresql/#{version}/main") }
  82   end
  83
  84   link "/var/lib/postgresql/#{version}/main/server.key" do
  85     to "/etc/ssl/private/ssl-cert-snakeoil.key"
  86     only_if { ::Dir.exist?("/var/lib/postgresql/#{version}/main") }
  87   end
  88
  89   if standby_mode == "on"
  90     file "/var/lib/postgresql/#{version}/main/standby.signal" do
  91       owner "postgres"
  92       group "postgres"
  93       mode "640"
  94     end
  95   else
  96     file "/var/lib/postgresql/#{version}/main/standby.signal" do
  97       action :delete
  98     end
  99   end
 100 end
 101
 102 service "postgresql" do
 103   action [:enable, :start]
 104   supports :status => true, :restart => true, :reload => true
 105 end
 106
 107 ohai_plugin "postgresql" do
 108   template "ohai.rb.erb"
 109 end
 110
 111 file "/etc/prometheus/exporters/postgres_queries.yml" do
 112   action :delete
 113 end
 114
 115 package "pgtop"
 116 package "libdbd-pg-perl"
 117
 118 clusters = node[:postgresql][:clusters] || []
 119 passwords = data_bag_item("postgresql", "passwords")
 120
 121 clusters.each do |name, details|
 122   prometheus_suffix = name.tr("/", "-")
 123   prometheus_database = node[:postgresql][:monitor_database]
 124
 125   postgresql_user "prometheus" do
 126     cluster name
 127     password passwords["prometheus"]
 128     roles "pg_monitor"
 129     not_if { ::File.exist?("/var/lib/postgresql/#{name}/standby.signal") }
 130   end
 131
 132   prometheus_exporter "postgres" do
 133     port 10000 + details[:port].to_i
 134     service "postgres-#{prometheus_suffix}"
 135     labels "cluster" => name
 136     scrape_interval "1m"
 137     scrape_timeout "1m"
 138     user "postgres"
 139     options %w[
 140       --collector.database_wraparound
 141       --collector.long_running_transactions
 142       --collector.process_idle
 143       --collector.stat_activity_autovacuum
 144       --collector.stat_wal_receiver
 145       --collector.statio_user_indexes
 146     ]
 147     environment "DATA_SOURCE_NAME" => "postgres:///#{prometheus_database}?host=/run/postgresql&port=#{details[:port]}"
 148     restrict_address_families "AF_UNIX"
 149     remove_ipc false
 150     subscribes :restart, "template[/etc/prometheus/exporters/postgres_queries.yml]"
 151   end
 152
 153   munin_suffix = name.tr("/", ":")
 154
 155   munin_plugin "postgres_bgwriter_#{munin_suffix}" do
 156     target "postgres_bgwriter"
 157     conf "munin.erb"
 158     conf_variables :port => details[:port]
 159   end
 160
 161   munin_plugin "postgres_checkpoints_#{munin_suffix}" do
 162     target "postgres_checkpoints"
 163     conf "munin.erb"
 164     conf_variables :port => details[:port]
 165   end
 166
 167   munin_plugin "postgres_connections_db_#{munin_suffix}" do
 168     target "postgres_connections_db"
 169     conf "munin.erb"
 170     conf_variables :port => details[:port]
 171   end
 172
 173   munin_plugin "postgres_users_#{munin_suffix}" do
 174     target "postgres_users"
 175     conf "munin.erb"
 176     conf_variables :port => details[:port]
 177   end
 178
 179   munin_plugin "postgres_xlog_#{munin_suffix}" do
 180     target "postgres_xlog"
 181     conf "munin.erb"
 182     conf_variables :port => details[:port]
 183   end
 184
 185   next unless File.exist?("/var/lib/postgresql/#{details[:version]}/main/recovery.conf")
 186
 187   munin_plugin "postgres_replication_#{munin_suffix}" do
 188     target "postgres_replication"
 189     conf "munin.erb"
 190     conf_variables :port => details[:port]
 191   end
 192 end
 193
 194 service "prometheus-postgres-exporter" do
 195   action [:stop, :disable]
 196 end
 197
 198 systemd_service "prometheus-postgres-exporter" do
 199   action :delete
 200 end