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

fix reroute check for single sql in trans

Browse Source
This commit is contained in:
dimstars
2023-07-11 15:42:04 +00:00
committed by ob-robot
parent c3f0fb704b
commit 06e697204e
3 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/observer/mysql/obmp_query.cpp
View File

@ -340,7 +340,8 @@ int ObMPQuery::process()
// 但是目前的代码实现难以在不同的线程处理同一个请求的回包，
// 因此这里只允许只有一个query的multi query请求异步回包。
force_sync_resp = queries.count() <= 1? false : true;
ret = process_single_stmt(ObMultiStmtItem(true, i, queries.at(i)),
bool is_single_stmt = queries.count() <= 1? true : false;
ret = process_single_stmt(ObMultiStmtItem(true, i, queries.at(i), is_single_stmt),
session,
has_more,
force_sync_resp,

3
src/sql/ob_sql.cpp
View File

@ -5129,7 +5129,8 @@ int ObSql::check_need_reroute(ObPlanCacheCtx &pc_ctx, ObSQLSessionInfo &session,
if (should_reroute && !session.is_inner() && session.is_in_transaction()) {
auto stmt_type = plan->get_stmt_type();
bool fixed_route = true;
if (pc_ctx.sql_ctx_.multi_stmt_item_.is_part_of_multi_stmt()) {
if (pc_ctx.sql_ctx_.multi_stmt_item_.is_part_of_multi_stmt()
&& !pc_ctx.sql_ctx_.multi_stmt_item_.is_single_stmt()) {
// current is multi-stmt
} else if (!STMT_SUPPORT_BY_TXN_FREE_ROUTE(stmt_type, false)) {
// stmt is not DML

18
src/sql/ob_sql_context.h
View File

@ -163,17 +163,19 @@ public:
batched_queries_(NULL),
is_ins_multi_val_opt_(false),
is_ps_mode_(false),
ab_cnt_(0)
ab_cnt_(0),
is_single_stmt_(false)
{
}
ObMultiStmtItem(bool is_part_of_multi, int64_t seq_num, const common::ObString &sql)
ObMultiStmtItem(bool is_part_of_multi, int64_t seq_num, const common::ObString &sql, bool is_single_stmt=false)
: is_part_of_multi_stmt_(is_part_of_multi),
seq_num_(seq_num),
sql_(sql),
batched_queries_(NULL),
is_ins_multi_val_opt_(false),
is_ps_mode_(false),
ab_cnt_(0)
ab_cnt_(0),
is_single_stmt_(is_single_stmt)
{
}
@ -188,7 +190,8 @@ public:
batched_queries_(queries),
is_ins_multi_val_opt_(is_multi_vas_opt),
is_ps_mode_(false),
ab_cnt_(0)
ab_cnt_(0),
is_single_stmt_(false)
{
}
virtual ~ObMultiStmtItem() {}
@ -200,6 +203,7 @@ public:
sql_.reset();
batched_queries_ = NULL;
is_ins_multi_val_opt_ = false;
is_single_stmt_ = false;
}
inline bool is_part_of_multi_stmt() const { return is_part_of_multi_stmt_; }
@ -234,9 +238,10 @@ public:
inline bool is_ps_mode() { return is_ps_mode_; }
inline void set_ab_cnt(int64_t cnt) { ab_cnt_ = cnt; }
inline int64_t get_ab_cnt() { return ab_cnt_; }
inline bool is_single_stmt() const { return is_single_stmt_; }
TO_STRING_KV(K_(is_part_of_multi_stmt), K_(seq_num), K_(sql),
KPC_(batched_queries), K_(is_ins_multi_val_opt), K_(is_ps_mode), K_(ab_cnt));
TO_STRING_KV(K_(is_part_of_multi_stmt), K_(seq_num), K_(sql), KPC_(batched_queries),
K_(is_ins_multi_val_opt), K_(is_ps_mode), K_(ab_cnt), K_(is_single_stmt));
private:
bool is_part_of_multi_stmt_; // 是否为multi stmt，非multi stmt也使用这个结构体，因此需要这个标记
@ -247,6 +252,7 @@ private:
bool is_ins_multi_val_opt_;
bool is_ps_mode_;
int64_t ab_cnt_;
bool is_single_stmt_;
};
class ObQueryRetryInfo