Skip to content

Commit

Permalink
Merge pull request #89 from basho-labs/dp-multi-fw
Browse files Browse the repository at this point in the history
Support of multiple frameworks improved
  • Loading branch information
sanmiguel authored Aug 18, 2016
2 parents 9a48d7a + b1617ae commit d87123d
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 9 deletions.
17 changes: 9 additions & 8 deletions src/rms_config.erl
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
constraints/0,
zk/0,
framework_name/0,
framework_role/0,
webui_url/0,
artifacts/0,
artifact_urls/0,
Expand All @@ -34,7 +35,7 @@
-export([get_value/2, get_value/3]).

-define(DEFAULT_NAME, "riak").
-define(DEFAULT_HOSTNAME, "riak.mesos").
-define(DEFAULT_HOSTNAME_SUFFIX, ".marathon.mesos").
-define(DEFAULT_MASTER, "master.mesos:5050").
-define(DEFAULT_ZK, "master.mesos:2181").
-define(DEFAULT_CONSTRAINTS, "[]").
Expand Down Expand Up @@ -86,17 +87,17 @@ zk() ->
framework_name() ->
get_value(name, ?DEFAULT_NAME, string).

-spec framework_role() -> string().
framework_role() ->
get_value(name, framework_name(), string).

-spec framework_hostname() -> string().
framework_hostname() ->
case get_value(hostname, undefined, string) of
undefined ->
{ok, LH} = inet:gethostname(),
case inet:gethostbyname(LH) of
{ok, {_, FullHostname, _, _, _, _}} ->
FullHostname;
_ -> ?DEFAULT_HOSTNAME
end;
HN -> HN
framework_name() ++ ?DEFAULT_HOSTNAME_SUFFIX;
Hostname ->
Hostname
end.

-spec webui_url() -> string().
Expand Down
2 changes: 1 addition & 1 deletion src/rms_sup.erl
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ init_rest() ->
end || Node <- ZkNodes],
FrameworkUser = rms_config:get_value(user, "root"),
FrameworkName = rms_config:framework_name(),
FrameworkRole = rms_config:get_value(role, "riak", string),
FrameworkRole = rms_config:framework_role(),
FrameworkHostname = rms_config:framework_hostname(),
FrameworkPrincipal = rms_config:get_value(principal, "riak", string),
FrameworkFailoverTimeout =
Expand Down

0 comments on commit d87123d

Please sign in to comment.