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

feat(tpl): switch basic connectors to `emqx_connector_template` 

Also avoid `filename:join/2` in HTTP connector since it's both OS specific
and an overkill.
This commit is contained in:
Andrew Mayorov 2023-04-18 15:34:38 +03:00
parent 28d55d72ca
commit 35902dc72d
No known key found for this signature in database
GPG Key ID: 2837C62ACFBFED5D
6 changed files with 280 additions and 324 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

120
apps/emqx_bridge_http/src/emqx_bridge_http_connector.erl
View File

@ -479,61 +479,47 @@ preprocess_request(
} = Req } = Req
) -> ) ->
#{ #{
method => emqx_placeholder:preproc_tmpl(to_bin(Method)), method => parse_template(to_bin(Method)),
path => emqx_placeholder:preproc_tmpl(Path), path => parse_template(Path),
body => maybe_preproc_tmpl(body, Req), body => maybe_parse_template(body, Req),
headers => wrap_auth_header(preproc_headers(Headers)), headers => parse_headers(Headers),
request_timeout => maps:get(request_timeout, Req, ?DEFAULT_REQUEST_TIMEOUT_MS), request_timeout => maps:get(request_timeout, Req, ?DEFAULT_REQUEST_TIMEOUT_MS),
max_retries => maps:get(max_retries, Req, 2) max_retries => maps:get(max_retries, Req, 2)
}. }.
preproc_headers(Headers) when is_map(Headers) -> parse_headers(Headers) when is_map(Headers) ->
maps:fold( maps:fold(
fun(K, V, Acc) -> fun(K, V, Acc) -> [parse_header(K, V) | Acc] end,
[
{
emqx_placeholder:preproc_tmpl(to_bin(K)),
emqx_placeholder:preproc_tmpl(to_bin(V))
}
| Acc
]
end,
[], [],
Headers Headers
); );
preproc_headers(Headers) when is_list(Headers) -> parse_headers(Headers) when is_list(Headers) ->
lists:map( lists:map(
fun({K, V}) -> fun({K, V}) -> parse_header(K, V) end,
{
emqx_placeholder:preproc_tmpl(to_bin(K)),
emqx_placeholder:preproc_tmpl(to_bin(V))
}
end,
Headers Headers
). ).
wrap_auth_header(Headers) -> parse_header(K, V) ->
lists:map(fun maybe_wrap_auth_header/1, Headers). KStr = to_bin(K),
VTpl = parse_template(to_bin(V)),
{parse_template(KStr), maybe_wrap_auth_header(KStr, VTpl)}.
maybe_wrap_auth_header({[{str, Key}] = StrKey, Val}) -> maybe_wrap_auth_header(Key, VTpl) when
{_, MaybeWrapped} = maybe_wrap_auth_header({Key, Val}), (byte_size(Key) =:= 19 orelse byte_size(Key) =:= 13)
{StrKey, MaybeWrapped};
maybe_wrap_auth_header({Key, Val} = Header) when
is_binary(Key), (size(Key) =:= 19 orelse size(Key) =:= 13)
-> ->
%% We check the size of potential keys in the guard above and consider only %% We check the size of potential keys in the guard above and consider only
%% those that match the number of characters of either "Authorization" or %% those that match the number of characters of either "Authorization" or
%% "Proxy-Authorization". %% "Proxy-Authorization".
case try_bin_to_lower(Key) of case try_bin_to_lower(Key) of
<<"authorization">> -> <<"authorization">> ->
{Key, emqx_secret:wrap(Val)}; emqx_secret:wrap(VTpl);
<<"proxy-authorization">> -> <<"proxy-authorization">> ->
{Key, emqx_secret:wrap(Val)}; emqx_secret:wrap(VTpl);
_Other -> _Other ->
Header VTpl
end; end;
maybe_wrap_auth_header(Header) -> maybe_wrap_auth_header(_Key, VTpl) ->
Header. VTpl.
try_bin_to_lower(Bin) -> try_bin_to_lower(Bin) ->
try iolist_to_binary(string:lowercase(Bin)) of try iolist_to_binary(string:lowercase(Bin)) of
@ -542,46 +528,57 @@ try_bin_to_lower(Bin) ->
_:_ -> Bin _:_ -> Bin
end. end.
maybe_preproc_tmpl(Key, Conf) -> maybe_parse_template(Key, Conf) ->
case maps:get(Key, Conf, undefined) of case maps:get(Key, Conf, undefined) of
undefined -> undefined; undefined -> undefined;
Val -> emqx_placeholder:preproc_tmpl(Val) Val -> parse_template(Val)
end. end.
parse_template(String) ->
emqx_connector_template:parse(String).
process_request( process_request(
#{ #{
method := MethodTks, method := MethodTemplate,
path := PathTks, path := PathTemplate,
body := BodyTks, body := BodyTemplate,
headers := HeadersTks, headers := HeadersTemplate,
request_timeout := ReqTimeout request_timeout := ReqTimeout
} = Conf, } = Conf,
Msg Msg
) -> ) ->
Conf#{ Conf#{
method => make_method(emqx_placeholder:proc_tmpl(MethodTks, Msg)), method => make_method(render_template_string(MethodTemplate, Msg)),
path => emqx_placeholder:proc_tmpl(PathTks, Msg), path => unicode:characters_to_list(render_template(PathTemplate, Msg)),
body => process_request_body(BodyTks, Msg), body => render_request_body(BodyTemplate, Msg),
headers => proc_headers(HeadersTks, Msg), headers => render_headers(HeadersTemplate, Msg),
request_timeout => ReqTimeout request_timeout => ReqTimeout
}. }.
process_request_body(undefined, Msg) -> render_request_body(undefined, Msg) ->
emqx_utils_json:encode(Msg); emqx_utils_json:encode(Msg);
process_request_body(BodyTks, Msg) -> render_request_body(BodyTks, Msg) ->
emqx_placeholder:proc_tmpl(BodyTks, Msg). render_template(BodyTks, Msg).
proc_headers(HeaderTks, Msg) -> render_headers(HeaderTks, Msg) ->
lists:map( lists:map(
fun({K, V}) -> fun({K, V}) ->
{ {
emqx_placeholder:proc_tmpl(K, Msg), render_template_string(K, Msg),
emqx_placeholder:proc_tmpl(emqx_secret:unwrap(V), Msg) render_template_string(emqx_secret:unwrap(V), Msg)
} }
end, end,
HeaderTks HeaderTks
). ).
render_template(Template, Msg) ->
% NOTE: ignoring errors here, missing variables will be rendered as `"undefined"`.
{String, _Errors} = emqx_connector_template:render(Template, Msg),
String.
render_template_string(Template, Msg) ->
unicode:characters_to_binary(render_template(Template, Msg)).
make_method(M) when M == <<"POST">>; M == <<"post">> -> post; make_method(M) when M == <<"POST">>; M == <<"post">> -> post;
make_method(M) when M == <<"PUT">>; M == <<"put">> -> put; make_method(M) when M == <<"PUT">>; M == <<"put">> -> put;
make_method(M) when M == <<"GET">>; M == <<"get">> -> get; make_method(M) when M == <<"GET">>; M == <<"get">> -> get;
@ -716,8 +713,6 @@ maybe_retry(Result, _Context, ReplyFunAndArgs) ->
emqx_resource:apply_reply_fun(ReplyFunAndArgs, Result). emqx_resource:apply_reply_fun(ReplyFunAndArgs, Result).
%% The HOCON schema system may generate sensitive keys with this format %% The HOCON schema system may generate sensitive keys with this format
is_sensitive_key([{str, StringKey}]) ->
is_sensitive_key(StringKey);
is_sensitive_key(Atom) when is_atom(Atom) -> is_sensitive_key(Atom) when is_atom(Atom) ->
is_sensitive_key(erlang:atom_to_binary(Atom)); is_sensitive_key(erlang:atom_to_binary(Atom));
is_sensitive_key(Bin) when is_binary(Bin), (size(Bin) =:= 19 orelse size(Bin) =:= 13) -> is_sensitive_key(Bin) when is_binary(Bin), (size(Bin) =:= 19 orelse size(Bin) =:= 13) ->
@ -742,25 +737,19 @@ redact(Data) ->
%% and we also can't know the body format and where the sensitive data will be %% and we also can't know the body format and where the sensitive data will be
%% so the easy way to keep data security is redacted the whole body %% so the easy way to keep data security is redacted the whole body
redact_request({Path, Headers}) -> redact_request({Path, Headers}) ->
{Path, redact(Headers)}; {Path, Headers};
redact_request({Path, Headers, _Body}) -> redact_request({Path, Headers, _Body}) ->
{Path, redact(Headers), <<"******">>}. {Path, Headers, <<"******">>}.
-ifdef(TEST). -ifdef(TEST).
-include_lib("eunit/include/eunit.hrl"). -include_lib("eunit/include/eunit.hrl").
redact_test_() -> redact_test_() ->
TestData1 = [ TestData = #{
{<<"content-type">>, <<"application/json">>}, headers => [
{<<"Authorization">>, <<"Basic YWxhZGRpbjpvcGVuc2VzYW1l">>} {<<"content-type">>, <<"application/json">>},
], {<<"Authorization">>, <<"Basic YWxhZGRpbjpvcGVuc2VzYW1l">>}
]
TestData2 = #{
headers =>
[
{[{str, <<"content-type">>}], [{str, <<"application/json">>}]},
{[{str, <<"Authorization">>}], [{str, <<"Basic YWxhZGRpbjpvcGVuc2VzYW1l">>}]}
]
}, },
[ [
?_assert(is_sensitive_key(<<"Authorization">>)), ?_assert(is_sensitive_key(<<"Authorization">>)),
@ -770,8 +759,7 @@ redact_test_() ->
?_assert(is_sensitive_key('PrOxy-authoRizaTion')), ?_assert(is_sensitive_key('PrOxy-authoRizaTion')),
?_assertNot(is_sensitive_key(<<"Something">>)), ?_assertNot(is_sensitive_key(<<"Something">>)),
?_assertNot(is_sensitive_key(89)), ?_assertNot(is_sensitive_key(89)),
?_assertNotEqual(TestData1, redact(TestData1)), ?_assertNotEqual(TestData, redact(TestData))
?_assertNotEqual(TestData2, redact(TestData2))
]. ].
join_paths_test_() -> join_paths_test_() ->

