78 lines
2.6 KiB
Erlang
78 lines
2.6 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2020-2024 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_utils_api).
|
|
|
|
-export([
|
|
with_node/2,
|
|
with_node_or_cluster/2
|
|
]).
|
|
|
|
-include("emqx_utils_api.hrl").
|
|
|
|
-define(NODE_NOT_FOUND(NODE), ?NOT_FOUND(<<"Node not found: ", NODE/binary>>)).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% exported API
|
|
%%--------------------------------------------------------------------
|
|
-spec with_node(binary() | atom(), fun((atom()) -> {ok, term()} | {error, term()})) ->
|
|
?OK(term()) | ?NOT_FOUND(binary()) | ?BAD_REQUEST(term()).
|
|
with_node(Node0, Fun) ->
|
|
case lookup_node(Node0) of
|
|
{ok, Node} ->
|
|
handle_result(Fun(Node));
|
|
not_found ->
|
|
?NODE_NOT_FOUND(Node0)
|
|
end.
|
|
|
|
-spec with_node_or_cluster(binary() | atom(), fun((atom()) -> {ok, term()} | {error, term()})) ->
|
|
?OK(term()) | ?NOT_FOUND(iolist()) | ?BAD_REQUEST(term()).
|
|
with_node_or_cluster(<<"all">>, Fun) ->
|
|
handle_result(Fun(all));
|
|
with_node_or_cluster(Node, Fun) ->
|
|
with_node(Node, Fun).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Internal
|
|
%%--------------------------------------------------------------------
|
|
|
|
-spec lookup_node(atom() | binary()) -> {ok, atom()} | not_found.
|
|
lookup_node(BinNode) when is_binary(BinNode) ->
|
|
case emqx_utils:safe_to_existing_atom(BinNode, utf8) of
|
|
{ok, Node} ->
|
|
is_running_node(Node);
|
|
_Error ->
|
|
not_found
|
|
end;
|
|
lookup_node(Node) when is_atom(Node) ->
|
|
is_running_node(Node).
|
|
|
|
-spec is_running_node(atom()) -> {ok, atom()} | not_found.
|
|
is_running_node(Node) ->
|
|
case lists:member(Node, mria:running_nodes()) of
|
|
true ->
|
|
{ok, Node};
|
|
false ->
|
|
not_found
|
|
end.
|
|
|
|
handle_result({ok, Result}) ->
|
|
?OK(Result);
|
|
handle_result({error, Reason}) ->
|
|
?BAD_REQUEST(Reason);
|
|
handle_result({HTTPCode, Content}) when is_integer(HTTPCode) ->
|
|
{HTTPCode, Content}.
|