173 lines
6.0 KiB
C++
173 lines
6.0 KiB
C++
/**
|
|
* Copyright (c) 2021 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
|
|
#ifndef TEST_SQL_UTILS_H_
|
|
#define TEST_SQL_UTILS_H_
|
|
|
|
#undef private
|
|
#undef protected
|
|
#include <fstream>
|
|
#include <dirent.h>
|
|
#include <getopt.h>
|
|
#include <gtest/gtest.h>
|
|
#define private public
|
|
#define protected public
|
|
#include "lib/allocator/page_arena.h"
|
|
#include "lib/allocator/ob_fifo_allocator.h"
|
|
#include "lib/utility/ob_test_util.h"
|
|
#include "lib/json/ob_json_print_utils.h" // for SJ
|
|
#include "share/inner_table/ob_inner_table_schema.h"
|
|
#include "share/ob_rpc_struct.h"
|
|
#include "share/schema/ob_schema_getter_guard.h"
|
|
#include "sql/parser/ob_parser.h"
|
|
#include "sql/resolver/ob_resolver.h"
|
|
#include "sql/resolver/ob_schema_checker.h"
|
|
#include "sql/resolver/ddl/ob_create_table_stmt.h"
|
|
#include "sql/resolver/ddl/ob_drop_table_stmt.h"
|
|
#include "sql/resolver/ddl/ob_create_index_stmt.h"
|
|
#include "sql/resolver/ddl/ob_create_database_stmt.h"
|
|
#include "sql/resolver/ddl/ob_use_database_stmt.h"
|
|
#include "sql/session/ob_sql_session_info.h"
|
|
#include "sql/resolver/dcl/ob_create_user_stmt.h"
|
|
#include "sql/optimizer/ob_log_plan_factory.h"
|
|
#include "sql/executor/ob_task_executor_ctx.h"
|
|
#include "sql/ob_sql_context.h"
|
|
#include "sql/engine/ob_exec_context.h"
|
|
#include "sql/plan_cache/ob_plan_cache_manager.h"
|
|
#include "../share/schema/mock_schema_service.h"
|
|
#include "ob_mock_partition_location_cache.h"
|
|
|
|
using namespace oceanbase;
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::share;
|
|
using namespace oceanbase::share::schema;
|
|
using namespace oceanbase::obrpc;
|
|
using namespace oceanbase::sql;
|
|
|
|
#define CSJ(x) (static_cast<const char*>(SJ((x))))
|
|
namespace test {
|
|
typedef Ob2DArray<ObObjParam, OB_MALLOC_BIG_BLOCK_SIZE, ObWrapperAllocator, false> ParamStore;
|
|
static const int32_t FILE_PATH_LEN = 512;
|
|
static const int32_t MAX_FILE_NUM = 128;
|
|
struct CmdLineParam {
|
|
CmdLineParam()
|
|
{
|
|
file_count = 0;
|
|
test_input_from_cmd = false;
|
|
print_schema_detail_info = false;
|
|
record_test_result = false;
|
|
sql_mode = DEFAULT_MYSQL_MODE;
|
|
}
|
|
char file_names[MAX_FILE_NUM][FILE_PATH_LEN];
|
|
int32_t file_count;
|
|
bool test_input_from_cmd;
|
|
bool print_schema_detail_info;
|
|
bool record_test_result;
|
|
ObSQLMode sql_mode;
|
|
std::vector<const char*> file_names_vector;
|
|
};
|
|
extern CmdLineParam clp;
|
|
|
|
extern bool comparisonFunc(const char* c1, const char* c2);
|
|
extern void load_sql_file(const char* file_name);
|
|
extern void load_all_sql_files(const char* directory_name);
|
|
extern void print_help_msg(const char* exe_name);
|
|
extern void parse_cmd_line_param(int argc, char* argv[], CmdLineParam& clp);
|
|
|
|
inline bool casesame_cstr(const char* a, const char* b)
|
|
{
|
|
size_t len1 = strlen(a);
|
|
size_t len2 = strlen(b);
|
|
return (len1 == len2) && (strncasecmp(a, b, len1) == 0);
|
|
}
|
|
|
|
enum ParserResultFormat { TREE_FORMAT, JSON_FORMAT };
|
|
|
|
class TestSqlUtils {
|
|
public:
|
|
TestSqlUtils();
|
|
virtual ~TestSqlUtils()
|
|
{}
|
|
virtual void init();
|
|
virtual void destroy();
|
|
|
|
public:
|
|
static const int64_t MAX_SCHEMA_FILE_PATH = 128 - 1;
|
|
// function members
|
|
void load_schema_from_file(const char* file_path);
|
|
void do_load_sql(const char* query_str, ObStmt*& stmt, bool is_print, ParserResultFormat format = TREE_FORMAT,
|
|
int64_t expect_error = OB_SUCCESS, int64_t case_line = 0);
|
|
void do_resolve(const char* query, ObStmt*& stmt, bool is_print = false, ParserResultFormat format = JSON_FORMAT,
|
|
int64_t expect_error = OB_SUCCESS, bool parameterized = true, bool need_replace_param_expr = true,
|
|
int64_t case_line = 0);
|
|
int create_system_table();
|
|
void create_system_db();
|
|
void do_create_table(const char* query_str);
|
|
void do_create_table(ObStmt*& stmt);
|
|
void do_drop_table(ObStmt*& stmt);
|
|
void do_create_index(ObStmt*& stmt);
|
|
void do_create_database(ObStmt*& stmt);
|
|
void do_use_database(ObStmt*& stmt);
|
|
void do_create_user(ObStmt*& stmt);
|
|
void generate_index_schema(ObCreateIndexStmt& stmt);
|
|
void generate_index_column_schema(ObCreateIndexStmt& stmt, ObTableSchema& index_schema);
|
|
int get_hidden_column_value(ObResolverParams& resolver_ctx, ParamStore& params);
|
|
void is_equal_content(const char* tmp_file, const char* result_file);
|
|
uint64_t get_next_table_id(const uint64_t user_tenant_id);
|
|
int add_table_schema(ObTableSchema& table_schema);
|
|
int add_database_schema(ObDatabaseSchema& database_schema);
|
|
int drop_table_schema(const ObTableSchema& table_schema);
|
|
int parse_row_from_json(const ObString& json_str, ObString& table_name, ObIArray<ObSEArray<ObObj, 3> >& row_array);
|
|
int parse_json_array(json::Value& value, ObIArray<ObSEArray<ObObj, 3> >& row_array);
|
|
ObSqlSchemaGuard& get_sql_schema_guard()
|
|
{
|
|
return sql_schema_guard_;
|
|
}
|
|
ObSchemaGetterGuard& get_schema_guard()
|
|
{
|
|
return schema_guard_;
|
|
}
|
|
|
|
public:
|
|
// table id
|
|
hash::ObHashMap<uint64_t, uint64_t> next_user_table_id_map_;
|
|
// user_id
|
|
uint64_t sys_user_id_;
|
|
uint64_t next_user_id_;
|
|
// database_id
|
|
uint64_t sys_database_id_;
|
|
uint64_t next_user_database_id_;
|
|
// tenant_id
|
|
uint64_t sys_tenant_id_;
|
|
ObSEArray<ObObj, 16> sys_view_bigint_param_list_;
|
|
int64_t schema_version_;
|
|
char schema_file_path_[MAX_SCHEMA_FILE_PATH + 1];
|
|
ObSQLSessionInfo session_info_;
|
|
ObSqlSchemaGuard sql_schema_guard_;
|
|
schema::ObSchemaGetterGuard schema_guard_;
|
|
MockSchemaService* schema_service_;
|
|
ObArenaAllocator allocator_;
|
|
ObRawExprFactory expr_factory_;
|
|
ObStmtFactory stmt_factory_;
|
|
ObLogPlanFactory log_plan_factory_;
|
|
ObSqlCtx sql_ctx_;
|
|
ObExecContext exec_ctx_;
|
|
ParamStore param_list_;
|
|
ObPlanCacheManager plan_cache_mgr_;
|
|
::test::MockPartitionLocationCache part_cache_;
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(TestSqlUtils);
|
|
};
|
|
} // namespace test
|
|
#endif
|