Current section

Files

Jump to
zotonic_launcher src command zotonic_cmd_wait.erl
Raw

src/command/zotonic_cmd_wait.erl

%%%-------------------------------------------------------------------
%%% @author Blaise
%%% @doc
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.
%%%
%%% @end
%%% Created : 18. Dec 2017 4:47 PM
%%%-------------------------------------------------------------------
-module(zotonic_cmd_wait).
-author("Blaise").
%% API
-export([info/0, run/1]).
-include("../../include/zotonic_command.hrl").
info() ->
"Wait for a max timeout till Zotonc is running and responding to a ping.".
run([ Timeout ]) ->
try
Tm = list_to_integer(Timeout),
wait(Tm)
catch
error:badarg ->
io:format("Timeout must be an integer~n"),
halt(1)
end;
run([]) ->
wait(?MAXWAIT);
run(_) ->
io:format("USAGE: wait [timeout] ~n"),
halt(1).
wait(Timeout) ->
case zotonic_command:net_start() of
ok ->
case zotonic_command:get_target_node() of
{ok, Target} ->
io:format("Waiting for zotonic: "),
wait( Target, timestamp() + Timeout, Timeout);
{error, _} = Error ->
zotonic_command:format_error(Error)
end;
{error, _} = Error ->
zotonic_command:format_error(Error)
end.
wait( Target, Till, Timeout ) ->
case timestamp() > Till of
false ->
case rpc:call(Target, zotonic, ping, []) of
pong ->
io:format(" OK~n");
_ ->
io:format("."),
timer:sleep(1000),
wait(Target, Till, Timeout)
end;
true ->
io:format(" No response after ~p seconds~n", [ Timeout ]),
halt()
end.
timestamp() ->
{A, B, _} = os:timestamp(),
A * 1000000 + B.