hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 2 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

96
unittest/sql/engine/set/set_data_generator.h
View File

@ -17,9 +17,12 @@
#include "sql/engine/ob_phy_operator.h"
#include "sql/engine/ob_exec_context.h"
namespace oceanbase {
namespace sql {
class SetDataGenerator : public ObPhyOperator {
namespace oceanbase
{
namespace sql
{
class SetDataGenerator : public ObPhyOperator
{
public:
const static int64_t ROW_ID_CELL = 0;
const static int64_t IDX_CELL = 1;
@ -28,50 +31,32 @@ public:
const static int64_t INT_VARCHAR_BUF_SIZE = 10;
class Ctx : public ObPhyOperatorCtx {
class Ctx : public ObPhyOperatorCtx
{
public:
Ctx(ObExecContext& c) : ObPhyOperatorCtx(c)
{}
virtual void destroy()
{
return ObPhyOperatorCtx::destroy_base();
}
Ctx(ObExecContext &c) : ObPhyOperatorCtx(c) {}
virtual void destroy() { return ObPhyOperatorCtx::destroy_base(); }
};
explicit SetDataGenerator(common::ObIAllocator& alloc) : ObPhyOperator(alloc)
explicit SetDataGenerator(common::ObIAllocator &alloc) : ObPhyOperator(alloc)
{
type_ = PHY_TABLE_SCAN;
}
virtual int set_child(int32_t, ObPhyOperator&)
{
return common::OB_SUCCESS;
}
virtual ObPhyOperator* get_child(int32_t) const
{
return NULL;
}
virtual int32_t get_child_num() const
{
return 0;
}
virtual int inner_get_next_row(ObExecContext&, const common::ObNewRow*&) const
{
return common::OB_SUCCESS;
}
int close(ObExecContext&) const
{
return common::OB_SUCCESS;
}
virtual int set_child(int32_t, ObPhyOperator &) { return common::OB_SUCCESS; }
virtual ObPhyOperator *get_child(int32_t) const { return NULL; }
virtual int32_t get_child_num() const { return 0; }
virtual int inner_get_next_row(ObExecContext &, const common::ObNewRow *&) const { return common::OB_SUCCESS; }
int close(ObExecContext &) const { return common::OB_SUCCESS; }
~SetDataGenerator()
{
if (str_) {
delete[] str_;
delete []str_;
str_ = NULL;
}
if (projector_) {
delete[] projector_;
delete []projector_;
projector_ = NULL;
}
}
@ -80,16 +65,16 @@ public:
int test_init()
{
int ret = OB_SUCCESS;
if (NULL == projector_) {
projector_ = new int32_t[CELL_CNT];
projector_size_ = CELL_CNT;
for (int i = 0; i < CELL_CNT; i++) {
projector_[i] = i;
}
}
if (NULL == projector_) {
projector_ = new int32_t[CELL_CNT];
projector_size_ = CELL_CNT;
for (int i = 0; i < CELL_CNT; i++) {
projector_[i] = i;
}
}
set_column_count(CELL_CNT);
row_id_ = 0;
row_id_ = 0;
if (reverse_) {
row_id_ = row_cnt_ - 1;
}
@ -102,13 +87,13 @@ public:
return ret;
}
virtual int init_op_ctx(ObExecContext& ec) const override
virtual int init_op_ctx(ObExecContext &ec) const override
{
Ctx* ctx = NULL;
Ctx *ctx = NULL;
return CREATE_PHY_OPERATOR_CTX(Ctx, ec, get_id(), get_type(), ctx);
}
virtual int inner_open(ObExecContext& ec) const override
virtual int inner_open(ObExecContext &ec) const override
{
return init_op_ctx(ec);
}
@ -120,7 +105,7 @@ public:
idx_cnt_ = 0;
}
virtual int get_next_row(ObExecContext&, const common::ObNewRow*& row) const override
virtual int get_next_row(ObExecContext &, const common::ObNewRow *&row) const override
{
while (true) {
if (row_id_ < 0 || row_id_ >= row_cnt_) {
@ -148,18 +133,21 @@ public:
int64_t rand = murmurhash64A(&row_id_, sizeof(row_id_), 0);
rand = murmurhash64A(&idx_, sizeof(idx_), (uint64_t)rand);
ObObj* objs[] = {&cells_[ROW_ID_CELL], &cells_[IDX_CELL]};
int64_t ints[] = {row_id_val_(row_id_), idx_val_(idx_), rand};
ObObj *objs[] = { &cells_[ROW_ID_CELL], &cells_[IDX_CELL] };
int64_t ints[] = { row_id_val_(row_id_), idx_val_(idx_), rand };
if (!gen_varchar_cell_) {
for (int64_t i = 0; i < ARRAYSIZEOF(objs); i++) {
objs[i]->set_int(ints[i]);
}
} else {
char* bufs[] = {
&int_varchar_buf_[ROW_ID_CELL * INT_VARCHAR_BUF_SIZE], &int_varchar_buf_[IDX_CELL * INT_VARCHAR_BUF_SIZE]};
char *bufs[] = {
&int_varchar_buf_[ROW_ID_CELL * INT_VARCHAR_BUF_SIZE],
&int_varchar_buf_[IDX_CELL * INT_VARCHAR_BUF_SIZE]
};
for (int64_t i = 0; i < ARRAYSIZEOF(objs); i++) {
snprintf(bufs[i], INT_VARCHAR_BUF_SIZE, "%0.*ld", (int)INT_VARCHAR_BUF_SIZE, ints[i]);
// '0' flag ignored with precision and ‘%d’ gnu_printf format [-Werror=format=]
snprintf(bufs[i], INT_VARCHAR_BUF_SIZE, "%.*ld", (int)INT_VARCHAR_BUF_SIZE, ints[i]);
objs[i]->set_varchar(bufs[i], INT_VARCHAR_BUF_SIZE);
objs[i]->set_collation_type(ObCollationType::CS_TYPE_UTF8MB4_BIN);
}
@ -186,11 +174,11 @@ private:
mutable int64_t row_id_ = 0;
mutable int64_t idx_ = 0;
mutable int64_t idx_cnt_ = 0;
mutable char* str_ = NULL;
mutable char *str_ = NULL;
mutable char int_varchar_buf_[CELL_CNT * INT_VARCHAR_BUF_SIZE];
};
} // namespace sql
} // namespace oceanbase
} // end sql
} // end oceanbase
#endif // OCEANBASE_SET_DATA_GENERATOR_H_
#endif // OCEANBASE_SET_DATA_GENERATOR_H_