chore(emqx_management): format code
This commit is contained in:
zhanghongtong
turtleDeng
parent
baa9fd8255
commit
28653bb457
|
|
@ -23,8 +23,6 @@
|
|||
|
||||
-define(PRINT_CMD(Cmd, Descr), io:format("~-48s# ~s~n", [Cmd, Descr])).
|
||||
|
||||
-import(lists, [foreach/2]).
|
||||
|
||||
-export([load/0]).
|
||||
|
||||
-export([ status/1
|
||||
|
|
@ -58,7 +56,7 @@
|
|||
-spec(load() -> ok).
|
||||
load() ->
|
||||
Cmds = [Fun || {Fun, _} <- ?MODULE:module_info(exports), is_cmd(Fun)],
|
||||
foreach(fun(Cmd) -> emqx_ctl:register_command(Cmd, {?MODULE, Cmd}, []) end, Cmds).
|
||||
lists:foreach(fun(Cmd) -> emqx_ctl:register_command(Cmd, {?MODULE, Cmd}, []) end, Cmds).
|
||||
|
||||
is_cmd(Fun) ->
|
||||
not lists:member(Fun, [init, load, module_info]).
|
||||
|
|
@ -100,7 +98,7 @@ mgmt(["delete", AppId]) ->
|
|||
end;
|
||||
|
||||
mgmt(["list"]) ->
|
||||
foreach(fun({AppId, AppSecret, Name, Desc, Status, Expired}) ->
|
||||
lists:foreach(fun({AppId, AppSecret, Name, Desc, Status, Expired}) ->
|
||||
emqx_ctl:print("app_id: ~s, secret: ~s, name: ~s, desc: ~s, status: ~s, expired: ~p~n",
|
||||
[AppId, AppSecret, Name, Desc, Status, Expired])
|
||||
end, emqx_mgmt_auth:list_apps());
|
||||
|
|
@ -229,7 +227,7 @@ routes(_) ->
|
|||
{"routes show <Topic>", "Show a route"}]).
|
||||
|
||||
subscriptions(["list"]) ->
|
||||
foreach(fun(Suboption) ->
|
||||
lists:foreach(fun(Suboption) ->
|
||||
print({emqx_suboption, Suboption})
|
||||
end, ets:tab2list(emqx_suboption));
|
||||
|
||||
|
|
@ -279,7 +277,7 @@ if_valid_qos(QoS, Fun) ->
|
|||
end.
|
||||
|
||||
plugins(["list"]) ->
|
||||
foreach(fun print/1, emqx_plugins:list());
|
||||
lists:foreach(fun print/1, emqx_plugins:list());
|
||||
|
||||
plugins(["load", Name]) ->
|
||||
case emqx_plugins:load(list_to_atom(Name)) of
|
||||
|
|
@ -421,7 +419,7 @@ log(_) ->
|
|||
%% @doc Trace Command
|
||||
|
||||
trace(["list"]) ->
|
||||
foreach(fun({{Who, Name}, {Level, LogFile}}) ->
|
||||
lists:foreach(fun({{Who, Name}, {Level, LogFile}}) ->
|
||||
emqx_ctl:print("Trace(~s=~s, level=~s, destination=~p)~n", [Who, Name, Level, LogFile])
|
||||
end, emqx_tracer:lookup_traces());
|
||||
|
||||
|
|
@ -470,7 +468,7 @@ trace_off(Who, Name) ->
|
|||
%% @doc Listeners Command
|
||||
|
||||
listeners([]) ->
|
||||
foreach(fun({{Protocol, ListenOn}, _Pid}) ->
|
||||
lists:foreach(fun({{Protocol, ListenOn}, _Pid}) ->
|
||||
Info = [{listen_on, {string, emqx_listeners:format_listen_on(ListenOn)}},
|
||||
{acceptors, esockd:get_acceptors({Protocol, ListenOn})},
|
||||
{max_conns, esockd:get_max_connections({Protocol, ListenOn})},
|
||||
|
|
@ -478,9 +476,9 @@ listeners([]) ->
|
|||
{shutdown_count, esockd:get_shutdown_count({Protocol, ListenOn})}
|
||||
],
|
||||
emqx_ctl:print("~s~n", [listener_identifier(Protocol, ListenOn)]),
|
||||
foreach(fun indent_print/1, Info)
|
||||
lists:foreach(fun indent_print/1, Info)
|
||||
end, esockd:listeners()),
|
||||
foreach(fun({Protocol, Opts}) ->
|
||||
lists:foreach(fun({Protocol, Opts}) ->
|
||||
Port = proplists:get_value(port, Opts),
|
||||
Info = [{listen_on, {string, emqx_listeners:format_listen_on(Port)}},
|
||||
{acceptors, maps:get(num_acceptors, proplists:get_value(transport_options, Opts, #{}), 0)},
|
||||
|
|
@ -488,7 +486,7 @@ listeners([]) ->
|
|||
{current_conn, proplists:get_value(all_connections, Opts)},
|
||||
{shutdown_count, []}],
|
||||
emqx_ctl:print("~s~n", [listener_identifier(Protocol, Port)]),
|
||||
foreach(fun indent_print/1, Info)
|
||||
lists:foreach(fun indent_print/1, Info)
|
||||
end, ranch:info());
|
||||
|
||||
listeners(["stop", Name = "http" ++ _N | _MaybePort]) ->
|
||||
|
|
|
|||
|
|
@ -447,96 +447,51 @@ to_version(Version) when is_binary(Version) ->
|
|||
to_version(Version) when is_list(Version) ->
|
||||
Version.
|
||||
|
||||
-ifdef(EMQX_ENTERPRISE).
|
||||
export() ->
|
||||
Modules = export_modules(),
|
||||
Rules = export_rules(),
|
||||
Resources = export_resources(),
|
||||
Blacklist = export_blacklist(),
|
||||
Apps = export_applications(),
|
||||
Users = export_users(),
|
||||
AuthMnesia = export_auth_mnesia(),
|
||||
AclMnesia = export_acl_mnesia(),
|
||||
Schemas = export_schemas(),
|
||||
{Configs, State} = export_confs(),
|
||||
Seconds = erlang:system_time(second),
|
||||
Data = do_export_data() ++ [{date, erlang:list_to_binary(emqx_mgmt_util:strftime(Seconds))}],
|
||||
{{Y, M, D}, {H, MM, S}} = emqx_mgmt_util:datetime(Seconds),
|
||||
Filename = io_lib:format("emqx-export-~p-~p-~p-~p-~p-~p.json", [Y, M, D, H, MM, S]),
|
||||
NFilename = filename:join([emqx:get_env(data_dir), Filename]),
|
||||
Version = string:sub_string(emqx_sys:version(), 1, 3),
|
||||
Data = [{version, erlang:list_to_binary(Version)},
|
||||
{date, erlang:list_to_binary(emqx_mgmt_util:strftime(Seconds))},
|
||||
{modules, Modules},
|
||||
{rules, Rules},
|
||||
{resources, Resources},
|
||||
{blacklist, Blacklist},
|
||||
{apps, Apps},
|
||||
{users, Users},
|
||||
{auth_mnesia, AuthMnesia},
|
||||
{acl_mnesia, AclMnesia},
|
||||
{schemas, Schemas},
|
||||
{configs, Configs},
|
||||
{listeners_state, State}],
|
||||
write_file(NFilename, Data).
|
||||
|
||||
import(Filename) ->
|
||||
case file:read_file(FullFilename) of
|
||||
{ok, Json} ->
|
||||
Data = emqx_json:decode(Json, [return_maps]),
|
||||
Version = to_version(maps:get(<<"version">>, Data)),
|
||||
case lists:member(Version, ?VERSIONS) of
|
||||
true ->
|
||||
try
|
||||
import_confs(maps:get(<<"configs">>, Data, []), maps:get(<<"listeners_state">>, Data, [])),
|
||||
import_resources(maps:get(<<"resources">>, Data, [])),
|
||||
import_rules(maps:get(<<"rules">>, Data, [])),
|
||||
import_blacklist(maps:get(<<"blacklist">>, Data, [])),
|
||||
import_applications(maps:get(<<"apps">>, Data, [])),
|
||||
import_users(maps:get(<<"users">>, Data, [])),
|
||||
import_modules(maps:get(<<"modules">>, Data, [])),
|
||||
import_auth_clientid(maps:get(<<"auth_clientid">>, Data, [])),
|
||||
import_auth_username(maps:get(<<"auth_username">>, Data, [])),
|
||||
import_auth_mnesia(maps:get(<<"auth_mnesia">>, Data, []), Version),
|
||||
import_acl_mnesia(maps:get(<<"acl_mnesia">>, Data, []), Version),
|
||||
import_schemas(maps:get(<<"schemas">>, Data, [])),
|
||||
logger:debug("The emqx data has been imported successfully"),
|
||||
ok
|
||||
catch Class:Reason:Stack ->
|
||||
logger:error("The emqx data import failed: ~0p", [{Class,Reason,Stack}]),
|
||||
{error, import_failed}
|
||||
ok = filelib:ensure_dir(NFilename),
|
||||
case file:write_file(NFilename, emqx_json:encode(Data)) of
|
||||
ok ->
|
||||
case file:read_file_info(NFilename) of
|
||||
{ok, #file_info{size = Size, ctime = {{Y, M, D}, {H, MM, S}}}} ->
|
||||
CreatedAt = io_lib:format("~p-~p-~p ~p:~p:~p", [Y, M, D, H, MM, S]),
|
||||
{ok, #{filename => list_to_binary(NFilename),
|
||||
size => Size,
|
||||
created_at => list_to_binary(CreatedAt),
|
||||
node => node()
|
||||
}};
|
||||
Error -> Error
|
||||
end;
|
||||
false ->
|
||||
logger:error("Unsupported version: ~p", [Version]),
|
||||
{error, unsupported_version}
|
||||
end;
|
||||
{error, Reason} ->
|
||||
{error, Reason}
|
||||
Error -> Error
|
||||
end.
|
||||
|
||||
-else.
|
||||
export() ->
|
||||
Rules = export_rules(),
|
||||
Resources = export_resources(),
|
||||
Blacklist = export_blacklist(),
|
||||
Apps = export_applications(),
|
||||
Users = export_users(),
|
||||
AuthMnesia = export_auth_mnesia(),
|
||||
AclMnesia = export_acl_mnesia(),
|
||||
Seconds = erlang:system_time(second),
|
||||
{{Y, M, D}, {H, MM, S}} = emqx_mgmt_util:datetime(Seconds),
|
||||
Filename = io_lib:format("emqx-export-~p-~p-~p-~p-~p-~p.json", [Y, M, D, H, MM, S]),
|
||||
NFilename = filename:join([emqx:get_env(data_dir), Filename]),
|
||||
do_export_data() ->
|
||||
Version = string:sub_string(emqx_sys:version(), 1, 3),
|
||||
Data = [{version, erlang:list_to_binary(Version)},
|
||||
{date, erlang:list_to_binary(emqx_mgmt_util:strftime(Seconds))},
|
||||
{rules, Rules},
|
||||
{resources, Resources},
|
||||
{blacklist, Blacklist},
|
||||
{apps, Apps},
|
||||
{users, Users},
|
||||
{auth_mnesia, AuthMnesia},
|
||||
{acl_mnesia, AclMnesia}],
|
||||
write_file(NFilename, Data).
|
||||
[{version, erlang:list_to_binary(Version)},
|
||||
{rules, export_rules()},
|
||||
{resources, export_resources()},
|
||||
{blacklist, export_blacklist()},
|
||||
{apps, export_applications()},
|
||||
{users, export_users()},
|
||||
{auth_mnesia, export_auth_mnesia()},
|
||||
{acl_mnesia, export_acl_mnesia()}
|
||||
] ++ do_export_extra_data().
|
||||
|
||||
-ifdef(EMQX_ENTERPRISE).
|
||||
do_export_extra_data() ->
|
||||
{Configs, State} = export_confs(),
|
||||
[{modules, export_modules()},
|
||||
{schemas, export_schemas()},
|
||||
{configs, Configs},
|
||||
{listeners_state, State}
|
||||
].
|
||||
-else.
|
||||
do_export_extra_data() -> [].
|
||||
-endif.
|
||||
|
||||
import(Filename) ->
|
||||
case file:read_file(Filename) of
|
||||
|
|
@ -546,14 +501,7 @@ import(Filename) ->
|
|||
case lists:member(Version, ?VERSIONS) of
|
||||
true ->
|
||||
try
|
||||
import_resources_and_rules(maps:get(<<"resources">>, Data, []), maps:get(<<"rules">>, Data, []), Version),
|
||||
import_blacklist(maps:get(<<"blacklist">>, Data, [])),
|
||||
import_applications(maps:get(<<"apps">>, Data, [])),
|
||||
import_users(maps:get(<<"users">>, Data, [])),
|
||||
import_auth_clientid(maps:get(<<"auth_clientid">>, Data, [])),
|
||||
import_auth_username(maps:get(<<"auth_username">>, Data, [])),
|
||||
import_auth_mnesia(maps:get(<<"auth_mnesia">>, Data, []), Version),
|
||||
import_acl_mnesia(maps:get(<<"acl_mnesia">>, Data, []), Version),
|
||||
do_import_data(Data, Version),
|
||||
logger:debug("The emqx data has been imported successfully"),
|
||||
ok
|
||||
catch Class:Reason:Stack ->
|
||||
|
|
@ -566,21 +514,25 @@ import(Filename) ->
|
|||
end;
|
||||
Error -> Error
|
||||
end.
|
||||
-endif.
|
||||
|
||||
write_file(Filename, Data) ->
|
||||
ok = filelib:ensure_dir(Filename),
|
||||
case file:write_file(Filename, emqx_json:encode(Data)) of
|
||||
ok ->
|
||||
case file:read_file_info(Filename) of
|
||||
{ok, #file_info{size = Size, ctime = {{Y, M, D}, {H, MM, S}}}} ->
|
||||
CreatedAt = io_lib:format("~p-~p-~p ~p:~p:~p", [Y, M, D, H, MM, S]),
|
||||
{ok, #{filename => list_to_binary(Filename),
|
||||
size => Size,
|
||||
created_at => list_to_binary(CreatedAt),
|
||||
node => node()
|
||||
}};
|
||||
Error -> Error
|
||||
end;
|
||||
Error -> Error
|
||||
end.
|
||||
do_import_data(Data, Version) ->
|
||||
import_blacklist(maps:get(<<"blacklist">>, Data, [])),
|
||||
import_applications(maps:get(<<"apps">>, Data, [])),
|
||||
import_users(maps:get(<<"users">>, Data, [])),
|
||||
import_auth_clientid(maps:get(<<"auth_clientid">>, Data, [])),
|
||||
import_auth_username(maps:get(<<"auth_username">>, Data, [])),
|
||||
import_auth_mnesia(maps:get(<<"auth_mnesia">>, Data, []), Version),
|
||||
import_acl_mnesia(maps:get(<<"acl_mnesia">>, Data, []), Version),
|
||||
do_import_extra_data(Data, Version).
|
||||
|
||||
-ifdef(EMQX_ENTERPRISE).
|
||||
do_import_extra_data(Data, Version) ->
|
||||
import_confs(maps:get(<<"configs">>, Data, []), maps:get(<<"listeners_state">>, Data, [])),
|
||||
import_resources(maps:get(<<"resources">>, Data, [])),
|
||||
import_rules(maps:get(<<"rules">>, Data, [])),
|
||||
import_modules(maps:get(<<"modules">>, Data, [])),
|
||||
import_schemas(maps:get(<<"schemas">>, Data, [])).
|
||||
-else.
|
||||
do_import_extra_data(Data, Version) ->
|
||||
import_resources_and_rules(maps:get(<<"resources">>, Data, []), maps:get(<<"rules">>, Data, []), Version).
|
||||
-endif.
|
||||
|
|
|
|||
Loading…
Reference in New Issue