3
apps/emqx_bridge_http/test/emqx_bridge_http_connector_tests.erl
View File

@ -83,7 +83,8 @@ is_wrapped(Secret) when is_function(Secret) ->
is_wrapped(_Other) -> is_wrapped(_Other) ->
false. false.
untmpl([{_, V} | _]) -> V. untmpl(Tpl) ->
iolist_to_binary(emqx_connector_template:render_strict(Tpl, #{})).
is_unwrapped_headers(Headers) -> is_unwrapped_headers(Headers) ->
lists:all(fun is_unwrapped_header/1, Headers). lists:all(fun is_unwrapped_header/1, Headers).

30
apps/emqx_bridge_mysql/test/emqx_bridge_mysql_SUITE.erl
View File

@ -565,8 +565,6 @@ t_simple_sql_query(Config) ->
ok. ok.
t_missing_data(Config) -> t_missing_data(Config) ->
BatchSize = ?config(batch_size, Config),
IsBatch = BatchSize > 1,
?assertMatch( ?assertMatch(
{ok, _}, {ok, _},
create_bridge(Config) create_bridge(Config)
@ -577,27 +575,13 @@ t_missing_data(Config) ->
), ),
send_message(Config, #{}), send_message(Config, #{}),
{ok, [Event]} = snabbkaffe:receive_events(SRef), {ok, [Event]} = snabbkaffe:receive_events(SRef),
case IsBatch of ?assertMatch(
true -> #{
?assertMatch( result :=
#{ {error, {unrecoverable_error, {1048, _, <<"Column 'arrived' cannot be null">>}}}
result := },
{error, Event
{unrecoverable_error, ),
{1292, _, <<"Truncated incorrect DOUBLE value: 'undefined'">>}}}
},
Event
);
false ->
?assertMatch(
#{
result :=
{error,
{unrecoverable_error, {1048, _, <<"Column 'arrived' cannot be null">>}}}
},
Event
)
end,
ok. ok.
t_bad_sql_parameter(Config) -> t_bad_sql_parameter(Config) ->

