./ct_report/coverage/ejabberd_app.COVER.html

1 %%%----------------------------------------------------------------------
2 %%% File : ejabberd_app.erl
3 %%% Author : Alexey Shchepin <alexey@process-one.net>
4 %%% Purpose : ejabberd's application callback module
5 %%% Created : 31 Jan 2003 by Alexey Shchepin <alexey@process-one.net>
6 %%%
7 %%%
8 %%% ejabberd, Copyright (C) 2002-2011 ProcessOne
9 %%%
10 %%% This program is free software; you can redistribute it and/or
11 %%% modify it under the terms of the GNU General Public License as
12 %%% published by the Free Software Foundation; either version 2 of the
13 %%% License, or (at your option) any later version.
14 %%%
15 %%% This program is distributed in the hope that it will be useful,
16 %%% but WITHOUT ANY WARRANTY; without even the implied warranty of
17 %%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 %%% General Public License for more details.
19 %%%
20 %%% You should have received a copy of the GNU General Public License
21 %%% along with this program; if not, write to the Free Software
22 %%% Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 %%%
24 %%%----------------------------------------------------------------------
25
26 -module(ejabberd_app).
27 -author('alexey@process-one.net').
28
29 -behaviour(application).
30
31 -export([start/2, prep_stop/1, stop/1]).
32
33 -ignore_xref([prep_stop/1]).
34
35 -include("mongoose.hrl").
36
37
38 %%%
39 %%% Application API
40 %%%
41
42 start(normal, _Args) ->
43 101 try
44 101 do_start()
45 catch Class:Reason:StackTrace ->
46 %% Log a stacktrace because while proc_lib:crash_report/4 would report a crash reason,
47 %% it would not report the stacktrace
48
:-(
?LOG_CRITICAL(#{what => app_failed_to_start,
49
:-(
class => Class, reason => Reason, stacktrace => StackTrace}),
50
:-(
erlang:raise(Class, Reason, StackTrace)
51 end;
52 start(_, _) ->
53
:-(
{error, badarg}.
54
55 do_start() ->
56 101 mongoose_fips:notify(),
57 101 write_pid_file(),
58 101 update_status_file(starting),
59 101 mongoose_config:start(),
60 101 mongoose_metrics:init(),
61 101 mongoose_internal_databases:init(),
62 101 mongoose_graphql:init(),
63 101 translate:start(),
64 101 mongoose_graphql_commands:start(),
65 101 mongoose_router:start(),
66 101 mongoose_logs:set_global_loglevel(mongoose_config:get_opt(loglevel)),
67 101 mongoose_deprecations:start(),
68 101 {ok, _} = Sup = ejabberd_sup:start_link(),
69 101 mongoose_wpool:ensure_started(),
70 101 mongoose_wpool:start_configured_pools(),
71 %% ejabberd_sm is started separately because it may use one of the outgoing_pools
72 %% but some outgoing_pools should be started only with ejabberd_sup already running
73 101 ejabberd_sm:start(),
74 101 ejabberd_auth:start(),
75 101 mongoose_cluster_id:start(),
76 101 mongoose_service:start(),
77 101 mongoose_modules:start(),
78 101 service_mongoose_system_metrics:verify_if_configured(),
79 101 mongoose_listener:start(),
80 101 mongoose_metrics:init_mongooseim_metrics(),
81 101 gen_hook:reload_hooks(),
82 101 update_status_file(started),
83 101 ?LOG_NOTICE(#{what => mongooseim_node_started, version => ?MONGOOSE_VERSION, node => node()}),
84 101 Sup.
85
86 %% @doc Prepare the application for termination.
87 %% This function is called when an application is about to be stopped,
88 %% before shutting down the processes of the application.
89 prep_stop(State) ->
90 101 mongoose_deprecations:stop(),
91 101 broadcast_c2s_shutdown_listeners(),
92 101 mongoose_listener:stop(),
93 101 mongoose_modules:stop(),
94 101 mongoose_service:stop(),
95 101 broadcast_c2s_shutdown_sup(),
96 101 mongoose_wpool:stop(),
97 101 mongoose_metrics:remove_all_metrics(),
98 101 mongoose_config:stop(),
99 101 mongoose_graphql_commands:stop(),
100 101 State.
101
102 %% All the processes were killed when this function is called
103 stop(_State) ->
104 101 ?LOG_NOTICE(#{what => mongooseim_node_stopped, version => ?MONGOOSE_VERSION, node => node()}),
105 101 delete_pid_file(),
106 101 update_status_file(stopped),
107 %% We cannot stop other applications inside of the stop callback
108 %% (because we would deadlock the application controller process).
109 %% That is why we call mnesia:stop() inside of db_init_mnesia() instead.
110 101 ok.
111
112 %%%
113 %%% Internal functions
114 %%%
115
116 -spec broadcast_c2s_shutdown_listeners() -> ok.
117 broadcast_c2s_shutdown_listeners() ->
118 101 Children = supervisor:which_children(mongoose_listener_sup),
119 101 Listeners = [Ref || {Ref, _, _, [mongoose_c2s_listener]} <- Children],
120 101 lists:foreach(
121 fun(Listener) ->
122 200 ranch:suspend_listener(Listener),
123 200 [mongoose_c2s:exit(Pid, system_shutdown) || Pid <- ranch:procs(Listener, connections)],
124 200 mongoose_lib:wait_until(
125 fun() ->
126 201 length(ranch:procs(Listener, connections))
127 end,
128 0)
129 end,
130 Listeners).
131
132 -spec broadcast_c2s_shutdown_sup() -> ok.
133 broadcast_c2s_shutdown_sup() ->
134 101 Children = supervisor:which_children(mongoose_c2s_sup),
135 101 lists:foreach(
136 fun({_, Pid, _, _}) ->
137
:-(
mongoose_c2s:exit(Pid, system_shutdown)
138 end,
139 Children),
140 101 mongoose_lib:wait_until(
141 fun() ->
142 101 Res = supervisor:count_children(mongoose_c2s_sup),
143 101 proplists:get_value(active, Res)
144 end,
145 0).
146
147 %%%
148 %%% PID file
149 %%%
150
151 -spec write_pid_file() -> 'ok' | {'error', atom()}.
152 write_pid_file() ->
153 101 case ejabberd:get_pid_file() of
154 false ->
155
:-(
ok;
156 PidFilename ->
157 101 write_pid_file(os:getpid(), PidFilename)
158 end.
159
160 -spec write_pid_file(Pid :: string(),
161 PidFilename :: nonempty_string()
162 ) -> 'ok' | {'error', atom()}.
163 write_pid_file(Pid, PidFilename) ->
164 101 case file:open(PidFilename, [write]) of
165 {ok, Fd} ->
166 101 io:format(Fd, "~s~n", [Pid]),
167 101 file:close(Fd);
168 {error, Reason} ->
169
:-(
?LOG_ERROR(#{what => cannot_write_to_pid_file,
170
:-(
pid_file => PidFilename, reason => Reason}),
171
:-(
throw({cannot_write_pid_file, PidFilename, Reason})
172 end.
173
174 update_status_file(Status) ->
175 303 case ejabberd:get_status_file() of
176 false ->
177
:-(
ok;
178 StatusFilename ->
179 303 file:write_file(StatusFilename, atom_to_list(Status))
180 end.
181
182 -spec delete_pid_file() -> 'ok' | {'error', atom()}.
183 delete_pid_file() ->
184 101 case ejabberd:get_pid_file() of
185 false ->
186
:-(
ok;
187 PidFilename ->
188 101 file:delete(PidFilename)
189 end.
Line Hits Source