1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
%%%----------------------------------------------------------------------
%%% This is the MIT license.
%%%
%%% Copyright (c) 2014 Peter Lemenkov <lemenkov@gmail.com>
%%%
%%% Permission is hereby granted, free of charge, to any person
%%% obtaining a copy of this software and associated documentation
%%% files (the "Software"), to deal in the Software without
%%% restriction, including without limitation the rights to use,
%%% copy, modify, merge, publish, distribute, sublicense, and/or sell
%%% copies of the Software, and to permit persons to whom the
%%% Software is furnished to do so, subject to the following
%%% conditions:
%%%
%%% The above copyright notice and this permission notice shall be
%%% included in all copies or substantial portions of the Software.
%%%
%%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
%%% EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
%%% OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
%%% NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
%%% HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
%%% WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
%%% FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
%%% OTHER DEALINGS IN THE SOFTWARE.
%%%----------------------------------------------------------------------
-module(sd_notify).
-export([sd_notify/2, sd_notifyf/3, sd_pid_notify/3, sd_pid_notifyf/4, sd_pid_notify_with_fds/4]).
-export([ready/0, reloading/0, stopping/0, watchdog/0]).
-export([start_link/0]).
-export([init/1, handle_info/2, terminate/2]).
% API helpers
ready() -> sd_pid_notify_with_fds(0, false, <<"READY=1">>, []).
reloading() -> sd_pid_notify_with_fds(0, false, <<"RELOADING=1">>, []).
stopping() -> sd_pid_notify_with_fds(0, false, <<"STOPPING=1">>, []).
watchdog() -> sd_pid_notify_with_fds(0, false, <<"WATCHDOG=1">>, []).
% gen_server API and callbacks
start_link() ->
gen_server:start_link({local,?MODULE}, ?MODULE, [], []).
init([]) ->
WatchdogMs = case os:getenv( "WATCHDOG_USEC" ) of
false -> none;
Value ->
Part = erlang:round(0.8 * erlang:list_to_integer(Value)),
erlang:convert_time_unit(Part, microsecond, millisecond)
end,
erlang:send_after(WatchdogMs, self(), watchdog),
error_logger:info_msg("watchdog: ~p ms", [WatchdogMs]),
{ok, WatchdogMs}.
handle_info(watchdog, none) ->
{noreply, none};
handle_info(watchdog, WatchdogMs) ->
watchdog(),
erlang:send_after(WatchdogMs, self(), watchdog),
{noreply, WatchdogMs}.
terminate(_,_) -> ok.
% Systemd API
sd_notify(UnsetEnv, Data) ->
sd_pid_notify_with_fds(0, UnsetEnv, Data, []).
sd_pid_notify(Pid, UnsetEnv, Data) ->
sd_pid_notify_with_fds(Pid, UnsetEnv, Data, []).
sd_notifyf(UnsetEnv, Format, Data) ->
sd_pid_notify_with_fds(0, UnsetEnv, lists:flatten(io_lib:format(Format, Data)), []).
sd_pid_notifyf(Pid, UnsetEnv, Format, Data) ->
sd_pid_notify_with_fds(Pid, UnsetEnv, lists:flatten(io_lib:format(Format, Data)), []).
sd_pid_notify_with_fds(_Pid, UnsetEnv, Call, _Fds) ->
error_logger:info_msg("systemd: ~p", [Call]),
case os:getenv("NOTIFY_SOCKET") of
false -> {error, not_configured};
Path ->
case gen_udp:open(0, [local]) of
{error, SocketError} ->
{error, SocketError};
{ok, Socket} ->
Result = gen_udp:send(Socket, {local,Path}, 0, Call),
gen_udp:close(Socket),
UnsetEnv == true andalso os:unsetenv("NOTIFY_SOCKET"),
Result
end
end.
|
