96 lines
2.7 KiB
Erlang
96 lines
2.7 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2020-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.
|
|
%%--------------------------------------------------------------------
|
|
|
|
-module(emqx_statsd_api).
|
|
|
|
-behaviour(minirest_api).
|
|
|
|
-include("emqx_statsd.hrl").
|
|
|
|
-include_lib("hocon/include/hoconsc.hrl").
|
|
-include_lib("typerefl/include/types.hrl").
|
|
|
|
-import(hoconsc, [mk/2, ref/2]).
|
|
|
|
-export([statsd/2]).
|
|
|
|
-export([
|
|
api_spec/0,
|
|
paths/0,
|
|
schema/1
|
|
]).
|
|
|
|
-define(API_TAG_STATSD, [<<"Monitor">>]).
|
|
-define(SCHEMA_MODULE, emqx_statsd_schema).
|
|
|
|
-define(INTERNAL_ERROR, 'INTERNAL_ERROR').
|
|
|
|
api_spec() ->
|
|
emqx_dashboard_swagger:spec(?MODULE, #{check_schema => true}).
|
|
|
|
paths() ->
|
|
["/statsd"].
|
|
|
|
schema("/statsd") ->
|
|
#{
|
|
'operationId' => statsd,
|
|
get =>
|
|
#{
|
|
description => ?DESC(get_statsd_config_api),
|
|
tags => ?API_TAG_STATSD,
|
|
responses =>
|
|
#{200 => statsd_config_schema()}
|
|
},
|
|
put =>
|
|
#{
|
|
description => ?DESC(update_statsd_config_api),
|
|
tags => ?API_TAG_STATSD,
|
|
'requestBody' => statsd_config_schema(),
|
|
responses =>
|
|
#{200 => statsd_config_schema()}
|
|
}
|
|
}.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Helper funcs
|
|
%%--------------------------------------------------------------------
|
|
|
|
statsd_config_schema() ->
|
|
emqx_dashboard_swagger:schema_with_example(
|
|
ref(?SCHEMA_MODULE, "statsd"),
|
|
statsd_example()
|
|
).
|
|
|
|
statsd_example() ->
|
|
#{
|
|
enable => true,
|
|
flush_time_interval => "30s",
|
|
sample_time_interval => "30s",
|
|
server => "127.0.0.1:8125",
|
|
tags => #{}
|
|
}.
|
|
|
|
statsd(get, _Params) ->
|
|
{200, emqx:get_raw_config([<<"statsd">>], #{})};
|
|
statsd(put, #{body := Body}) ->
|
|
case emqx_statsd_config:update(Body) of
|
|
{ok, NewConfig} ->
|
|
{200, NewConfig};
|
|
{error, Reason} ->
|
|
Message = list_to_binary(io_lib:format("Update config failed ~p", [Reason])),
|
|
{500, ?INTERNAL_ERROR, Message}
|
|
end.
|