capistrano-sidekiq 1.0.1 → 2.0.0.beta3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 26aff29f42bea923b68df1bb882bd0ae1a4289ee
-  data.tar.gz: b512116ee2722510550958339899136abb399958
+SHA256:
+  metadata.gz: 01e37e98d391172380ba15ee0c478e8669ad869cf4be1b83a8c61e2ef4ba9dfb
+  data.tar.gz: 418a99c90061cababdd5f0e3907b967f44fda2a7c25460f749f7738e12e072df
 SHA512:
-  metadata.gz: 8b8b90a8bfd058f771270b665bf6c76d43a99312f85ce64ffaf8e6ae01ce14e3f6029f1831f99cb9cf6df926df2ce4c9bade86ee6283a06b1093f9e426d0db05
-  data.tar.gz: 6ab3ce241dcc7d8c66ee5e017290187b7d8ed6afc8198c416262c5f3d48d17012d3d970db8a5f4a85773566ed009b0924618a3b28031903470bae19759e428f6
+  metadata.gz: 184f40bb8a1d41fd346cd32c5f55dff603888b462aa38a68b39faac4ce25836232109ca92e4bb9ba458ca4915cbf9c06a6228dabd5eb1a6b464dc389bebd9eaa
+  data.tar.gz: b616d1322ca5d835e808265d3303852428da7d152be27d6472954a02d3a894536c45c29f6c086e2e8cb2b32f0276cb342294766d7f9e99c4e62864344a597793

data/.gitignore

@@ -16,3 +16,4 @@ test/tmp
 test/version_tmp
 tmp
 .idea
+*.iml

data/.tool-versions

@@ -0,0 +1 @@
+ruby 2.6.5

data/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## 1.0.3
+- Add upstart support
+- Fix monit service name
+
+## 1.0.2
+- Fix deployment flow hooks
+
 ## 1.0.1
 - Add deploy failure handling @phillbaker
 - Support custom monit filename @zocoi

data/README.md