3
apps/emqx_bridge_pgsql/test/emqx_bridge_pgsql_SUITE.erl
View File

@ -324,6 +324,7 @@ connect_and_drop_table(Config) ->
connect_and_clear_table(Config) -> connect_and_clear_table(Config) ->
Con = connect_direct_pgsql(Config), Con = connect_direct_pgsql(Config),
_ = epgsql:squery(Con, ?SQL_CREATE_TABLE),
{ok, _} = epgsql:squery(Con, ?SQL_DELETE), {ok, _} = epgsql:squery(Con, ?SQL_DELETE),
ok = epgsql:close(Con). ok = epgsql:close(Con).
@ -668,7 +669,7 @@ t_missing_table(Config) ->
ok ok
end, end,
fun(Trace) -> fun(Trace) ->
?assertMatch([_, _, _], ?of_kind(pgsql_undefined_table, Trace)), ?assertMatch([_], ?of_kind(pgsql_undefined_table, Trace)),
ok ok
end end
), ),

248
apps/emqx_mysql/src/emqx_mysql.erl
View File

@ -46,16 +46,12 @@
default_port => ?MYSQL_DEFAULT_PORT default_port => ?MYSQL_DEFAULT_PORT
}). }).
-type prepares() :: #{atom() => binary()}. -type template() :: {unicode:chardata(), emqx_connector_template:str()}.
-type params_tokens() :: #{atom() => list()}.
-type sqls() :: #{atom() => binary()}.
-type state() :: -type state() ::
#{ #{
pool_name := binary(), pool_name := binary(),
prepare_statement := prepares(), prepares := ok | {error, _},
params_tokens := params_tokens(), templates := #{{atom(), batch | prepstmt} => template()}
batch_inserts := sqls(),
batch_params_tokens := params_tokens()
}. }.
%%===================================================================== %%=====================================================================
@ -154,13 +150,13 @@ on_query(InstId, {TypeOrKey, SQLOrKey, Params}, State) ->
on_query( on_query(
InstId, InstId,
{TypeOrKey, SQLOrKey, Params, Timeout}, {TypeOrKey, SQLOrKey, Params, Timeout},
#{pool_name := PoolName, prepare_statement := Prepares} = State State
) -> ) ->
MySqlFunction = mysql_function(TypeOrKey), MySqlFunction = mysql_function(TypeOrKey),
{SQLOrKey2, Data} = proc_sql_params(TypeOrKey, SQLOrKey, Params, State), {SQLOrKey2, Data} = proc_sql_params(TypeOrKey, SQLOrKey, Params, State),
case on_sql_query(InstId, MySqlFunction, SQLOrKey2, Data, Timeout, State) of case on_sql_query(InstId, MySqlFunction, SQLOrKey2, Data, Timeout, State) of
{error, not_prepared} -> {error, not_prepared} ->
case maybe_prepare_sql(SQLOrKey2, Prepares, PoolName) of case maybe_prepare_sql(SQLOrKey2, State) of
ok -> ok ->
?tp( ?tp(
mysql_connector_on_query_prepared_sql, mysql_connector_on_query_prepared_sql,
@ -187,23 +183,27 @@ on_query(
on_batch_query( on_batch_query(
InstId, InstId,
BatchReq, BatchReq = [{Key, _} | _],
#{batch_inserts := Inserts, batch_params_tokens := ParamsTokens} = State #{query_templates := Templates} = State
) -> ) ->
case hd(BatchReq) of case maps:get({Key, batch}, Templates, undefined) of
{Key, _} -> undefined ->
case maps:get(Key, Inserts, undefined) of {error, {unrecoverable_error, batch_select_not_implemented}};
undefined -> Template ->
{error, {unrecoverable_error, batch_select_not_implemented}}; on_batch_insert(InstId, BatchReq, Template, State)
InsertSQL -> end;
Tokens = maps:get(Key, ParamsTokens), on_batch_query(
on_batch_insert(InstId, BatchReq, InsertSQL, Tokens, State) InstId,
end; BatchReq,
Request -> State
LogMeta = #{connector => InstId, first_request => Request, state => State}, ) ->
?SLOG(error, LogMeta#{msg => "invalid request"}), ?SLOG(error, #{
{error, {unrecoverable_error, invalid_request}} msg => "invalid request",
end. connector => InstId,
request => BatchReq,
state => State
}),
{error, {unrecoverable_error, invalid_request}}.
mysql_function(sql) -> mysql_function(sql) ->
query; query;
@ -222,8 +222,8 @@ on_get_status(_InstId, #{pool_name := PoolName} = State) ->
{ok, NState} -> {ok, NState} ->
%% return new state with prepared statements %% return new state with prepared statements
{connected, NState}; {connected, NState};
{error, {undefined_table, NState}} -> {error, undefined_table} ->
{disconnected, NState, unhealthy_target}; {disconnected, State, unhealthy_target};
{error, _Reason} -> {error, _Reason} ->
%% do not log error, it is logged in prepare_sql_to_conn %% do not log error, it is logged in prepare_sql_to_conn
connecting connecting
@ -238,8 +238,8 @@ do_get_status(Conn) ->
do_check_prepares( do_check_prepares(
#{ #{
pool_name := PoolName, pool_name := PoolName,
prepare_statement := #{send_message := SQL} templates := #{{send_message, prepstmt} := SQL}
} = State }
) -> ) ->
% it's already connected. Verify if target table still exists % it's already connected. Verify if target table still exists
Workers = [Worker || {_WorkerName, Worker} <- ecpool:workers(PoolName)], Workers = [Worker || {_WorkerName, Worker} <- ecpool:workers(PoolName)],
@ -250,7 +250,7 @@ do_check_prepares(
{ok, Conn} -> {ok, Conn} ->
case mysql:prepare(Conn, get_status, SQL) of case mysql:prepare(Conn, get_status, SQL) of
{error, {1146, _, _}} -> {error, {1146, _, _}} ->
{error, {undefined_table, State}}; {error, undefined_table};
{ok, Statement} -> {ok, Statement} ->
mysql:unprepare(Conn, Statement); mysql:unprepare(Conn, Statement);
_ -> _ ->
@ -265,17 +265,14 @@ do_check_prepares(
ok, ok,
Workers Workers
); );
do_check_prepares(#{prepare_statement := Statement}) when is_map(Statement) -> do_check_prepares(#{prepares := ok}) ->
ok; ok;
do_check_prepares(State = #{pool_name := PoolName, prepare_statement := {error, Prepares}}) -> do_check_prepares(#{prepares := {error, _}} = State) ->
%% retry to prepare %% retry to prepare
case prepare_sql(Prepares, PoolName) of case prepare_sql(State) of
ok -> ok ->
%% remove the error %% remove the error
{ok, State#{prepare_statement => Prepares}}; {ok, State#{prepares => ok}};
{error, undefined_table} ->
%% indicate the error
{error, {undefined_table, State#{prepare_statement => {error, Prepares}}}};
{error, Reason} -> {error, Reason} ->
{error, Reason} {error, Reason}
end. end.
@ -285,41 +282,44 @@ do_check_prepares(State = #{pool_name := PoolName, prepare_statement := {error,
connect(Options) -> connect(Options) ->
mysql:start_link(Options). mysql:start_link(Options).
init_prepare(State = #{prepare_statement := Prepares, pool_name := PoolName}) -> init_prepare(State = #{query_templates := Templates}) ->
case maps:size(Prepares) of case maps:size(Templates) of
0 -> 0 ->
State; State#{prepares => ok};
_ -> _ ->
case prepare_sql(Prepares, PoolName) of case prepare_sql(State) of
ok -> ok ->
State; State#{prepares => ok};
{error, Reason} -> {error, Reason} ->
LogMeta = #{msg => <<"mysql_init_prepare_statement_failed">>, reason => Reason}, ?SLOG(error, #{
?SLOG(error, LogMeta), msg => <<"MySQL init prepare statement failed">>,
reason => Reason
}),
%% mark the prepare_statement as failed %% mark the prepare_statement as failed
State#{prepare_statement => {error, Prepares}} State#{prepares => {error, Reason}}
end end
end. end.
maybe_prepare_sql(SQLOrKey, Prepares, PoolName) -> maybe_prepare_sql(SQLOrKey, State = #{query_templates := Templates}) ->
case maps:is_key(SQLOrKey, Prepares) of case maps:is_key({SQLOrKey, prepstmt}, Templates) of
true -> prepare_sql(Prepares, PoolName); true -> prepare_sql(State);
false -> {error, {unrecoverable_error, prepared_statement_invalid}} false -> {error, {unrecoverable_error, prepared_statement_invalid}}
end. end.
prepare_sql(Prepares, PoolName) when is_map(Prepares) -> prepare_sql(#{query_templates := Templates, pool_name := PoolName}) ->
prepare_sql(maps:to_list(Prepares), PoolName); prepare_sql(maps:to_list(Templates), PoolName).
prepare_sql(Prepares, PoolName) ->
case do_prepare_sql(Prepares, PoolName) of prepare_sql(Templates, PoolName) ->
case do_prepare_sql(Templates, PoolName) of
ok -> ok ->
%% prepare for reconnect %% prepare for reconnect
ecpool:add_reconnect_callback(PoolName, {?MODULE, prepare_sql_to_conn, [Prepares]}), ecpool:add_reconnect_callback(PoolName, {?MODULE, prepare_sql_to_conn, [Templates]}),
ok; ok;
{error, R} -> {error, R} ->
{error, R} {error, R}
end. end.
do_prepare_sql(Prepares, PoolName) -> do_prepare_sql(Templates, PoolName) ->
Conns = Conns =
[ [
begin begin
@ -328,33 +328,30 @@ do_prepare_sql(Prepares, PoolName) ->
end end
|| {_Name, Worker} <- ecpool:workers(PoolName) || {_Name, Worker} <- ecpool:workers(PoolName)
], ],
prepare_sql_to_conn_list(Conns, Prepares). prepare_sql_to_conn_list(Conns, Templates).
prepare_sql_to_conn_list([], _PrepareList) -> prepare_sql_to_conn_list([], _Templates) ->
ok; ok;
prepare_sql_to_conn_list([Conn | ConnList], PrepareList) -> prepare_sql_to_conn_list([Conn | ConnList], Templates) ->
case prepare_sql_to_conn(Conn, PrepareList) of case prepare_sql_to_conn(Conn, Templates) of
ok -> ok ->
prepare_sql_to_conn_list(ConnList, PrepareList); prepare_sql_to_conn_list(ConnList, Templates);
{error, R} -> {error, R} ->
%% rollback %% rollback
Fun = fun({Key, _}) -> _ = [unprepare_sql_to_conn(Conn, Template) || Template <- Templates],
_ = unprepare_sql_to_conn(Conn, Key),
ok
end,
lists:foreach(Fun, PrepareList),
{error, R} {error, R}
end. end.
prepare_sql_to_conn(Conn, []) when is_pid(Conn) -> ok; prepare_sql_to_conn(_Conn, []) ->
prepare_sql_to_conn(Conn, [{Key, SQL} | PrepareList]) when is_pid(Conn) -> ok;
LogMeta = #{msg => "mysql_prepare_statement", name => Key, prepare_sql => SQL}, prepare_sql_to_conn(Conn, [{{Key, prepstmt}, {SQL, _RowTemplate}} | Rest]) ->
LogMeta = #{msg => "MySQL Prepare Statement", name => Key, prepare_sql => SQL},
?SLOG(info, LogMeta), ?SLOG(info, LogMeta),
_ = unprepare_sql_to_conn(Conn, Key), _ = unprepare_sql_to_conn(Conn, Key),
case mysql:prepare(Conn, Key, SQL) of case mysql:prepare(Conn, Key, SQL) of
{ok, _Key} -> {ok, _Key} ->
?SLOG(info, LogMeta#{result => success}), ?SLOG(info, LogMeta#{result => success}),
prepare_sql_to_conn(Conn, PrepareList); prepare_sql_to_conn(Conn, Rest);
{error, {1146, _, _} = Reason} -> {error, {1146, _, _} = Reason} ->
%% Target table is not created %% Target table is not created
?tp(mysql_undefined_table, #{}), ?tp(mysql_undefined_table, #{}),
@ -365,84 +362,85 @@ prepare_sql_to_conn(Conn, [{Key, SQL} | PrepareList]) when is_pid(Conn) ->
% syntax failures. Retrying syntax failures is not very productive. % syntax failures. Retrying syntax failures is not very productive.
?SLOG(error, LogMeta#{result => failed, reason => Reason}), ?SLOG(error, LogMeta#{result => failed, reason => Reason}),
{error, Reason} {error, Reason}
end. end;
prepare_sql_to_conn(Conn, [{_Key, _Template} | Rest]) ->
prepare_sql_to_conn(Conn, Rest).
unprepare_sql_to_conn(Conn, PrepareSqlKey) -> unprepare_sql_to_conn(Conn, {{Key, prepstmt}, _}) ->
mysql:unprepare(Conn, PrepareSqlKey). mysql:unprepare(Conn, Key);
unprepare_sql_to_conn(Conn, Key) when is_atom(Key) ->
mysql:unprepare(Conn, Key);
unprepare_sql_to_conn(_Conn, _) ->
ok.
parse_prepare_sql(Config) -> parse_prepare_sql(Config) ->
SQL = Queries =
case maps:get(prepare_statement, Config, undefined) of case Config of
undefined -> #{prepare_statement := Qs} ->
case maps:get(sql, Config, undefined) of Qs;
undefined -> #{}; #{sql := Query} ->
Template -> #{send_message => Template} #{send_message => Query};
end; _ ->
Any -> #{}
Any
end, end,
parse_prepare_sql(maps:to_list(SQL), #{}, #{}, #{}, #{}). Templates = maps:fold(fun parse_prepare_sql/3, #{}, Queries),
#{query_templates => Templates}.
parse_prepare_sql([{Key, H} | _] = L, Prepares, Tokens, BatchInserts, BatchTks) -> parse_prepare_sql(Key, Query, Acc) ->
{PrepareSQL, ParamsTokens} = emqx_placeholder:preproc_sql(H), Template = emqx_connector_template_sql:parse_prepstmt(Query, #{parameters => '?'}),
parse_batch_prepare_sql( AccNext = Acc#{{Key, prepstmt} => Template},
L, Prepares#{Key => PrepareSQL}, Tokens#{Key => ParamsTokens}, BatchInserts, BatchTks parse_batch_sql(Key, Query, AccNext).
);
parse_prepare_sql([], Prepares, Tokens, BatchInserts, BatchTks) ->
#{
prepare_statement => Prepares,
params_tokens => Tokens,
batch_inserts => BatchInserts,
batch_params_tokens => BatchTks
}.
parse_batch_prepare_sql([{Key, H} | T], Prepares, Tokens, BatchInserts, BatchTks) -> parse_batch_sql(Key, Query, Acc) ->
case emqx_utils_sql:get_statement_type(H) of case emqx_connector_sql:get_statement_type(Query) of
select ->
parse_prepare_sql(T, Prepares, Tokens, BatchInserts, BatchTks);
insert -> insert ->
case emqx_utils_sql:parse_insert(H) of case emqx_connector_sql:parse_insert(Query) of
{ok, {InsertSQL, Params}} -> {ok, {Insert, Params}} ->
ParamsTks = emqx_placeholder:preproc_tmpl(Params), RowTemplate = emqx_connector_template_sql:parse(Params),
parse_prepare_sql( Acc#{{Key, batch} => {Insert, RowTemplate}};
T,
Prepares,
Tokens,
BatchInserts#{Key => InsertSQL},
BatchTks#{Key => ParamsTks}
);
{error, Reason} -> {error, Reason} ->
?SLOG(error, #{msg => "split_sql_failed", sql => H, reason => Reason}), ?SLOG(error, #{
parse_prepare_sql(T, Prepares, Tokens, BatchInserts, BatchTks) msg => "parse insert sql statement failed",
sql => Query,
reason => Reason
}),
Acc
end; end;
Type when is_atom(Type) -> select ->
?SLOG(error, #{msg => "detect_sql_type_unsupported", sql => H, type => Type}), Acc;
parse_prepare_sql(T, Prepares, Tokens, BatchInserts, BatchTks); Otherwise ->
{error, Reason} -> ?SLOG(error, #{
?SLOG(error, #{msg => "detect_sql_type_failed", sql => H, reason => Reason}), msg => "invalid sql statement type",
parse_prepare_sql(T, Prepares, Tokens, BatchInserts, BatchTks) sql => Query,
type => Otherwise
}),
Acc
end. end.
proc_sql_params(query, SQLOrKey, Params, _State) -> proc_sql_params(query, SQLOrKey, Params, _State) ->
{SQLOrKey, Params}; {SQLOrKey, Params};
proc_sql_params(prepared_query, SQLOrKey, Params, _State) -> proc_sql_params(prepared_query, SQLOrKey, Params, _State) ->
{SQLOrKey, Params}; {SQLOrKey, Params};
proc_sql_params(TypeOrKey, SQLOrData, Params, #{params_tokens := ParamsTokens}) -> proc_sql_params(TypeOrKey, SQLOrData, Params, #{query_templates := Templates}) ->
case maps:get(TypeOrKey, ParamsTokens, undefined) of case maps:get({TypeOrKey, prepstmt}, Templates, undefined) of
undefined -> undefined ->
{SQLOrData, Params}; {SQLOrData, Params};
Tokens -> {_InsertPart, RowTemplate} ->
{TypeOrKey, emqx_placeholder:proc_sql(Tokens, SQLOrData)} % NOTE: ignoring errors here, missing variables are set to `null`.
{Row, _Errors} = emqx_connector_template_sql:render_prepstmt(RowTemplate, SQLOrData),
{TypeOrKey, Row}
end. end.
on_batch_insert(InstId, BatchReqs, InsertPart, Tokens, State) -> on_batch_insert(InstId, BatchReqs, {InsertPart, RowTemplate}, State) ->
ValuesPart = lists:join($,, [ Rows = [render_row(RowTemplate, Msg) || {_, Msg} <- BatchReqs],
emqx_placeholder:proc_param_str(Tokens, Msg, fun emqx_placeholder:quote_mysql/1) Query = [InsertPart, <<" values ">> | lists:join($,, Rows)],
|| {_, Msg} <- BatchReqs
]),
Query = [InsertPart, <<" values ">> | ValuesPart],
on_sql_query(InstId, query, Query, no_params, default_timeout, State). on_sql_query(InstId, query, Query, no_params, default_timeout, State).
render_row(RowTemplate, Data) ->
% NOTE: ignoring errors here, missing variables are set to "NULL".
{Row, _Errors} = emqx_connector_template_sql:render(RowTemplate, Data, #{escaping => mysql}),
Row.
on_sql_query( on_sql_query(
InstId, InstId,
SQLFunc, SQLFunc,

200
apps/emqx_postgresql/src/emqx_postgresql.erl
View File

@ -52,15 +52,12 @@
default_port => ?PGSQL_DEFAULT_PORT default_port => ?PGSQL_DEFAULT_PORT
}). }).
-type prepares() :: #{atom() => binary()}. -type template() :: {unicode:chardata(), emqx_connector_template_sql:row_template()}.
-type params_tokens() :: #{atom() => list()}.
-type state() :: -type state() ::
#{ #{
pool_name := binary(), pool_name := binary(),
prepare_sql := prepares(), query_templates := #{binary() => template()},
params_tokens := params_tokens(), prepares := #{binary() => epgsql:statement()} | {error, _}
prepare_statement := epgsql:statement()
}. }.
%% FIXME: add `{error, sync_required}' to `epgsql:execute_batch' %% FIXME: add `{error, sync_required}' to `epgsql:execute_batch'
@ -142,7 +139,7 @@ on_start(
State = parse_prepare_sql(Config), State = parse_prepare_sql(Config),
case emqx_resource_pool:start(InstId, ?MODULE, Options ++ SslOpts) of case emqx_resource_pool:start(InstId, ?MODULE, Options ++ SslOpts) of
ok -> ok ->
{ok, init_prepare(State#{pool_name => InstId, prepare_statement => #{}})}; {ok, init_prepare(State#{pool_name => InstId, prepares => #{}})};
{error, Reason} -> {error, Reason} ->
?tp( ?tp(
pgsql_connector_start_failed, pgsql_connector_start_failed,
@ -189,55 +186,50 @@ pgsql_query_type(_) ->
on_batch_query( on_batch_query(
InstId, InstId,
BatchReq, [{Key, _} = Request | _] = BatchReq,
#{pool_name := PoolName, params_tokens := Tokens, prepare_statement := Sts} = State #{pool_name := PoolName, query_templates := Templates, prepares := PrepStatements} = State
) -> ) ->
case BatchReq of BinKey = to_bin(Key),
[{Key, _} = Request | _] -> case maps:get(BinKey, Templates, undefined) of
BinKey = to_bin(Key), undefined ->
case maps:get(BinKey, Tokens, undefined) of
undefined ->
Log = #{
connector => InstId,
first_request => Request,
state => State,
msg => "batch_prepare_not_implemented"
},
?SLOG(error, Log),
{error, {unrecoverable_error, batch_prepare_not_implemented}};
TokenList ->
{_, Datas} = lists:unzip(BatchReq),
Datas2 = [emqx_placeholder:proc_sql(TokenList, Data) || Data <- Datas],
St = maps:get(BinKey, Sts),
case on_sql_query(InstId, PoolName, execute_batch, St, Datas2) of
{error, _Error} = Result ->
handle_result(Result);
{_Column, Results} ->
handle_batch_result(Results, 0)
end
end;
_ ->
Log = #{ Log = #{
connector => InstId, connector => InstId,
request => BatchReq, first_request => Request,
state => State, state => State,
msg => "invalid_request" msg => "batch prepare not implemented"
}, },
?SLOG(error, Log), ?SLOG(error, Log),
{error, {unrecoverable_error, invalid_request}} {error, {unrecoverable_error, batch_prepare_not_implemented}};
end. {_Statement, RowTemplate} ->
PrepStatement = maps:get(BinKey, PrepStatements),
Rows = [render_prepare_sql_row(RowTemplate, Data) || {_Key, Data} <- BatchReq],
case on_sql_query(InstId, PoolName, execute_batch, PrepStatement, Rows) of
{error, _Error} = Result ->
handle_result(Result);
{_Column, Results} ->
handle_batch_result(Results, 0)
end
end;
on_batch_query(InstId, BatchReq, State) ->
?SLOG(error, #{
connector => InstId,
request => BatchReq,
state => State,
msg => "invalid request"
}),
{error, {unrecoverable_error, invalid_request}}.
proc_sql_params(query, SQLOrKey, Params, _State) -> proc_sql_params(query, SQLOrKey, Params, _State) ->
{SQLOrKey, Params}; {SQLOrKey, Params};
proc_sql_params(prepared_query, SQLOrKey, Params, _State) -> proc_sql_params(prepared_query, SQLOrKey, Params, _State) ->
{SQLOrKey, Params}; {SQLOrKey, Params};
proc_sql_params(TypeOrKey, SQLOrData, Params, #{params_tokens := ParamsTokens}) -> proc_sql_params(TypeOrKey, SQLOrData, Params, #{query_templates := Templates}) ->
Key = to_bin(TypeOrKey), Key = to_bin(TypeOrKey),
case maps:get(Key, ParamsTokens, undefined) of case maps:get(Key, Templates, undefined) of
undefined -> undefined ->
{SQLOrData, Params}; {SQLOrData, Params};
Tokens -> {_Statement, RowTemplate} ->
{Key, emqx_placeholder:proc_sql(Tokens, SQLOrData)} {Key, render_prepare_sql_row(RowTemplate, SQLOrData)}
end. end.
on_sql_query(InstId, PoolName, Type, NameOrSQL, Data) -> on_sql_query(InstId, PoolName, Type, NameOrSQL, Data) ->
@ -297,9 +289,9 @@ on_get_status(_InstId, #{pool_name := PoolName} = State) ->
{ok, NState} -> {ok, NState} ->
%% return new state with prepared statements %% return new state with prepared statements
{connected, NState}; {connected, NState};
{error, {undefined_table, NState}} -> {error, undefined_table} ->
%% return new state indicating that we are connected but the target table is not created %% return new state indicating that we are connected but the target table is not created
{disconnected, NState, unhealthy_target}; {disconnected, State, unhealthy_target};
{error, _Reason} -> {error, _Reason} ->
%% do not log error, it is logged in prepare_sql_to_conn %% do not log error, it is logged in prepare_sql_to_conn
connecting connecting
@ -314,8 +306,8 @@ do_get_status(Conn) ->
do_check_prepares( do_check_prepares(
#{ #{
pool_name := PoolName, pool_name := PoolName,
prepare_sql := #{<<"send_message">> := SQL} query_templates := #{<<"send_message">> := {SQL, _RowTemplate}}
} = State }
) -> ) ->
WorkerPids = [Worker || {_WorkerName, Worker} <- ecpool:workers(PoolName)], WorkerPids = [Worker || {_WorkerName, Worker} <- ecpool:workers(PoolName)],
case validate_table_existence(WorkerPids, SQL) of case validate_table_existence(WorkerPids, SQL) of
@ -324,19 +316,16 @@ do_check_prepares(
{error, undefined_table} -> {error, undefined_table} ->
{error, {undefined_table, State}} {error, {undefined_table, State}}
end; end;
do_check_prepares(#{prepare_sql := Prepares}) when is_map(Prepares) -> do_check_prepares(#{prepares := Prepares}) when is_map(Prepares) ->
ok; ok;
do_check_prepares(State = #{pool_name := PoolName, prepare_sql := {error, Prepares}}) -> do_check_prepares(#{prepares := {error, _}} = State) ->
%% retry to prepare %% retry to prepare
case prepare_sql(Prepares, PoolName) of case prepare_sql(State) of
{ok, Sts} -> {ok, PrepStatements} ->
%% remove the error %% remove the error
{ok, State#{prepare_sql => Prepares, prepare_statement := Sts}}; {ok, State#{prepares := PrepStatements}};
{error, undefined_table} -> {error, Reason} ->
%% indicate the error {error, Reason}
{error, {undefined_table, State#{prepare_sql => {error, Prepares}}}};
Error ->
{error, Error}
end. end.
-spec validate_table_existence([pid()], binary()) -> ok | {error, undefined_table}. -spec validate_table_existence([pid()], binary()) -> ok | {error, undefined_table}.
@ -426,69 +415,63 @@ conn_opts([_Opt | Opts], Acc) ->
conn_opts(Opts, Acc). conn_opts(Opts, Acc).
parse_prepare_sql(Config) -> parse_prepare_sql(Config) ->
SQL = Queries =
case maps:get(prepare_statement, Config, undefined) of case Config of
undefined -> #{prepare_statement := Qs} ->
case maps:get(sql, Config, undefined) of Qs;
undefined -> #{}; #{sql := Query} ->
Template -> #{<<"send_message">> => Template} #{<<"send_message">> => Query};
end; #{} ->
Any -> #{}
Any
end, end,
parse_prepare_sql(maps:to_list(SQL), #{}, #{}). Templates = maps:fold(fun parse_prepare_sql/3, #{}, Queries),
#{query_templates => Templates}.
parse_prepare_sql([{Key, H} | T], Prepares, Tokens) -> parse_prepare_sql(Key, Query, Acc) ->
{PrepareSQL, ParamsTokens} = emqx_placeholder:preproc_sql(H, '$n'), Template = emqx_connector_template_sql:parse_prepstmt(Query, #{parameters => '$n'}),
parse_prepare_sql( Acc#{Key => Template}.
T, Prepares#{Key => PrepareSQL}, Tokens#{Key => ParamsTokens}
);
parse_prepare_sql([], Prepares, Tokens) ->
#{
prepare_sql => Prepares,
params_tokens => Tokens
}.
init_prepare(State = #{prepare_sql := Prepares, pool_name := PoolName}) -> render_prepare_sql_row(RowTemplate, Data) ->
case maps:size(Prepares) of % NOTE: ignoring errors here, missing variables will be replaced with `null`.
0 -> {Row, _Errors} = emqx_connector_template_sql:render_prepstmt(RowTemplate, Data),
State; Row.
_ ->
case prepare_sql(Prepares, PoolName) of init_prepare(State = #{query_templates := Templates}) when map_size(Templates) == 0 ->
{ok, Sts} -> State;
State#{prepare_statement := Sts}; init_prepare(State = #{}) ->
Error -> case prepare_sql(State) of
LogMsg = {ok, PrepStatements} ->
maps:merge( State#{prepares => PrepStatements};
#{msg => <<"postgresql_init_prepare_statement_failed">>}, Error ->
translate_to_log_context(Error) ?SLOG(error, maps:merge(
), #{msg => <<"postgresql_init_prepare_statement_failed">>},
?SLOG(error, LogMsg), translate_to_log_context(Error)
%% mark the prepare_sql as failed )),
State#{prepare_sql => {error, Prepares}} %% mark the prepares failed
end State#{prepares => Error}
end. end.
prepare_sql(Prepares, PoolName) when is_map(Prepares) -> prepare_sql(#{query_templates := Templates, pool_name := PoolName}) ->
prepare_sql(maps:to_list(Prepares), PoolName); prepare_sql(maps:to_list(Templates), PoolName).
prepare_sql(Prepares, PoolName) ->
case do_prepare_sql(Prepares, PoolName) of prepare_sql(Templates, PoolName) ->
case do_prepare_sql(Templates, PoolName) of
{ok, _Sts} = Ok -> {ok, _Sts} = Ok ->
%% prepare for reconnect %% prepare for reconnect
ecpool:add_reconnect_callback(PoolName, {?MODULE, prepare_sql_to_conn, [Prepares]}), ecpool:add_reconnect_callback(PoolName, {?MODULE, prepare_sql_to_conn, [Templates]}),
Ok; Ok;
Error -> Error ->
Error Error
end. end.
do_prepare_sql(Prepares, PoolName) -> do_prepare_sql(Templates, PoolName) ->
do_prepare_sql(ecpool:workers(PoolName), Prepares, #{}). do_prepare_sql(ecpool:workers(PoolName), Templates, #{}).
do_prepare_sql([{_Name, Worker} | T], Prepares, _LastSts) -> do_prepare_sql([{_Name, Worker} | Rest], Templates, _LastSts) ->
{ok, Conn} = ecpool_worker:client(Worker), {ok, Conn} = ecpool_worker:client(Worker),
case prepare_sql_to_conn(Conn, Prepares) of case prepare_sql_to_conn(Conn, Templates) of
{ok, Sts} -> {ok, Sts} ->
do_prepare_sql(T, Prepares, Sts); do_prepare_sql(Rest, Templates, Sts);
Error -> Error ->
Error Error
end; end;
@ -498,13 +481,14 @@ do_prepare_sql([], _Prepares, LastSts) ->
prepare_sql_to_conn(Conn, Prepares) -> prepare_sql_to_conn(Conn, Prepares) ->
prepare_sql_to_conn(Conn, Prepares, #{}). prepare_sql_to_conn(Conn, Prepares, #{}).
prepare_sql_to_conn(Conn, [], Statements) when is_pid(Conn) -> {ok, Statements}; prepare_sql_to_conn(Conn, [], Statements) when is_pid(Conn) ->
prepare_sql_to_conn(Conn, [{Key, SQL} | PrepareList], Statements) when is_pid(Conn) -> {ok, Statements};
LogMeta = #{msg => "postgresql_prepare_statement", name => Key, prepare_sql => SQL}, prepare_sql_to_conn(Conn, [{Key, {SQL, _RowTemplate}} | Rest], Statements) when is_pid(Conn) ->
LogMeta = #{msg => "PostgreSQL Prepare Statement", name => Key, sql => SQL},
?SLOG(info, LogMeta), ?SLOG(info, LogMeta),
case epgsql:parse2(Conn, Key, SQL, []) of case epgsql:parse2(Conn, Key, SQL, []) of
{ok, Statement} -> {ok, Statement} ->
prepare_sql_to_conn(Conn, PrepareList, Statements#{Key => Statement}); prepare_sql_to_conn(Conn, Rest, Statements#{Key => Statement});
{error, {error, error, _, undefined_table, _, _} = Error} -> {error, {error, error, _, undefined_table, _, _} = Error} ->
%% Target table is not created %% Target table is not created
?tp(pgsql_undefined_table, #{}), ?tp(pgsql_undefined_table, #{}),