rubber-jcarnegie 0.0.1

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

@@ -0,0 +1,9 @@
+redis_server_version: 2.0.4
+
+redis_db_dir: /mnt/redis
+
+gems: [redis]
+
+roles:
+  redis:
+    packages: [redis-server]

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

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

data/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/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/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/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 = "/bin/bash -l -c '<%= start_program %>'"
+  stop program = "/bin/bash -l -c '<%= stop_program %>'"
+
+  if totalmem > 350.0 MB for 15 cycles then restart

data/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 = "/usr/bin/sudo -H -u #{rubber_env.app_user} bash -l -c 'cd #{RUBBER_ROOT} && ./script/resque_worker_management.rb start #{i}'"
+    stop_program  = "/usr/bin/sudo -H -u #{rubber_env.app_user} bash -l -c 'cd #{RUBBER_ROOT} && ./script/resque_worker_management.rb stop #{i} TERM'"
+  %>
+
+  check process resque_worker_<%= i %> with pidfile <%= PIDFILE %>
+    group resque_worker
+    start program = "<%= start_program %>"
+    stop program = "<%= stop_program %>"
+
+    if totalmem > 350.0 MB for 15 cycles then restart
+<% end %>

data/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/generators/vulcanize/templates/resque/script/resque_worker_management.rb

@@ -0,0 +1,144 @@
+#!/usr/bin/env ruby
+
+STDOUT.sync = true
+STDERR.sync = true
+
+def load_env(rubber_only=true)
+  env = ENV["RUBBER_ENV"] ||= "development"
+  root = File.expand_path('../..', __FILE__)
+  rails_env_file = File.join(root, 'config', 'environment.rb')
+
+  if ! rubber_only && File.exists?(rails_env_file)
+    require(rails_env_file)
+  else
+    require "bundler/setup" if File.exist?(File.join(root, "Gemfile"))
+    require "rubber"
+    Rubber::initialize(root, env)
+  end
+end
+
+
+def start_all(workers)
+  puts "Starting all workers"
+  # fork then load rails env so script runs quick, yet we don't overload
+  # machine by loading env for each worker
+  daemonize(log_file('all')) do
+    load_env(false)
+    puts "Preloaded environment for all workers"
+
+    workers.each_with_index do |worker, i|
+      start(worker, i)
+    end
+  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
+    # load env for each worker, if starting multiple, this will be a no-op due
+    # to start_all preloading the env
+    load_env(false)
+    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, signal)
+  puts "Stopping all workers"
+  workers.size.times do |i|
+    stop(i, signal)
+  end
+end
+
+# Resque workers respond to a few different signals:
+#
+# QUIT - Wait for child to finish processing then exit
+# TERM / INT - Immediately kill child then exit
+# USR1 - Immediately kill child but don't exit
+# USR2 - Don't start to process any new jobs
+# CONT - Start to process new jobs again after a USR2
+def stop(index, signal)
+  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} - #{signal}"
+    begin
+      Process.kill(signal, 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=nil)
+  return if fork
+  Process::setsid
+  exit!(0) if fork
+  Dir::chdir(Rubber.root)
+  File.umask 0000
+  FileUtils.touch log_file
+  STDIN.reopen    log_file
+  STDOUT.reopen   log_file, "a"
+  STDERR.reopen   log_file, "a"
+  STDOUT.sync = true
+  STDERR.sync = true
+
+  File.open(pid_file, 'w') {|f| f.write("#{Process.pid}") } if pid_file
+
+  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
+signal = ARGV[2] || "QUIT"
+if action.nil? || ! %w[start stop restart].include?(action)
+  puts "Usage: script/resque_worker_management.rb [start|stop|restart]"
+end
+
+# load just the rubber env so things run quickly
+load_env(true)
+workers = Rubber.config.resque_workers
+
+case action
+  when 'start'
+    worker_index ? start(workers[worker_index], worker_index) : start_all(workers)
+    # sleep a bit to allow daemonization to complete
+    sleep 0.5
+  when 'stop'
+    worker_index ? stop(worker_index, signal) : stop_all(workers, signal)
+  when 'restart'
+    if worker_index
+      stop(worker_index, signal)
+      start(workers[worker_index], worker_index)
+    else
+      stop_all(workers, signal)
+      start_all(workers)
+    end
+    # sleep a bit to allow daemonization to complete
+    sleep 0.5
+end

data/generators/vulcanize/templates/resque/templates.rb

@@ -0,0 +1 @@
+gem "resque" if Rubber::Util::is_bundler?

data/generators/vulcanize/templates/resque/templates.yml

@@ -0,0 +1,3 @@
+description: The resque background queue system
+dependent_templates:
+  - redis

data/generators/vulcanize/templates/sphinx/config/rubber/common/sphinx.yml

