chore: apply code review

This commit is contained in:
zhongwencool 2024-07-17 21:27:07 +08:00
parent 3c832db13d
commit 3381eecd6f
3 changed files with 17 additions and 14 deletions

View File

@ -46,6 +46,4 @@
) )
). ).
-define(READONLY_KEYS, [rpc, node]).
-endif. -endif.

View File

@ -45,6 +45,9 @@
-define(LOCAL_OPTIONS, #{rawconf_with_defaults => true, persistent => false}). -define(LOCAL_OPTIONS, #{rawconf_with_defaults => true, persistent => false}).
-define(TIMEOUT, 30000). -define(TIMEOUT, 30000).
%% All 'cluster.*' keys, except for 'cluster.link', should also be treated as read-only.
-define(READONLY_ROOT_KEYS, [rpc, node]).
-dialyzer({no_match, [load/0]}). -dialyzer({no_match, [load/0]}).
load() -> load() ->
@ -315,13 +318,13 @@ get_config(Key) ->
load_config(Path, Opts) when is_list(Path) -> load_config(Path, Opts) when is_list(Path) ->
case hocon:files([Path]) of case hocon:files([Path]) of
{ok, RawConf} when RawConf =:= #{} -> {ok, RawConf} when RawConf =:= #{} ->
case filelib:is_file(Path) of case filelib:is_regular(Path) of
true -> true ->
emqx_ctl:warning("load ~ts is empty~n", [Path]), emqx_ctl:warning("load ~ts is empty~n", [Path]),
{error, empty_hocon_file}; {error, #{cause => empty_hocon_file, path => Path}};
false -> false ->
emqx_ctl:warning("~ts is not found~n", [Path]), emqx_ctl:warning("~ts file is not found~n", [Path]),
{error, not_found_hocon_file} {error, #{cause => not_a_file, path => Path}}
end; end;
{ok, RawConf} -> {ok, RawConf} ->
load_config_from_raw(RawConf, Opts); load_config_from_raw(RawConf, Opts);
@ -540,7 +543,7 @@ check_config(Conf0, Opts) ->
check_keys_is_not_readonly(Conf, Opts) -> check_keys_is_not_readonly(Conf, Opts) ->
IgnoreReadonly = maps:get(ignore_readonly, Opts, false), IgnoreReadonly = maps:get(ignore_readonly, Opts, false),
Keys = maps:keys(Conf), Keys = maps:keys(Conf),
ReadOnlyKeys = [atom_to_binary(K) || K <- ?READONLY_KEYS], ReadOnlyKeys = [atom_to_binary(K) || K <- ?READONLY_ROOT_KEYS],
case lists:filter(fun(K) -> lists:member(K, Keys) end, ReadOnlyKeys) of case lists:filter(fun(K) -> lists:member(K, Keys) end, ReadOnlyKeys) of
[] -> [] ->
{ok, Conf}; {ok, Conf};
@ -561,7 +564,7 @@ check_cluster_keys(Conf = #{<<"cluster">> := Cluster}, Opts) ->
?SLOG(info, #{msg => "readonly_root_keys_ignored", keys => Keys}), ?SLOG(info, #{msg => "readonly_root_keys_ignored", keys => Keys}),
{ok, Conf#{<<"cluster">> => maps:with([<<"links">>], Cluster)}}; {ok, Conf#{<<"cluster">> => maps:with([<<"links">>], Cluster)}};
Keys -> Keys ->
BadKeys = ["cluster." ++ K || K <- Keys], BadKeys = [<<"cluster.", K/binary>> || K <- Keys],
BadKeysStr = lists:join(<<",">>, BadKeys), BadKeysStr = lists:join(<<",">>, BadKeys),
{error, ?UPDATE_READONLY_KEYS_PROHIBITED, BadKeysStr} {error, ?UPDATE_READONLY_KEYS_PROHIBITED, BadKeysStr}
end; end;
@ -626,7 +629,7 @@ filter_readonly_config(Raw) ->
try try
RawDefault = fill_defaults(Raw), RawDefault = fill_defaults(Raw),
_ = emqx_config:check_config(SchemaMod, RawDefault), _ = emqx_config:check_config(SchemaMod, RawDefault),
{ok, maps:without([atom_to_binary(K) || K <- ?READONLY_KEYS], Raw)} {ok, maps:without([atom_to_binary(K) || K <- ?READONLY_ROOT_KEYS], Raw)}
catch catch
throw:Error -> throw:Error ->
?SLOG(error, #{ ?SLOG(error, #{
@ -798,7 +801,7 @@ find_running_confs() ->
lists:map( lists:map(
fun(Node) -> fun(Node) ->
Conf = emqx_conf_proto_v4:get_config(Node, []), Conf = emqx_conf_proto_v4:get_config(Node, []),
{Node, maps:without(?READONLY_KEYS, Conf)} {Node, maps:without(?READONLY_ROOT_KEYS, Conf)}
end, end,
mria:running_nodes() mria:running_nodes()
). ).

View File

@ -21,9 +21,11 @@
-include_lib("eunit/include/eunit.hrl"). -include_lib("eunit/include/eunit.hrl").
-include_lib("common_test/include/ct.hrl"). -include_lib("common_test/include/ct.hrl").
-include_lib("snabbkaffe/include/snabbkaffe.hrl"). -include_lib("snabbkaffe/include/snabbkaffe.hrl").
-include("emqx_conf.hrl").
-import(emqx_config_SUITE, [prepare_conf_file/3]). -import(emqx_config_SUITE, [prepare_conf_file/3]).
-define(READONLY_ROOT_KEYS, [rpc, node]).
all() -> all() ->
emqx_common_test_helpers:all(?MODULE). emqx_common_test_helpers:all(?MODULE).
@ -81,10 +83,10 @@ t_load_config(Config) ->
Conf#{<<"sources">> => [emqx_authz_schema:default_authz()]}, Conf#{<<"sources">> => [emqx_authz_schema:default_authz()]},
emqx_conf:get_raw([Authz]) emqx_conf:get_raw([Authz])
), ),
?assertEqual({error, not_found_hocon_file}, emqx_conf_cli:conf(["load", "non-exist-file"])), ?assertMatch({error, #{cause := not_a_file}}, emqx_conf_cli:conf(["load", "non-exist-file"])),
EmptyFile = "empty_file.conf", EmptyFile = "empty_file.conf",
ok = file:write_file(EmptyFile, <<>>), ok = file:write_file(EmptyFile, <<>>),
?assertEqual({error, empty_hocon_file}, emqx_conf_cli:conf(["load", EmptyFile])), ?assertMatch({error, #{cause := empty_hocon_file}}, emqx_conf_cli:conf(["load", EmptyFile])),
ok = file:delete(EmptyFile), ok = file:delete(EmptyFile),
ok. ok.
@ -207,7 +209,7 @@ t_load_readonly(Config) ->
%% Don't update readonly key %% Don't update readonly key
?assertEqual(Conf, emqx_conf:get_raw([Key])) ?assertEqual(Conf, emqx_conf:get_raw([Key]))
end, end,
?READONLY_KEYS ?READONLY_ROOT_KEYS
), ),
ok. ok.