rubber 1.3.2 → 1.5.0

data/lib/generators/vulcanize/templates/postgresql/config/rubber/rubber-postgresql.yml

@@ -0,0 +1,45 @@
+
+# REQUIRED: The credentials for creating/accessong your app's database
+#
+db_user: "#{app_name[0,16]}"
+db_pass:
+db_name: "#{app_name}_#{RUBBER_ENV}"
+
+db_slave_user: "#{('slave_' + db_user)[0,16]}"
+db_replicator_user: "#{('replicator_' + db_user)[0,16]}"
+
+# REQUIRED: The command to use to do a full backup of your database using
+#           "rake rubber:backup_db"
+db_backup_cmd: "nice sudo -u postgres pg_dumpall -U postgres | gzip -c > %backup_file%"
+
+# REQUIRED: The command to use to restore the database using
+#           "rake rubber:restore_db_s3"
+#           This command will receive the data generated by db_backup_cmd via
+#           its standard input
+db_restore_cmd: "zcat - | psql -Uroot -h %host% %name%"
+
+postgresql_ver: 8.4
+postgresql_conf_dir: "/etc/postgresql/#{postgresql_ver}/main"
+postgresql_data_dir: /mnt/postgresql/data
+postgresql_archive_dir: /mnt/postgresql/archive
+postgresql_pid_file: "/var/run/postgresql/#{postgresql_ver}-main.pid"
+postgresql_ctl: "/usr/sbin/service postgresql-#{postgresql_ver}"
+
+# Capistrano needs db:primary role for migrate to work,
+# so we might as well just make consistent across the board
+role_dependencies:
+  postgresql_master: [db:primary=true]
+  postgresql_slave: [db]
+  db:primary=true: [postgresql_master]
+  db: [postgresql_slave]
+
+packages: [postgresql-client, libpq-dev]
+gems: [pg]
+
+roles:
+  postgresql_master:
+    db_server_id: 1
+  postgresql_slave:
+    db_server_id: 2
+  db:
+    packages: [postgresql]

data/lib/generators/vulcanize/templates/postgresql/templates.yml

@@ -0,0 +1 @@
+description: The PostgreSQL database system

data/lib/generators/vulcanize/templates/redis/config/rubber/deploy-redis.rb

@@ -0,0 +1,37 @@
+namespace :rubber do
+
+  namespace :redis do
+
+    rubber.allow_optional_tasks(self)
+
+    after "rubber:install_packages", "rubber:redis:custom_install"
+
+    task :custom_install, :roles => :redis do
+      custom_package('http://ftp.us.debian.org/debian/pool/main/r/redis/', 'redis-server', '1.2.6-1', '! -x /usr/bin/redis-server')
+      # force redis config regen so db dir gets recreated when creating staging from a bundle
+      rsudo "rm -f /etc/redis/redis.conf"
+    end
+
+    desc "Stops the redis server"
+    task :stop, :roles => :redis do
+      rsudo "/etc/init.d/redis-server stop; exit 0"
+    end
+
+    desc "Starts the redis server"
+    task :start, :roles => :redis do
+      rsudo "/etc/init.d/redis-server start"
+    end
+
+    desc "Restarts the redis server"
+    task :restart, :roles => :redis do
+      rsudo "/etc/init.d/redis-server restart"
+    end
+
+    desc "Reloads the redis server"
+    task :reload, :roles => :redis do
+      rsudo "/etc/init.d/redis-server restart"
+    end
+
+  end
+
+end

data/lib/generators/vulcanize/templates/redis/config/rubber/role/redis/crontab

@@ -0,0 +1,8 @@
+<%
+  @read_cmd = 'crontab -l'
+  @write_cmd = 'crontab -'
+  @additive = ['# start-redis-crontab', '# end-redis-crontab']
+%>
+
+# backup redis DBs every hour
+0 * * * * BACKUP_DIR=/mnt/redis_backups BACKUP_NAME=redis BACKUP_CMD="tar -czf \%dir\%/redis_\%time_stamp\%.tgz <%= rubber_env.redis_db_dir %>" <%= RUBBER_ROOT %>/script/cron-rake rubber:backup

