|
1 | 1 | # frozen_string_literal: true |
2 | 2 |
|
3 | 3 | module Capistrano |
4 | | - class Sidekiq::Systemd < Capistrano::Plugin |
5 | | - include SidekiqCommon |
6 | | - def define_tasks |
7 | | - eval_rakefile File.expand_path('../tasks/systemd.rake', __dir__) |
8 | | - eval_rakefile File.expand_path('../tasks/helpers.rake', __dir__) |
9 | | - end |
10 | | - def set_defaults |
11 | | - set_if_empty :systemctl_bin, '/bin/systemctl' |
12 | | - set_if_empty :service_unit_user, :user |
13 | | - set_if_empty :systemctl_user, -> { fetch(:service_unit_user, :user) == :user } |
14 | | - |
15 | | - set_if_empty :sidekiq_systemctl_bin, -> { fetch(:systemctl_bin) } |
16 | | - set_if_empty :sidekiq_service_unit_name, -> { "#{fetch(:application)}_sidekiq_#{fetch(:stage)}" } |
17 | | - |
18 | | - set_if_empty :sidekiq_systemctl_user, -> { fetch(:service_unit_user) } |
19 | | - set_if_empty :sidekiq_enable_lingering, -> { fetch(:sidekiq_systemctl_user) != :system } |
20 | | - set_if_empty :sidekiq_lingering_user, -> { fetch(:lingering_user, fetch(:user)) } |
| 4 | + class Sidekiq |
| 5 | + class Systemd < Capistrano::Plugin |
| 6 | + include SidekiqCommon |
| 7 | + def define_tasks |
| 8 | + eval_rakefile File.expand_path('../tasks/systemd.rake', __dir__) |
| 9 | + eval_rakefile File.expand_path('../tasks/helpers.rake', __dir__) |
| 10 | + end |
21 | 11 |
|
22 | | - ## Sidekiq could have a stripped down or more complex version of the environment variables |
23 | | - set_if_empty :sidekiq_service_unit_env_files, -> { fetch(:service_unit_env_files, []) } |
24 | | - set_if_empty :sidekiq_service_unit_env_vars, -> { fetch(:service_unit_env_vars, []) } |
| 12 | + def set_defaults |
| 13 | + set_if_empty :systemctl_bin, '/bin/systemctl' |
| 14 | + set_if_empty :service_unit_user, :user |
| 15 | + set_if_empty :systemctl_user, -> { fetch(:service_unit_user, :user) == :user } |
25 | 16 |
|
26 | | - set_if_empty :sidekiq_service_templates_path, fetch(:service_templates_path, 'config/deploy/templates') |
27 | | - |
28 | | - # Allow customization of the sidekiq command |
29 | | - set_if_empty :sidekiq_command, 'sidekiq' |
30 | | - set_if_empty :sidekiq_command_args, -> { "-e #{fetch(:sidekiq_env)}" } |
31 | | - |
32 | | - # Deployment tracking for Sidekiq 7+ metrics |
33 | | - set_if_empty :sidekiq_mark_deploy, false |
34 | | - set_if_empty :sidekiq_deploy_label, nil |
35 | | - |
36 | | - # Login shell option for loading environment |
37 | | - set_if_empty :sidekiq_use_login_shell, false |
38 | | - end |
| 17 | + set_if_empty :sidekiq_systemctl_bin, -> { fetch(:systemctl_bin) } |
| 18 | + set_if_empty :sidekiq_service_unit_name, -> { "#{fetch(:application)}_sidekiq_#{fetch(:stage)}" } |
39 | 19 |
|
40 | | - def fetch_systemd_unit_path |
41 | | - if fetch(:sidekiq_systemctl_user) == :system |
42 | | - "/etc/systemd/system/" |
43 | | - else |
44 | | - home_dir = backend.capture :pwd |
45 | | - File.join(home_dir, ".config", "systemd", "user") |
46 | | - end |
47 | | - end |
| 20 | + set_if_empty :sidekiq_systemctl_user, -> { fetch(:service_unit_user) } |
| 21 | + set_if_empty :sidekiq_enable_lingering, -> { fetch(:sidekiq_systemctl_user) != :system } |
| 22 | + set_if_empty :sidekiq_lingering_user, -> { fetch(:lingering_user, fetch(:user)) } |
| 23 | + |
| 24 | + ## Sidekiq could have a stripped down or more complex version of the environment variables |
| 25 | + set_if_empty :sidekiq_service_unit_env_files, -> { fetch(:service_unit_env_files, []) } |
| 26 | + set_if_empty :sidekiq_service_unit_env_vars, -> { fetch(:service_unit_env_vars, []) } |
| 27 | + |
| 28 | + set_if_empty :sidekiq_service_templates_path, fetch(:service_templates_path, 'config/deploy/templates') |
| 29 | + |
| 30 | + # Allow customization of the sidekiq command |
| 31 | + set_if_empty :sidekiq_command, 'sidekiq' |
| 32 | + set_if_empty :sidekiq_command_args, -> { "-e #{fetch(:sidekiq_env)}" } |
48 | 33 |
|
49 | | - def systemd_command(*args) |
50 | | - command = [fetch(:sidekiq_systemctl_bin)] |
| 34 | + # Deployment tracking for Sidekiq 7+ metrics |
| 35 | + set_if_empty :sidekiq_mark_deploy, false |
| 36 | + set_if_empty :sidekiq_deploy_label, nil |
51 | 37 |
|
52 | | - unless fetch(:sidekiq_systemctl_user) == :system |
53 | | - command << "--user" |
| 38 | + # Login shell option for loading environment |
| 39 | + set_if_empty :sidekiq_use_login_shell, false |
54 | 40 | end |
55 | 41 |
|
56 | | - command + args |
57 | | - end |
| 42 | + def fetch_systemd_unit_path |
| 43 | + if fetch(:sidekiq_systemctl_user) == :system |
| 44 | + '/etc/systemd/system/' |
| 45 | + else |
| 46 | + home_dir = backend.capture :pwd |
| 47 | + File.join(home_dir, '.config', 'systemd', 'user') |
| 48 | + end |
| 49 | + end |
| 50 | + |
| 51 | + def systemd_command(*args) |
| 52 | + command = [fetch(:sidekiq_systemctl_bin)] |
| 53 | + |
| 54 | + command << '--user' unless fetch(:sidekiq_systemctl_user) == :system |
58 | 55 |
|
59 | | - def sudo_if_needed(*command) |
60 | | - if fetch(:sidekiq_systemctl_user) == :system |
61 | | - backend.sudo command.map(&:to_s).join(" ") |
62 | | - else |
63 | | - backend.execute(*command) |
| 56 | + command + args |
| 57 | + end |
| 58 | + |
| 59 | + def sudo_if_needed(*command) |
| 60 | + if fetch(:sidekiq_systemctl_user) == :system |
| 61 | + backend.sudo command.map(&:to_s).join(' ') |
| 62 | + else |
| 63 | + backend.execute(*command) |
| 64 | + end |
64 | 65 | end |
65 | | - end |
66 | 66 |
|
67 | | - def execute_systemd(*args) |
68 | | - sudo_if_needed(*systemd_command(*args)) |
| 67 | + def execute_systemd(*) |
| 68 | + sudo_if_needed(*systemd_command(*)) |
| 69 | + end |
69 | 70 | end |
70 | 71 | end |
71 | 72 | end |
0 commit comments