75 lines
2.6 KiB
Erlang
75 lines
2.6 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_rewrite_api).
|
|
|
|
-behaviour(minirest_api).
|
|
|
|
-export([api_spec/0]).
|
|
|
|
-export([topic_rewrite/2]).
|
|
|
|
-define(MAX_RULES_LIMIT, 20).
|
|
|
|
-define(EXCEED_LIMIT, 'EXCEED_LIMIT').
|
|
|
|
-import(emqx_mgmt_util, [ object_array_schema/1
|
|
, object_array_schema/2
|
|
, error_schema/2
|
|
, properties/1
|
|
]).
|
|
|
|
api_spec() ->
|
|
{[rewrite_api()], []}.
|
|
|
|
properties() ->
|
|
properties([{action, string, <<"Node">>, [subscribe, publish]},
|
|
{source_topic, string, <<"Topic">>},
|
|
{re, string, <<"Regular expressions">>},
|
|
{dest_topic, string, <<"Destination topic">>}]).
|
|
|
|
rewrite_api() ->
|
|
Path = "/mqtt/topic_rewrite",
|
|
Metadata = #{
|
|
get => #{
|
|
description => <<"List topic rewrite">>,
|
|
responses => #{
|
|
<<"200">> => object_array_schema(properties(), <<"List all rewrite rules">>)
|
|
}
|
|
},
|
|
put => #{
|
|
description => <<"Update topic rewrite">>,
|
|
'requestBody' => object_array_schema(properties()),
|
|
responses => #{
|
|
<<"200">> =>object_array_schema(properties(), <<"Update topic rewrite success">>),
|
|
<<"413">> => error_schema(<<"Rules count exceed max limit">>, [?EXCEED_LIMIT])
|
|
}
|
|
}
|
|
},
|
|
{Path, Metadata, topic_rewrite}.
|
|
|
|
topic_rewrite(get, _Params) ->
|
|
{200, emqx_rewrite:list()};
|
|
|
|
topic_rewrite(put, #{body := Body}) ->
|
|
case length(Body) < ?MAX_RULES_LIMIT of
|
|
true ->
|
|
ok = emqx_rewrite:update(Body),
|
|
{200, emqx_rewrite:list()};
|
|
_ ->
|
|
Message = iolist_to_binary(io_lib:format("Max rewrite rules count is ~p", [?MAX_RULES_LIMIT])),
|
|
{413, #{code => ?EXCEED_LIMIT, message => Message}}
|
|
end.
|