data/lib/generators/vulcanize/templates/redis/config/rubber/role/redis/monit-redis.conf

@@ -0,0 +1,9 @@
+<%
+  @path = '/etc/monit/monit.d/monit-redis.conf'
+%>
+<% PIDFILE = "/var/run/redis.pid" %>
+check process redis with pidfile <%= PIDFILE %>
+  group redis-<%= RUBBER_ENV %>
+  start program = "/etc/init.d/redis-server start"
+  stop program = "/etc/init.d/redis-server stop"
+  if failed host <%= rubber_env.host %> port 6379 with timeout 10 seconds for 10 cycles then restart

data/lib/generators/vulcanize/templates/redis/config/rubber/role/redis/redis.conf

@@ -0,0 +1,141 @@
+<%
+  @path = '/etc/redis/redis.conf'
+  @post = <<-SCRIPT
+    mkdir -p #{rubber_env.redis_db_dir}
+    chown -R redis:redis #{rubber_env.redis_db_dir}
+    /etc/init.d/redis-server restart
+  SCRIPT
+%>
+
+# Redis configuration file example
+
+# By default Redis does not run as a daemon. Use 'yes' if you need it.
+# Note that Redis will write a pid file in /var/run/redis.pid when daemonized.
+daemonize yes
+
+# When run as a daemon, Redis write a pid file in /var/run/redis.pid by default.
+# You can specify a custom pid file location here.
+pidfile /var/run/redis.pid
+
+# Accept connections on the specified port, default is 6379
+port 6379
+
+# If you want you can bind a single interface, if the bind option is not
+# specified all the interfaces will listen for connections.
+#
+# bind 127.0.0.1
+
+# Close the connection after a client is idle for N seconds (0 to disable)
+timeout 300
+
+# Save the DB on disk:
+#
+#   save <seconds> <changes>
+#
+#   Will save the DB if both the given number of seconds and the given
+#   number of write operations against the DB occurred.
+#
+#   In the example below the behaviour will be to save:
+#   after 900 sec (15 min) if at least 1 key changed
+#   after 300 sec (5 min) if at least 10 keys changed
+#   after 60 sec if at least 10000 keys changed
+save 900 1
+save 300 10
+save 60 10000
+
+# The filename where to dump the DB
+dbfilename dump.rdb
+
+# For default save/load DB in/from the working directory
+# Note that you must specify a directory not a file name.
+dir <%= rubber_env.redis_db_dir %>
+
+# Set server verbosity to 'debug'
+# it can be one of:
+# debug (a lot of information, useful for development/testing)
+# notice (moderately verbose, what you want in production probably)
+# warning (only very important / critical messages are logged)
+loglevel notice
+
+# Specify the log file name. Also 'stdout' can be used to force
+# the demon to log on the standard output. Note that if you use standard
+# output for logging but daemonize, logs will be sent to /dev/null
+logfile /var/log/redis/redis-server.log
+
+# Set the number of databases. The default database is DB 0, you can select
+# a different one on a per-connection basis using SELECT <dbid> where
+# dbid is a number between 0 and 'databases'-1
+databases 16
+
+################################# REPLICATION #################################
+
+# Master-Slave replication. Use slaveof to make a Redis instance a copy of
+# another Redis server. Note that the configuration is local to the slave
+# so for example it is possible to configure the slave to save the DB with a
+# different interval, or to listen to another port, and so on.
+
+# slaveof <masterip> <masterport>
+
+################################## SECURITY ###################################
+
+# Require clients to issue AUTH <PASSWORD> before processing any other
+# commands.  This might be useful in environments in which you do not trust
+# others with access to the host running redis-server.
+#
+# This should stay commented out for backward compatibility and because most
+# people do not need auth (e.g. they run their own servers).
+
+# requirepass foobared
+
+################################### LIMITS ####################################
+
+# Set the max number of connected clients at the same time. By default there
+# is no limit, and it's up to the number of file descriptors the Redis process
+# is able to open. The special value '0' means no limts.
+# Once the limit is reached Redis will close all the new connections sending
+# an error 'max number of clients reached'.
+
+# maxclients 128
+
+# Don't use more memory than the specified amount of bytes.
+# When the memory limit is reached Redis will try to remove keys with an
+# EXPIRE set. It will try to start freeing keys that are going to expire
+# in little time and preserve keys with a longer time to live.
+# Redis will also try to remove objects from free lists if possible.
+#
+# If all this fails, Redis will start to reply with errors to commands
+# that will use more memory, like SET, LPUSH, and so on, and will continue
+# to reply to most read-only commands like GET.
+#
+# WARNING: maxmemory can be a good idea mainly if you want to use Redis as a
+# 'state' server or cache, not as a real DB. When Redis is used as a real
+# database the memory usage will grow over the weeks, it will be obvious if
+# it is going to use too much memory in the long run, and you'll have the time
+# to upgrade. With maxmemory after the limit is reached you'll start to get
+# errors for write operations, and this may even lead to DB inconsistency.
+
+# maxmemory <bytes>
+
+############################### ADVANCED CONFIG ###############################
+
+# Glue small output buffers together in order to send small replies in a
+# single TCP packet. Uses a bit more CPU but most of the times it is a win
+# in terms of number of queries per second. Use 'yes' if unsure.
+glueoutputbuf yes
+
+# Use object sharing. Can save a lot of memory if you have many common
+# string in your dataset, but performs lookups against the shared objects
+# pool so it uses more CPU and can be a bit slower. Usually it's a good
+# idea.
+#
+# When object sharing is enabled (shareobjects yes) you can use
+# shareobjectspoolsize to control the size of the pool used in order to try
+# object sharing. A bigger pool size will lead to better sharing capabilities.
+# In general you want this value to be at least the double of the number of
+# very common strings you have in your dataset.
+#
+# WARNING: object sharing is experimental, don't enable this feature
+# in production before of Redis 1.0-stable. Still please try this feature in
+# your development environment so that we can test it better.
+shareobjects no
+shareobjectspoolsize 1024

