install_upgrade.escript 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. #!/usr/bin/env escript
  2. %%! -noshell -noinput
  3. %% -*- mode: erlang;erlang-indent-level: 4;indent-tabs-mode: nil -*-
  4. %% ex: ft=erlang ts=4 sw=4 et
  5. -define(TIMEOUT, 60000).
  6. -define(INFO(Fmt,Args), io:format(Fmt,Args)).
  7. main([NodeName, Cookie, ReleasePackage]) ->
  8. TargetNode = start_distribution(NodeName, Cookie),
  9. {ok, Vsn} = rpc:call(TargetNode, release_handler, unpack_release,
  10. [ReleasePackage], ?TIMEOUT),
  11. ?INFO("Unpacked Release ~p~n", [Vsn]),
  12. {ok, OtherVsn, Desc} = rpc:call(TargetNode, release_handler,
  13. check_install_release, [Vsn], ?TIMEOUT),
  14. {ok, OtherVsn, Desc} = rpc:call(TargetNode, release_handler,
  15. install_release, [Vsn], ?TIMEOUT),
  16. ?INFO("Installed Release ~p~n", [Vsn]),
  17. ok = rpc:call(TargetNode, release_handler, make_permanent, [Vsn], ?TIMEOUT),
  18. ?INFO("Made Release ~p Permanent~n", [Vsn]);
  19. main(_) ->
  20. init:stop(1).
  21. start_distribution(NodeName, Cookie) ->
  22. MyNode = make_script_node(NodeName),
  23. {ok, _Pid} = net_kernel:start([MyNode, shortnames]),
  24. erlang:set_cookie(node(), list_to_atom(Cookie)),
  25. TargetNode = make_target_node(NodeName),
  26. case {net_kernel:hidden_connect_node(TargetNode),
  27. net_adm:ping(TargetNode)} of
  28. {true, pong} ->
  29. ok;
  30. {_, pang} ->
  31. io:format("Node ~p not responding to pings.\n", [TargetNode]),
  32. init:stop(1)
  33. end,
  34. TargetNode.
  35. make_target_node(Node) ->
  36. [_, Host] = string:tokens(atom_to_list(node()), "@"),
  37. list_to_atom(lists:concat([Node, "@", Host])).
  38. make_script_node(Node) ->
  39. list_to_atom(lists:concat([Node, "_upgrader_", os:getpid()])).