Merge remote-tracking branch 'github/pull/447'

[chef.git] / cookbooks / planet / recipes / replication.rb

diff --git a/cookbooks/planet/recipes/replication.rb b/cookbooks/planet/recipes/replication.rb
index f1be1d0d0b45fe8676d1c8b3ad37b5854b85b851..d25197171412538e41ef77475d5a96429f2b20b5 100644 (file)
--- a/cookbooks/planet/recipes/replication.rb
+++ b/cookbooks/planet/recipes/replication.rb
@@ -17,31 +17,40 @@
 # limitations under the License.
 #
 
 # limitations under the License.
 #
 

+require "yaml"
+

 include_recipe "accounts"
 include_recipe "accounts"

+include_recipe "apt"

 include_recipe "osmosis"
 
 db_passwords = data_bag_item("db", "passwords")
 
 include_recipe "osmosis"
 
 db_passwords = data_bag_item("db", "passwords")
 

-package "postgresql-client"
-
-package "ruby"
-package "ruby-dev"
-package "ruby-libxml"
+## Install required packages

 
 

-package "make"
-package "gcc"
-package "libpq-dev"
+package %w[
+  postgresql-client
+  ruby
+  ruby-dev
+  ruby-libxml
+  make
+  gcc
+  libc6-dev
+  libpq-dev
+  osmdbt
+]

 
 gem_package "pg"
 
 
 gem_package "pg"
 

+## Build preload library to flush files
+

 remote_directory "/opt/flush" do
   source "flush"
   owner "root"
   group "root"
 remote_directory "/opt/flush" do
   source "flush"
   owner "root"
   group "root"

-  mode 0o755
+  mode "755"

   files_owner "root"
   files_group "root"
   files_owner "root"
   files_group "root"

-  files_mode 0o755
+  files_mode "755"

 end
 
 execute "/opt/flush/Makefile" do
 end
 
 execute "/opt/flush/Makefile" do


@@ -53,196 +62,327 @@ execute "/opt/flush/Makefile" do
   subscribes :run, "remote_directory[/opt/flush]"
 end
 
   subscribes :run, "remote_directory[/opt/flush]"
 end
 

+## Install scripts
+

 remote_directory "/usr/local/bin" do
   source "replication-bin"
   owner "root"
   group "root"
 remote_directory "/usr/local/bin" do
   source "replication-bin"
   owner "root"
   group "root"

-  mode 0o755
+  mode "755"

   files_owner "root"
   files_group "root"
   files_owner "root"
   files_group "root"

-  files_mode 0o755
+  files_mode "755"
+end
+
+template "/usr/local/bin/replicate-minute" do
+  source "replicate-minute.erb"
+  owner "root"
+  group "root"
+  mode "755"

 end
 
 template "/usr/local/bin/users-agreed" do
   source "users-agreed.erb"
   owner "root"
   group "root"
 end
 
 template "/usr/local/bin/users-agreed" do
   source "users-agreed.erb"
   owner "root"
   group "root"

-  mode 0o755
+  mode "755"

 end
 
 template "/usr/local/bin/users-deleted" do
   source "users-deleted.erb"
   owner "root"
   group "root"
 end
 
 template "/usr/local/bin/users-deleted" do
   source "users-deleted.erb"
   owner "root"
   group "root"

-  mode 0o755
+  mode "755"

 end
 
 end
 

+## Published deleted users directory
+

 remote_directory "/store/planet/users_deleted" do
   source "users_deleted"
   owner "planet"
   group "planet"
 remote_directory "/store/planet/users_deleted" do
   source "users_deleted"
   owner "planet"
   group "planet"

-  mode 0o755
+  mode "755"

   files_owner "root"
   files_group "root"
   files_owner "root"
   files_group "root"

-  files_mode 0o644
+  files_mode "644"

 end
 
 end
 

+## Published replication directory
+

 remote_directory "/store/planet/replication" do
   source "replication-cgi"
   owner "root"
   group "root"
 remote_directory "/store/planet/replication" do
   source "replication-cgi"
   owner "root"
   group "root"

-  mode 0o755
+  mode "755"

   files_owner "root"
   files_group "root"
   files_owner "root"
   files_group "root"

-  files_mode 0o755
+  files_mode "755"

 end
 
 end
 

-directory "/store/planet/replication/changesets" do
-  owner "planet"
-  group "planet"
-  mode 0o755
+## Configuration directory
+
+directory "/etc/replication" do
+  owner "root"
+  group "root"
+  mode "755"

 end
 
 end
 

-directory "/store/planet/replication/day" do
+## Transient state directory
+
+systemd_tmpfile "/run/replication" do
+  type "d"

   owner "planet"
   group "planet"
   owner "planet"
   group "planet"