data/lib/generators/vulcanize/templates/redis/config/rubber/rubber-redis.yml

@@ -0,0 +1,4 @@
+
+redis_db_dir: /mnt/redis
+
+gems: [redis]

data/lib/generators/vulcanize/templates/redis/templates.yml

@@ -0,0 +1 @@
+description: The redis module

data/lib/generators/vulcanize/templates/resque/config/initializers/resque.rb

@@ -0,0 +1,15 @@
+require 'yaml'
+require 'resque'
+
+rails_root = ENV['RAILS_ROOT'] || File.dirname(__FILE__) + '/../..'
+rails_env = ENV['RAILS_ENV'] || 'development'
+
+redis_host = 'localhost:6379'
+
+resque_yml = rails_root + '/config/resque.yml'
+if File.exist? resque_yml
+  resque_config = YAML.load_file(resque_yml)
+  redis_host = resque_config[rails_env]
+end
+
+Resque.redis = redis_host

data/lib/generators/vulcanize/templates/resque/config/rubber/common/resque.yml

@@ -0,0 +1,5 @@
+<%
+  @path = "#{RUBBER_ROOT}/config/resque.yml"
+%>
+
+<%= RUBBER_ENV %>: <%= rubber_instances.for_role('redis').first.full_name rescue 'localhost' %>:6379

data/lib/generators/vulcanize/templates/resque/config/rubber/deploy-resque.rb

