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

patch 4.0

Browse Source
This commit is contained in:
wangzelin.wzl
2022-10-24 10:34:53 +08:00
parent 4ad6e00ec3
commit 93a1074b0c
10533 changed files with 2588271 additions and 2299373 deletions
Download Patch File Download Diff File Expand all files Collapse all files

740
src/sql/resolver/ob_stmt.h
View File

@ -21,15 +21,18 @@
#include "share/ob_autoincrement_param.h"
#include "sql/resolver/expr/ob_raw_expr.h"
#include "sql/resolver/ob_stmt_type.h"
#include "share/stat/ob_opt_stat_manager.h"
namespace oceanbase {
namespace sql {
#include "share/schema/ob_dependency_info.h" // ObReferenceObjTable
namespace oceanbase
{
namespace sql
{
class ObStmt;
struct ObStmtHint;
struct ObQueryCtx;
class ObSelectStmt;
struct ObStmtLevelRefSet : public common::ObBitSet<common::OB_DEFAULT_STATEMEMT_LEVEL_COUNT> {
struct ObStmtLevelRefSet: public common::ObBitSet<common::OB_DEFAULT_STATEMEMT_LEVEL_COUNT>
{
virtual int64_t to_string(char* buf, const int64_t buf_len) const
{
int64_t pos = 0;
@ -49,165 +52,144 @@ struct ObStmtLevelRefSet : public common::ObBitSet<common::OB_DEFAULT_STATEMEMT_
}
};
struct expr_hash_func {
uint64_t operator()(ObRawExpr* expr) const
struct expr_hash_func
{
uint64_t operator()(ObRawExpr *expr) const
{
return reinterpret_cast<uint64_t>(expr);
}
};
struct expr_equal_to {
bool operator()(ObRawExpr* a, ObRawExpr* b) const
struct expr_equal_to
{
bool operator()(ObRawExpr *a, ObRawExpr *b) const
{
return a == b;
}
};
/// the base class of all statements
class ObStmt {
class ObStmt
{
public:
typedef common::ObSEArray<uint64_t, 8, common::ModulePageAllocator, true> ObSynonymIds;
typedef common::ObSEArray<uint64_t, 8, common::ModulePageAllocator, true> ObSynonymIds;
public:
ObStmt()
: stmt_type_(stmt::T_NONE),
literal_stmt_type_(stmt::T_NONE),
sql_stmt_(),
sql_stmt_coll_type_(CS_TYPE_INVALID),
query_ctx_(NULL),
stmt_id_(0),
prepare_param_count_(0),
is_prepare_stmt_(false),
has_nested_sql_(false),
synonym_id_store_(),
tz_info_(NULL)
{}
stmt_id_(OB_INVALID_STMT_ID)
{
}
explicit ObStmt(const stmt::StmtType stmt_type)
: stmt_type_(stmt_type),
literal_stmt_type_(stmt::T_NONE),
sql_stmt_(),
sql_stmt_coll_type_(CS_TYPE_INVALID),
query_ctx_(NULL),
stmt_id_(0),
prepare_param_count_(0),
is_prepare_stmt_(false),
has_nested_sql_(false),
synonym_id_store_(),
tz_info_(NULL)
{}
ObStmt(common::ObIAllocator* name_pool, stmt::StmtType type)
stmt_id_(OB_INVALID_STMT_ID)
{
}
ObStmt(common::ObIAllocator *name_pool, stmt::StmtType type)
: stmt_type_(type),
literal_stmt_type_(stmt::T_NONE),
sql_stmt_(),
sql_stmt_coll_type_(CS_TYPE_INVALID),
query_ctx_(NULL),
stmt_id_(0),
prepare_param_count_(0),
is_prepare_stmt_(false),
has_nested_sql_(false),
synonym_id_store_(),
tz_info_(NULL)
stmt_id_(OB_INVALID_STMT_ID)
{
UNUSED(name_pool);
}
virtual ~ObStmt();
void set_stmt_id();
int64_t get_stmt_id() const
{
return stmt_id_;
}
int get_stmt_name_by_id(int64_t stmt_id, common::ObString& stmt_name) const;
int get_stmt_org_name_by_id(int64_t stmt_id, common::ObString& org_name) const;
int get_stmt_name(common::ObString& stmt_name) const;
int get_stmt_org_name(common::ObString& org_name) const;
virtual int get_first_stmt(common::ObString& first_stmt);
int set_stmt_id();
int64_t get_stmt_id() const { return stmt_id_; }
virtual int get_first_stmt(common::ObString &first_stmt);
void set_stmt_type(const stmt::StmtType stmt_type);
stmt::StmtType get_stmt_type() const;
void set_literal_stmt_type(const stmt::StmtType type)
{
literal_stmt_type_ = type;
}
stmt::StmtType get_literal_stmt_type() const
{
return literal_stmt_type_;
}
// 因为对于show，字面type是SHOW，实际stmt_type_是SELECT
// 所以这里实现成通用方法
static bool is_diagnostic_stmt(const stmt::StmtType type)
{
return stmt::T_SHOW_WARNINGS == type || stmt::T_SHOW_ERRORS == type;
return stmt::T_SHOW_WARNINGS == type || stmt::T_SHOW_ERRORS == type || stmt::T_DIAGNOSTICS == type;
}
static bool is_show_trace_stmt(const stmt::StmtType type)
{
return stmt::T_SHOW_TRACE == type;
}
virtual bool has_global_variable() const
{
return false;
}
virtual bool has_global_variable() const { return false; }
virtual bool is_show_stmt() const;
inline bool is_select_stmt() const
{
return is_select_stmt(stmt_type_);
}
inline bool is_insert_stmt() const
{
return stmt::T_INSERT == stmt_type_ || stmt::T_REPLACE == stmt_type_;
}
inline bool is_merge_stmt() const
{
return stmt::T_MERGE == stmt_type_;
}
inline bool is_update_stmt() const
{
return stmt::T_UPDATE == stmt_type_;
}
inline bool is_delete_stmt() const
{
return stmt::T_DELETE == stmt_type_;
}
inline bool is_explain_stmt() const
{
return stmt::T_EXPLAIN == stmt_type_;
}
inline bool is_select_stmt() const { return is_select_stmt(stmt_type_); }
inline bool is_insert_stmt() const { return stmt::T_INSERT == stmt_type_ || stmt::T_REPLACE == stmt_type_; }
inline bool is_insert_all_stmt() const { return stmt::T_INSERT_ALL == stmt_type_; }
inline bool is_merge_stmt() const { return stmt::T_MERGE == stmt_type_; }
inline bool is_update_stmt() const { return stmt::T_UPDATE == stmt_type_; }
inline bool is_delete_stmt() const { return stmt::T_DELETE == stmt_type_; }
inline bool is_explain_stmt() const { return stmt::T_EXPLAIN == stmt_type_; }
inline bool is_help_stmt() const { return stmt::T_HELP == stmt_type_; }
bool is_dml_stmt() const;
bool is_pdml_supported_stmt() const;
bool is_px_dml_supported_stmt() const;
bool is_dml_write_stmt() const
{ return is_dml_write_stmt(stmt_type_); }
bool is_support_batch_exec_stmt() const
{
return is_dml_write_stmt(stmt_type_);
return stmt_type_ == stmt::T_INSERT
|| stmt_type_ == stmt::T_UPDATE
|| stmt_type_ == stmt::T_DELETE;
}
bool is_valid_transform_stmt() const
{
return stmt_type_ == stmt::T_SELECT || stmt_type_ == stmt::T_DELETE || stmt_type_ == stmt::T_UPDATE ||
stmt_type_ == stmt::T_INSERT || stmt_type_ == stmt::T_REPLACE || stmt_type_ == stmt::T_MERGE;
return stmt_type_ == stmt::T_SELECT
|| stmt_type_ == stmt::T_DELETE
|| stmt_type_ == stmt::T_UPDATE
|| stmt_type_ == stmt::T_INSERT_ALL
|| stmt_type_ == stmt::T_INSERT
|| stmt_type_ == stmt::T_REPLACE
|| stmt_type_ == stmt::T_MERGE;
}
bool is_sel_del_upd() const
{
return stmt_type_ == stmt::T_SELECT || stmt_type_ == stmt::T_DELETE || stmt_type_ == stmt::T_UPDATE;
return stmt_type_ == stmt::T_SELECT
|| stmt_type_ == stmt::T_DELETE
|| stmt_type_ == stmt::T_UPDATE;
}
bool is_allowed_reroute() const
{
return (stmt_type_ == stmt::T_SELECT || stmt_type_ == stmt::T_DELETE || stmt_type_ == stmt::T_UPDATE ||
stmt_type_ == stmt::T_INSERT || stmt_type_ == stmt::T_REPLACE || stmt_type_ == stmt::T_MERGE);
return (stmt_type_ == stmt::T_SELECT
|| stmt_type_ == stmt::T_DELETE
|| stmt_type_ == stmt::T_UPDATE
|| stmt_type_ == stmt::T_INSERT
|| stmt_type_ == stmt::T_INSERT_ALL
|| stmt_type_ == stmt::T_REPLACE
|| stmt_type_ == stmt::T_MERGE);
}
inline bool is_support_instead_of_trigger_stmt() const {
return stmt::T_DELETE == stmt_type_
|| stmt::T_UPDATE == stmt_type_
|| stmt::T_INSERT == stmt_type_;
}
static inline bool is_show_stmt(stmt::StmtType stmt_type)
{
return (stmt_type >= stmt::T_SHOW_TABLES && stmt_type <= stmt::T_SHOW_GRANTS);
return (stmt_type >= stmt::T_SHOW_TABLES && stmt_type <= stmt::T_SHOW_GRANTS)
|| stmt_type == stmt::T_SHOW_TRIGGERS;
}
static inline bool is_dml_write_stmt(stmt::StmtType stmt_type)
{
return (stmt_type == stmt::T_INSERT || stmt_type == stmt::T_REPLACE || stmt_type == stmt::T_DELETE ||
stmt_type == stmt::T_UPDATE || stmt_type == stmt::T_MERGE);
return (stmt_type == stmt::T_INSERT
|| stmt_type == stmt::T_INSERT_ALL
|| stmt_type == stmt::T_REPLACE
|| stmt_type == stmt::T_DELETE
|| stmt_type == stmt::T_UPDATE
|| stmt_type == stmt::T_MERGE);
}
static inline bool is_write_stmt(stmt::StmtType stmt_type, bool has_global_variable)
{
return is_ddl_stmt(stmt_type, has_global_variable) ||
(stmt_type == stmt::T_INSERT || stmt_type == stmt::T_REPLACE || stmt_type == stmt::T_DELETE ||
stmt_type == stmt::T_UPDATE);
(stmt_type == stmt::T_INSERT ||
stmt_type == stmt::T_INSERT_ALL ||
stmt_type == stmt::T_REPLACE ||
stmt_type == stmt::T_DELETE ||
stmt_type == stmt::T_UPDATE);
}
static inline bool is_select_stmt(stmt::StmtType stmt_type)
@ -217,9 +199,15 @@ public:
static inline bool is_dml_stmt(stmt::StmtType stmt_type)
{
return (stmt_type == stmt::T_SELECT || stmt_type == stmt::T_INSERT || stmt_type == stmt::T_REPLACE ||
stmt_type == stmt::T_MERGE || stmt_type == stmt::T_DELETE || stmt_type == stmt::T_UPDATE ||
stmt_type == stmt::T_EXPLAIN || is_show_stmt(stmt_type));
return (stmt_type == stmt::T_SELECT
|| stmt_type == stmt::T_INSERT
|| stmt_type == stmt::T_INSERT_ALL
|| stmt_type == stmt::T_REPLACE
|| stmt_type == stmt::T_MERGE
|| stmt_type == stmt::T_DELETE
|| stmt_type == stmt::T_UPDATE
|| stmt_type == stmt::T_EXPLAIN
|| is_show_stmt(stmt_type));
}
static inline bool is_execute_stmt(stmt::StmtType stmt_type)
@ -229,19 +217,32 @@ public:
static inline bool is_pdml_supported_stmt(stmt::StmtType stmt_type)
{
return (stmt_type == stmt::T_INSERT || stmt_type == stmt::T_DELETE || stmt_type == stmt::T_UPDATE);
return (stmt_type == stmt::T_INSERT
|| stmt_type == stmt::T_DELETE
|| stmt_type == stmt::T_UPDATE
|| stmt_type == stmt::T_MERGE);
}
static inline bool is_px_dml_supported_stmt(stmt::StmtType stmt_type)
{
return (stmt_type == stmt::T_INSERT || stmt_type == stmt::T_DELETE || stmt_type == stmt::T_UPDATE ||
stmt_type == stmt::T_REPLACE || stmt_type == stmt::T_MERGE);
return (stmt_type == stmt::T_INSERT
|| stmt_type == stmt::T_INSERT_ALL
|| stmt_type == stmt::T_DELETE
|| stmt_type == stmt::T_UPDATE
|| stmt_type == stmt::T_REPLACE
|| stmt_type == stmt::T_MERGE);
}
static bool is_dynamic_supported_stmt(stmt::StmtType stmt_type)
{
return !(stmt::T_KILL == stmt_type);
}
static inline bool is_savepoint_stmt(stmt::StmtType stmt_type)
{
return (stmt::T_CREATE_SAVEPOINT == stmt_type || stmt::T_ROLLBACK_SAVEPOINT == stmt_type ||
stmt::T_RELEASE_SAVEPOINT == stmt_type);
return (stmt::T_CREATE_SAVEPOINT == stmt_type
|| stmt::T_ROLLBACK_SAVEPOINT == stmt_type
|| stmt::T_RELEASE_SAVEPOINT == stmt_type);
}
static inline bool is_tcl_stmt(stmt::StmtType stmt_type)
@ -253,177 +254,252 @@ public:
{
return (
// tenant resource
stmt_type == stmt::T_CREATE_RESOURCE_POOL || stmt_type == stmt::T_DROP_RESOURCE_POOL ||
stmt_type == stmt::T_ALTER_RESOURCE_POOL || stmt_type == stmt::T_SPLIT_RESOURCE_POOL ||
stmt_type == stmt::T_MERGE_RESOURCE_POOL || stmt_type == stmt::T_CREATE_RESOURCE_UNIT ||
stmt_type == stmt::T_ALTER_RESOURCE_UNIT || stmt_type == stmt::T_DROP_RESOURCE_UNIT ||
stmt_type == stmt::T_CREATE_TENANT || stmt_type == stmt::T_DROP_TENANT || stmt_type == stmt::T_MODIFY_TENANT ||
stmt_type == stmt::T_LOCK_TENANT
// database
|| stmt_type == stmt::T_CREATE_DATABASE || stmt_type == stmt::T_ALTER_DATABASE ||
stmt_type == stmt::T_DROP_DATABASE
// tablegroup
|| stmt_type == stmt::T_CREATE_TABLEGROUP || stmt_type == stmt::T_ALTER_TABLEGROUP ||
stmt_type == stmt::T_DROP_TABLEGROUP
// table
|| stmt_type == stmt::T_CREATE_TABLE || stmt_type == stmt::T_DROP_TABLE || stmt_type == stmt::T_RENAME_TABLE ||
stmt_type == stmt::T_TRUNCATE_TABLE || stmt_type == stmt::T_CREATE_TABLE_LIKE ||
stmt_type == stmt::T_ALTER_TABLE ||
stmt_type == stmt::T_SET_TABLE_COMMENT
// column
|| stmt_type == stmt::T_SET_COLUMN_COMMENT
// audit and noaudit
|| stmt_type == stmt::T_AUDIT
// optimize
|| stmt_type == stmt::T_OPTIMIZE_TABLE || stmt_type == stmt::T_OPTIMIZE_TENANT ||
stmt_type == stmt::T_OPTIMIZE_ALL
// view
|| stmt_type == stmt::T_CREATE_VIEW || stmt_type == stmt::T_ALTER_VIEW ||
stmt_type == stmt::T_DROP_VIEW
// index
|| stmt_type == stmt::T_CREATE_INDEX ||
stmt_type == stmt::T_DROP_INDEX
// flashback
|| stmt_type == stmt::T_FLASHBACK_TENANT
|| stmt_type == stmt::T_FLASHBACK_DATABASE
|| stmt_type == stmt::T_FLASHBACK_TABLE_FROM_RECYCLEBIN
|| stmt_type == stmt::T_FLASHBACK_INDEX
// purge
|| stmt_type == stmt::T_PURGE_RECYCLEBIN || stmt_type == stmt::T_PURGE_TENANT ||
stmt_type == stmt::T_PURGE_DATABASE || stmt_type == stmt::T_PURGE_TABLE ||
stmt_type == stmt::T_PURGE_INDEX
// outline
|| stmt_type == stmt::T_CREATE_OUTLINE || stmt_type == stmt::T_ALTER_OUTLINE ||
stmt_type == stmt::T_DROP_OUTLINE
// sequence
|| stmt_type == stmt::T_CREATE_SEQUENCE || stmt_type == stmt::T_ALTER_SEQUENCE ||
stmt_type == stmt::T_DROP_SEQUENCE
stmt_type == stmt::T_CREATE_RESOURCE_POOL
|| stmt_type == stmt::T_DROP_RESOURCE_POOL
|| stmt_type == stmt::T_ALTER_RESOURCE_POOL
|| stmt_type == stmt::T_SPLIT_RESOURCE_POOL
|| stmt_type == stmt::T_MERGE_RESOURCE_POOL
|| stmt_type == stmt::T_CREATE_RESOURCE_UNIT
|| stmt_type == stmt::T_ALTER_RESOURCE_UNIT
|| stmt_type == stmt::T_DROP_RESOURCE_UNIT
|| stmt_type == stmt::T_CREATE_TENANT
|| stmt_type == stmt::T_DROP_TENANT
|| stmt_type == stmt::T_MODIFY_TENANT
|| stmt_type == stmt::T_LOCK_TENANT
// database
|| stmt_type == stmt::T_CREATE_DATABASE
|| stmt_type == stmt::T_ALTER_DATABASE
|| stmt_type == stmt::T_DROP_DATABASE
// tablegroup
|| stmt_type == stmt::T_CREATE_TABLEGROUP
|| stmt_type == stmt::T_ALTER_TABLEGROUP
|| stmt_type == stmt::T_DROP_TABLEGROUP
// table
|| stmt_type == stmt::T_CREATE_TABLE
|| stmt_type == stmt::T_DROP_TABLE
|| stmt_type == stmt::T_RENAME_TABLE
|| stmt_type == stmt::T_TRUNCATE_TABLE
|| stmt_type == stmt::T_CREATE_TABLE_LIKE
|| stmt_type == stmt::T_ALTER_TABLE
|| stmt_type == stmt::T_SET_TABLE_COMMENT
// column
|| stmt_type == stmt::T_SET_COLUMN_COMMENT
// audit and noaudit
|| stmt_type == stmt::T_AUDIT
// analyze, 这个在 oracle 里属于 ddl，但是 ob 判定其为 ddl 时会有一些问题
// TODO:待溪峰处理完 analyze 的问题后放开
//|| stmt_type == stmt::T_ANALYZE
// optimize
|| stmt_type == stmt::T_OPTIMIZE_TABLE
|| stmt_type == stmt::T_OPTIMIZE_TENANT
|| stmt_type == stmt::T_OPTIMIZE_ALL
// view
|| stmt_type == stmt::T_CREATE_VIEW
|| stmt_type == stmt::T_ALTER_VIEW
|| stmt_type == stmt::T_DROP_VIEW
// index
|| stmt_type == stmt::T_CREATE_INDEX
|| stmt_type == stmt::T_DROP_INDEX
// flashback
|| stmt_type == stmt::T_FLASHBACK_TENANT
|| stmt_type == stmt::T_FLASHBACK_DATABASE
|| stmt_type == stmt::T_FLASHBACK_TABLE_FROM_RECYCLEBIN
|| stmt_type == stmt::T_FLASHBACK_TABLE_TO_SCN
|| stmt_type == stmt::T_FLASHBACK_INDEX
// purge
|| stmt_type == stmt::T_PURGE_RECYCLEBIN
|| stmt_type == stmt::T_PURGE_TENANT
|| stmt_type == stmt::T_PURGE_DATABASE
|| stmt_type == stmt::T_PURGE_TABLE
|| stmt_type == stmt::T_PURGE_INDEX
// outline
|| stmt_type == stmt::T_CREATE_OUTLINE
|| stmt_type == stmt::T_ALTER_OUTLINE
|| stmt_type == stmt::T_DROP_OUTLINE
// sequence
|| stmt_type == stmt::T_CREATE_SEQUENCE
|| stmt_type == stmt::T_ALTER_SEQUENCE
|| stmt_type == stmt::T_DROP_SEQUENCE
// grant and revoke
|| stmt_type == stmt::T_GRANT ||
stmt_type == stmt::T_REVOKE
// grant and revoke
|| stmt_type == stmt::T_GRANT
|| stmt_type == stmt::T_REVOKE
// synonym
|| stmt_type == stmt::T_CREATE_SYNONYM ||
stmt_type == stmt::T_DROP_SYNONYM
//synonym
|| stmt_type == stmt::T_CREATE_SYNONYM
|| stmt_type == stmt::T_DROP_SYNONYM
// variable
|| (stmt_type == stmt::T_VARIABLE_SET && has_global_variable)
// variable
//目前只有set global variable才是DDL操作，session级别的variable变更不是DDL
|| (stmt_type == stmt::T_VARIABLE_SET && has_global_variable)
// plan baseline
|| stmt_type == stmt::T_ALTER_BASELINE
// stored procedure
|| stmt_type == stmt::T_CREATE_ROUTINE
|| stmt_type == stmt::T_DROP_ROUTINE
|| stmt_type == stmt::T_ALTER_ROUTINE
// stored procedure
|| stmt_type == stmt::T_CREATE_ROUTINE || stmt_type == stmt::T_DROP_ROUTINE ||
stmt_type == stmt::T_ALTER_ROUTINE
// package
|| stmt_type == stmt::T_CREATE_PACKAGE
|| stmt_type == stmt::T_CREATE_PACKAGE_BODY
|| stmt_type == stmt::T_ALTER_PACKAGE
|| stmt_type == stmt::T_DROP_PACKAGE
// package
|| stmt_type == stmt::T_CREATE_PACKAGE || stmt_type == stmt::T_CREATE_PACKAGE_BODY ||
stmt_type == stmt::T_ALTER_PACKAGE ||
stmt_type == stmt::T_DROP_PACKAGE
// trigger
|| stmt_type == stmt::T_CREATE_TRIGGER
|| stmt_type == stmt::T_DROP_TRIGGER
|| stmt_type == stmt::T_ALTER_TRIGGER
// trigger
|| stmt_type == stmt::T_CREATE_TRIGGER || stmt_type == stmt::T_DROP_TRIGGER ||
stmt_type == stmt::T_ALTER_TRIGGER
// user define type
|| stmt_type == stmt::T_CREATE_TYPE
|| stmt_type == stmt::T_DROP_TYPE
// user define type
|| stmt_type == stmt::T_CREATE_TYPE ||
stmt_type == stmt::T_DROP_TYPE
// trigger
|| stmt_type == stmt::T_CREATE_TRIGGER
|| stmt_type == stmt::T_DROP_TRIGGER
|| stmt_type == stmt::T_ALTER_TRIGGER
|| stmt_type == stmt::T_CREATE_DBLINK
|| stmt_type == stmt::T_DROP_DBLINK
// trigger
|| stmt_type == stmt::T_CREATE_DBLINK ||
stmt_type == stmt::T_DROP_DBLINK
// user function
|| stmt_type == stmt::T_CREATE_FUNC || stmt_type == stmt::T_DROP_FUNC ||
(stmt_type == stmt::T_CREATE_USER && lib::is_oracle_mode()));
// keystore
|| stmt_type == stmt::T_CREATE_KEYSTORE
|| stmt_type == stmt::T_ALTER_KEYSTORE
// tablespace
|| stmt_type == stmt::T_CREATE_TABLESPACE
|| stmt_type == stmt::T_ALTER_TABLESPACE
|| stmt_type == stmt::T_DROP_TABLESPACE
// user function
|| stmt_type == stmt::T_CREATE_FUNC
|| stmt_type == stmt::T_DROP_FUNC
|| (stmt_type == stmt::T_CREATE_USER && lib::is_oracle_mode())
// directory
|| stmt_type == stmt::T_CREATE_DIRECTORY
|| stmt_type == stmt::T_DROP_DIRECTORY
// application context
|| stmt_type == stmt::T_CREATE_CONTEXT
|| stmt_type == stmt::T_DROP_CONTEXT
);
}
static inline bool is_ddl_stmt_allowed_in_dropping_tenant(stmt::StmtType stmt_type, bool has_global_variable)
{
return ( // tenant resource
stmt_type == stmt::T_CREATE_RESOURCE_POOL || stmt_type == stmt::T_DROP_RESOURCE_POOL ||
stmt_type == stmt::T_ALTER_RESOURCE_POOL || stmt_type == stmt::T_SPLIT_RESOURCE_POOL ||
stmt_type == stmt::T_MERGE_RESOURCE_POOL || stmt_type == stmt::T_CREATE_RESOURCE_UNIT ||
stmt_type == stmt::T_ALTER_RESOURCE_UNIT || stmt_type == stmt::T_DROP_RESOURCE_UNIT ||
stmt_type == stmt::T_CREATE_TENANT || stmt_type == stmt::T_DROP_TENANT || stmt_type == stmt::T_MODIFY_TENANT ||
stmt_type == stmt::T_LOCK_TENANT
// database
|| stmt_type == stmt::T_ALTER_DATABASE ||
stmt_type == stmt::T_DROP_DATABASE
// tablegroup
|| stmt_type == stmt::T_ALTER_TABLEGROUP ||
stmt_type == stmt::T_DROP_TABLEGROUP
// table
|| stmt_type == stmt::T_DROP_TABLE ||
stmt_type == stmt::T_ALTER_TABLE
// view
|| stmt_type == stmt::T_DROP_VIEW
// index
|| stmt_type == stmt::T_DROP_INDEX
// flashback
|| stmt_type == stmt::T_FLASHBACK_TENANT
|| stmt_type == stmt::T_FLASHBACK_DATABASE
|| stmt_type == stmt::T_FLASHBACK_TABLE_FROM_RECYCLEBIN
|| stmt_type == stmt::T_FLASHBACK_INDEX
// purge
|| stmt_type == stmt::T_PURGE_RECYCLEBIN || stmt_type == stmt::T_PURGE_TENANT ||
stmt_type == stmt::T_PURGE_DATABASE || stmt_type == stmt::T_PURGE_TABLE ||
stmt_type == stmt::T_PURGE_INDEX
// outline
|| stmt_type == stmt::T_DROP_OUTLINE
// sequence
|| stmt_type == stmt::T_DROP_SEQUENCE
// synonym
|| stmt_type == stmt::T_DROP_SYNONYM
// variable
|| (stmt_type == stmt::T_VARIABLE_SET && has_global_variable)
// stored procedure
|| stmt_type == stmt::T_DROP_ROUTINE
// package
|| stmt_type == stmt::T_DROP_PACKAGE
// user define type
|| stmt_type == stmt::T_DROP_TYPE
// udf
|| stmt_type == stmt::T_DROP_FUNC);
return (// tenant resource
stmt_type == stmt::T_CREATE_RESOURCE_POOL
|| stmt_type == stmt::T_DROP_RESOURCE_POOL
|| stmt_type == stmt::T_ALTER_RESOURCE_POOL
|| stmt_type == stmt::T_SPLIT_RESOURCE_POOL
|| stmt_type == stmt::T_MERGE_RESOURCE_POOL
|| stmt_type == stmt::T_CREATE_RESOURCE_UNIT
|| stmt_type == stmt::T_ALTER_RESOURCE_UNIT
|| stmt_type == stmt::T_DROP_RESOURCE_UNIT
|| stmt_type == stmt::T_CREATE_TENANT
|| stmt_type == stmt::T_DROP_TENANT
|| stmt_type == stmt::T_MODIFY_TENANT
|| stmt_type == stmt::T_LOCK_TENANT
// database
|| stmt_type == stmt::T_ALTER_DATABASE
|| stmt_type == stmt::T_DROP_DATABASE
// tablegroup
|| stmt_type == stmt::T_ALTER_TABLEGROUP
|| stmt_type == stmt::T_DROP_TABLEGROUP
// table
|| stmt_type == stmt::T_DROP_TABLE
|| stmt_type == stmt::T_ALTER_TABLE
// view
|| stmt_type == stmt::T_DROP_VIEW
// index
|| stmt_type == stmt::T_DROP_INDEX
// purge
|| stmt_type == stmt::T_PURGE_RECYCLEBIN
|| stmt_type == stmt::T_PURGE_TENANT
|| stmt_type == stmt::T_PURGE_DATABASE
|| stmt_type == stmt::T_PURGE_TABLE
|| stmt_type == stmt::T_PURGE_INDEX
// outline
|| stmt_type == stmt::T_DROP_OUTLINE
// sequence
|| stmt_type == stmt::T_DROP_SEQUENCE
//synonym
|| stmt_type == stmt::T_DROP_SYNONYM
// variable
//目前只有set global variable才是DDL操作，session级别的variable变更不是DDL
|| (stmt_type == stmt::T_VARIABLE_SET && has_global_variable)
// stored procedure
|| stmt_type == stmt::T_DROP_ROUTINE
// package
|| stmt_type == stmt::T_DROP_PACKAGE
// user define type
|| stmt_type == stmt::T_DROP_TYPE
//tablespace
|| stmt_type == stmt::T_DROP_TABLESPACE
//udf
|| stmt_type == stmt::T_DROP_FUNC
//trigger
|| stmt_type == stmt::T_DROP_TRIGGER
);
}
static inline bool is_ddl_stmt_allowed_in_creating_tenant(stmt::StmtType stmt_type, bool has_global_variable)
{
return (
// tenant resource
stmt_type == stmt::T_CREATE_RESOURCE_POOL || stmt_type == stmt::T_DROP_RESOURCE_POOL ||
stmt_type == stmt::T_ALTER_RESOURCE_POOL || stmt_type == stmt::T_SPLIT_RESOURCE_POOL ||
stmt_type == stmt::T_MERGE_RESOURCE_POOL || stmt_type == stmt::T_CREATE_RESOURCE_UNIT ||
stmt_type == stmt::T_ALTER_RESOURCE_UNIT || stmt_type == stmt::T_DROP_RESOURCE_UNIT ||
stmt_type == stmt::T_CREATE_TENANT || stmt_type == stmt::T_DROP_TENANT || stmt_type == stmt::T_MODIFY_TENANT ||
stmt_type == stmt::T_LOCK_TENANT
// variable
|| (stmt_type == stmt::T_VARIABLE_SET && !has_global_variable));
stmt_type == stmt::T_CREATE_RESOURCE_POOL
|| stmt_type == stmt::T_DROP_RESOURCE_POOL
|| stmt_type == stmt::T_ALTER_RESOURCE_POOL
|| stmt_type == stmt::T_SPLIT_RESOURCE_POOL
|| stmt_type == stmt::T_MERGE_RESOURCE_POOL
|| stmt_type == stmt::T_CREATE_RESOURCE_UNIT
|| stmt_type == stmt::T_ALTER_RESOURCE_UNIT
|| stmt_type == stmt::T_DROP_RESOURCE_UNIT
|| stmt_type == stmt::T_CREATE_TENANT
|| stmt_type == stmt::T_DROP_TENANT
|| stmt_type == stmt::T_MODIFY_TENANT
|| stmt_type == stmt::T_LOCK_TENANT
// variable
//目前只有set global variable才是DDL操作，session级别的variable变更不是DDL
|| (stmt_type == stmt::T_VARIABLE_SET && !has_global_variable));
}
static bool is_dcl_stmt(stmt::StmtType stmt_type)
{
return (stmt_type >= stmt::T_CREATE_USER && stmt_type <= stmt::T_REVOKE)
// user profile
|| stmt_type == stmt::T_USER_PROFILE || stmt_type == stmt::T_ALTER_USER_PROFILE ||
stmt_type == stmt::T_ALTER_USER_PRIMARY_ZONE ||
stmt_type == stmt::T_ALTER_USER
//
|| stmt_type == stmt::T_CREATE_ROLE || stmt_type == stmt::T_DROP_ROLE || stmt_type == stmt::T_SET_ROLE ||
stmt_type == stmt::T_ALTER_ROLE || stmt_type == stmt::T_GRANT_ROLE ||
stmt_type == stmt::T_REVOKE_ROLE
//
|| stmt_type == stmt::T_SYSTEM_GRANT || stmt_type == stmt::T_SYSTEM_REVOKE;
{ return (stmt_type >= stmt::T_CREATE_USER && stmt_type <= stmt::T_REVOKE)
// user profile
|| stmt_type == stmt::T_CREATE_PROFILE
|| stmt_type == stmt::T_ALTER_PROFILE
|| stmt_type == stmt::T_DROP_PROFILE
|| stmt_type == stmt::T_ALTER_USER_PROFILE
|| stmt_type == stmt::T_ALTER_USER_PRIMARY_ZONE
|| stmt_type == stmt::T_ALTER_USER
//
|| stmt_type == stmt::T_CREATE_ROLE
|| stmt_type == stmt::T_DROP_ROLE
|| stmt_type == stmt::T_SET_ROLE
|| stmt_type == stmt::T_ALTER_ROLE
|| stmt_type == stmt::T_GRANT_ROLE
|| stmt_type == stmt::T_REVOKE_ROLE
//
|| stmt_type == stmt::T_SYSTEM_GRANT
|| stmt_type == stmt::T_SYSTEM_REVOKE;
}
static bool check_change_tenant_stmt(stmt::StmtType stmt_type)
{
return is_dml_stmt(stmt_type) || is_tcl_stmt(stmt_type) || stmt_type == stmt::T_VARIABLE_SET ||
stmt_type == stmt::T_USE_DATABASE || stmt_type == stmt::T_EMPTY_QUERY || stmt_type == stmt::T_CHANGE_TENANT;
return is_dml_stmt(stmt_type)
|| is_tcl_stmt(stmt_type)
|| stmt_type == stmt::T_HELP
|| stmt_type == stmt::T_VARIABLE_SET
|| stmt_type == stmt::T_USE_DATABASE
|| stmt_type == stmt::T_EMPTY_QUERY
// TODO: When T_LOCK_TABLE is actually implemented, needs to be checked for legitimacy
|| stmt_type == stmt::T_LOCK_TABLE
|| stmt_type == stmt::T_CHANGE_TENANT;
}
virtual int64_t to_string(char* buf, const int64_t buf_len) const
// following stmt don't do retry
static bool force_skip_retry_stmt(stmt::StmtType stmt_type)
{
return stmt_type == stmt::T_CHANGE_TENANT;
}
virtual int64_t to_string(char *buf, const int64_t buf_len) const
{
int64_t pos = 0;
J_OBJ_START();
@ -431,114 +507,42 @@ public:
J_OBJ_END();
return pos;
}
int assign(const ObStmt& other);
int deep_copy(const ObStmt& other);
int assign(const ObStmt &other);
int deep_copy(const ObStmt &other);
bool get_fetch_cur_time() const;
int64_t get_pre_param_size() const;
void increase_question_marks_count();
int64_t get_question_marks_count() const;
void set_query_ctx(ObQueryCtx* query_ctx)
{
query_ctx_ = query_ctx;
}
ObQueryCtx* get_query_ctx()
{
return query_ctx_;
}
const ObQueryCtx* get_query_ctx() const
{
return query_ctx_;
}
int distribute_hint_in_query_ctx(common::ObIAllocator* allocator);
int add_calculable_item(const ObHiddenColumnItem& calcuable_item);
const common::ObIArray<ObHiddenColumnItem>& get_calculable_exprs() const;
common::ObIArray<ObHiddenColumnItem>& get_calculable_exprs();
int check_table_id_exists(uint64_t table_id, bool& is_exist);
int check_synonym_id_exist(uint64_t synonym_id, bool& is_exist);
common::ObIArray<ObRawExpr*>& get_exec_param_ref_exprs() const;
common::ObIArray<ObRawExpr*>& get_exec_param_ref_exprs();
inline common::ObString& get_sql_stmt()
{
return sql_stmt_;
}
inline const common::ObString& get_sql_stmt() const
{
return sql_stmt_;
}
inline void set_sql_stmt(const char* sql, int32_t sql_len)
{
sql_stmt_.assign_ptr(sql, sql_len);
}
inline void set_sql_stmt(const common::ObString sql_stmt)
{
sql_stmt_ = sql_stmt;
}
void set_prepare_param_count(const int64_t prepare_param_count)
{
prepare_param_count_ = prepare_param_count;
}
int64_t get_prepare_param_count() const
{
return prepare_param_count_;
}
bool is_prepare_stmt() const
{
return is_prepare_stmt_;
}
void set_is_prepare_stmt(bool is_prepare)
{
is_prepare_stmt_ = is_prepare;
}
bool has_nested_sql() const
{
return has_nested_sql_;
}
void set_has_nested_sql(bool has_nested_sql)
{
has_nested_sql_ = has_nested_sql;
}
inline const ObSynonymIds& get_synonym_id_store() const
{
return synonym_id_store_;
}
int add_synonym_ids(const common::ObIArray<uint64_t>& synonym_ids, bool error_with_repeate);
int add_global_dependency_table(const share::schema::ObSchemaObjVersion& dependency_table);
const common::ObIArray<share::schema::ObSchemaObjVersion>* get_global_dependency_table() const;
common::ObIArray<share::schema::ObSchemaObjVersion>* get_global_dependency_table();
int add_table_stat_version(const common::ObOptTableStatVersion& table_stat_version);
const common::ObIArray<common::ObOptTableStatVersion>* get_table_stat_versions() const;
void set_sql_stmt_coll_type(common::ObCollationType coll_type)
{
sql_stmt_coll_type_ = coll_type;
}
common::ObCollationType get_sql_stmt_coll_type()
{
return sql_stmt_coll_type_;
}
void set_query_ctx(ObQueryCtx *query_ctx) { query_ctx_ = query_ctx; }
ObQueryCtx *get_query_ctx() const { return query_ctx_; }
int add_calculable_item(const ObHiddenColumnItem &calcuable_item);
const common::ObIArray<ObHiddenColumnItem> &get_calculable_exprs() const;
common::ObIArray<ObHiddenColumnItem> &get_calculable_exprs();
int check_synonym_id_exist(uint64_t synonym_id, bool &is_exist);
int add_global_dependency_table(const share::schema::ObSchemaObjVersion &dependency_table);
const common::ObIArray<share::schema::ObSchemaObjVersion> *get_global_dependency_table() const;
common::ObIArray<share::schema::ObSchemaObjVersion> *get_global_dependency_table();
int add_ref_obj_version(const uint64_t dep_obj_id,
const uint64_t dep_db_id,
const share::schema::ObObjectType dep_obj_type,
const share::schema::ObSchemaObjVersion &ref_obj_version,
common::ObIAllocator &allocator);
const share::schema::ObReferenceObjTable *get_ref_obj_table() const;
share::schema::ObReferenceObjTable *get_ref_obj_table();
protected:
void print_indentation(FILE* fp, int32_t level) const;
void print_indentation(FILE *fp, int32_t level) const;
public:
static const int64_t MAX_PRINTABLE_SIZE = 2 * 1024 * 1024;
static const int64_t MAX_PRINTABLE_SIZE = 2*1024*1024;
private:
DISALLOW_COPY_AND_ASSIGN(ObStmt);
// protected:
//protected:
public:
stmt::StmtType stmt_type_;
stmt::StmtType literal_stmt_type_;
common::ObString sql_stmt_;
common::ObCollationType sql_stmt_coll_type_;
ObQueryCtx* query_ctx_;
// 实际stmt类型，即：resolver改写后的类型
stmt::StmtType stmt_type_;
// 字面stmt类型，例如show语句的字面类型为show，而stmt_type_为SELECT
ObQueryCtx *query_ctx_;
int64_t stmt_id_;
int64_t prepare_param_count_;
bool is_prepare_stmt_;
bool has_nested_sql_;
// used for plan cache check synonym version
ObSynonymIds synonym_id_store_;
const common::ObTimeZoneInfo* tz_info_;
};
inline void ObStmt::set_stmt_type(stmt::StmtType stmt_type)
@ -551,7 +555,7 @@ inline stmt::StmtType ObStmt::get_stmt_type() const
return stmt_type_;
}
inline void ObStmt::print_indentation(FILE* fp, int32_t level) const
inline void ObStmt::print_indentation(FILE *fp, int32_t level) const
{
for (int i = 0; i < level; ++i) {
fprintf(fp, " ");
@ -578,28 +582,30 @@ inline bool ObStmt::is_px_dml_supported_stmt() const
return is_px_dml_supported_stmt(stmt_type_);
}
class ObStmtFactory {
class ObStmtFactory
{
public:
explicit ObStmtFactory(common::ObIAllocator& alloc)
: allocator_(alloc), stmt_store_(alloc), free_list_(alloc), query_ctx_(NULL)
{}
~ObStmtFactory()
explicit ObStmtFactory(common::ObIAllocator &alloc)
: allocator_(alloc),
stmt_store_(alloc),
free_list_(alloc),
query_ctx_(NULL)
{
destory();
}
~ObStmtFactory() { destory(); }
template <typename StmtType>
inline int create_stmt(StmtType*& stmt)
inline int create_stmt(StmtType *&stmt)
{
int ret = common::OB_SUCCESS;
void* ptr = allocator_.alloc(sizeof(StmtType));
void *ptr = allocator_.alloc(sizeof(StmtType));
stmt = NULL;
if (OB_UNLIKELY(NULL == ptr)) {
ret = common::OB_ALLOCATE_MEMORY_FAILED;
SQL_RESV_LOG(ERROR, "no more memory to stmt");
} else {
stmt = new (ptr) StmtType();
stmt = new(ptr) StmtType();
if (OB_FAIL(stmt_store_.store_obj(stmt))) {
SQL_RESV_LOG(WARN, "store stmt failed", K(ret));
stmt->~StmtType();
@ -617,25 +623,17 @@ public:
* so, query_ctx is globally unique in the single query
* @return query_ctx_
*/
ObQueryCtx* get_query_ctx();
inline common::ObIAllocator& get_allocator()
{
return allocator_;
}
ObQueryCtx *get_query_ctx();
inline common::ObIAllocator &get_allocator() { return allocator_; }
private:
common::ObIAllocator& allocator_;
common::ObIAllocator &allocator_;
common::ObObjStore<ObStmt*, common::ObIAllocator&, true> stmt_store_;
common::ObObjStore<ObSelectStmt*, common::ObIAllocator&, true> free_list_;
ObQueryCtx* query_ctx_;
ObQueryCtx *query_ctx_;
private:
DISALLOW_COPY_AND_ASSIGN(ObStmtFactory);
};
}
}
template <>
int ObStmtFactory::create_stmt<ObSelectStmt>(ObSelectStmt*& stmt);
} // namespace sql
} // namespace oceanbase
#endif // OCEANBASE_SQL_OB_STMT_H_
#endif //OCEANBASE_SQL_OB_STMT_H_