-  mode 0o755
+  mode "755"

 end
 
 end
 

-directory "/store/planet/replication/hour" do
+## Persistent state directory
+
+directory "/var/lib/replication" do

   owner "planet"
   group "planet"
   owner "planet"
   group "planet"

-  mode 0o755
+  mode "755"

 end
 
 end
 

-directory "/store/planet/replication/minute" do
+## Temporary directory
+
+directory "/store/replication" do

   owner "planet"
   group "planet"
   owner "planet"
   group "planet"

-  mode 0o755
+  mode "755"

 end
 
 end
 

-directory "/etc/replication" do
-  owner "root"
-  group "root"
-  mode 0o755
+## Users replication
+
+template "/etc/replication/users-agreed.conf" do
+  source "users-agreed.conf.erb"
+  user "planet"
+  group "planet"
+  mode "600"
+  variables :password => db_passwords["planetdiff"]

 end
 
 end
 

-directory "/var/run/lock/changeset-replication/" do
+## Changeset replication
+
+directory "/store/planet/replication/changesets" do

   owner "planet"
   group "planet"
   owner "planet"
   group "planet"

-  mode 0o750
+  mode "755"

 end
 
 end
 

-template "/etc/replication/auth.conf" do
-  source "replication.auth.erb"
+template "/etc/replication/changesets.conf" do
+  source "changesets.conf.erb"

   user "root"
   group "planet"
   user "root"
   group "planet"

-  mode 0o640
+  mode "640"

   variables :password => db_passwords["planetdiff"]
 end
 
   variables :password => db_passwords["planetdiff"]
 end
 

-template "/etc/replication/changesets.conf" do
-  source "changesets.conf.erb"
+## Minutely replication
+
+directory "/store/planet/replication/minute" do
+  owner "planet"
+  group "planet"
+  mode "755"
+end
+
+directory "/var/lib/replication/minute" do
+  owner "planet"
+  group "planet"
+  mode "755"
+end
+
+directory "/store/replication/minute" do
+  owner "planet"
+  group "planet"
+  mode "755"
+end
+
+osmdbt_config = {
+  "database" => {
+    "host" => node[:web][:database_host],
+    "dbname" => "openstreetmap",
+    "user" => "planetdiff",
+    "password" => db_passwords["planetdiff"],
+    "replication_slot" => "osmdbt"
+  },
+  "log_dir" => "/var/lib/replication/minute",
+  "changes_dir" => "/store/planet/replication/minute",
+  "tmp_dir" => "/store/replication/minute",
+  "run_dir" => "/run/replication"
+}
+
+file "/etc/replication/osmdbt-config.yaml" do

   user "root"
   group "planet"
   user "root"
   group "planet"

-  mode 0o640
-  variables :password => db_passwords["planetdiff"]
+  mode "640"
+  content YAML.dump(osmdbt_config)

 end
 
 end
 

-template "/etc/replication/users-agreed.conf" do
-  source "users-agreed.conf.erb"
+systemd_service "replication-minutely" do
+  description "Minutely replication"

   user "planet"
   user "planet"

-  group "planet"
-  mode 0o600
-  variables :password => db_passwords["planetdiff"]
+  working_directory "/etc/replication"
+  exec_start "/usr/local/bin/replicate-minute"
+  private_tmp true
+  private_devices true
+  protect_system "full"
+  protect_home true
+  restrict_address_families %w[AF_INET AF_INET6]
+  no_new_privileges true

 end
 
 end
 

-directory "/var/lib/replication" do
+systemd_timer "replication-minutely" do
+  description "Minutely replication"
+  on_boot_sec 60
+  on_unit_active_sec 60
+  accuracy_sec 5
+end
+
+## Hourly replication
+
+directory "/store/planet/replication/hour" do

   owner "planet"
   group "planet"
   owner "planet"
   group "planet"

-  mode 0o755
+  mode "755"

 end
 
 directory "/var/lib/replication/hour" do
   owner "planet"
   group "planet"
 end
 
 directory "/var/lib/replication/hour" do
   owner "planet"
   group "planet"

-  mode 0o755
+  mode "755"
+end
+
+link "/var/lib/replication/hour/data" do
+  to "/store/planet/replication/hour"

 end
 
 template "/var/lib/replication/hour/configuration.txt" do
   source "replication.config.erb"
   owner "planet"
   group "planet"
 end
 
 template "/var/lib/replication/hour/configuration.txt" do
   source "replication.config.erb"
   owner "planet"
   group "planet"

-  mode 0o644
+  mode "644"

   variables :base => "minute", :interval => 3600
 end
 
   variables :base => "minute", :interval => 3600
 end
 

