unicorn.rb in caploy-1.0.0

- old
+ new
@@ -1,137 +1,138 @@
-Capistrano::Configuration.instance.load do
+# Check if remote file exists
+#
+def remote_file_exists?(full_path)
+  'true' == capture("if [ -e #{full_path} ]; then echo 'true'; fi").strip
+end
 
-  # Check if remote file exists
-  #
-  def remote_file_exists?(full_path)
-    'true' == capture("if [ -e #{full_path} ]; then echo 'true'; fi").strip
-  end
+# Check if process is running
+#
+def process_exists?(pid_file)
+  capture("ps -p $(cat #{pid_file}) ; true").strip.split("\n").size == 2
+end
 
-  # Check if process is running
-  #
-  def process_exists?(pid_file)
-    capture("ps -p $(cat #{pid_file}) ; true").strip.split("\n").size == 2
-  end
+# Set unicorn vars
+#
+_cset (:app_env) { fetch(:rails_env, 'production') }
+_cset (:unicorn_bin) { "unicorn" }
+_cset (:unicorn_pid) { "#{fetch(:current_path)}/tmp/pids/unicorn.pid" }
+_cset (:unicorn_env) { app_env }
+_cset (:unicorn_std_log) { "log/unicorn.stderr.log" }
+_cset (:unicorn_err_log) { "log/unicorn.stderr.log" }
+_cset (:unicorn_worker_count) { 2 }
+_cset (:unicorn_timeout) { 30 }
+_cset (:unicorn_listen_backlog) { 2048 }
+_cset (:sidekiq_redis_url) { nil }
+_cset (:sidekiq_redis_count) { 1 }
+_cset (:unicorn_hard_restart) { false }
 
-  # Set unicorn vars
-  #
-  _cset (:app_env) { fetch(:rails_env, 'production') }
-  _cset (:unicorn_bin) { "unicorn" }
-  _cset (:unicorn_pid) { "#{fetch(:current_path)}/tmp/pids/unicorn.pid" }
-  _cset (:unicorn_env) { app_env }
-  _cset (:unicorn_std_log) { "log/unicorn.stderr.log" }
-  _cset (:unicorn_err_log) { "log/unicorn.stderr.log" }
-  _cset (:unicorn_worker_processes) { 2 }
-  _cset (:unicorn_timeout) { 30 }
-  _cset (:unicorn_listen_backlog) { 2048 }
-  _cset (:sidekiq_redis_url) { nil }
-  _cset (:sidekiq_redis_count) { 1 }
-  _cset (:unicorn_hard_restart) { false }
-
-  namespace :unicorn do
-    desc 'Start Unicorn'
-    task :start, :roles => :app, :except => {:no_release => true} do
-      if remote_file_exists?(unicorn_pid)
-        if process_exists?(unicorn_pid)
-          logger.important("Unicorn is already running!", "Unicorn")
-          next
-        else
-          run "rm #{unicorn_pid}"
-        end
-      end
-
-      config_path = "#{current_path}/config/unicorn/#{unicorn_env}.rb"
-      if remote_file_exists?(config_path)
-        logger.important("Starting...", "Unicorn - #{bundle_cmd} ")
-        run "cd #{current_path} && #{bundle_cmd} exec #{unicorn_bin} -c #{config_path} -E #{app_env} -D"
+namespace :unicorn do
+  desc 'Start Unicorn'
+  task :start, :roles => :app, :except => { :no_release => true } do
+    if remote_file_exists?(unicorn_pid)
+      if process_exists?(unicorn_pid)
+        logger.important("Unicorn is already running!", "Unicorn")
+        next
       else
-        logger.important("Config file for \"#{unicorn_env}\" environment was not found at \"#{config_path}\"", "Unicorn")
+        run "rm #{unicorn_pid}"
       end
     end
 
-    desc 'Stop Unicorn'
-    task :stop, :roles => :app, :except => {:no_release => true} do
-      if remote_file_exists?(unicorn_pid)
-        if process_exists?(unicorn_pid)
-          logger.important("Stopping...", "Unicorn")
-          run "#{try_sudo} kill `cat #{unicorn_pid}`"
-        else
-          run "rm #{unicorn_pid}"
-          logger.important("Unicorn is not running.", "Unicorn")
-        end
-      else
-        logger.important("No PIDs found. Check if unicorn is running.", "Unicorn")
-      end
+    config_path = "#{current_path}/config/unicorn/#{unicorn_env}.rb"
+    if remote_file_exists?(config_path)
+      logger.important("Starting...", "Unicorn - #{bundle_cmd} ")
+      run "cd #{current_path} && #{bundle_cmd} exec #{unicorn_bin} -c #{config_path} -E #{app_env} -D"
+    else
+      logger.important("Config file for \"#{unicorn_env}\" environment was not found at \"#{config_path}\"", "Unicorn")
     end
+  end
 
-    desc 'Unicorn graceful shutdown'
-    task :graceful_stop, :roles => :app, :except => {:no_release => true} do
-      if remote_file_exists?(unicorn_pid)
-        if process_exists?(unicorn_pid)
-          logger.important("Stopping...", "Unicorn")
-          run "#{try_sudo} kill -s QUIT `cat #{unicorn_pid}`"
-        else
-          run "rm #{unicorn_pid}"
-          logger.important("Unicorn is not running.", "Unicorn")
-        end
+  desc 'Stop Unicorn'
+  task :stop, :roles => :app, :except => { :no_release => true } do
+    if remote_file_exists?(unicorn_pid)
+      if process_exists?(unicorn_pid)
+        logger.important("Stopping...", "Unicorn")
+        run "#{try_sudo} kill `cat #{unicorn_pid}`"
       else
