feat(exhook): export basic usage info for telemetry
This commit is contained in:
parent
22e2ad1c2e
commit
72b87a6a7e
|
@ -23,6 +23,9 @@
|
||||||
, call_fold/3
|
, call_fold/3
|
||||||
]).
|
]).
|
||||||
|
|
||||||
|
%% exported for `emqx_telemetry'
|
||||||
|
-export([get_basic_usage_info/0]).
|
||||||
|
|
||||||
%%--------------------------------------------------------------------
|
%%--------------------------------------------------------------------
|
||||||
%% Dispatch APIs
|
%% Dispatch APIs
|
||||||
%%--------------------------------------------------------------------
|
%%--------------------------------------------------------------------
|
||||||
|
@ -81,3 +84,39 @@ deny_action_result('message.publish', Msg) ->
|
||||||
%% TODO: Not support to deny a message
|
%% TODO: Not support to deny a message
|
||||||
%% maybe we can put the 'allow_publish' into message header
|
%% maybe we can put the 'allow_publish' into message header
|
||||||
Msg.
|
Msg.
|
||||||
|
|
||||||
|
%%--------------------------------------------------------------------
|
||||||
|
%% APIs for `emqx_telemetry'
|
||||||
|
%%--------------------------------------------------------------------
|
||||||
|
|
||||||
|
-spec get_basic_usage_info() ->
|
||||||
|
#{ num_servers => non_neg_integer()
|
||||||
|
, servers =>
|
||||||
|
[#{ driver => Driver
|
||||||
|
, hooks => [emqx_exhook_server:hookpoint()]
|
||||||
|
}]
|
||||||
|
} when Driver :: grpc.
|
||||||
|
get_basic_usage_info() ->
|
||||||
|
try
|
||||||
|
Servers = emqx_exhook_mgr:running(),
|
||||||
|
NumServers = length(Servers),
|
||||||
|
ServerInfo =
|
||||||
|
lists:map(
|
||||||
|
fun(ServerName) ->
|
||||||
|
Hooks = emqx_exhook_mgr:hooks(ServerName),
|
||||||
|
HookNames = lists:map(fun(#{name := Name}) -> Name end, Hooks),
|
||||||
|
#{ hooks => HookNames
|
||||||
|
, %% currently, only grpc driver exists.
|
||||||
|
driver => grpc
|
||||||
|
}
|
||||||
|
end,
|
||||||
|
Servers),
|
||||||
|
#{ num_servers => NumServers
|
||||||
|
, servers => ServerInfo
|
||||||
|
}
|
||||||
|
catch
|
||||||
|
_:_ ->
|
||||||
|
#{ num_servers => 0
|
||||||
|
, servers => []
|
||||||
|
}
|
||||||
|
end.
|
||||||
|
|
|
@ -231,6 +231,38 @@ t_misc_test(_) ->
|
||||||
_ = emqx_exhook_server:format(#{name => <<"test">>, hookspec => #{}}),
|
_ = emqx_exhook_server:format(#{name => <<"test">>, hookspec => #{}}),
|
||||||
ok.
|
ok.
|
||||||
|
|
||||||
|
t_get_basic_usage_info(_Config) ->
|
||||||
|
#{ num_servers := NumServers
|
||||||
|
, servers := Servers
|
||||||
|
} = emqx_exhook:get_basic_usage_info(),
|
||||||
|
?assertEqual(1, NumServers),
|
||||||
|
?assertMatch([_], Servers),
|
||||||
|
[#{driver := Driver, hooks := Hooks}] = Servers,
|
||||||
|
?assertEqual(grpc, Driver),
|
||||||
|
?assertEqual(
|
||||||
|
[
|
||||||
|
'client.authenticate',
|
||||||
|
'client.authorize',
|
||||||
|
'client.connack',
|
||||||
|
'client.connect',
|
||||||
|
'client.connected',
|
||||||
|
'client.disconnected',
|
||||||
|
'client.subscribe',
|
||||||
|
'client.unsubscribe',
|
||||||
|
'message.acked',
|
||||||
|
'message.delivered',
|
||||||
|
'message.dropped',
|
||||||
|
'message.publish',
|
||||||
|
'session.created',
|
||||||
|
'session.discarded',
|
||||||
|
'session.resumed',
|
||||||
|
'session.subscribed',
|
||||||
|
'session.takenover',
|
||||||
|
'session.terminated',
|
||||||
|
'session.unsubscribed'
|
||||||
|
],
|
||||||
|
lists:sort(Hooks)).
|
||||||
|
|
||||||
%%--------------------------------------------------------------------
|
%%--------------------------------------------------------------------
|
||||||
%% Utils
|
%% Utils
|
||||||
%%--------------------------------------------------------------------
|
%%--------------------------------------------------------------------
|
||||||
|
|
Loading…
Reference in New Issue