-link "/var/lib/replication/hour/data" do
-  to "/store/planet/replication/hour"
+systemd_service "replication-hourly" do
+  description "Hourly replication"
+  user "planet"
+  exec_start "/usr/local/bin/osmosis -q --merge-replication-files workingDirectory=/var/lib/replication/hour"
+  private_tmp true
+  private_devices true
+  protect_system "full"
+  protect_home true
+  restrict_address_families %w[AF_INET AF_INET6]
+  no_new_privileges true
+end
+
+systemd_timer "replication-hourly" do
+  description "Daily replication"
+  on_calendar "*-*-* *:02/15:00"
+end
+
+## Daily replication
+
+directory "/store/planet/replication/day" do
+  owner "planet"
+  group "planet"
+  mode "755"

 end
 
 directory "/var/lib/replication/day" do
   owner "planet"
   group "planet"
 end
 
 directory "/var/lib/replication/day" do
   owner "planet"
   group "planet"

-  mode 0o755
+  mode "755"
+end
+
+link "/var/lib/replication/day/data" do
+  to "/store/planet/replication/day"

 end
 
 template "/var/lib/replication/day/configuration.txt" do
   source "replication.config.erb"
   owner "planet"
   group "planet"
 end
 
 template "/var/lib/replication/day/configuration.txt" do
   source "replication.config.erb"
   owner "planet"
   group "planet"

-  mode 0o644
+  mode "644"

   variables :base => "hour", :interval => 86400
 end
 
   variables :base => "hour", :interval => 86400
 end
 

-link "/var/lib/replication/day/data" do
-  to "/store/planet/replication/day"
+systemd_service "replication-daily" do
+  description "Daily replication"
+  user "planet"
+  exec_start "/usr/local/bin/osmosis -q --merge-replication-files workingDirectory=/var/lib/replication/day"
+  private_tmp true
+  private_devices true
+  protect_system "full"
+  protect_home true
+  restrict_address_families %w[AF_INET AF_INET6]
+  no_new_privileges true

 end
 
 end
 

+systemd_timer "replication-daily" do
+  description "Daily replication"
+  on_calendar "*-*-* *:02/15:00"
+end
+
+## Enable/disable feeds
+

 if node[:planet][:replication] == "enabled"
 if node[:planet][:replication] == "enabled"

-  template "/etc/cron.d/replication" do
-    source "replication.cron.erb"
-    owner "root"
-    group "root"
-    mode 0o644
+  cron_d "users-agreed" do
+    minute "0"
+    hour "7"
+    user "planet"
+    command "/usr/local/bin/users-agreed"
+    mailto "zerebubuth@gmail.com"
+  end
+
+  cron_d "users-deleted" do
+    minute "0"
+    hour "17"
+    user "planet"
+    command "/usr/local/bin/users-deleted"
+    mailto "zerebubuth@gmail.com"
+  end
+
+  cron_d "replication-changesets" do
+    user "planet"
+    command "/usr/local/bin/replicate-changesets /etc/replication/changesets.conf"
+    mailto "zerebubuth@gmail.com"
+  end
+
+  service "replication-minutely.timer" do
+    action [:enable, :start]
+  end
+
+  service "replication-hourly.timer" do
+    action [:enable, :start]
+  end
+
+  service "replication-daily.timer" do
+    action [:enable, :start]

   end
 else
   end
 else

-  file "/etc/cron.d/replication" do
+  cron_d "users-agreed" do

     action :delete
   end
     action :delete
   end

-end

 
 

-# directory "/var/lib/replication/streaming" do
-#   owner "planet"
-#   group "planet"
-#   mode 0o755
-# end
-#
-# directory "/var/log/replication" do
-#   owner "planet"
-#   group "planet"
-#   mode 0o755
-# end
-#
-# ["streaming-replicator", "streaming-server"].each do |name|
-#   template "/etc/init.d/#{name}" do
-#     source "streaming.init.erb"
-#     owner "root"
-#     group "root"
-#     mode 0o755
-#     variables :service => name
-#   end
-#
-#   if node[:planet][:replication] == "enabled"
-#     service name do
-#       action [:enable, :start]
-#       supports :restart => true, :status => true
-#       subscribes :restart, "template[/etc/init.d/#{name}]"
-#     end
-#   else
-#     service name do
-#       action [:disable, :stop]
-#       supports :restart => true, :status => true
-#     end
-#   end
-# end
+  cron_d "users-deleted" do
+    action :delete
+  end
+
+  cron_d "replication-changesets" do
+    action :delete
+  end
+
+  service "replication-minutely.timer" do
+    action [:stop, :disable]
+  end
+
+  service "replication-hourly.timer" do
+    action [:stop, :disable]
+  end
+
+  service "replication-daily.timer" do
+    action [:stop, :disable]
+  end
+end