@@ -1,5 +1,4 @@
 [![Gem Version](https://badge.fury.io/rb/capistrano-sidekiq.svg)](http://badge.fury.io/rb/capistrano-sidekiq)
-[![Dependency Status](https://gemnasium.com/seuros/capistrano-sidekiq.svg)](https://gemnasium.com/seuros/capistrano-sidekiq)
 
 # Capistrano::Sidekiq
 
@@ -7,12 +6,6 @@ Sidekiq integration for Capistrano
 
 ## Installation
 
-Add this line to your application's Gemfile:
-
-    gem 'capistrano-sidekiq', github: 'seuros/capistrano-sidekiq'
-
-or:
-
     gem 'capistrano-sidekiq', group: :development
 
 And then execute:
@@ -22,29 +15,34 @@ And then execute:
 
 ## Usage
 ```ruby
-# Capfile
-require 'capistrano/sidekiq'
-require 'capistrano/sidekiq/monit' #to require monit tasks # Only for capistrano3
+    # Capfile
+
+    require 'capistrano/sidekiq'
+    install_plugin Capistrano::Sidekiq  # Default sidekiq tasks
+    # Then select your service manager
+    install_plugin Capistrano::Sidekiq::Systemd 
+    # or  
+    install_plugin Capistrano::Sidekiq::Upstart  # tests needed
+    # or  
+    install_plugin Capistrano::Sidekiq::Monit  # tests needed
 ```
 
 
 Configurable options, shown here with defaults:
 
 ```ruby
+:sidekiq_roles => :app
 :sidekiq_default_hooks => true
 :sidekiq_pid => File.join(shared_path, 'tmp', 'pids', 'sidekiq.pid') # ensure this path exists in production before deploying.
 :sidekiq_env => fetch(:rack_env, fetch(:rails_env, fetch(:stage)))
 :sidekiq_log => File.join(shared_path, 'log', 'sidekiq.log')
-:sidekiq_options => nil
-:sidekiq_require => nil
-:sidekiq_tag => nil
-:sidekiq_config => nil # if you have a config/sidekiq.yml, do not forget to set this. 
-:sidekiq_queue => nil
-:sidekiq_timeout => 10
-:sidekiq_roles => :app
-:sidekiq_processes => 1
-:sidekiq_options_per_process => nil
-:sidekiq_concurrency => nil
+
+# sidekiq systemd options
+:sidekiq_service_unit_name => 'sidekiq'
+:sidekiq_service_unit_user => :user # :system
+:sidekiq_enable_lingering => true
+:sidekiq_lingering_user => nil
+
 # sidekiq monit
 :sidekiq_monit_templates_path => 'config/deploy/templates'
 :sidekiq_monit_conf_dir => '/etc/monit/conf.d'
@@ -52,10 +50,8 @@ Configurable options, shown here with defaults:
 :monit_bin => '/usr/bin/monit'
 :sidekiq_monit_default_hooks => true
 :sidekiq_monit_group => nil
-:sidekiq_service_name => "sidekiq_#{fetch(:application)}_#{fetch(:sidekiq_env)}" + (index ? "_#{index}" : '') 
+:sidekiq_service_name => "sidekiq_#{fetch(:application)}" 
 
-:sidekiq_cmd => "#{fetch(:bundle_cmd, "bundle")} exec sidekiq" # Only for capistrano2.5
-:sidekiqctl_cmd => "#{fetch(:bundle_cmd, "bundle")} exec sidekiqctl" # Only for capistrano2.5
 :sidekiq_user => nil #user to run sidekiq as
 ```
 
@@ -75,50 +71,6 @@ SSHKit.config.command_map[:sidekiq] = "bundle exec sidekiq"
 SSHKit.config.command_map[:sidekiqctl] = "bundle exec sidekiqctl"
 ```
 
-## Multiple processes
-
-You can configure sidekiq to start with multiple processes. Just set the proper amount in `sidekiq_processes`.
-
-You can also customize the configuration for every process. If you want to do that, just set
-`sidekiq_options_per_process` with an array of the configuration options that you want in string format.
-This example should boot the first process with the queue `high` and the second one with the queues `default`
-and `low`:
-
-```ruby
-set :sidekiq_options_per_process, ["--queue high", "--queue default --queue low"]
-```
-
-## Different number of processes per role
-
-You can configure how many processes you want to run on each host next way:
-
-```ruby
-set :sidekiq_roles, [:sidekiq_small, :sidekiq_big]
-set :sidekiq_small_processes, 1
-set :sidekiq_big_processes, 4
-server 'example-small.com', roles: [:sidekiq_small]
-server 'example-big.com', roles: [:sidekiq_big]
-```
-
-## Integration with systemd
-
-Set init system to systemd in the cap deploy config:
-
-```ruby
-set :init_system, :systemd
-```
-
-Install systemd.service template file and enable the service with:
-
-```
-bundle exec cap sidekiq:install
-```
-
-Default name for the service file is `sidekiq-stage.service`. This can be changed as needed, for example:
-
-```ruby
-set :service_unit_name, "sidekiq-#{fetch(:application)}-#{fetch(:stage)}.service"
-```
 
 ## Customizing the monit sidekiq templates
 

data/capistrano-sidekiq.gemspec

@@ -18,5 +18,6 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
 
   spec.add_dependency 'capistrano', '>= 3.9.0'
-  spec.add_dependency 'sidekiq', '>= 3.4'
+  spec.add_dependency 'capistrano-bundler'
+  spec.add_dependency 'sidekiq', '>= 6.0'
 end

data/lib/capistrano/sidekiq.rb

@@ -1 +1,30 @@
-load File.expand_path('../tasks/sidekiq.rake', __FILE__)
+require 'capistrano/bundler'
+require "capistrano/plugin"
+
+module Capistrano
+  class Sidekiq < Capistrano::Plugin
+    def define_tasks
+      eval_rakefile File.expand_path('../tasks/sidekiq.rake', __FILE__)
+    end
+
+    def set_defaults
+      set_if_empty :sidekiq_default_hooks, true
+
+      set_if_empty :sidekiq_env, -> { fetch(:rack_env, fetch(:rails_env, fetch(:rake_env, fetch(:stage)))) }
+      set_if_empty :sidekiq_roles, fetch(:sidekiq_role, :app)
+      set_if_empty :sidekiq_log, -> { File.join(shared_path, 'log', 'sidekiq.log') }
+      set_if_empty :sidekiq_error_log, -> { File.join(shared_path, 'log', 'sidekiq.error.log') }
+      # Rbenv, Chruby, and RVM integration
+      append :rbenv_map_bins, 'sidekiq', 'sidekiqctl'
+      append :rvm_map_bins, 'sidekiq', 'sidekiqctl'
+      append :chruby_map_bins, 'sidekiq', 'sidekiqctl'
+      # Bundler integration
+      append :bundle_bins, 'sidekiq', 'sidekiqctl'
+    end
+
+  end
+end
+
+require_relative 'sidekiq/systemd'
+require_relative 'sidekiq/upstart'
+require_relative 'sidekiq/monit'

data/lib/capistrano/sidekiq/monit.rb

@@ -1 +1,17 @@
-load File.expand_path('../../tasks/monit.rake', __FILE__)
+module Capistrano
+  class Sidekiq::Monit < Capistrano::Plugin
+    def set_defaults
+      set_if_empty :monit_bin, '/usr/bin/monit'
+      set_if_empty :sidekiq_monit_conf_dir, '/etc/monit/conf.d'
+      set_if_empty :sidekiq_monit_conf_file, -> { "#{sidekiq_service_name}.conf" }
+      set_if_empty :sidekiq_monit_use_sudo, true
+      set_if_empty :sidekiq_monit_default_hooks, true
+      set_if_empty :sidekiq_monit_templates_path, 'config/deploy/templates'
+      set_if_empty :sidekiq_monit_group, nil
+    end
+
+    def define_tasks
+      eval_rakefile File.expand_path('../../tasks/monit.rake', __FILE__)
+    end
+  end
+end

data/lib/capistrano/sidekiq/systemd.rb

@@ -0,0 +1,14 @@
+module Capistrano
+  class Sidekiq::Systemd < Capistrano::Plugin
+    def set_defaults
+      set_if_empty :sidekiq_service_unit_name, 'sidekiq'
+      set_if_empty :sidekiq_service_unit_user, :user # :system
+      set_if_empty :sidekiq_enable_lingering, true
+      set_if_empty :sidekiq_lingering_user, nil
+    end
+
+    def define_tasks
+      eval_rakefile File.expand_path('../../tasks/systemd.rake', __FILE__)
+    end
+  end
+end

data/lib/capistrano/sidekiq/upstart.rb

@@ -0,0 +1,11 @@
+module Capistrano
+  class Sidekiq::Upstart < Capistrano::Plugin
+    def set_defaults
+      set_if_empty :sidekiq_service_unit_name, 'sidekiq'
+    end
+
+    def define_tasks
+      eval_rakefile File.expand_path('../../tasks/upstart.rake', __FILE__)
+    end
+  end
+end

data/lib/capistrano/sidekiq/version.rb

@@ -1,3 +1,3 @@
 module Capistrano
-  SidekiqVERSION = '1.0.1'
+  SidekiqVERSION = '2.0.0.beta3'
 end

data/lib/capistrano/tasks/monit.rake

@@ -1,7 +1,7 @@
 namespace :load do
   task :defaults do
     set :sidekiq_monit_conf_dir, '/etc/monit/conf.d'
-    set :sidekiq_monit_conf_file, "#{sidekiq_service_name}.conf"
+    set :sidekiq_monit_conf_file, -> { "#{sidekiq_service_name}.conf" }
     set :sidekiq_monit_use_sudo, true
     set :monit_bin, '/usr/bin/monit'
     set :sidekiq_monit_default_hooks, true

data/lib/capistrano/tasks/sidekiq.rake

@@ -1,285 +1,20 @@
-namespace :load do
-  task :defaults do
-    set :sidekiq_default_hooks, true
-
-    set :sidekiq_pid, -> { File.join(shared_path, 'tmp', 'pids', 'sidekiq.pid') }
-    set :sidekiq_env, -> { fetch(:rack_env, fetch(:rails_env, fetch(:stage))) }
-    set :sidekiq_log, -> { File.join(shared_path, 'log', 'sidekiq.log') }
-    set :sidekiq_timeout, 10
-    set :sidekiq_roles, fetch(:sidekiq_role, :app)
-    set :sidekiq_processes, 1
-    set :sidekiq_options_per_process, nil
-    set :sidekiq_user, nil
-    # Rbenv, Chruby, and RVM integration
-    set :rbenv_map_bins, fetch(:rbenv_map_bins).to_a.concat(%w[sidekiq sidekiqctl])
-    set :rvm_map_bins, fetch(:rvm_map_bins).to_a.concat(%w[sidekiq sidekiqctl])
-    set :chruby_map_bins, fetch(:chruby_map_bins).to_a.concat(%w[sidekiq sidekiqctl])
-    # Bundler integration
-    set :bundle_bins, fetch(:bundle_bins).to_a.concat(%w[sidekiq sidekiqctl])
-    # Init system integration
-    set :init_system, -> { nil }
-    # systemd integration
-    set :service_unit_name, "sidekiq-#{fetch(:stage)}.service"
-  end
-end
-
 namespace :deploy do
   before :starting, :check_sidekiq_hooks do
     invoke 'sidekiq:add_default_hooks' if fetch(:sidekiq_default_hooks)
   end
-  after :publishing, :restart_sidekiq do
-    invoke 'sidekiq:restart' if fetch(:sidekiq_default_hooks)
-  end
-  after :failed, :restart_sidekiq do
-    invoke 'sidekiq:restart' if fetch(:sidekiq_default_hooks)
-  end
 end
 
 namespace :sidekiq do
   task :add_default_hooks do
-    after 'deploy:starting',  'sidekiq:quiet'
-    after 'deploy:updated',   'sidekiq:stop'
-    after 'deploy:reverted',  'sidekiq:stop'
+    after 'deploy:starting', 'sidekiq:quiet'
+    after 'deploy:updated', 'sidekiq:stop'
     after 'deploy:published', 'sidekiq:start'
-  end
-
-  desc 'Quiet sidekiq (stop fetching new tasks from Redis)'
-  task :quiet do
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        case fetch(:init_system)
-        when :systemd
-          execute :systemctl, "--user", "reload", fetch(:service_unit_name), raise_on_non_zero_exit: false
-        else
-          if test("[ -d #{release_path} ]")
-            each_process_with_index(reverse: true) do |pid_file, _idx|
-              if pid_file_exists?(pid_file) && process_exists?(pid_file)
-                quiet_sidekiq(pid_file)
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-
-  desc 'Stop sidekiq (graceful shutdown within timeout, put unfinished tasks back to Redis)'
-  task :stop do
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        case fetch(:init_system)
-        when :systemd
-          execute :systemctl, "--user", "stop", fetch(:service_unit_name)
-        else
-          if test("[ -d #{release_path} ]")
-            each_process_with_index(reverse: true) do |pid_file, _idx|
-              if pid_file_exists?(pid_file) && process_exists?(pid_file)
-                stop_sidekiq(pid_file)
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-
-  desc 'Start sidekiq'
-  task :start do
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        case fetch(:init_system)
-        when :systemd
-          execute :systemctl, '--user', 'start', fetch(:service_unit_name)
-        else
-          each_process_with_index do |pid_file, idx|
-            unless pid_file_exists?(pid_file) && process_exists?(pid_file)
-              start_sidekiq(pid_file, idx)
-            end
-          end
-        end
-      end
-    end
+    after 'deploy:failed', 'sidekiq:restart'
   end
 
   desc 'Restart sidekiq'
   task :restart do
     invoke! 'sidekiq:stop'
-    invoke 'sidekiq:start'
-  end
-
-  desc 'Rolling-restart sidekiq'
-  task :rolling_restart do
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        each_process_with_index(reverse: true) do |pid_file, idx|
-          if pid_file_exists?(pid_file) && process_exists?(pid_file)
-            stop_sidekiq(pid_file)
-          end
-          start_sidekiq(pid_file, idx)
-        end
-      end
-    end
-  end
-
-  desc 'Delete any pid file not in use'
-  task :cleanup do
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        each_process_with_index do |pid_file, _idx|
-          unless process_exists?(pid_file)
-            next unless pid_file_exists?(pid_file)
-            execute "rm #{pid_file}"
-          end
-        end
-      end
-    end
-  end
-
-  # TODO: Don't start if all processes are off, raise warning.
-  desc 'Respawn missing sidekiq processes'
-  task :respawn do
-    invoke 'sidekiq:cleanup'
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        each_process_with_index do |pid_file, idx|
-          start_sidekiq(pid_file, idx) unless pid_file_exists?(pid_file)
-        end
-      end
-    end
-  end
-
-  task :install do
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        case fetch(:init_system)
-        when :systemd
-          create_systemd_template
-          execute :systemctl, "--user", "enable", fetch(:service_unit_name)
-        end
-      end
-    end
-  end
-
-  task :uninstall do
-    on roles fetch(:sidekiq_roles) do |role|
-      switch_user(role) do
-        case fetch(:init_system)
-        when :systemd
-          execute :systemctl, "--user", "disable", fetch(:service_unit_name)
-          execute :rm, File.join(fetch(:service_unit_path, fetch_systemd_unit_path),fetch(:service_unit_name))
-        end
-      end
-    end
-  end
-
-  def each_process_with_index(reverse: false)
-    pid_file_list = pid_files
-    pid_file_list.reverse! if reverse
-    pid_file_list.each_with_index do |pid_file, idx|
-      within release_path do
-        yield(pid_file, idx)
-      end
-    end
-  end
-
-  def fetch_systemd_unit_path
-    home_dir = capture :pwd
-    File.join(home_dir, ".config", "systemd", "user")
-  end
-
-  def create_systemd_template
-    search_paths = [
-      File.expand_path(
-        File.join(*%w[.. .. .. generators capistrano sidekiq systemd templates sidekiq.service.capistrano.erb]),
-        __FILE__
-      ),
-    ]
-    template_path = search_paths.detect {|path| File.file?(path)}
-    template = File.read(template_path)
-    systemd_path = fetch(:service_unit_path, fetch_systemd_unit_path)
-    execute :mkdir, "-p", systemd_path
-    upload!(
-      StringIO.new(ERB.new(template).result(binding)),
-      "#{systemd_path}/#{fetch :service_unit_name}"
-    )
-    execute :systemctl, "--user", "daemon-reload"
-  end
-
-  def pid_files
-    sidekiq_roles = Array(fetch(:sidekiq_roles)).dup
-    sidekiq_roles.select! { |role| host.roles.include?(role) }
-    sidekiq_roles.flat_map do |role|
-      processes = fetch(:"#{ role }_processes") || fetch(:sidekiq_processes)
-      Array.new(processes) { |idx| fetch(:sidekiq_pid).gsub(/\.pid$/, "-#{idx}.pid") }
-    end
-  end
-
-  def pid_file_exists?(pid_file)
-    test(*("[ -f #{pid_file} ]").split(' '))
-  end
-
-  def process_exists?(pid_file)
-    test(*("kill -0 $( cat #{pid_file} )").split(' '))
-  end
-
-  def quiet_sidekiq(pid_file)
-    begin
-      execute :sidekiqctl, 'quiet', pid_file.to_s
-    rescue SSHKit::Command::Failed
-      # If gems are not installed (first deploy) and sidekiq_default_hooks is active
-      warn 'sidekiqctl not found (ignore if this is the first deploy)'
-    end
-  end
-
-  def stop_sidekiq(pid_file)
-    execute :sidekiqctl, 'stop', pid_file.to_s, fetch(:sidekiq_timeout)
-  end
-
-  def start_sidekiq(pid_file, idx = 0)
-    args = []
-    args.push "--index #{idx}"
-    args.push "--pidfile #{pid_file}"
-    args.push "--environment #{fetch(:sidekiq_env)}"
-    args.push "--logfile #{fetch(:sidekiq_log)}" if fetch(:sidekiq_log)
-    args.push "--require #{fetch(:sidekiq_require)}" if fetch(:sidekiq_require)
-    args.push "--tag #{fetch(:sidekiq_tag)}" if fetch(:sidekiq_tag)
-    Array(fetch(:sidekiq_queue)).each do |queue|
-      args.push "--queue #{queue}"
-    end
-    args.push "--config #{fetch(:sidekiq_config)}" if fetch(:sidekiq_config)
-    args.push "--concurrency #{fetch(:sidekiq_concurrency)}" if fetch(:sidekiq_concurrency)
-    if (process_options = fetch(:sidekiq_options_per_process))
-      args.push process_options[idx]
-    end
-    # use sidekiq_options for special options
-    args.push fetch(:sidekiq_options) if fetch(:sidekiq_options)
-
-    if defined?(JRUBY_VERSION)
-      args.push '>/dev/null 2>&1 &'
-      warn 'Since JRuby doesn\'t support Process.daemon, Sidekiq will not be running as a daemon.'
-    else
-      args.push '--daemon'
-    end
-
-    execute :sidekiq, args.compact.join(' ')
-  end
-
-  def switch_user(role)
-    su_user = sidekiq_user(role)
-    if su_user == role.user
-      yield
-    else
-      as su_user do
-        yield
-      end
-    end
-  end
-
-  def sidekiq_user(role)
-    properties = role.properties
-    properties.fetch(:sidekiq_user) || # local property for sidekiq only
-      fetch(:sidekiq_user) ||
-      properties.fetch(:run_as) || # global property across multiple capistrano gems
-      role.user
+    invoke! 'sidekiq:start'
   end
 end

data/lib/capistrano/tasks/systemd.rake

@@ -0,0 +1,135 @@
+git_plugin = self
+
+namespace :sidekiq do
+  desc 'Quiet sidekiq (stop fetching new tasks from Redis)'
+  task :quiet do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        if fetch(:sidekiq_service_unit_user) == :system
+          execute :sudo, :systemctl, "reload", fetch(:sidekiq_service_unit_name), raise_on_non_zero_exit: false
+        else
+          execute :systemctl, "--user", "reload", fetch(:sidekiq_service_unit_name), raise_on_non_zero_exit: false
+        end
+      end
+    end
+  end
+
+  desc 'Stop sidekiq (graceful shutdown within timeout, put unfinished tasks back to Redis)'
+  task :stop do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        if fetch(:sidekiq_service_unit_user) == :system
+          execute :sudo, :systemctl, "stop", fetch(:sidekiq_service_unit_name)
+        else
+          execute :systemctl, "--user", "stop", fetch(:sidekiq_service_unit_name)
+        end
+      end
+    end
+  end
+
+  desc 'Start sidekiq'
+  task :start do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        if fetch(:sidekiq_service_unit_user) == :system
+          execute :sudo, :systemctl, 'start', fetch(:sidekiq_service_unit_name)
+        else
+          execute :systemctl, '--user', 'start', fetch(:sidekiq_service_unit_name)
+        end
+      end
+    end
+  end
+
+  desc 'Install systemd sidekiq service'
+  task :install do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        git_plugin.create_systemd_template
+        if fetch(:sidekiq_service_unit_user) == :system
+          execute :sudo, :systemctl, "enable", fetch(:sidekiq_service_unit_name)
+        else
+          execute :systemctl, "--user", "enable", fetch(:sidekiq_service_unit_name)
+          execute :loginctl, "enable-linger", fetch(:sidekiq_lingering_user) if fetch(:sidekiq_enable_lingering)
+        end
+      end
+    end
+  end
+
+  desc 'UnInstall systemd sidekiq service'
+  task :uninstall do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        if fetch(:sidekiq_service_unit_user) == :system
+          execute :sudo, :systemctl, "disable", fetch(:sidekiq_service_unit_name)
+        else
+          execute :systemctl, "--user", "disable", fetch(:sidekiq_service_unit_name)
+        end
+        execute :rm, '-f', File.join(fetch(:service_unit_path, fetch_systemd_unit_path), fetch(:sidekiq_service_unit_name))
+      end
+    end
+  end
+
+  desc 'Generate service_locally'
+  task :generate_service_locally do
+    run_locally do
+      File.write('sidekiq', git_plugin.compiled_template)
+    end
+  end
+
+  def fetch_systemd_unit_path
+    if fetch(:sidekiq_service_unit_user) == :system
+      # if the path is not standard `set :service_unit_path`
+      "/etc/systemd/system/"
+    else
+      home_dir = backend.capture :pwd
+      File.join(home_dir, ".config", "systemd", "user")
+    end
+  end
+
+  def compiled_template
+    search_paths = [
+      File.expand_path(
+          File.join(*%w[.. .. .. generators capistrano sidekiq systemd templates sidekiq.service.capistrano.erb]),
+          __FILE__
+      ),
+    ]
+    template_path = search_paths.detect { |path| File.file?(path) }
+    template = File.read(template_path)
+    ERB.new(template).result(binding)
+  end
+
+  def create_systemd_template
+    ctemplate = compiled_template
+    systemd_path = fetch(:service_unit_path, fetch_systemd_unit_path)
+
+    if fetch(:sidekiq_service_unit_user) == :user
+      backend.execute :mkdir, "-p", systemd_path
+    end
+    backend.upload!(
+        StringIO.new(ctemplate),
+        "/tmp/#{fetch :sidekiq_service_unit_name}.service"
+    )
+    if fetch(:sidekiq_service_unit_user) == :system
+      backend.execute :sudo, :mv, "/tmp/#{fetch :sidekiq_service_unit_name}.service", "#{systemd_path}/#{fetch :sidekiq_service_unit_name}.service"
+      backend.execute :sudo, :systemctl, "daemon-reload"
+    else
+      backend.execute :sudo, :mv, "/tmp/#{fetch :sidekiq_service_unit_name}.service", "#{systemd_path}/#{fetch :sidekiq_service_unit_name}.service"
+      backend.execute :systemctl, "--user", "daemon-reload"
+    end
+  end
+
+  def switch_user(role)
+    su_user = sidekiq_user
+    if su_user != role.user
+      yield
+    else
+      as su_user do
+        yield
+      end
+    end
+  end
+
+  def sidekiq_user
+    fetch(:sidekiq_user, fetch(:run_as))
+  end
+end

data/lib/capistrano/tasks/upstart.rake

@@ -0,0 +1,47 @@
+namespace :sidekiq do
+  desc 'Quiet sidekiq (stop fetching new tasks from Redis)'
+  task :quiet do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        sudo :service, fetch(:sidekiq_service_unit_name), :reload
+      end
+    end
+  end
+
+  desc 'Stop sidekiq (graceful shutdown within timeout, put unfinished tasks back to Redis)'
+  task :stop do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        sudo :service, fetch(:sidekiq_service_unit_name), :stop
+      end
+    end
+  end
+
+  desc 'Start sidekiq'
+  task :start do
+    on roles fetch(:sidekiq_roles) do |role|
+      git_plugin.switch_user(role) do
+        sudo :service, fetch(:sidekiq_service_unit_name), :start
+      end
+    end
+  end
+
+  def switch_user(role)
+    su_user = sidekiq_user(role)
+    if su_user == role.user
+      yield
+    else
+      as su_user do
+        yield
+      end
+    end
+  end
+
+  def sidekiq_user(role)
+    properties = role.properties
+    properties.fetch(:sidekiq_user) || # local property for sidekiq only
+        fetch(:sidekiq_user) ||
+        properties.fetch(:run_as) || # global property across multiple capistrano gems
+        role.user
+  end
+end

data/lib/generators/capistrano/sidekiq/monit/templates/sidekiq_monit.conf.erb

@@ -5,6 +5,6 @@ check process <%= sidekiq_service_name(idx) %>
   start program = "/bin/su - <%= sidekiq_user(@role) %> -c 'cd <%= current_path %> && <%= SSHKit.config.command_map[:sidekiq] %> <%= sidekiq_config %> --index <%= idx %> --pidfile <%= pid_file %> --environment <%= fetch(:sidekiq_env) %> <%= sidekiq_concurrency %> <%= sidekiq_logfile %> <%= sidekiq_require %> <%= sidekiq_queues %> <%= sidekiq_options_per_process[idx] %> -d'" with timeout 30 seconds
 
   stop program = "/bin/su - <%= sidekiq_user(@role) %> -c 'cd <%= current_path %> && <%= SSHKit.config.command_map[:sidekiqctl] %> stop <%= pid_file %>'" with timeout <%= fetch(:sidekiq_timeout).to_i + 10  %> seconds
-  group <%= fetch(:sidekiq_monit_group, fetch(:application)) %>-sidekiq
+  group <%= fetch(:sidekiq_monit_group) || fetch(:application) %>-sidekiq
 
 <% end %>

data/lib/generators/capistrano/sidekiq/systemd/templates/sidekiq.service.capistrano.erb

@@ -4,11 +4,18 @@ After=syslog.target network.target
 
 [Service]
 Type=simple
-Environment=RAILS_ENV=<%= fetch(:rails_env) %>
-WorkingDirectory=<%= fetch(:deploy_to) %>/current
-ExecStart=<%= fetch(:bundler_path, '/usr/local/bin/bundler') %> exec sidekiq -e <%= fetch(:rails_env) %>
+WorkingDirectory=<%= File.join(fetch(:deploy_to), 'current') %>
+ExecStart=<%= SSHKit.config.command_map[:bundler] %> exec sidekiq -e <%= fetch(:sidekiq_env) %>
 ExecReload=/bin/kill -TSTP $MAINPID
 ExecStop=/bin/kill -TERM $MAINPID
+<%="StandardOutput=append:#{fetch(:sidekiq_log)}" if fetch(:sidekiq_log) %>
+<%="StandardError=append:#{fetch(:sidekiq_error_log)}" if fetch(:sidekiq_error_log) %>
+<%="User=#{sidekiq_user}" if sidekiq_user %>
+<%="EnvironmentFile=#{fetch(:sidekiq_service_unit_env_file)}" if fetch(:sidekiq_service_unit_env_file) %>
+
+<% fetch(:sidekiq_service_unit_env_vars, []).each do |environment_variable| %>
+  <%="Environment=#{environment_variable}" %>
+<% end %>
 
 RestartSec=1
 Restart=on-failure

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-sidekiq
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 2.0.0.beta3
 platform: ruby
 authors:
 - Abdelkader Boudih
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-04-04 00:00:00.000000000 Z
+date: 2020-05-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: capistrano
@@ -24,20 +24,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 3.9.0
+- !ruby/object:Gem::Dependency
+  name: capistrano-bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: sidekiq
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '3.4'
+        version: '6.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '3.4'
+        version: '6.0'
 description: Sidekiq integration for Capistrano
 email:
 - terminale@gmail.com
@@ -46,6 +60,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".tool-versions"
 - CHANGELOG.md
 - CONTRIBUTORS.md
 - Gemfile
@@ -56,9 +71,13 @@ files:
 - lib/capistrano-sidekiq.rb
 - lib/capistrano/sidekiq.rb
 - lib/capistrano/sidekiq/monit.rb
+- lib/capistrano/sidekiq/systemd.rb
+- lib/capistrano/sidekiq/upstart.rb
 - lib/capistrano/sidekiq/version.rb
 - lib/capistrano/tasks/monit.rake
 - lib/capistrano/tasks/sidekiq.rake
+- lib/capistrano/tasks/systemd.rake
+- lib/capistrano/tasks/upstart.rake
 - lib/generators/capistrano/sidekiq/monit/template_generator.rb
 - lib/generators/capistrano/sidekiq/monit/templates/sidekiq_monit.conf.erb
 - lib/generators/capistrano/sidekiq/systemd/templates/sidekiq.service.capistrano.erb
@@ -77,12 +96,11 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: 2.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">="
+  - - ">"
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 1.3.1
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.6.14
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: Sidekiq integration for Capistrano