emqx/apps/emqx_statsd/src/emqx_statsd_api.erl

68 lines
2.0 KiB
Erlang

%%--------------------------------------------------------------------
%% Copyright (c) 2020-2021 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.
%%--------------------------------------------------------------------
-module(emqx_statsd_api).
-behaviour(minirest_api).
-include("emqx_statsd.hrl").
-import(emqx_mgmt_util, [ schema/1
, bad_request/0]).
-export([api_spec/0]).
-export([ statsd/2
]).
api_spec() ->
{statsd_api(), []}.
conf_schema() ->
emqx_mgmt_api_configs:gen_schema(emqx:get_raw_config([statsd])).
statsd_api() ->
Metadata = #{
get => #{
description => <<"Get statsd info">>,
responses => #{<<"200">> => schema(conf_schema())}
},
put => #{
description => <<"Update Statsd">>,
'requestBody' => schema(conf_schema()),
responses => #{
<<"200">> => schema(conf_schema()),
<<"400">> => bad_request()
}
}
},
[{"/statsd", Metadata, statsd}].
statsd(get, _Params) ->
{200, emqx:get_raw_config([<<"statsd">>], #{})};
statsd(put, #{body := Body}) ->
{ok, Config} = emqx:update_config([statsd], Body),
case maps:get(<<"enable">>, Body) of
true ->
_ = emqx_statsd_sup:stop_child(?APP),
emqx_statsd_sup:start_child(?APP, maps:get(config, Config));
false ->
_ = emqx_statsd_sup:stop_child(?APP),
ok
end,
{200, emqx:get_raw_config([<<"statsd">>], #{})}.