@@ -0,0 +1,46 @@
+<%
+  @path = "#{RUBBER_ROOT}/config/sphinx.yml"
+%>
+
+# Taken from: http://ts.freelancing-gods.com/rdoc/classes/ThinkingSphinx/Configuration.html
+#
+# Here are the default settings, relative to RUBBER_ROOT where relevant:
+#
+# config file::           config/#{environment}.sphinx.conf
+# searchd log file::      log/searchd.log
+# query log file::        log/searchd.query.log
+# pid file::              log/searchd.#{environment}.pid
+# searchd files::         db/sphinx/#{environment}/
+# address::               127.0.0.1
+# port::                  3312
+# allow star::            false
+# min prefix length::     1
+# min infix length::      1
+# mem limit::             64M
+# max matches::           1000
+# morphology::            stem_en
+# charset type::          utf-8
+# charset table::         nil
+# ignore chars::          nil
+# html strip::            false
+# html remove elements::  ''
+#
+# If you want to change these settings, create a YAML file at
+# config/sphinx.yml with settings for each environment, in a similar
+# fashion to database.yml - using the following keys: config_file,
+# searchd_log_file, query_log_file, pid_file, searchd_file_path, port,
+# allow_star, enable_star, min_prefix_len, min_infix_len, mem_limit,
+# max_matches, # morphology, charset_type, charset_table, ignore_chars,
+# html_strip, # html_remove_elements. I think you've got the idea.
+#
+# Each setting in the YAML file is optional - so only put in the ones you
+# want to change.
+
+<%= RUBBER_ENV %>:
+  address:            <%= rubber_instances.for_role('sphinx').first.full_name rescue 'localhost' %>
+  mem_limit:          <%= RUBBER_ENV == 'production' ? '1024M' : '64M' %>
+  config_file:        sphinx/config/<%= RUBBER_ENV %>.sphinx.conf
+  searchd_file_path:  sphinx/db/<%= RUBBER_ENV %>/
+  pid_file:           <%= RUBBER_ROOT %>/log/searchd.<%= RUBBER_ENV %>.pid
+  searchd_log_file:   <%= RUBBER_ROOT %>/log/searchd.log
+  query_log_file:     <%= RUBBER_ROOT %>/log/searchd.query.log

data/generators/vulcanize/templates/sphinx/config/rubber/deploy-sphinx.rb

@@ -0,0 +1,112 @@
+# installs, starts and stops sphinx
+#
+# Please note that all tasks are executed as runner. So sphinx will run under
+# the same userid as mongrel. This is important to allow delta indexes (mongrel
+# has to send a sighup to searchd).
+#
+# * installation is ubuntu specific
+# * start and stop tasks are using the thinking sphinx plugin
+
+namespace :rubber do
+
+  namespace :sphinx do
+
+    rubber.allow_optional_tasks(self)
+
+    after "rubber:install_packages", "rubber:sphinx:custom_install"
+
+    task :custom_install, :roles => :sphinx do
+      # install sphinx from source
+      ver = "0.9.8.1"
+      rubber.sudo_script 'install_sphinx', <<-ENDSCRIPT
+        # check if already installed
+        if [ -x /usr/local/bin/searchd ]
+          then echo 'Found sphinx searchd on system'
+          if /usr/local/bin/searchd --help | grep 'Sphinx #{ver}'
+            then echo 'Sphinx version matches, no further steps needed'
+            exit 0
+          fi
+        fi
+
+        echo 'Installing / Upgrading sphinx #{ver}'
+        TMPDIR=`mktemp -d` || exit 1
+        cd $TMPDIR
+        echo 'Downloading'
+        wget -qN http://www.sphinxsearch.com/downloads/sphinx-#{ver}.tar.gz
+        echo 'Unpacking'
+        tar xf sphinx-#{ver}.tar.gz
+        cd sphinx-#{ver}
+        ./configure
+        make
+        make install
+        cd ; rm -rf $TMPDIR
+      ENDSCRIPT
+    end
+
+    set :sphinx_root, Proc.new {"#{shared_path}/sphinx"}
+    after "deploy:setup", "rubber:sphinx:setup"
+    after "deploy:symlink", "rubber:sphinx:config_dir"
+
+    before "deploy:stop", "rubber:sphinx:stop"
+    after "deploy:start", "rubber:sphinx:start"
+    after "deploy:restart", "rubber:sphinx:restart"
+    before "deploy:cold" do
+      before "rubber:sphinx:start", "rubber:sphinx:index"
+    end
+    before "rubber:create_staging" do
+      before "rubber:sphinx:start", "rubber:sphinx:index"
+    end
+
+    desc "Do sphinx setup tasks"
+    task :setup, :roles => :sphinx do
+      # Setup links to sphinx config/index as they need to persist between deploys
+      rsudo "mkdir -p #{sphinx_root} #{sphinx_root}/config #{sphinx_root}/db"
+      rsudo "chown -R #{runner}:#{runner} #{sphinx_root}"
+    end
+
+    desc "Setup paths for sphinx runtime"
+    task :config_dir, :roles => :sphinx do
+      rsudo "rm -rf #{current_path}/sphinx"
+      rsudo "ln -sf #{sphinx_root} #{current_path}/sphinx"
+    end
+
+    # runs the given ultrasphinx rake tasks
+    def run_sphinx task
+      rsudo "cd #{current_path} && RAILS_ENV=#{RUBBER_ENV} rake #{task}", :as => runner
+    end
+
+
+    desc "Stops sphinx searchd"
+    task :stop, :roles => :sphinx do
+      run_sphinx 'ts:stop; exit 0'
+    end
+
+    desc "Starts sphinx searchd"
+    task :start, :roles => :sphinx do
+      # rake tasks that load rails env can be slow, so
+      # do multiple here as a performance tweak
+      # config always needs to be run before start as
+      # rubber generates a sphinx config file with new paths
+      run_sphinx 'ts:config ts:start'
+    end
+
+    desc "Restarts sphinx searchd"
+    task :restart, :roles => :sphinx do
+      # rake tasks that load rails env can be slow, so
+      # do multiple here as a performance tweak
+      run_sphinx 'ts:config ts:stop ts:start'
+    end
+
+    desc "Configures sphinx index"
+    task :config, :roles => :sphinx do
+      run_sphinx 'ts:config'
+    end
+
+    desc "Builds sphinx index"
+    task :index, :roles => :sphinx do
+      run_sphinx 'ts:index'
+    end
+
+  end
+
+end