100 lines
2.9 KiB
C++
100 lines
2.9 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.
|
|
*/
|
|
|
|
#include "sql/optimizer/ob_log_plan.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
|
|
#define BUF_LEN 102400 // 100K
|
|
|
|
using namespace oceanbase::sql;
|
|
using namespace oceanbase::common;
|
|
|
|
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)
|
|
{
|
|
|
|
// create a plan
|
|
ObSQLSessionInfo session_info
|
|
ASSERT_EQ(OB_SUCCESS, session_info.test_init(0, 0, 0, NULL));
|
|
ObOptimizerContext ctx(NULL, NULL);
|
|
ObSelectStmt *select_stmt = NULL;
|
|
ObSelectLogPlan plan(ctx, select_stmt);
|
|
|
|
// create the operator tree
|
|
ObLogTableScan *left =
|
|
reinterpret_cast<ObLogTableScan *>(log_op_factory_.allocate(LOG_TABLE_SCAN));
|
|
ObLogTableScan *right =
|
|
reinterpret_cast<ObLogTableScan *>(log_op_factory_.allocate(LOG_TABLE_SCAN));
|
|
ObLogJoin *join =
|
|
reinterpret_cast<ObLogJoin *>(log_op_factory_.allocate(LOG_JOIN));
|
|
ObLogGroupBy *group_by =
|
|
reinterpret_cast<ObLogGroupBy *>(log_op_factory_.allocate(LOG_GROUP_BY));
|
|
ObLogOrderBy *order_by =
|
|
reinterpret_cast<ObLogOrderBy *>(log_op_factory_.allocate(LOG_ORDER_BY));
|
|
ObLogLimit *limit =
|
|
reinterpret_cast<ObLogLimit *>(log_op_factory_.allocate(LOG_LIMIT));
|
|
|
|
ASSERT_EQ(OB_SUCCESS, plan.set_plan_root(limit));
|
|
ASSERT_EQ(OB_SUCCESS, limit.set_child(first_child, order_by));
|
|
ASSERT_EQ(OB_SUCCESS, order_by.set_parent(limit));
|
|
ASSERT_EQ(OB_SUCCESS, order_by->set_child(first_child, group_by));
|
|
ASSERT_EQ(OB_SUCCESS, group_by->set_parent(order_by));
|
|
ASSERT_EQ(OB_SUCCESS, group_by->set_child(first_child, join));
|
|
ASSERT_EQ(OB_SUCCESS, join->set_parent(group_by));
|
|
ASSERT_EQ(OB_SUCCESS, join->set_child(first_child, left));
|
|
ASSERT_EQ(OB_SUCCESS, left->set_parent(join));
|
|
ASSERT_EQ(OB_SUCCESS, join->set_child(second_child, right));
|
|
ASSERT_EQ(OB_SUCCESS, right->set_parent(join));
|
|
|
|
// print a plan
|
|
char buf[BUF_LEN];
|
|
ASSERT_EQ(OB_SUCCESS, plan.to_string(buf, BUF_LEN));
|
|
printf("%s\n", buf);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
::testing::InitGoogleTest(&argc,argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|