Add CLI set/get env
This commit is contained in:
turtled
parent
7c1ee6610d
commit
701ee3e8dc
4
Makefile
4
Makefile
|
|
@ -2,7 +2,7 @@ PROJECT = emqttd
|
|||
PROJECT_DESCRIPTION = Erlang MQTT Broker
|
||||
PROJECT_VERSION = 2.3
|
||||
|
||||
DEPS = goldrush gproc lager esockd ekka mochiweb pbkdf2 lager_syslog bcrypt
|
||||
DEPS = goldrush gproc lager esockd ekka mochiweb pbkdf2 lager_syslog bcrypt clique jsx
|
||||
|
||||
dep_goldrush = git https://github.com/basho/goldrush 0.1.9
|
||||
dep_gproc = git https://github.com/uwiger/gproc
|
||||
|
|
@ -14,6 +14,8 @@ dep_mochiweb = git https://github.com/emqtt/mochiweb master
|
|||
dep_pbkdf2 = git https://github.com/emqtt/pbkdf2 2.0.1
|
||||
dep_lager_syslog = git https://github.com/basho/lager_syslog
|
||||
dep_bcrypt = git https://github.com/smarkets/erlang-bcrypt master
|
||||
dep_clique = git https://github.com/turtleDeng/clique
|
||||
dep_jsx = git https://github.com/talentdeficit/jsx
|
||||
|
||||
ERLC_OPTS += +debug_info
|
||||
ERLC_OPTS += +'{parse_transform, lager_transform}'
|
||||
|
|
|
|||
|
|
@ -48,7 +48,8 @@
|
|||
|
||||
load() ->
|
||||
Cmds = [Fun || {Fun, _} <- ?MODULE:module_info(exports), is_cmd(Fun)],
|
||||
[emqttd_ctl:register_cmd(Cmd, {?MODULE, Cmd}, []) || Cmd <- Cmds].
|
||||
[emqttd_ctl:register_cmd(Cmd, {?MODULE, Cmd}, []) || Cmd <- Cmds],
|
||||
emqttd_cli_config:register_config().
|
||||
|
||||
is_cmd(Fun) ->
|
||||
not lists:member(Fun, [init, load, module_info]).
|
||||
|
|
|
|||
|
|
@ -0,0 +1,235 @@
|
|||
%%--------------------------------------------------------------------
|
||||
%% Copyright (c) 2013-2017 EMQ Enterprise, Inc. (http://emqtt.io)
|
||||
%%
|
||||
%% 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.
|
||||
%%--------------------------------------------------------------------
|
||||
|
||||
-module (emqttd_cli_config).
|
||||
|
||||
-export ([register_config_cli/0, register_config/0, run/1]).
|
||||
|
||||
-define(APP, emqttd).
|
||||
|
||||
register_config() ->
|
||||
F = fun() -> ekka_mnesia:running_nodes() end,
|
||||
clique:register_node_finder(F),
|
||||
emqttd_cli_config:register_config_cli().
|
||||
|
||||
run(Cmd) ->
|
||||
clique:run(Cmd).
|
||||
|
||||
register_config_cli() ->
|
||||
ok = clique_config:load_schema([code:priv_dir(?APP)], ?APP),
|
||||
register_protocol_formatter(),
|
||||
register_client_formatter(),
|
||||
register_session_formatter(),
|
||||
register_queue_formatter(),
|
||||
register_lager_formatter(),
|
||||
|
||||
register_auth_config(),
|
||||
register_protocol_config(),
|
||||
register_connection_config(),
|
||||
register_client_config(),
|
||||
register_session_config(),
|
||||
register_queue_config(),
|
||||
register_broker_config(),
|
||||
register_lager_config().
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% Auth/Acl
|
||||
%%--------------------------------------------------------------------
|
||||
register_auth_config() ->
|
||||
ConfigKeys = ["mqtt.allow_anonymous",
|
||||
"mqtt.acl_nomatch",
|
||||
"mqtt.acl_file",
|
||||
"mqtt.cache_acl"],
|
||||
[clique:register_config(Key , fun auth_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
auth_config_callback([_, KeyStr], Value) ->
|
||||
application:set_env(?APP, l2a(KeyStr), Value),
|
||||
" successfully\n".
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% MQTT Protocol
|
||||
%%--------------------------------------------------------------------
|
||||
register_protocol_formatter() ->
|
||||
ConfigKeys = ["max_clientid_len",
|
||||
"max_packet_size",
|
||||
"websocket_protocol_header",
|
||||
"keepalive_backoff"],
|
||||
[clique:register_formatter(["mqtt", Key], fun protocol_formatter_callback/2) || Key <- ConfigKeys].
|
||||
|
||||
protocol_formatter_callback([_, Key], Params) ->
|
||||
proplists:get_value(l2a(Key), Params).
|
||||
|
||||
register_protocol_config() ->
|
||||
ConfigKeys = ["mqtt.max_clientid_len",
|
||||
"mqtt.max_packet_size",
|
||||
"mqtt.websocket_protocol_header",
|
||||
"mqtt.keepalive_backoff"],
|
||||
[clique:register_config(Key , fun protocol_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
protocol_config_callback([_AppStr, KeyStr], Value) ->
|
||||
protocol_config_callback(protocol, l2a(KeyStr), Value).
|
||||
protocol_config_callback(App, Key, Value) ->
|
||||
{ok, Env} = emqttd:env(App),
|
||||
application:set_env(?APP, App, lists:keyreplace(Key, 1, Env, {Key, Value})),
|
||||
" successfully\n".
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% MQTT Connection
|
||||
%%--------------------------------------------------------------------
|
||||
register_connection_config() ->
|
||||
ConfigKeys = ["mqtt.conn.force_gc_count"],
|
||||
[clique:register_config(Key , fun connection_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
connection_config_callback([_, KeyStr0, KeyStr1], Value) ->
|
||||
KeyStr = lists:concat([KeyStr0, "_", KeyStr1]),
|
||||
application:set_env(?APP, l2a(KeyStr), Value),
|
||||
" successfully\n".
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% MQTT Client
|
||||
%%--------------------------------------------------------------------
|
||||
register_client_formatter() ->
|
||||
ConfigKeys = ["max_publish_rate",
|
||||
"idle_timeout",
|
||||
"enable_stats"],
|
||||
[clique:register_formatter(["mqtt", "client", Key], fun client_formatter_callback/2) || Key <- ConfigKeys].
|
||||
|
||||
client_formatter_callback([_, _, Key], Params) ->
|
||||
proplists:get_value(list_to_atom(Key), Params).
|
||||
|
||||
register_client_config() ->
|
||||
ConfigKeys = ["mqtt.client.max_publish_rate",
|
||||
"mqtt.client.idle_timeout",
|
||||
"mqtt.client.enable_stats"],
|
||||
[clique:register_config(Key , fun client_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
client_config_callback([_, AppStr, KeyStr], Value) ->
|
||||
client_config_callback(l2a(AppStr), l2a(KeyStr), Value).
|
||||
client_config_callback(App, Key, Value) ->
|
||||
{ok, Env} = emqttd:env(App),
|
||||
application:set_env(?APP, App, lists:keyreplace(Key, 1, Env, {Key, Value})),
|
||||
" successfully\n".
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% session
|
||||
%%--------------------------------------------------------------------
|
||||
register_session_formatter() ->
|
||||
ConfigKeys = ["max_subscriptions",
|
||||
"upgrade_qos",
|
||||
"max_inflight",
|
||||
"retry_interval",
|
||||
"max_awaiting_rel",
|
||||
"await_rel_timeout",
|
||||
"enable_stats",
|
||||
"expiry_interval",
|
||||
"ignore_loop_deliver"],
|
||||
[clique:register_formatter(["mqtt", "session", Key], fun session_formatter_callback/2) || Key <- ConfigKeys].
|
||||
|
||||
session_formatter_callback([_, _, Key], Params) ->
|
||||
proplists:get_value(list_to_atom(Key), Params).
|
||||
|
||||
register_session_config() ->
|
||||
ConfigKeys = ["mqtt.session.max_subscriptions",
|
||||
"mqtt.session.upgrade_qos",
|
||||
"mqtt.session.max_inflight",
|
||||
"mqtt.session.retry_interval",
|
||||
"mqtt.session.max_awaiting_rel",
|
||||
"mqtt.session.await_rel_timeout",
|
||||
"mqtt.session.enable_stats",
|
||||
"mqtt.session.expiry_interval",
|
||||
"mqtt.session.ignore_loop_deliver"],
|
||||
[clique:register_config(Key , fun session_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
session_config_callback([_, AppStr, KeyStr], Value) ->
|
||||
session_config_callback(l2a(AppStr), l2a(KeyStr), Value).
|
||||
session_config_callback(App, Key, Value) ->
|
||||
{ok, Env} = emqttd:env(App),
|
||||
application:set_env(?APP, App, lists:keyreplace(Key, 1, Env, {Key, Value})),
|
||||
" successfully\n".
|
||||
|
||||
l2a(List) -> list_to_atom(List).
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% MQTT MQueue
|
||||
%%--------------------------------------------------------------------
|
||||
register_queue_formatter() ->
|
||||
ConfigKeys = ["type",
|
||||
"priority",
|
||||
"max_length",
|
||||
"low_watermark",
|
||||
"high_watermark",
|
||||
"store_qos0"],
|
||||
[clique:register_formatter(["mqtt", "mqueue", Key], fun queue_formatter_callback/2) || Key <- ConfigKeys].
|
||||
|
||||
queue_formatter_callback([_, _, Key], Params) ->
|
||||
proplists:get_value(list_to_atom(Key), Params).
|
||||
|
||||
register_queue_config() ->
|
||||
ConfigKeys = ["mqtt.mqueue.type",
|
||||
"mqtt.mqueue.priority",
|
||||
"mqtt.mqueue.max_length",
|
||||
"mqtt.mqueue.low_watermark",
|
||||
"mqtt.mqueue.high_watermark",
|
||||
"mqtt.mqueue.store_qos0"],
|
||||
[clique:register_config(Key , fun queue_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
queue_config_callback([_, AppStr, KeyStr], Value) ->
|
||||
queue_config_callback(l2a(AppStr), l2a(KeyStr), Value).
|
||||
queue_config_callback(App, Key, Value) ->
|
||||
{ok, Env} = emqttd:env(App),
|
||||
application:set_env(?APP, App, lists:keyreplace(Key, 1, Env, {Key, Value})),
|
||||
" successfully\n".
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% MQTT Broker
|
||||
%%--------------------------------------------------------------------
|
||||
register_broker_config() ->
|
||||
ConfigKeys = ["mqtt.broker.sys_interval"],
|
||||
[clique:register_config(Key , fun broker_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
broker_config_callback([_, KeyStr0, KeyStr1], Value) ->
|
||||
KeyStr = lists:concat([KeyStr0, "_", KeyStr1]),
|
||||
application:set_env(?APP, l2a(KeyStr), Value),
|
||||
" successfully\n".
|
||||
|
||||
%%--------------------------------------------------------------------
|
||||
%% MQTT Lager
|
||||
%%--------------------------------------------------------------------
|
||||
register_lager_formatter() ->
|
||||
ConfigKeys = ["level"],
|
||||
[clique:register_formatter(["log", "console", Key], fun lager_formatter_callback/2) || Key <- ConfigKeys].
|
||||
|
||||
lager_formatter_callback(_, Params) ->
|
||||
proplists:get_value(lager_console_backend, Params).
|
||||
|
||||
register_lager_config() ->
|
||||
ConfigKeys = ["log.console.level"],
|
||||
[clique:register_config(Key , fun lager_config_callback/2) || Key <- ConfigKeys],
|
||||
ok = register_config_whitelist(ConfigKeys).
|
||||
|
||||
lager_config_callback(_, Value) ->
|
||||
lager:set_loglevel(lager_console_backend, Value),
|
||||
" successfully\n".
|
||||
|
||||
register_config_whitelist(ConfigKeys) ->
|
||||
clique:register_config_whitelist(ConfigKeys, ?APP).
|
||||
|
|
@ -68,6 +68,12 @@ run([]) -> usage(), ok;
|
|||
|
||||
run(["help"]) -> usage(), ok;
|
||||
|
||||
run(["set" | _] = CmdS) ->
|
||||
emqttd_cli_config:run(["config" | CmdS]), ok;
|
||||
|
||||
run(["show" | _] = CmdS) ->
|
||||
emqttd_cli_config:run(["config" | CmdS]), ok;
|
||||
|
||||
run([CmdS|Args]) ->
|
||||
case lookup(list_to_atom(CmdS)) of
|
||||
[{Mod, Fun}] ->
|
||||
|
|
|
|||
Loading…
Reference in New Issue