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

107
unittest/sql/optimizer/test_explain_json_format.cpp
View File

@ -30,40 +30,44 @@
#include "sql/ob_sql.h"
#include "sql/resolver/ob_resolver.h"
#define BUF_LEN 102400 // 100K
#define BUF_LEN 102400 // 100K
using namespace oceanbase::sql;
using namespace oceanbase::common;
using namespace oceanbase::json;
namespace test {
class ObLogPlanTest : public ::testing::Test {
public:
ObLogPlanTest();
virtual ~ObLogPlanTest();
virtual void SetUp();
virtual void TearDown();
private:
// disallow copy
ObLogPlanTest(const ObLogPlanTest& other);
ObLogPlanTest& operator=(const ObLogPlanTest& other);
private:
// data members
namespace test
{
class ObLogPlanTest: public ::testing::Test
{
public:
ObLogPlanTest();
virtual ~ObLogPlanTest();
virtual void SetUp();
virtual void TearDown();
private:
// disallow copy
ObLogPlanTest(const ObLogPlanTest &other);
ObLogPlanTest& operator=(const ObLogPlanTest &other);
private:
// data members
};
ObLogPlanTest::ObLogPlanTest()
{}
{
}
ObLogPlanTest::~ObLogPlanTest()
{}
{
}
void ObLogPlanTest::SetUp()
{}
{
}
void ObLogPlanTest::TearDown()
{}
{
}
TEST_F(ObLogPlanTest, ob_explain_test)
{
@ -72,39 +76,32 @@ TEST_F(ObLogPlanTest, ob_explain_test)
ObSQLSessionInfo session_info;
ASSERT_EQ(OB_SUCCESS, session_info.test_init(0, 0, 0, NULL));
ObArenaAllocator mempool(ObModIds::OB_SQL_COMPILE, OB_MALLOC_NORMAL_BLOCK_SIZE);
ObExecContext exec_ctx;
ObExecContext exec_ctx(mempool);
ObAddr addr;
ObQueryHint query_hint;
ObGlobalHint global_hint;
ObRawExprFactory expr_factory(mempool);
ObLogPlanFactory log_plan_factory_(mempool);
ObOptimizerContext ctx(&session_info,
&exec_ctx,
NULL,
NULL,
NULL,
NULL,
static_cast<ObIAllocator&>(mempool),
NULL,
NULL,
addr,
NULL,
OB_MERGED_VERSION_INIT,
query_hint,
expr_factory,
NULL);
ObSelectStmt* select_stmt = NULL;
ObSelectLogPlan plan(ctx, select_stmt);
ObOptimizerContext ctx(&session_info, &exec_ctx, NULL, NULL,
static_cast<ObIAllocator &>(mempool),
NULL, addr, NULL, global_hint,
expr_factory, NULL, false);
ObSelectStmt *select_stmt = NULL;
ObSelectLogPlan plan(ctx, select_stmt);
// create the operator tree
ObLogTableScan* left =
reinterpret_cast<ObLogTableScan*>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_TABLE_SCAN));
ObLogTableScan* right =
reinterpret_cast<ObLogTableScan*>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_TABLE_SCAN));
ObLogJoin* join = reinterpret_cast<ObLogJoin*>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_JOIN));
ObLogGroupBy* group_by =
reinterpret_cast<ObLogGroupBy*>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_GROUP_BY));
ObLogSort* order_by = reinterpret_cast<ObLogSort*>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_SORT));
ObLogLimit* limit = reinterpret_cast<ObLogLimit*>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_LIMIT));
ObLogTableScan *left =
reinterpret_cast<ObLogTableScan *>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_TABLE_SCAN));
ObLogTableScan *right =
reinterpret_cast<ObLogTableScan *>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_TABLE_SCAN));
ObLogJoin *join =
reinterpret_cast<ObLogJoin *>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_JOIN));
ObLogGroupBy *group_by =
reinterpret_cast<ObLogGroupBy *>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_GROUP_BY));
ObLogSort *order_by =
reinterpret_cast<ObLogSort *>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_SORT));
ObLogLimit *limit =
reinterpret_cast<ObLogLimit *>(plan.log_op_factory_.allocate(plan, log_op_def::LOG_LIMIT));
plan.set_plan_root(reinterpret_cast<ObLogicalOperator*>(limit));
limit->set_child(ObLogicalOperator::first_child, order_by);
@ -120,29 +117,29 @@ TEST_F(ObLogPlanTest, ob_explain_test)
// print a plan
char buf[BUF_LEN];
plan.to_string(buf, BUF_LEN);
printf("%s\n", buf);
plan.to_string(buf, BUF_LEN);
printf("%s\n", buf);
// print plan as json
char output_buf[OB_MAX_LOG_BUFFER_SIZE];
int64_t pos = 0;
Value* val = NULL;
Value *val = NULL;
plan.get_plan_root()->to_json(output_buf, OB_MAX_LOG_BUFFER_SIZE, pos, val);
Tidy tidy(val);
LOG_DEBUG("succ to generate json object", K(pos));
output_buf[0] = '\n';
pos = tidy.to_string(output_buf + pos, OB_MAX_LOG_BUFFER_SIZE - pos);
if (pos < OB_MAX_LOG_BUFFER_SIZE - 2) {
if (pos < OB_MAX_LOG_BUFFER_SIZE -2) {
output_buf[pos + 1] = '\n';
_OB_LOG(INFO, "%.*s", static_cast<int32_t>(pos + 2), output_buf);
}
}
} // namespace test
int main(int argc, char** argv)
}
int main(int argc, char **argv)
{
// OB_LOGGER.set_log_level("DEBUG");
::testing::InitGoogleTest(&argc, argv);
::testing::InitGoogleTest(&argc,argv);
return RUN_ALL_TESTS();
}