yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add stop after render and after render trace to greptimedb action

This commit is contained in:
Kjell Winblad 2024-04-24 10:37:52 +02:00
parent 7922d5d422
commit 7c7590fbc8
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/emqx_bridge_greptimedb/src/emqx_bridge_greptimedb_connector.erl
View File

@ -128,7 +128,7 @@ on_query(InstId, {Channel, Message}, State) ->
greptimedb_connector_send_query, greptimedb_connector_send_query,
#{points => Points, batch => false, mode => sync} #{points => Points, batch => false, mode => sync}
), ),
do_query(InstId, Client, Points); do_query(InstId, Channel, Client, Points);
{error, ErrorPoints} -> {error, ErrorPoints} ->
?tp( ?tp(
greptimedb_connector_send_query_error, greptimedb_connector_send_query_error,
@ -152,7 +152,7 @@ on_batch_query(InstId, [{Channel, _} | _] = BatchData, State) ->
greptimedb_connector_send_query, greptimedb_connector_send_query,
#{points => Points, batch => true, mode => sync} #{points => Points, batch => true, mode => sync}
), ),
do_query(InstId, Client, Points); do_query(InstId, Channel, Client, Points);
{error, Reason} -> {error, Reason} ->
?tp( ?tp(
greptimedb_connector_send_query_error, greptimedb_connector_send_query_error,
@ -173,7 +173,7 @@ on_query_async(InstId, {Channel, Message}, {ReplyFun, Args}, State) ->
greptimedb_connector_send_query, greptimedb_connector_send_query,
#{points => Points, batch => false, mode => async} #{points => Points, batch => false, mode => async}
), ),
do_async_query(InstId, Client, Points, {ReplyFun, Args}); do_async_query(InstId, Channel, Client, Points, {ReplyFun, Args});
{error, ErrorPoints} = Err -> {error, ErrorPoints} = Err ->
?tp( ?tp(
greptimedb_connector_send_query_error, greptimedb_connector_send_query_error,
@ -195,7 +195,7 @@ on_batch_query_async(InstId, [{Channel, _} | _] = BatchData, {ReplyFun, Args}, S
greptimedb_connector_send_query, greptimedb_connector_send_query,
#{points => Points, batch => true, mode => async} #{points => Points, batch => true, mode => async}
), ),
do_async_query(InstId, Client, Points, {ReplyFun, Args}); do_async_query(InstId, Channel, Client, Points, {ReplyFun, Args});
{error, Reason} -> {error, Reason} ->
?tp( ?tp(
greptimedb_connector_send_query_error, greptimedb_connector_send_query_error,
@ -420,7 +420,8 @@ is_auth_key(_) ->
%% ------------------------------------------------------------------------------------------------- %% -------------------------------------------------------------------------------------------------
%% Query %% Query
do_query(InstId, Client, Points) -> do_query(InstId, Channel, Client, Points) ->
emqx_trace:rendered_action_template(Channel, #{points => Points, is_async => false}),
case greptimedb:write_batch(Client, Points) of case greptimedb:write_batch(Client, Points) of
{ok, #{response := {affected_rows, #{value := Rows}}}} -> {ok, #{response := {affected_rows, #{value := Rows}}}} ->
?SLOG(debug, #{ ?SLOG(debug, #{
@ -452,12 +453,13 @@ do_query(InstId, Client, Points) ->
end end
end. end.
do_async_query(InstId, Client, Points, ReplyFunAndArgs) -> do_async_query(InstId, Channel, Client, Points, ReplyFunAndArgs) ->
?SLOG(info, #{ ?SLOG(info, #{
msg => "greptimedb_write_point_async", msg => "greptimedb_write_point_async",
connector => InstId, connector => InstId,
points => Points points => Points
}), }),
emqx_trace:rendered_action_template(Channel, #{points => Points, is_async => true}),
WrappedReplyFunAndArgs = {fun ?MODULE:reply_callback/2, [ReplyFunAndArgs]}, WrappedReplyFunAndArgs = {fun ?MODULE:reply_callback/2, [ReplyFunAndArgs]},
ok = greptimedb:async_write_batch(Client, Points, WrappedReplyFunAndArgs). ok = greptimedb:async_write_batch(Client, Points, WrappedReplyFunAndArgs).