hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Ps prepare store query for sql audit

Browse Source
This commit is contained in:
obdev
2023-05-26 03:46:42 +00:00
committed by ob-robot
parent 57cbdb67ed
commit f4de9eb665
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/sql/ob_sql.cpp
View File

@ -1519,6 +1519,7 @@ int ObSql::handle_ps_prepare(const ObString &stmt,
// open_cursors is 0 to indicate a special state, no limit is set // open_cursors is 0 to indicate a special state, no limit is set
#define NEED_CHECK_SESS_MAX_PS_HANDLE_LIMIT(v) (0 == v ? false : true) #define NEED_CHECK_SESS_MAX_PS_HANDLE_LIMIT(v) (0 == v ? false : true)
int ret = OB_SUCCESS; int ret = OB_SUCCESS;
ObString cur_query;
// trimed_stmt仅用于query empty检查, prepare语句需要用原始语句, 避免checksum不一致 // trimed_stmt仅用于query empty检查, prepare语句需要用原始语句, 避免checksum不一致
ObString trimed_stmt = const_cast<ObString &>(stmt).trim(); ObString trimed_stmt = const_cast<ObString &>(stmt).trim();
if (trimed_stmt.empty()) { if (trimed_stmt.empty()) {
@ -1532,6 +1533,7 @@ int ObSql::handle_ps_prepare(const ObString &stmt,
ObSQLSessionInfo &session = result.get_session(); ObSQLSessionInfo &session = result.get_session();
ObPsCache *ps_cache = session.get_ps_cache(); ObPsCache *ps_cache = session.get_ps_cache();
ObExecContext &ectx = result.get_exec_context(); ObExecContext &ectx = result.get_exec_context();
ObIAllocator &allocator = result.get_mem_pool();
ObPhysicalPlanCtx *pctx = ectx.get_physical_plan_ctx(); ObPhysicalPlanCtx *pctx = ectx.get_physical_plan_ctx();
ObSchemaGetterGuard *schema_guard = context.schema_guard_; ObSchemaGetterGuard *schema_guard = context.schema_guard_;
ectx.set_is_ps_prepare_stage(true); ectx.set_is_ps_prepare_stage(true);
@ -1545,6 +1547,8 @@ int ObSql::handle_ps_prepare(const ObString &stmt,
ret = OB_INVALID_ARGUMENT; ret = OB_INVALID_ARGUMENT;
LOG_ERROR("physical plan context or ps plan cache is NULL or schema_guard is null", LOG_ERROR("physical plan context or ps plan cache is NULL or schema_guard is null",
K(ret), K(pctx), K(ps_cache)); K(ret), K(pctx), K(ps_cache));
} else if (OB_FAIL(ob_write_string(allocator, session.get_current_query_string(), cur_query))) {
LOG_WARN("failed to write string", K(ret));
} else if (OB_FAIL(session.store_query_string(trimed_stmt))) { } else if (OB_FAIL(session.store_query_string(trimed_stmt))) {
LOG_WARN("store query string fail", K(ret)); LOG_WARN("store query string fail", K(ret));
} else { } else {
@ -1660,6 +1664,7 @@ int ObSql::handle_ps_prepare(const ObString &stmt,
} }
} }
} }
OZ (session.store_query_string(cur_query));
} }
return ret; return ret;
} }