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

support vector index query use plan parameterization

Browse Source
This commit is contained in:
obdev 2024-09-18 06:56:32 +00:00 committed by ob-robot
parent cd4968488a
commit b95d1e3ce5
2 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/sql/plan_cache/ob_sql_parameterization.cpp
View File

@ -325,7 +325,9 @@ bool ObSqlParameterization::is_tree_not_param(const ParseNode *tree)
// oracle模式下，select a from t group by 1这种语法被禁止，所以可以开放group by的参数化
ret_bool = true;
} else if (T_SORT_LIST == tree->type_) {
ret_bool = true;
// vector index query always use order by vec_func() approx limit, we should open Parameterization for this situation
bool is_vec_idx_query = is_vector_index_query(tree);
ret_bool = is_vec_idx_query ? false : true;
} else if (T_HINT_OPTION_LIST == tree->type_) {
ret_bool = true;
} else if (T_COLLATION == tree->type_) {
@ -2455,3 +2457,25 @@ int ObSqlParameterization::find_leftest_const_node(ParseNode &cur_node, ParseNod
}
return ret;
}
bool ObSqlParameterization::is_vector_index_query(const ParseNode *tree)
{
bool bret = false;
// for vector index query
if (T_SORT_LIST == tree->type_) {
if (tree->num_child_ == 1 && OB_NOT_NULL(tree->children_[0])) { // only one sort key
ParseNode *curr = tree->children_[0];
if (curr->type_ == T_SORT_KEY && curr->num_child_ == 2 && OB_NOT_NULL(curr->children_[0])) {
curr = curr->children_[0];
if (curr->type_ == T_FUN_SYS && curr->num_child_ == 2 && OB_NOT_NULL(curr->children_[0])) {
curr = curr->children_[0]; // sort key is sys func
if (curr->type_ == T_IDENT && OB_NOT_NULL(curr->str_value_)) {
ObString func_name(curr->str_len_, curr->str_value_);
bret = func_name.case_compare("l2_distance") == 0;
}
}
}
}
}
return bret;
}

1
src/sql/plan_cache/ob_sql_parameterization.h
View File

@ -221,6 +221,7 @@ private:
static int find_leftest_const_node(ParseNode &cur_node, ParseNode *&const_node);
static bool need_fast_parser(const ObString &sql);
static bool is_vector_index_query(const ParseNode *tree);
};
}