-        logger.important("No PIDs found. Check if unicorn is running.", "Unicorn")
+        run "rm #{unicorn_pid}"
+        logger.important("Unicorn is not running.", "Unicorn")
       end
+    else
+      logger.important("No PIDs found. Check if unicorn is running.", "Unicorn")
     end
+  end
 
-    desc 'Reload Unicorn'
-    task :reload, :roles => :app, :except => {:no_release => true} do
-      if remote_file_exists?(unicorn_pid)
+  desc 'Unicorn graceful shutdown'
+  task :graceful_stop, :roles => :app, :except => { :no_release => true } do
+    if remote_file_exists?(unicorn_pid)
+      if process_exists?(unicorn_pid)
         logger.important("Stopping...", "Unicorn")
-        if unicorn_hard_restart
-          unicorn.stop
-          sleep(2)
-          unicorn.start
-        else
-          run "#{try_sudo} kill -s USR2 `cat #{unicorn_pid}`"
-        end
+        run "#{try_sudo} kill -s QUIT `cat #{unicorn_pid}`"
       else
-        logger.important("No PIDs found. Starting Unicorn server...", "Unicorn")
-        config_path = "#{current_path}/config/unicorn/#{unicorn_env}.rb"
-        if remote_file_exists?(config_path)
-          run "cd #{current_path} && #{bundle_cmd} exec #{unicorn_bin} -c #{config_path} -E #{app_env} -D"
-        else
-          logger.important("Config file for \"#{unicorn_env}\" environment was not found at \"#{config_path}\"", "Unicorn")
-        end
+        run "rm #{unicorn_pid}"
+        logger.important("Unicorn is not running.", "Unicorn")
       end
+    else
+      logger.important("No PIDs found. Check if unicorn is running.", "Unicorn")
     end
+  end
 
-    desc "Setup unicorn"
-    task :setup, :roles => :app, :except => {:no_release => true} do
-      run "mkdir -p \"#{shared_path}/config/unicorn\""
-      config_path = "#{shared_path}/config/unicorn/#{rails_env}.rb"
-      template_path = File.expand_path('../../templates/unicorn/unicorn.rb.erb', __FILE__)
-      vars = {
-          'application' => application,
-          'current_path' => current_path,
-          'unicorn_pid' => unicorn_pid,
-          'unicorn_std_log' => unicorn_std_log,
-          'unicorn_err_log' => unicorn_err_log,
-          'stage' => stage,
-          'unicorn_listen_backlog' => unicorn_listen_backlog,
-          'unicorn_worker_processes' => unicorn_worker_processes,
-          'unicorn_timeout' => unicorn_timeout,
-          'sidekiq_redis_count' => sidekiq_redis_count,
-          'sidekiq_redis_url' => sidekiq_redis_url
-      }
-      put(render_erb_template(template_path, vars), config_path)
+  desc 'Reload Unicorn'
+  task :reload, :roles => :app, :except => { :no_release => true } do
+    if remote_file_exists?(unicorn_pid)
+      logger.important("Stopping...", "Unicorn")
+      if unicorn_hard_restart
+        unicorn.stop
+        sleep(2)
+        unicorn.start
+      else
+        run "#{try_sudo} kill -s USR2 `cat #{unicorn_pid}`"
+      end
+    else
+      logger.important("No PIDs found. Starting Unicorn server...", "Unicorn")
+      config_path = "#{current_path}/config/unicorn/#{unicorn_env}.rb"
+      if remote_file_exists?(config_path)
+        run "cd #{current_path} && #{bundle_cmd} exec #{unicorn_bin} -c #{config_path} -E #{app_env} -D"
+      else
+        logger.important("Config file for \"#{unicorn_env}\" environment was not found at \"#{config_path}\"", "Unicorn")
+      end
     end
   end
 
-  after :"deploy:restart", :"unicorn:reload"
-  after :"deploy:setup", :"unicorn:setup";
+  desc "Setup unicorn"
+  task :setup, :roles => :app, :except => { :no_release => true } do
+    run "mkdir -p \"#{shared_path}/config/unicorn\""
+    config_path = "#{shared_path}/config/unicorn/#{rails_env}.rb"
+    template_path = File.expand_path('../../templates/unicorn/unicorn.rb.erb', __FILE__)
+    vars = {
+        'application' => application,
+        'current_path' => current_path,
+        'unicorn_pid' => unicorn_pid,
+        'unicorn_std_log' => unicorn_std_log,
+        'unicorn_err_log' => unicorn_err_log,
+        'stage' => stage,
+        'unicorn_listen_backlog' => unicorn_listen_backlog,
+        'unicorn_worker_count' => unicorn_worker_count,
+        'unicorn_timeout' => unicorn_timeout,
+        'sidekiq_redis_count' => sidekiq_redis_count,
+        'sidekiq_redis_url' => sidekiq_redis_url
+    }
+    put(render_erb_template(template_path, vars), config_path)
+  end
+end
 
-  namespace :deploy do
-    task :start, :roles => :app do
+after :"deploy:restart", :"unicorn:reload"
+after :"deploy:setup", :"unicorn:setup";
+
+namespace :deploy do
+  task :start do
+    on roles(:app) do
       unicorn.start
     end
+  end
 
-    task :stop, :roles => :app do
+  task :stop do
+    on roles(:app) do
       unicorn.stop
     end
   end
 end