fix(limiter): remove the `enable` field
This commit is contained in:
parent
d13055be43
commit
7e3c8162ea
|
@ -1,16 +1,5 @@
|
||||||
emqx_limiter_schema {
|
emqx_limiter_schema {
|
||||||
|
|
||||||
enable {
|
|
||||||
desc {
|
|
||||||
en: """Enable"""
|
|
||||||
zh: """是否开启"""
|
|
||||||
}
|
|
||||||
label: {
|
|
||||||
en: """Enable"""
|
|
||||||
zh: """是否开启"""
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
failure_strategy {
|
failure_strategy {
|
||||||
desc {
|
desc {
|
||||||
en: """The strategy when all the retries failed."""
|
en: """The strategy when all the retries failed."""
|
||||||
|
|
|
@ -119,19 +119,9 @@ post_config_update([limiter, Type], _Config, NewConf, _OldConf, _AppEnvs) ->
|
||||||
Config = maps:get(Type, NewConf),
|
Config = maps:get(Type, NewConf),
|
||||||
case emqx_limiter_server:whereis(Type) of
|
case emqx_limiter_server:whereis(Type) of
|
||||||
undefined ->
|
undefined ->
|
||||||
case Config of
|
start_server(Type, Config);
|
||||||
#{enable := false} ->
|
|
||||||
ok;
|
|
||||||
_ ->
|
|
||||||
start_server(Type)
|
|
||||||
end;
|
|
||||||
_ ->
|
_ ->
|
||||||
case Config of
|
emqx_limiter_server:update_config(Type, Config)
|
||||||
#{enable := false} ->
|
|
||||||
stop_server(Type);
|
|
||||||
_ ->
|
|
||||||
emqx_limiter_server:update_config(Type, Config)
|
|
||||||
end
|
|
||||||
end.
|
end.
|
||||||
|
|
||||||
%%--------------------------------------------------------------------
|
%%--------------------------------------------------------------------
|
||||||
|
|
|
@ -30,8 +30,7 @@
|
||||||
namespace/0,
|
namespace/0,
|
||||||
get_bucket_cfg_path/2,
|
get_bucket_cfg_path/2,
|
||||||
desc/1,
|
desc/1,
|
||||||
types/0,
|
types/0
|
||||||
is_enable/1
|
|
||||||
]).
|
]).
|
||||||
|
|
||||||
-define(KILOBYTE, 1024).
|
-define(KILOBYTE, 1024).
|
||||||
|
@ -89,13 +88,12 @@ fields(limiter) ->
|
||||||
{Type,
|
{Type,
|
||||||
?HOCON(?R_REF(limiter_opts), #{
|
?HOCON(?R_REF(limiter_opts), #{
|
||||||
desc => ?DESC(Type),
|
desc => ?DESC(Type),
|
||||||
default => #{<<"enable">> => false}
|
default => #{}
|
||||||
})}
|
})}
|
||||||
|| Type <- types()
|
|| Type <- types()
|
||||||
];
|
];
|
||||||
fields(limiter_opts) ->
|
fields(limiter_opts) ->
|
||||||
[
|
[
|
||||||
{enable, ?HOCON(boolean(), #{desc => ?DESC(enable), default => true})},
|
|
||||||
{rate, ?HOCON(rate(), #{desc => ?DESC(rate), default => "infinity"})},
|
{rate, ?HOCON(rate(), #{desc => ?DESC(rate), default => "infinity"})},
|
||||||
{burst,
|
{burst,
|
||||||
?HOCON(burst_rate(), #{
|
?HOCON(burst_rate(), #{
|
||||||
|
@ -202,10 +200,6 @@ to_rate(Str) ->
|
||||||
get_bucket_cfg_path(Type, BucketName) ->
|
get_bucket_cfg_path(Type, BucketName) ->
|
||||||
[limiter, Type, bucket, BucketName].
|
[limiter, Type, bucket, BucketName].
|
||||||
|
|
||||||
-spec is_enable(limiter_type()) -> boolean().
|
|
||||||
is_enable(Type) ->
|
|
||||||
emqx:get_config([limiter, Type, enable], false).
|
|
||||||
|
|
||||||
types() ->
|
types() ->
|
||||||
[bytes_in, message_in, connection, message_routing, batch].
|
[bytes_in, message_in, connection, message_routing, batch].
|
||||||
|
|
||||||
|
|
|
@ -113,13 +113,10 @@
|
||||||
connect(_Type, undefined) ->
|
connect(_Type, undefined) ->
|
||||||
{ok, emqx_htb_limiter:make_infinity_limiter()};
|
{ok, emqx_htb_limiter:make_infinity_limiter()};
|
||||||
connect(Type, BucketName) when is_atom(BucketName) ->
|
connect(Type, BucketName) when is_atom(BucketName) ->
|
||||||
case check_enable_and_get_bucket_cfg(Type, BucketName) of
|
case get_bucket_cfg(Type, BucketName) of
|
||||||
undefined ->
|
undefined ->
|
||||||
?SLOG(error, #{msg => "bucket_config_not_found", type => Type, bucket => BucketName}),
|
?SLOG(error, #{msg => "bucket_config_not_found", type => Type, bucket => BucketName}),
|
||||||
{error, config_not_found};
|
{error, config_not_found};
|
||||||
limiter_not_started ->
|
|
||||||
?SLOG(error, #{msg => "limiter_not_started", type => Type, bucket => BucketName}),
|
|
||||||
{error, limiter_not_started};
|
|
||||||
#{
|
#{
|
||||||
rate := AggrRate,
|
rate := AggrRate,
|
||||||
capacity := AggrSize,
|
capacity := AggrSize,
|
||||||
|
@ -478,7 +475,7 @@ init_tree(Type, #{bucket := Buckets} = Cfg) ->
|
||||||
|
|
||||||
State2 = State#{
|
State2 = State#{
|
||||||
root := Root,
|
root := Root,
|
||||||
counter := counters:new(CounterNum, [write_concurrency])
|
counter := counters:new(CounterNum, [atomics])
|
||||||
},
|
},
|
||||||
|
|
||||||
lists:foldl(fun(F, Acc) -> F(Acc) end, State2, DelayBuckets).
|
lists:foldl(fun(F, Acc) -> F(Acc) end, State2, DelayBuckets).
|
||||||
|
@ -605,13 +602,8 @@ call(Type, Msg) ->
|
||||||
gen_server:call(Pid, Msg)
|
gen_server:call(Pid, Msg)
|
||||||
end.
|
end.
|
||||||
|
|
||||||
-spec check_enable_and_get_bucket_cfg(limiter_type(), bucket_name()) ->
|
-spec get_bucket_cfg(limiter_type(), bucket_name()) ->
|
||||||
undefined | limiter_not_started | hocons:config().
|
undefined | limiter_not_started | hocons:config().
|
||||||
check_enable_and_get_bucket_cfg(Type, Bucket) ->
|
get_bucket_cfg(Type, Bucket) ->
|
||||||
case emqx_limiter_schema:is_enable(Type) of
|
Path = emqx_limiter_schema:get_bucket_cfg_path(Type, Bucket),
|
||||||
false ->
|
emqx:get_config(Path, undefined).
|
||||||
limiter_not_started;
|
|
||||||
_ ->
|
|
||||||
Path = emqx_limiter_schema:get_bucket_cfg_path(Type, Bucket),
|
|
||||||
emqx:get_config(Path, undefined)
|
|
||||||
end.
|
|
||||||
|
|
|
@ -101,14 +101,4 @@ make_child(Type, Cfg) ->
|
||||||
}.
|
}.
|
||||||
|
|
||||||
childs() ->
|
childs() ->
|
||||||
Conf = emqx:get_config([limiter]),
|
[make_child(Type) || Type <- emqx_limiter_schema:types()].
|
||||||
lists:foldl(
|
|
||||||
fun
|
|
||||||
({Type, #{enable := true}}, Acc) ->
|
|
||||||
[make_child(Type) | Acc];
|
|
||||||
(_, Acc) ->
|
|
||||||
Acc
|
|
||||||
end,
|
|
||||||
[],
|
|
||||||
maps:to_list(Conf)
|
|
||||||
).
|
|
||||||
|
|
Loading…
Reference in New Issue