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

fix: authz mysql prepare query

This commit is contained in:
DDDHuang 2022-04-20 22:15:05 +08:00
parent 365ca67078
commit 2aedd38a43
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apps/emqx_authz/src/emqx_authz_mysql.erl
View File

@ -23,6 +23,8 @@
-behaviour(emqx_authz).
-define(PREPARE_KEY, ?MODULE).
%% AuthZ Callbacks
-export([
description/0,
@ -52,10 +54,11 @@ init(#{query := SQL} = Source) ->
{error, Reason} ->
error({load_config_error, Reason});
{ok, Id} ->
{PrepareKey, PrepareStatement} = emqx_authz_utils:parse_sql(SQL, '?', ?PLACEHOLDERS),
case emqx_resource:query(Id, {prepare_sql, [{PrepareKey, PrepareStatement}]}) of
{PrepareSQL, TmplToken} = emqx_authz_utils:parse_sql(SQL, '?', ?PLACEHOLDERS),
case emqx_resource:query(Id, {prepare_sql, [{?MODULE, PrepareSQL}]}) of
ok ->
Source#{annotations => #{id => Id, prepare => {PrepareKey, PrepareStatement}}};
Source#{annotations => #{
id => Id, tmpl_oken => TmplToken}};
{error, Reason} ->
error({load_config_error, Reason})
end
@ -71,12 +74,12 @@ authorize(
#{
annotations := #{
id := ResourceID,
prepare := {PrepareKey, PrepareStatement}
tmpl_oken := TmplToken
}
}
) ->
RenderParams = emqx_authz_utils:render_sql_params(PrepareStatement, Client),
case emqx_resource:query(ResourceID, {sql, PrepareKey, RenderParams}) of
RenderParams = emqx_authz_utils:render_sql_params(TmplToken, Client),
case emqx_resource:query(ResourceID, {prepared_query, ?MODULE, RenderParams}) of
{ok, _Columns, []} ->
nomatch;
{ok, Columns, Rows} ->
@ -85,7 +88,7 @@ authorize(
?SLOG(error, #{
msg => "query_mysql_error",
reason => Reason,
prepare => {PrepareKey, PrepareStatement},
tmpl_oken => TmplToken,
params => RenderParams,
resource_id => ResourceID
}),