106 lines
4.3 KiB
C++
106 lines
4.3 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 OCEANBASE_UNITEST_BLOCKSSTABLE_OB_ROW_GENERATE_
|
|
#define OCEANBASE_UNITEST_BLOCKSSTABLE_OB_ROW_GENERATE_
|
|
|
|
#include "share/schema/ob_table_schema.h"
|
|
#include "storage/ob_i_store.h"
|
|
#include "lib/container/ob_array.h"
|
|
#include "lib/allocator/ob_malloc.h"
|
|
#include "share/ob_errno.h"
|
|
#include "common/object/ob_object.h"
|
|
#include "common/object/ob_obj_type.h"
|
|
#include "storage/blocksstable/ob_datum_row.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
|
|
namespace blocksstable
|
|
{
|
|
|
|
class ObRowGenerate
|
|
{
|
|
public:
|
|
ObRowGenerate();
|
|
~ObRowGenerate();
|
|
inline int init(const share::schema::ObTableSchema &src_schema, bool is_multi_version_row = false);
|
|
inline int init(
|
|
const share::schema::ObTableSchema &src_schema,
|
|
common::ObArenaAllocator *allocator,
|
|
bool is_multi_version_row = false);
|
|
inline int get_next_row(ObDatumRow &row);
|
|
inline int get_next_row(const int64_t seed, ObDatumRow &row);
|
|
inline int get_next_row(const int64_t seed,
|
|
const int64_t trans_version,
|
|
const ObDmlFlag dml_flag,
|
|
ObDatumRow &row);
|
|
inline int get_next_row(const int64_t seed,
|
|
const int64_t trans_version,
|
|
const ObDmlFlag dml_flag,
|
|
const bool is_compacted_row,
|
|
const bool is_last_row,
|
|
const bool is_first_row,
|
|
ObDatumRow &row);
|
|
|
|
inline int get_next_row(const int64_t seed, const common::ObArray<uint64_t> &nop_column_idxs, ObDatumRow &row);
|
|
inline int check_one_row(const ObDatumRow &row, bool &exist);
|
|
// TODO @hanhui to be removed
|
|
inline int get_next_row(storage::ObStoreRow &row);
|
|
inline int get_next_row(const int64_t seed, storage::ObStoreRow &row);
|
|
inline int get_next_row(const int64_t seed,
|
|
const int64_t trans_version,
|
|
const ObDmlFlag dml_flag,
|
|
storage::ObStoreRow &row);
|
|
inline int get_next_row(const int64_t seed,
|
|
const int64_t trans_version,
|
|
const ObDmlFlag dml_flag,
|
|
const bool is_compacted_row,
|
|
const bool is_last_row,
|
|
const bool is_first_row,
|
|
storage::ObStoreRow &row);
|
|
|
|
inline int get_next_row(const int64_t seed, const common::ObArray<uint64_t> &nop_column_idxs, storage::ObStoreRow &row);
|
|
inline int check_one_row(const storage::ObStoreRow &row, bool &exist);
|
|
inline void reset() { seed_ = 0; column_list_.reset(); is_inited_ = false; allocator_.reset(); is_multi_version_row_ = false; }
|
|
const share::schema::ObTableSchema &get_schema() const { return schema_; }
|
|
|
|
private:
|
|
int generate_one_row(ObDatumRow& row, const int64_t seed, const ObDmlFlag dml_flag = DF_INSERT, const int64_t trans_version = 0);
|
|
int generate_one_row(storage::ObStoreRow& row, const int64_t seed, const ObDmlFlag dml_flag = DF_INSERT, const int64_t trans_version = 0);
|
|
int set_obj(const common::ObObjType &column_type, const uint64_t column_id, const int64_t seed, common::ObObj &obj, const int64_t trans_version);
|
|
int compare_obj(const common::ObObjType &column_type, const int64_t value, const common::ObObj obj, bool &exist);
|
|
int get_seed(const common::ObObjType &column_type, const common::ObObj obj, int64_t &seed);
|
|
int generate_urowid_obj(const int64_t rowkey_pos,
|
|
const int64_t seed,
|
|
const int64_t value,
|
|
ObObj &urowid_obj);
|
|
|
|
private:
|
|
common::ObArenaAllocator allocator_;
|
|
common::ObArenaAllocator *p_allocator_;
|
|
share::schema::ObTableSchema schema_;
|
|
storage::ObColDescArray column_list_;
|
|
int64_t seed_;
|
|
bool is_multi_version_row_;
|
|
bool is_inited_;
|
|
bool is_reused_;
|
|
};
|
|
|
|
}//storage
|
|
}//oceanbase
|
|
|
|
//.ipp file implement the inline function, no need link
|
|
#include "ob_row_generate.ipp"
|
|
|
|
#endif //OCEANBASE_UNITEST_BLOCKSSTABLE_OB_ROW_GENERATE_
|