53 lines
1.6 KiB
Erlang
53 lines
1.6 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2018-2023 EMQ Technologies Co., Ltd. All Rights Reserved.
|
|
%%
|
|
%% 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.
|
|
%%--------------------------------------------------------------------
|
|
|
|
%% @doc The utility functions for erlang process dictionary.
|
|
-module(emqx_pd).
|
|
|
|
-include("types.hrl").
|
|
|
|
-export([ get_counters/1
|
|
, get_counter/1
|
|
, inc_counter/2
|
|
, reset_counter/1
|
|
]).
|
|
|
|
-compile({inline,
|
|
[ get_counters/1
|
|
, get_counter/1
|
|
, inc_counter/2
|
|
, reset_counter/1
|
|
]}).
|
|
|
|
-type(key() :: term()).
|
|
|
|
-spec(get_counters(list(key())) -> list({key(), number()})).
|
|
get_counters(Keys) when is_list(Keys) ->
|
|
[{Key, emqx_pd:get_counter(Key)} || Key <- Keys].
|
|
|
|
-spec(get_counter(key()) -> number()).
|
|
get_counter(Key) ->
|
|
case get(Key) of undefined -> 0; Cnt -> Cnt end.
|
|
|
|
-spec(inc_counter(key(), number()) -> maybe(number())).
|
|
inc_counter(Key, Inc) ->
|
|
put(Key, get_counter(Key) + Inc).
|
|
|
|
-spec(reset_counter(key()) -> number()).
|
|
reset_counter(Key) ->
|
|
case put(Key, 0) of undefined -> 0; Cnt -> Cnt end.
|
|
|