Module: Sidekiq::SdNotify

Defined in:
lib/sidekiq/sd_notify.rb

Defined Under Namespace

Classes: NotifyError

Constant Summary collapse

READY =
"READY=1"
RELOADING =
"RELOADING=1"
STOPPING =
"STOPPING=1"
STATUS =
"STATUS="
ERRNO =
"ERRNO="
MAINPID =
"MAINPID="
WATCHDOG =
"WATCHDOG=1"
FDSTORE =
"FDSTORE=1"

Class Method Summary collapse

Class Method Details

.errno(errno, unset_env = false) ⇒ Object

Parameters:

  • errno (Integer)


71
72
73
# File 'lib/sidekiq/sd_notify.rb', line 71

def self.errno(errno, unset_env = false)
  notify("#{ERRNO}#{errno}", unset_env)
end

.fdstore(unset_env = false) ⇒ Object



84
85
86
# File 'lib/sidekiq/sd_notify.rb', line 84

def self.fdstore(unset_env = false)
  notify(FDSTORE, unset_env)
end

.mainpid(pid, unset_env = false) ⇒ Object

Parameters:

  • pid (Integer)


76
77
78
# File 'lib/sidekiq/sd_notify.rb', line 76

def self.mainpid(pid, unset_env = false)
  notify("#{MAINPID}#{pid}", unset_env)
end

.notify(state, unset_env = false) ⇒ Fixnum?

Notify systemd with the provided state, via the notification socket, if any.

Generally this method will be used indirectly through the other methods of the library.

Parameters:

  • state (String)
  • unset_env (Boolean) (defaults to: false)

Returns:

  • (Fixnum, nil)

    the number of bytes written to the notification socket or nil if there was no socket to report to (eg. the program wasn’t started by systemd)

Raises:

  • (NotifyError)

    if there was an error communicating with the systemd socket

See Also:



132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
# File 'lib/sidekiq/sd_notify.rb', line 132

def self.notify(state, unset_env = false)
  sock = ENV["NOTIFY_SOCKET"]

  return nil unless sock

  ENV.delete("NOTIFY_SOCKET") if unset_env

  begin
    Addrinfo.unix(sock, :DGRAM).connect do |s|
      s.close_on_exec = true
      s.write(state)
    end
  rescue => e
    raise NotifyError, "#{e.class}: #{e.message}", e.backtrace
  end
end

.ready(unset_env = false) ⇒ Object



52
53
54
# File 'lib/sidekiq/sd_notify.rb', line 52

def self.ready(unset_env = false)
  notify(READY, unset_env)
end

.reloading(unset_env = false) ⇒ Object



56
57
58
# File 'lib/sidekiq/sd_notify.rb', line 56

def self.reloading(unset_env = false)
  notify(RELOADING, unset_env)
end

.status(status, unset_env = false) ⇒ Object

Parameters:

  • status (String)

    a custom status string that describes the current state of the service



66
67
68
# File 'lib/sidekiq/sd_notify.rb', line 66

def self.status(status, unset_env = false)
  notify("#{STATUS}#{status}", unset_env)
end

.stopping(unset_env = false) ⇒ Object



60
61
62
# File 'lib/sidekiq/sd_notify.rb', line 60

def self.stopping(unset_env = false)
  notify(STOPPING, unset_env)
end

.watchdog(unset_env = false) ⇒ Object



80
81
82
# File 'lib/sidekiq/sd_notify.rb', line 80

def self.watchdog(unset_env = false)
  notify(WATCHDOG, unset_env)
end

.watchdog?Boolean

Note:

Unlike sd_watchdog_enabled(3), this method does not mutate the environment.

If the $WATCHDOG_USEC environment variable is set, and the $WATCHDOG_PID variable is unset or set to the PID of the current process

Returns:

  • (Boolean)

    true if the service manager expects watchdog keep-alive notification messages to be sent from this process.



97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# File 'lib/sidekiq/sd_notify.rb', line 97

def self.watchdog?
  wd_usec = ENV["WATCHDOG_USEC"]
  wd_pid = ENV["WATCHDOG_PID"]

  return false unless wd_usec

  begin
    wd_usec = Integer(wd_usec)
  rescue
    return false
  end

  return false if wd_usec <= 0
  return true if !wd_pid || wd_pid == $$.to_s

  false
end