fix(limiter): fix test case error
This commit is contained in:
parent
7e3c8162ea
commit
f5c1221ec8
|
@ -475,7 +475,7 @@ init_tree(Type, #{bucket := Buckets} = Cfg) ->
|
||||||
|
|
||||||
State2 = State#{
|
State2 = State#{
|
||||||
root := Root,
|
root := Root,
|
||||||
counter := counters:new(CounterNum, [atomics])
|
counter := counters:new(CounterNum, [write_concurrency])
|
||||||
},
|
},
|
||||||
|
|
||||||
lists:foldl(fun(F, Acc) -> F(Acc) end, State2, DelayBuckets).
|
lists:foldl(fun(F, Acc) -> F(Acc) end, State2, DelayBuckets).
|
||||||
|
|
|
@ -49,32 +49,33 @@ init_per_testcase(Case, Config) when
|
||||||
undefined -> ok;
|
undefined -> ok;
|
||||||
Listeners -> emqx_config:put([listeners], maps:remove(quic, Listeners))
|
Listeners -> emqx_config:put([listeners], maps:remove(quic, Listeners))
|
||||||
end,
|
end,
|
||||||
PrevListeners = emqx_config:get([listeners, tcp], #{}),
|
|
||||||
PrevRateLimit = emqx_config:get([rate_limit], #{}),
|
PrevListeners = emqx_config:get([listeners]),
|
||||||
emqx_config:put(
|
PureListeners = remove_default_limiter(PrevListeners),
|
||||||
[listeners, tcp],
|
PureListeners2 = PureListeners#{
|
||||||
#{
|
tcp => #{
|
||||||
listener_test => #{
|
listener_test => #{
|
||||||
bind => {"127.0.0.1", 9999},
|
bind => {"127.0.0.1", 9999},
|
||||||
max_connections => 4321,
|
max_connections => 4321,
|
||||||
limiter => #{}
|
limiter => #{}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
),
|
},
|
||||||
emqx_config:put([rate_limit], #{max_conn_rate => 1000}),
|
emqx_config:put([listeners], PureListeners2),
|
||||||
|
|
||||||
ok = emqx_listeners:start(),
|
ok = emqx_listeners:start(),
|
||||||
[
|
[
|
||||||
{prev_listener_conf, PrevListeners},
|
{prev_listener_conf, PrevListeners}
|
||||||
{prev_rate_limit_conf, PrevRateLimit}
|
|
||||||
| Config
|
| Config
|
||||||
];
|
];
|
||||||
init_per_testcase(t_wss_conn, Config) ->
|
init_per_testcase(t_wss_conn, Config) ->
|
||||||
catch emqx_config_handler:stop(),
|
catch emqx_config_handler:stop(),
|
||||||
{ok, _} = emqx_config_handler:start_link(),
|
{ok, _} = emqx_config_handler:start_link(),
|
||||||
PrevListeners = emqx_config:get([listeners, wss], #{}),
|
|
||||||
emqx_config:put(
|
PrevListeners = emqx_config:get([listeners]),
|
||||||
[listeners, wss],
|
PureListeners = remove_default_limiter(PrevListeners),
|
||||||
#{
|
PureListeners2 = PureListeners#{
|
||||||
|
wss => #{
|
||||||
listener_test => #{
|
listener_test => #{
|
||||||
bind => {{127, 0, 0, 1}, 9998},
|
bind => {{127, 0, 0, 1}, 9998},
|
||||||
limiter => #{},
|
limiter => #{},
|
||||||
|
@ -85,7 +86,9 @@ init_per_testcase(t_wss_conn, Config) ->
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
),
|
},
|
||||||
|
emqx_config:put([listeners], PureListeners2),
|
||||||
|
|
||||||
ok = emqx_listeners:start(),
|
ok = emqx_listeners:start(),
|
||||||
[
|
[
|
||||||
{prev_listener_conf, PrevListeners}
|
{prev_listener_conf, PrevListeners}
|
||||||
|
@ -94,25 +97,31 @@ init_per_testcase(t_wss_conn, Config) ->
|
||||||
init_per_testcase(_, Config) ->
|
init_per_testcase(_, Config) ->
|
||||||
catch emqx_config_handler:stop(),
|
catch emqx_config_handler:stop(),
|
||||||
{ok, _} = emqx_config_handler:start_link(),
|
{ok, _} = emqx_config_handler:start_link(),
|
||||||
Config.
|
PrevListeners = emqx_config:get([listeners]),
|
||||||
|
PureListeners = remove_default_limiter(PrevListeners),
|
||||||
|
emqx_config:put([listeners], PureListeners),
|
||||||
|
[
|
||||||
|
{prev_listener_conf, PrevListeners}
|
||||||
|
| Config
|
||||||
|
].
|
||||||
|
|
||||||
end_per_testcase(Case, Config) when
|
end_per_testcase(Case, Config) when
|
||||||
Case =:= t_max_conns_tcp; Case =:= t_current_conns_tcp
|
Case =:= t_max_conns_tcp; Case =:= t_current_conns_tcp
|
||||||
->
|
->
|
||||||
PrevListener = ?config(prev_listener_conf, Config),
|
PrevListener = ?config(prev_listener_conf, Config),
|
||||||
PrevRateLimit = ?config(prev_rate_limit_conf, Config),
|
|
||||||
emqx_listeners:stop(),
|
emqx_listeners:stop(),
|
||||||
emqx_config:put([listeners, tcp], PrevListener),
|
emqx_config:put([listeners], PrevListener),
|
||||||
emqx_config:put([rate_limit], PrevRateLimit),
|
|
||||||
_ = emqx_config_handler:stop(),
|
_ = emqx_config_handler:stop(),
|
||||||
ok;
|
ok;
|
||||||
end_per_testcase(t_wss_conn, Config) ->
|
end_per_testcase(t_wss_conn, Config) ->
|
||||||
PrevListener = ?config(prev_listener_conf, Config),
|
PrevListener = ?config(prev_listener_conf, Config),
|
||||||
emqx_listeners:stop(),
|
emqx_listeners:stop(),
|
||||||
emqx_config:put([listeners, wss], PrevListener),
|
emqx_config:put([listeners], PrevListener),
|
||||||
_ = emqx_config_handler:stop(),
|
_ = emqx_config_handler:stop(),
|
||||||
ok;
|
ok;
|
||||||
end_per_testcase(_, _Config) ->
|
end_per_testcase(_, Config) ->
|
||||||
|
PrevListener = ?config(prev_listener_conf, Config),
|
||||||
|
emqx_config:put([listeners], PrevListener),
|
||||||
_ = emqx_config_handler:stop(),
|
_ = emqx_config_handler:stop(),
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
@ -184,3 +193,16 @@ get_base_dir(Module) ->
|
||||||
|
|
||||||
get_base_dir() ->
|
get_base_dir() ->
|
||||||
get_base_dir(?MODULE).
|
get_base_dir(?MODULE).
|
||||||
|
|
||||||
|
remove_default_limiter(Listeners) ->
|
||||||
|
maps:map(
|
||||||
|
fun(_, X) ->
|
||||||
|
maps:map(
|
||||||
|
fun(_, E) ->
|
||||||
|
maps:remove(limiter, E)
|
||||||
|
end,
|
||||||
|
X
|
||||||
|
)
|
||||||
|
end,
|
||||||
|
Listeners
|
||||||
|
).
|
||||||
|
|
|
@ -29,7 +29,6 @@
|
||||||
"\n"
|
"\n"
|
||||||
"limiter {\n"
|
"limiter {\n"
|
||||||
" bytes_in {\n"
|
" bytes_in {\n"
|
||||||
" enable = true\n"
|
|
||||||
" bucket.default {\n"
|
" bucket.default {\n"
|
||||||
" rate = infinity\n"
|
" rate = infinity\n"
|
||||||
" capacity = infinity\n"
|
" capacity = infinity\n"
|
||||||
|
@ -37,7 +36,6 @@
|
||||||
" }\n"
|
" }\n"
|
||||||
"\n"
|
"\n"
|
||||||
" message_in {\n"
|
" message_in {\n"
|
||||||
" enable = true\n"
|
|
||||||
" bucket.default {\n"
|
" bucket.default {\n"
|
||||||
" rate = infinity\n"
|
" rate = infinity\n"
|
||||||
" capacity = infinity\n"
|
" capacity = infinity\n"
|
||||||
|
@ -45,7 +43,6 @@
|
||||||
" }\n"
|
" }\n"
|
||||||
"\n"
|
"\n"
|
||||||
" connection {\n"
|
" connection {\n"
|
||||||
" enable = true\n"
|
|
||||||
" bucket.default {\n"
|
" bucket.default {\n"
|
||||||
" rate = infinity\n"
|
" rate = infinity\n"
|
||||||
" capacity = infinity\n"
|
" capacity = infinity\n"
|
||||||
|
@ -53,7 +50,6 @@
|
||||||
" }\n"
|
" }\n"
|
||||||
"\n"
|
"\n"
|
||||||
" message_routing {\n"
|
" message_routing {\n"
|
||||||
" enable = true\n"
|
|
||||||
" bucket.default {\n"
|
" bucket.default {\n"
|
||||||
" rate = infinity\n"
|
" rate = infinity\n"
|
||||||
" capacity = infinity\n"
|
" capacity = infinity\n"
|
||||||
|
@ -61,7 +57,6 @@
|
||||||
" }\n"
|
" }\n"
|
||||||
"\n"
|
"\n"
|
||||||
" batch {\n"
|
" batch {\n"
|
||||||
" enable = true\n"
|
|
||||||
" bucket.retainer {\n"
|
" bucket.retainer {\n"
|
||||||
" rate = infinity\n"
|
" rate = infinity\n"
|
||||||
" capacity = infinity\n"
|
" capacity = infinity\n"
|
||||||
|
|
Loading…
Reference in New Issue