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
|
#!/usr/bin/env escript
%%! -noshell -noinput
%% -*- mode: erlang;erlang-indent-level: 4;indent-tabs-mode: nil -*-
%% ex: ft=erlang ts=4 sw=4 et
%% This file is left for backward-compatibility.
%% You, probably, shouldn't include it to new projects.
main([NodeName, Cookie, ReleasePackage]) ->
io:format("WARNING: 'install_upgrade.escript' is deprecated! "
"Use 'nodetool upgrade' instead.~n"),
NodeRoot = filename:dirname(filename:dirname(escript:script_name())),
NodeTool = which_nodetool(NodeRoot),
process_flag(trap_exit, true),
Port = erlang:open_port(
{spawn_executable, NodeTool},
[{args, ["-sname", NodeName,
"-setcookie", Cookie,
"upgrade", ReleasePackage]},
binary, exit_status, use_stdio, stderr_to_stdout, hide]),
port_loop(Port);
main(_) ->
halt(1).
which_nodetool(NodeRoot) ->
%% ${RELEASE_ROOT}/
%% bin/install_upgrade.escript
%% bin/nodetool ?
%% erts-<erts_ver>/bin/nodetool ?
%% releases/<app_ver>/nodetool ?
%% releases/start_erl.data
{ok, Content} = file:read_file(filename:join([NodeRoot, "releases", "start_erl.data"])),
[ErtsVsn, AppVsn] = binary:split(Content, <<" ">>),
Probes = [
filename:join([NodeRoot, "bin", "nodetool"]),
filename:join([NodeRoot, <<"erts-", ErtsVsn/binary>>, "bin", "nodetool"]),
filename:join([NodeRoot, "releases", AppVsn, "bin", "nodetool"])
],
case lists:dropwhile(fun(Path) -> not filelib:is_regular(Path) end, Probes) of
[] ->
io:format("ERROR: can't find 'nodetool' in ~p.~n", [Probes]),
halt(2);
[Path | _] ->
Path
end.
port_loop(Port) ->
receive
{Port, {data, Data}} ->
io:put_chars(Data),
port_loop(Port);
{Port, {exit_status, Status}} ->
halt(Status)
end.
|
