647 lines
21 KiB
C++
647 lines
21 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 <gtest/gtest.h>
|
|
#include "common/object/ob_object.h"
|
|
#include "common/row/ob_row.h"
|
|
#include "sql/ob_sql_init.h"
|
|
#include "sql/engine/expr/ob_sql_expression.h"
|
|
#include "sql/engine/ob_phy_operator.h"
|
|
#include "sql/engine/ob_single_child_phy_operator.h"
|
|
#include "sql/engine/ob_no_children_phy_operator.h"
|
|
#include "sql/engine/ob_physical_plan.h"
|
|
#include "sql/engine/aggregate/ob_merge_groupby.h"
|
|
#include "sql/engine/ob_exec_context.h"
|
|
#include "sql/session/ob_sql_session_info.h"
|
|
#include "sql/engine/set/ob_merge_union.h"
|
|
using namespace oceanbase;
|
|
using namespace oceanbase::sql;
|
|
using namespace oceanbase::common;
|
|
using namespace ::testing;
|
|
using namespace oceanbase::share;
|
|
|
|
int64_t row_size = 3;
|
|
static ObArenaAllocator alloc_;
|
|
class ObSingleChildOperatorFake : public ObSingleChildPhyOperator {
|
|
friend class ObPhyOperatorTest;
|
|
|
|
protected:
|
|
class ObSingleOpCtx : public ObPhyOperator::ObPhyOperatorCtx {
|
|
public:
|
|
ObSingleOpCtx(ObExecContext& ctx) : ObPhyOperatorCtx(ctx)
|
|
{}
|
|
virtual void destroy()
|
|
{
|
|
return ObPhyOperatorCtx::destroy_base();
|
|
}
|
|
};
|
|
|
|
public:
|
|
ObSingleChildOperatorFake() : ObSingleChildPhyOperator(alloc_)
|
|
{}
|
|
ObPhyOperatorType get_type() const
|
|
{
|
|
return static_cast<ObPhyOperatorType>(0);
|
|
}
|
|
|
|
int inner_open(ObExecContext& ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
if (OB_SUCCESS != (ret = init_op_ctx(ctx))) {
|
|
_OB_LOG(WARN, "init operator context failed, ret=%d", ret);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int inner_close(ObExecContext& ctx) const
|
|
{
|
|
UNUSED(ctx);
|
|
int ret = OB_SUCCESS;
|
|
return ret;
|
|
}
|
|
|
|
protected:
|
|
virtual int inner_get_next_row(ObExecContext& ctx, const ObNewRow*& row) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
const ObNewRow* input_row = NULL;
|
|
ObPhyOperator* child_op = NULL;
|
|
ObSingleOpCtx* phy_op_ctx = NULL;
|
|
|
|
if (NULL == (child_op = get_child(0))) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
_OB_LOG(WARN, "get child operator failed");
|
|
} else if (OB_SUCCESS != (ret = child_op->get_next_row(ctx, input_row))) {
|
|
if (OB_ITER_END != ret) {
|
|
_OB_LOG(WARN, "get_next_row failed, ret=%d", ret);
|
|
}
|
|
} else if (NULL == (phy_op_ctx = GET_PHY_OPERATOR_CTX(ObSingleOpCtx, ctx, get_id()))) {
|
|
ret = OB_ERR_UNEXPECTED;
|
|
_OB_LOG(WARN, "get physical operator context failed, ret=%d", ret);
|
|
} else {
|
|
_OB_LOG(DEBUG, "inner_get_next_row, row=%s", to_cstring(*input_row));
|
|
ObNewRow& cur_row = phy_op_ctx->get_cur_row();
|
|
OB_ASSERT(input_row->count_ <= cur_row.count_);
|
|
for (int64_t i = 0; i < input_row->count_; ++i) {
|
|
cur_row.cells_[i] = input_row->cells_[i];
|
|
}
|
|
row = &cur_row;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
virtual int init_op_ctx(ObExecContext& ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObPhyOperatorCtx* op_ctx = NULL;
|
|
|
|
if (OB_SUCCESS != (ret = CREATE_PHY_OPERATOR_CTX(ObSingleOpCtx, ctx, get_id(), get_type(), op_ctx))) {
|
|
_OB_LOG(WARN, "create physical operator context failed, ret=%d", ret);
|
|
} else if (OB_SUCCESS != (ret = op_ctx->create_cur_row(get_column_count(), projector_, projector_size_))) {
|
|
_OB_LOG(WARN, "create current row failed, ret=%d", ret);
|
|
}
|
|
return ret;
|
|
}
|
|
};
|
|
class ObTableScanFake : public ObNoChildrenPhyOperator {
|
|
protected:
|
|
class ObTableScanFakeCtx : public ObPhyOperatorCtx {
|
|
public:
|
|
ObTableScanFakeCtx(ObExecContext& ctx) : ObPhyOperatorCtx(ctx)
|
|
{}
|
|
virtual void destroy()
|
|
{
|
|
return ObPhyOperatorCtx::destroy_base();
|
|
}
|
|
};
|
|
|
|
public:
|
|
ObTableScanFake() : ObNoChildrenPhyOperator(alloc_), row_store_(NULL), store_size_(0), cur_index_(0)
|
|
{}
|
|
ObPhyOperatorType get_type() const
|
|
{
|
|
return static_cast<ObPhyOperatorType>(0);
|
|
}
|
|
|
|
int inner_open(ObExecContext& ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
if (OB_SUCCESS != (ret = init_op_ctx(ctx))) {
|
|
_OB_LOG(WARN, "init operator context failed, ret=%d", ret);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int inner_close(ObExecContext& ctx) const
|
|
{
|
|
UNUSED(ctx);
|
|
return OB_SUCCESS;
|
|
}
|
|
|
|
void load_data_row(ObExecContext& ctx, int64_t store_size)
|
|
{
|
|
int64_t cell_val = -1;
|
|
void* ptr = NULL;
|
|
|
|
store_size_ = store_size;
|
|
ASSERT_GT(store_size_, 0);
|
|
ptr = ctx.get_allocator().alloc(store_size * sizeof(ObNewRow));
|
|
ASSERT_FALSE(NULL == ptr);
|
|
row_store_ = static_cast<ObNewRow*>(ptr);
|
|
for (int64_t i = 0; i < store_size; ++i) {
|
|
void* cells = ctx.get_allocator().alloc(row_size * sizeof(ObObj));
|
|
ASSERT_FALSE(NULL == cells);
|
|
row_store_[i].cells_ = static_cast<ObObj*>(cells);
|
|
row_store_[i].count_ = row_size;
|
|
for (int64_t j = 0; j < row_size; ++j) {
|
|
row_store_[i].cells_[j].set_int(++cell_val);
|
|
}
|
|
}
|
|
}
|
|
|
|
protected:
|
|
virtual int inner_get_next_row(ObExecContext& ctx, const ObNewRow*& row) const
|
|
{
|
|
UNUSED(ctx);
|
|
int ret = OB_SUCCESS;
|
|
if (cur_index_ < store_size_) {
|
|
row = &row_store_[cur_index_++];
|
|
} else {
|
|
ret = OB_ITER_END;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
virtual int init_op_ctx(ObExecContext& ctx) const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObPhyOperatorCtx* op_ctx = NULL;
|
|
ret = CREATE_PHY_OPERATOR_CTX(ObTableScanFakeCtx, ctx, get_id(), get_type(), op_ctx);
|
|
UNUSED(op_ctx);
|
|
return ret;
|
|
}
|
|
|
|
private:
|
|
ObNewRow* row_store_;
|
|
int64_t store_size_;
|
|
mutable int64_t cur_index_;
|
|
};
|
|
|
|
class ObPhyOperatorTest : public ::testing::Test {
|
|
public:
|
|
ObPhyOperatorTest();
|
|
virtual ~ObPhyOperatorTest();
|
|
virtual void SetUp();
|
|
virtual void TearDown();
|
|
|
|
int test_calculate_row(ObExecContext& ctx, ObSingleChildOperatorFake& phy_op)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObExprCtx expr_ctx;
|
|
ObSingleChildOperatorFake::ObSingleOpCtx single_op_ctx(ctx);
|
|
|
|
if (OB_SUCCESS != (ret = single_op_ctx.create_cur_row(2, NULL, 0))) {
|
|
_OB_LOG(WARN, "create current row failed, ret=%d", ret);
|
|
} else {
|
|
ret = phy_op.calculate_row(expr_ctx, single_op_ctx.get_cur_row());
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int test_wrap_expr_ctx(ObExecContext& exec_ctx, ObExprCtx& expr_ctx, ObSingleChildOperatorFake& phy_op)
|
|
{
|
|
return phy_op.wrap_expr_ctx(exec_ctx, expr_ctx);
|
|
}
|
|
|
|
private:
|
|
// disallow copy
|
|
ObPhyOperatorTest(const ObPhyOperatorTest& other);
|
|
ObPhyOperatorTest& operator=(const ObPhyOperatorTest& other);
|
|
|
|
private:
|
|
// data members
|
|
};
|
|
ObPhyOperatorTest::ObPhyOperatorTest()
|
|
{}
|
|
|
|
ObPhyOperatorTest::~ObPhyOperatorTest()
|
|
{}
|
|
|
|
void ObPhyOperatorTest::SetUp()
|
|
{}
|
|
|
|
void ObPhyOperatorTest::TearDown()
|
|
{}
|
|
|
|
/**
|
|
* @brief test operator only get row
|
|
*/
|
|
TEST_F(ObPhyOperatorTest, test_get_row)
|
|
{
|
|
const ObNewRow* row = NULL;
|
|
ObExecContext ctx;
|
|
ObSQLSessionInfo my_session;
|
|
ASSERT_EQ(OB_SUCCESS, my_session.test_init(0, 0, 0, NULL));
|
|
ctx.set_my_session(&my_session);
|
|
ObTableScanFake table_scan;
|
|
ObSingleChildOperatorFake root;
|
|
ObPhysicalPlan physical_plan;
|
|
int64_t op_size = 2;
|
|
|
|
ASSERT_EQ(OB_SUCCESS, ctx.init_phy_op(op_size));
|
|
ASSERT_EQ(OB_SUCCESS, ctx.create_physical_plan_ctx());
|
|
|
|
table_scan.load_data_row(ctx, 1);
|
|
table_scan.set_id(0);
|
|
table_scan.set_column_count(row_size);
|
|
table_scan.set_phy_plan(&physical_plan);
|
|
|
|
root.set_id(1);
|
|
root.set_column_count(row_size);
|
|
root.set_phy_plan(&physical_plan);
|
|
ASSERT_EQ(OB_SUCCESS, root.set_child(0, table_scan));
|
|
ASSERT_EQ(OB_SUCCESS, root.open(ctx));
|
|
ASSERT_EQ(OB_SUCCESS, root.get_next_row(ctx, row));
|
|
for (int64_t i = 0; i < row_size; ++i) {
|
|
ObObj res;
|
|
res.set_int(i);
|
|
ASSERT_TRUE(res == row->cells_[i]);
|
|
}
|
|
printf("row=%s\n", to_cstring(*row));
|
|
ASSERT_EQ(OB_SUCCESS, root.close(ctx));
|
|
root.reset();
|
|
root.reuse();
|
|
table_scan.reset();
|
|
table_scan.reuse();
|
|
}
|
|
|
|
TEST_F(ObPhyOperatorTest, test_filter_and_calc_row)
|
|
{
|
|
const ObNewRow* row = NULL;
|
|
ObExecContext ctx;
|
|
ObTableScanFake table_scan;
|
|
ObSingleChildOperatorFake root;
|
|
ObPhysicalPlan physical_plan;
|
|
int64_t op_size = 2;
|
|
ObSQLSessionInfo my_session;
|
|
my_session.test_init(0, 0, 0, NULL);
|
|
ctx.set_my_session(&my_session);
|
|
|
|
physical_plan.set_main_query(&root);
|
|
ASSERT_EQ(OB_SUCCESS, ctx.init_phy_op(op_size));
|
|
ASSERT_EQ(OB_SUCCESS, ctx.create_physical_plan_ctx());
|
|
|
|
table_scan.load_data_row(ctx, 1);
|
|
table_scan.set_id(0);
|
|
table_scan.set_column_count(row_size);
|
|
table_scan.set_phy_plan(&physical_plan);
|
|
|
|
root.set_id(1);
|
|
root.set_column_count(row_size);
|
|
root.set_phy_plan(&physical_plan);
|
|
ASSERT_EQ(OB_SUCCESS, root.set_child(0, table_scan));
|
|
ASSERT_EQ(OB_SUCCESS, root.open(ctx));
|
|
ASSERT_EQ(OB_SUCCESS, root.get_next_row(ctx, row));
|
|
for (int64_t i = 0; i < row_size; ++i) {
|
|
ObObj res;
|
|
res.set_int(i);
|
|
ASSERT_TRUE(res == row->cells_[i]);
|
|
}
|
|
printf("row=%s\n", to_cstring(*row));
|
|
ASSERT_EQ(OB_SUCCESS, root.close(ctx));
|
|
root.reset();
|
|
root.reuse();
|
|
table_scan.reset();
|
|
table_scan.reuse();
|
|
}
|
|
|
|
TEST_F(ObPhyOperatorTest, test_filter_and_calc_row_1)
|
|
{
|
|
const ObNewRow* row = NULL;
|
|
ObExecContext ctx;
|
|
ObTableScanFake table_scan;
|
|
ObSingleChildOperatorFake root;
|
|
ObPhysicalPlan physical_plan;
|
|
ObSQLSessionInfo my_session;
|
|
int64_t op_size = 2;
|
|
|
|
my_session.test_init(0, 0, 0, NULL);
|
|
ctx.set_my_session(&my_session);
|
|
|
|
physical_plan.set_main_query(&root);
|
|
ASSERT_EQ(OB_SUCCESS, ctx.init_phy_op(op_size));
|
|
ASSERT_EQ(OB_SUCCESS, ctx.create_physical_plan_ctx());
|
|
|
|
table_scan.set_id(0);
|
|
table_scan.load_data_row(ctx, 2);
|
|
table_scan.set_column_count(row_size);
|
|
table_scan.set_phy_plan(&physical_plan);
|
|
|
|
ObExprOperator* expr_op = NULL;
|
|
ObExprResType res_type;
|
|
res_type.set_calc_type(ObIntType);
|
|
res_type.set_type(ObIntType);
|
|
/*
|
|
* filter c3 - 1 = 1
|
|
*/
|
|
ObSqlExpression* filer_expr = NULL;
|
|
ASSERT_EQ(OB_SUCCESS, ObSqlExpressionUtil::make_sql_expr(&physical_plan, filer_expr));
|
|
ASSERT_FALSE(NULL == filer_expr);
|
|
filer_expr->set_item_count(5);
|
|
ObPostExprItem expr_item;
|
|
|
|
expr_item.set_int(1);
|
|
expr_item.set_item_type(T_INT);
|
|
ASSERT_EQ(OB_SUCCESS, filer_expr->add_expr_item(expr_item));
|
|
expr_item.set_column(2);
|
|
ASSERT_EQ(OB_SUCCESS, filer_expr->add_expr_item(expr_item));
|
|
expr_item.set_int(1);
|
|
expr_item.set_item_type(T_INT);
|
|
ASSERT_EQ(OB_SUCCESS, filer_expr->add_expr_item(expr_item));
|
|
expr_item.set_op(physical_plan.get_allocator(), "-", 2);
|
|
expr_op = expr_item.get_expr_operator();
|
|
expr_op->set_result_type(res_type);
|
|
ASSERT_EQ(OB_SUCCESS, filer_expr->add_expr_item(expr_item));
|
|
expr_item.set_op(physical_plan.get_allocator(), "=", 2);
|
|
ASSERT_EQ(OB_SUCCESS, filer_expr->add_expr_item(expr_item));
|
|
expr_op = expr_item.get_expr_operator();
|
|
expr_op->set_result_type(res_type);
|
|
/*
|
|
* calculate c1 + 1
|
|
*/
|
|
ObColumnExpression* calc_expr = NULL;
|
|
ASSERT_EQ(OB_SUCCESS, ObSqlExpressionUtil::make_sql_expr(&physical_plan, calc_expr));
|
|
ASSERT_FALSE(NULL == calc_expr);
|
|
calc_expr->set_item_count(3);
|
|
calc_expr->set_result_index(row_size);
|
|
ObPostExprItem calc_item;
|
|
calc_item.set_int(1);
|
|
calc_item.set_item_type(T_INT);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
// column c1
|
|
calc_item.set_column(0);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
calc_item.set_op(physical_plan.get_allocator(), "+", 2);
|
|
expr_op = calc_item.get_expr_operator();
|
|
expr_op->set_result_type(res_type);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
|
|
root.set_id(1);
|
|
root.set_column_count(row_size + 1);
|
|
root.set_phy_plan(&physical_plan);
|
|
ASSERT_EQ(OB_SUCCESS, root.add_filter(filer_expr));
|
|
ASSERT_EQ(OB_SUCCESS, root.add_compute(calc_expr));
|
|
ASSERT_EQ(OB_SUCCESS, root.set_child(0, table_scan));
|
|
ASSERT_EQ(OB_SUCCESS, root.open(ctx));
|
|
ASSERT_EQ(OB_SUCCESS, root.get_next_row(ctx, row));
|
|
|
|
ObObj cmp;
|
|
cmp.set_int(1);
|
|
ASSERT_TRUE(cmp == row->cells_[row_size]);
|
|
for (int64_t i = 1; i < row_size; ++i) {
|
|
ObObj res;
|
|
res.set_int(i);
|
|
ASSERT_TRUE(res == row->cells_[i]);
|
|
}
|
|
printf("row=%s\n", to_cstring(*row));
|
|
ASSERT_EQ(OB_ITER_END, root.get_next_row(ctx, row));
|
|
ASSERT_EQ(OB_SUCCESS, root.close(ctx));
|
|
ASSERT_FALSE(NULL == root.get_phy_plan());
|
|
root.reset();
|
|
root.reuse();
|
|
table_scan.reset();
|
|
table_scan.reuse();
|
|
}
|
|
|
|
TEST_F(ObPhyOperatorTest, test_get_op_name)
|
|
{
|
|
ObArenaAllocator alloc;
|
|
ObMergeUnion op(alloc);
|
|
op.set_type(PHY_HASH_UNION);
|
|
const char* name = "PHY_HASH_UNION";
|
|
ASSERT_EQ(0, strcmp(name, op.get_name()));
|
|
int64_t size = strlen(name);
|
|
ASSERT_EQ(size, strlen(op.get_name()));
|
|
}
|
|
|
|
TEST_F(ObPhyOperatorTest, test_invalid_argument)
|
|
{
|
|
ObExecContext ctx;
|
|
ObExprCtx expr_ctx;
|
|
ObSingleChildOperatorFake root;
|
|
ObPhysicalPlan physical_plan;
|
|
ObMergeGroupBy* groupby = NULL;
|
|
ObSQLSessionInfo my_session;
|
|
ASSERT_EQ(OB_SUCCESS, my_session.test_init(0, 0, 0, NULL));
|
|
ctx.set_my_session(&my_session);
|
|
|
|
ASSERT_EQ(OB_SUCCESS, physical_plan.alloc_operator_by_type(PHY_MERGE_GROUP_BY, groupby));
|
|
|
|
// invalid argument calculate_row() function
|
|
ObColumnExpression* calc_expr = NULL;
|
|
ASSERT_EQ(OB_SUCCESS, ObSqlExpressionUtil::make_sql_expr(&physical_plan, calc_expr));
|
|
ASSERT_FALSE(NULL == calc_expr);
|
|
calc_expr->set_item_count(3);
|
|
calc_expr->set_result_index(3);
|
|
ObPostExprItem calc_item;
|
|
calc_item.set_int(1);
|
|
calc_item.set_item_type(T_INT);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
// column c1
|
|
calc_item.set_column(0);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
calc_item.set_op(physical_plan.get_allocator(), "+", 2);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
root.set_phy_plan(&physical_plan);
|
|
ASSERT_EQ(OB_SUCCESS, root.add_compute(calc_expr));
|
|
ASSERT_EQ(OB_SUCCESS, ctx.init_phy_op(1));
|
|
ASSERT_EQ(OB_INVALID_ARGUMENT, ObPhyOperatorTest::test_calculate_row(ctx, root));
|
|
|
|
// invalid argument wrap_expr_ctx()
|
|
ObExecContext invalid_ctx;
|
|
invalid_ctx.set_my_session(&my_session);
|
|
root.reset();
|
|
ASSERT_EQ(OB_ERR_UNEXPECTED, ObPhyOperatorTest::test_wrap_expr_ctx(invalid_ctx, expr_ctx, root));
|
|
ASSERT_EQ(OB_SUCCESS, invalid_ctx.init_phy_op(1));
|
|
ASSERT_EQ(OB_SUCCESS, invalid_ctx.create_physical_plan_ctx());
|
|
ASSERT_EQ(OB_ERR_UNEXPECTED, ObPhyOperatorTest::test_wrap_expr_ctx(invalid_ctx, expr_ctx, root));
|
|
|
|
// invalid argument for add filter
|
|
root.reset();
|
|
ASSERT_EQ(OB_INVALID_ARGUMENT, root.add_filter(NULL));
|
|
ASSERT_EQ(OB_INVALID_ARGUMENT, root.add_compute(NULL));
|
|
}
|
|
|
|
TEST_F(ObPhyOperatorTest, test_serialize_and_deserialize)
|
|
{
|
|
/*
|
|
int64_t pos = 0;
|
|
const int64_t MAX_SERIALIZE_BUF_LEN = 1024;
|
|
char buf[MAX_SERIALIZE_BUF_LEN] = {'\0'};
|
|
ObTableScanFake table_scan1;
|
|
ObTableScanFake table_scan2;
|
|
ObPhysicalPlan physical_plan;
|
|
ObSqlExpression *filter_expr = NULL;
|
|
ASSERT_EQ(OB_SUCCESS, ObSqlExpressionUtil::make_sql_expr(&physical_plan, filter_expr));
|
|
ASSERT_FALSE(NULL == filter_expr);
|
|
filter_expr->set_item_count(5);
|
|
ObPostExprItem expr_item;
|
|
expr_item.set_int(1);
|
|
expr_item.set_item_type(T_INT);
|
|
ASSERT_EQ(OB_SUCCESS, filter_expr->add_expr_item(expr_item));
|
|
expr_item.set_column(2);
|
|
ASSERT_EQ(OB_SUCCESS, filter_expr->add_expr_item(expr_item));
|
|
expr_item.set_int(1);
|
|
expr_item.set_item_type(T_INT);
|
|
ASSERT_EQ(OB_SUCCESS, filter_expr->add_expr_item(expr_item));
|
|
expr_item.set_op(physical_plan.get_allocator(), "-", 2);
|
|
ASSERT_EQ(OB_SUCCESS, filter_expr->add_expr_item(expr_item));
|
|
expr_item.set_op(physical_plan.get_allocator(), "=", 2);
|
|
ASSERT_EQ(OB_SUCCESS, filter_expr->add_expr_item(expr_item));
|
|
ASSERT_EQ(OB_SUCCESS, filter_expr->serialize(buf, MAX_SERIALIZE_BUF_LEN, pos));
|
|
ASSERT_EQ(pos, filter_expr->get_serialize_size());
|
|
ObColumnExpression *calc_expr = NULL;
|
|
ASSERT_EQ(OB_SUCCESS, ObSqlExpressionUtil::make_sql_expr(&physical_plan, calc_expr));
|
|
ASSERT_FALSE(NULL == calc_expr);
|
|
calc_expr->set_item_count(3);
|
|
calc_expr->set_result_index(row_size);
|
|
ObPostExprItem calc_item;
|
|
calc_item.set_int(1);
|
|
calc_item.set_item_type(T_INT);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
//column c1
|
|
calc_item.set_column(0);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
calc_item.set_op(physical_plan.get_allocator(), "+", 2);
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->add_expr_item(calc_item));
|
|
pos = 0;
|
|
ASSERT_EQ(OB_SUCCESS, calc_expr->serialize(buf, MAX_SERIALIZE_BUF_LEN, pos));
|
|
ASSERT_EQ(pos, calc_expr->get_serialize_size());
|
|
|
|
table_scan1.set_id(1);
|
|
table_scan1.set_column_count(row_size + 1);
|
|
table_scan1.set_phy_plan(&physical_plan);
|
|
ASSERT_EQ(OB_SUCCESS, table_scan1.add_filter(filter_expr));
|
|
ASSERT_EQ(OB_SUCCESS, table_scan1.add_compute(calc_expr));
|
|
|
|
pos = 0;
|
|
ASSERT_EQ(OB_SUCCESS, table_scan1.serialize(buf, MAX_SERIALIZE_BUF_LEN, pos));
|
|
ASSERT_EQ(pos, table_scan1.get_serialize_size());
|
|
int64_t data_len = pos;
|
|
pos = 0;
|
|
table_scan2.set_phy_plan(&physical_plan);
|
|
ASSERT_EQ(OB_SUCCESS, table_scan2.deserialize(buf, data_len, pos));
|
|
ASSERT_EQ(pos, data_len);
|
|
ASSERT_EQ(0, strcmp(to_cstring(table_scan1), to_cstring(table_scan2)));
|
|
printf("table_scan serialize: %s\ntable_scan deserialize: %s\n",
|
|
to_cstring(table_scan1), to_cstring(table_scan2));
|
|
*/
|
|
}
|
|
|
|
TEST_F(ObPhyOperatorTest, physical_plan_size)
|
|
{
|
|
typedef oceanbase::common::ObFixedArray<common::ObField, common::ObIAllocator> ParamsFieldArray;
|
|
typedef oceanbase::common::ObFixedArray<common::ObField, common::ObIAllocator> ColumnsFieldArray;
|
|
struct MockPhysicalPlan {
|
|
common::ObArenaAllocator inner_alloc_;
|
|
common::ObArenaAllocator& allocator_;
|
|
int64_t prepare_count_;
|
|
volatile int64_t ref_count_;
|
|
int64_t schema_version_;
|
|
uint64_t plan_id_;
|
|
int64_t hint_timeout_us_;
|
|
ObPhyOperator* main_query_;
|
|
int64_t param_count_;
|
|
int64_t merged_version_;
|
|
common::ObDList<ObSqlExpression> pre_calc_exprs_;
|
|
common::ObFixedArray<ObParamInfo, common::ObIAllocator> params_info_;
|
|
uint64_t signature_;
|
|
ColumnsFieldArray field_columns_;
|
|
ParamsFieldArray param_columns_;
|
|
common::ObFixedArray<share::AutoincParam, common::ObIAllocator> autoinc_params_; // auto-incrment param
|
|
share::schema::ObStmtNeedPrivs stmt_need_privs_;
|
|
common::ObFixedArray<ObVarInfo, common::ObIAllocator> vars_;
|
|
ObPhyOperatorFactory op_factory_;
|
|
ObSqlExpressionFactory sql_expression_factory_;
|
|
ObExprOperatorFactory expr_op_factory_;
|
|
stmt::StmtType stmt_type_;
|
|
stmt::StmtType literal_stmt_type_;
|
|
ObPhyPlanType plan_type_;
|
|
common::ObConsistencyLevel hint_consistency_;
|
|
uint32_t next_phy_operator_id_; // share val
|
|
// for regexp expression's compilation
|
|
int16_t regexp_op_count_;
|
|
bool is_sfu_;
|
|
bool fetch_cur_time_;
|
|
bool is_contain_virtual_table_;
|
|
bool is_require_sys_tenant_priv_;
|
|
// if the stmt contains user variable assignment
|
|
// such as @a:=123
|
|
// we may need to serialize the map to remote server
|
|
bool is_contains_assignment_;
|
|
bool affected_last_insert_id_;
|
|
bool is_affect_found_row_; // not need serialize
|
|
};
|
|
|
|
printf("output mock physical plan = %ld, ModulePageAllocator=%ld\n",
|
|
sizeof(MockPhysicalPlan),
|
|
sizeof(ModulePageAllocator));
|
|
ObPhysicalPlan plan;
|
|
printf("output plan size=%ld\n", sizeof(plan));
|
|
common::ObConsistencyLevel hint_consistency;
|
|
printf("output enum size = %ld\n", sizeof(hint_consistency));
|
|
common::ObDList<ObSqlExpression> sql_expression;
|
|
printf("output dlist size=%ld\n", sizeof(sql_expression));
|
|
// printf("output fixarray = %ld, IArray=%ld\n", sizeof(common::ObFixedArray), sizeof(common::ObIArray));
|
|
printf("output IArray=%ld\n", sizeof(common::ObIArray<int>));
|
|
printf("output fixarray = %ld\n", sizeof(common::ObFixedArray<int, common::ObArenaAllocator>));
|
|
printf("output physical plan size=%ld\n", sizeof(ObPhysicalPlan));
|
|
printf("output factory=%ld, sql_expression=%ld\n", sizeof(ObPhyOperatorFactory), sizeof(ObSqlExpression));
|
|
printf("output sql_factory=%ld, expr_op=%ld\n", sizeof(ObSqlExpressionFactory), sizeof(ObExprOperatorFactory));
|
|
common::ObArenaAllocator alloc;
|
|
common::ObArenaAllocator& a = alloc;
|
|
common::ObArenaAllocator* p = &alloc;
|
|
printf(
|
|
"output referrence =%ld, Obiallocator=%ld\n", sizeof(common::ObArenaAllocator&), sizeof(common::ObIAllocator&));
|
|
printf("output alloc =%ld, refference=%ld, point=%ld\n", sizeof(alloc), sizeof(a), sizeof(p));
|
|
printf("output pramInfo before=%ld\n", sizeof(ObParamInfo));
|
|
struct MockParamInfo {
|
|
common::ObObjType type_;
|
|
common::ObScale scale_;
|
|
common::ParamFlag flag_;
|
|
};
|
|
printf("output MockParamInfo after=%ld\n", sizeof(MockParamInfo));
|
|
printf("output ObPostItem size before=%ld, obj=%ld\n", sizeof(ObPostExprItem), sizeof(common::ObObj));
|
|
class MockPostExprItem {
|
|
common::ObObj v1_; // const
|
|
union {
|
|
int64_t query_id_;
|
|
int64_t cell_index_; // column reference, aka cell index in ObRow
|
|
ObExprOperator* op_; // expression operator
|
|
ObSetVar::SetScopeType sys_var_scope_;
|
|
} v2_;
|
|
common::ObAccuracy accuracy_; // for const, column, questionmark
|
|
ObItemType item_type_;
|
|
};
|
|
printf("output MockPostItem size=%ld\n", sizeof(MockPostExprItem));
|
|
printf("output ObSqlExpression size=%ld, ObPostfixExpression=%ld\n",
|
|
sizeof(ObSqlExpression),
|
|
sizeof(ObPostfixExpression));
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
init_sql_factories();
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|