@@ -0,0 +1,58 @@
+
+namespace :rubber do
+
+  namespace :resque do
+  
+    rubber.allow_optional_tasks(self)
+
+    namespace :worker do
+
+      rubber.allow_optional_tasks(self)
+
+      before "deploy:stop", "rubber:resque:worker:stop"
+      after "deploy:start", "rubber:resque:worker:start"
+      after "deploy:restart", "rubber:resque:worker:restart"
+
+      desc "Starts resque workers"
+      task :start, :roles => :resque_worker do
+        rsudo "cd #{current_path} && RUBBER_ENV=#{rails_env} ./script/resque_worker_management.rb start", :as => rubber_env.app_user
+      end
+
+      desc "Stops resque workers"
+      task :stop, :roles => :resque_worker do
+        rsudo "cd #{current_path} && RUBBER_ENV=#{rails_env} ./script/resque_worker_management.rb stop", :as => rubber_env.app_user
+      end
+
+      desc "Restarts resque workers"
+      task :restart, :roles => :resque_worker do
+        rsudo "cd #{current_path} && RUBBER_ENV=#{rails_env} ./script/resque_worker_management.rb restart", :as => rubber_env.app_user
+      end
+    end
+
+    namespace :web do
+      rubber.allow_optional_tasks(self)
+
+      before "deploy:stop", "rubber:resque:web:stop"
+      after "deploy:start", "rubber:resque:web:start"
+      after "deploy:restart", "rubber:resque:web:restart"
+
+      desc "Starts resque web tools"
+      task :start, :roles => :resque_web do
+        rsudo "RAILS_ENV=#{RUBBER_ENV} resque-web --pid-file /var/run/resque_web.pid --port #{rubber_env.resque_web_port} --no-launch #{current_path}/config/initializers/resque.rb"
+      end
+
+      desc "Stops resque web tools"
+      task :stop, :roles => :resque_web do
+        rsudo "RAILS_ENV=#{RUBBER_ENV} resque-web --pid-file /var/run/resque_web.pid --kill"
+      end
+
+      desc "Restarts resque web tools"
+      task :restart, :roles => :resque_web do
+        rubber.resque.web.stop
+        rubber.resque.web.start
+      end
+
+    end
+
+  end
+end

data/lib/generators/vulcanize/templates/resque/config/rubber/role/resque_web/monit-resque_web.conf

@@ -0,0 +1,14 @@
+<%
+  @path = '/etc/monit/monit.d/monit-resque_web.conf'
+
+  PIDFILE = "/var/run/resque_web.pid"
+  start_program = "HOME=/root resque-web --pid-file #{PIDFILE} --port #{rubber_env.resque_web_port} --no-launch #{RUBBER_ROOT}/config/initializers/resque.rb"
+  stop_program  = "HOME=/root resque-web --pid-file #{PIDFILE} --kill"
+%>
+
+check process resque_web with pidfile <%= PIDFILE %>
+  group resque_web
+  start program = "bash -l -c '<%= start_program %>'"
+  stop program = "bash -l -c '<%= stop_program %>'"
+
+  if totalmem > 350.0 MB for 15 cycles then restart

data/lib/generators/vulcanize/templates/resque/config/rubber/role/resque_worker/monit-resque_worker.conf

@@ -0,0 +1,19 @@
+<%
+  @path = '/etc/monit/monit.d/monit-resque_worker.conf'
+%>
+
+<% rubber_env.resque_workers.each_with_index do |worker, i| %>
+  <%
+    PIDFILE = "#{RUBBER_ROOT}/tmp/pids/resque_worker_#{i}.pid"
+
+    start_program = "cd #{RUBBER_ROOT} && ./script/resque_worker_management.rb start #{i}"
+    stop_program  = "cd #{RUBBER_ROOT} && ./script/resque_worker_management.rb stop #{i}"
+  %>
+
+  check process resque_worker_<%= i %> with pidfile <%= PIDFILE %>
+    group resque_worker
+    start program = "/usr/bin/sudo -H -u <%= rubber_env.app_user %> bash -l -c '<%= start_program %>'"
+    stop program = "/usr/bin/sudo -H -u <%= rubber_env.app_user %> bash -l -c '<%= stop_program %>'"
+
+    if totalmem > 350.0 MB for 15 cycles then restart
+<% end %>

data/lib/generators/vulcanize/templates/resque/config/rubber/rubber-resque.yml

@@ -0,0 +1,18 @@
+resque_web_port: 5678
+
+# list of workers with the queues to be run on each - on each worker instance,
+# a worker is run for each item in this list to handle the queues specified
+#
+resque_workers:
+  -
+    queues: *
+    poll_interval: 1
+  -
+    queues: *
+    poll_interval: 1
+
+gems: [yajl-ruby resque]
+
+role_dependencies:
+  web_tools: [resque_web]
+

data/lib/generators/vulcanize/templates/resque/script/resque_worker_management.rb

@@ -0,0 +1,112 @@
+#!/usr/bin/env ruby
+
+env = ENV["RUBBER_ENV"] ||= "development"
+root = File.join(File.dirname(__FILE__), '..')
+rails_env_file = File.join(root, 'config', 'environment.rb')
+
+if File.exists? rails_env_file
+  require(rails_env_file)
+else
+  require "rubber"
+  Rubber::initialize(root, env)
+  require 'resque'
+  require "#{root}/config/initializers/resque.rb"
+end
+
+def start_all(workers)
+  puts "Starting all workers"
+  workers.each_with_index do |worker, i|
+    start(worker, i)
+  end
+end
+
+def start(worker, index)
+  puts "Starting worker #{index}/#{worker.queues}"
+
+  log_file = log_file(index)
+  pid_file = pid_file(index)
+
+  queues = worker.queues.to_s.split(',')
+
+  daemonize(log_file, pid_file) do
+    resque_worker = Resque::Worker.new(*queues)
+    resque_worker.verbose = ENV['LOGGING'] || ENV['VERBOSE']
+    resque_worker.very_verbose = ENV['VVERBOSE']
+
+    puts "*** Starting worker #{resque_worker}"
+    resque_worker.work(worker.poll_interval.to_i || 5) # interval, will block
+  end
+end
+
+def stop_all(workers)
+  puts "Stopping all workers"
+  workers.size.times do |i|
+    stop(i)
+  end
+end
+
+def stop(index)
+  puts "Stopping worker #{index}"
+  
+  pid_file = pid_file(index)
+  pid = File.read(pid_file).to_i rescue nil
+  if pid
+    puts "Killing worker #{index}: pid #{pid}"
+    begin
+      Process.kill("QUIT", pid)
+    rescue Exception => e
+      puts e
+    end
+    File.delete(pid_file) if File.exist?(pid_file)
+  else
+    puts "No pid file for worker #{index}: #{pid_file}"
+  end
+end
+
+def daemonize(log_file, pid_file)
+  return if fork
+  Process::setsid
+  exit!(0) if fork
+  Dir::chdir("/")
+  File.umask 0000
+  FileUtils.touch log_file
+  STDIN.reopen    log_file
+  STDOUT.reopen   log_file, "a"
+  STDERR.reopen   log_file, "a"
+
+  File.open(pid_file, 'w') {|f| f.write("#{Process.pid}") }
+  
+  yield if block_given?
+  exit(0)
+end
+
+def pid_file(index)
+  File.expand_path "#{Rubber.root}/tmp/pids/resque_worker_#{index}.pid"
+end
+
+def log_file(index)
+  File.expand_path "#{Rubber.root}/log/resque_worker_#{index}.log"
+end
+
+action = ARGV[0]
+worker_index = ARGV[1] ? ARGV[1].to_i : nil
+if action.nil? || ! %w[start stop restart].include?(action)
+  puts "Usage: script/resque_worker_management.rb [start|stop|restart]"
+end
+
+workers = Rubber.config.resque_workers
+
+case action
+  when 'start'
+    worker_index ? start(workers[worker_index], worker_index) : start_all(workers)
+  when 'stop'
+    worker_index ? stop(worker_index) : stop_all(workers)
+  when 'restart'
+    if worker_index
+      stop(worker_index)
+      start(workers[worker_index], worker_index)
+    else
+      stop_all(workers)
+      start_all(workers)
+    end
+end