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

130
unittest/sql/optimizer/ob_mock_partition_service.h
View File

@ -15,103 +15,83 @@
#undef private
#undef protected
#include <gmock/gmock.h>
#include "ob_mock_partition.h"
#define private public
#define protected public
#include "storage/ob_partition_service.h"
#include "storage/ob_i_partition_storage.h"
#include "storage/ob_dml_param.h"
#include "../../storage/mockcontainer/ob_partition_service.h"
#include "storage/access/ob_dml_param.h"
#include "share/ob_server_locality_cache.h"
#include "common/row/ob_row_iterator.h"
using namespace oceanbase;
namespace test {
class MockPartitionService : public oceanbase::storage::ObPartitionService {
namespace test
{
class MockPartitionService : public oceanbase::storage::ObPartitionService
{
public:
MockPartitionService()
{}
virtual ~MockPartitionService()
{}
MockPartitionService() {}
virtual ~MockPartitionService() {}
virtual int get_server_locality_array(
common::ObIArray<share::ObServerLocality>& server_locality_array, bool& has_readonly_zone) const
{
int ret = OB_SUCCESS;
UNUSED(server_locality_array);
UNUSED(has_readonly_zone);
return ret;
}
//int get_scan_cost(const oceanbase::storage::ObTableScanParam &param,
//oceanbase::storage::ObPartitionEst &cost_estimate)
//{
//UNUSED(param);
//cost_estimate.logical_row_count_ = 100.0;
//cost_estimate.physical_row_count_ = 100.0;
//return OB_SUCCESS;
//}
int get_scan_cost(
const oceanbase::storage::ObTableScanParam& param, oceanbase::storage::ObPartitionEst& cost_estimate)
{
UNUSED(param);
cost_estimate.logical_row_count_ = 100.0;
cost_estimate.physical_row_count_ = 100.0;
return OB_SUCCESS;
}
int get_partition(const common::ObPartitionKey& pkey, oceanbase::storage::ObIPartitionGroup*& partition) const
{
UNUSED(pkey);
partition = const_cast<MockPartition*>(&partition_);
return OB_SUCCESS;
}
int get_partition(const common::ObPartitionKey& pkey, oceanbase::storage::ObIPartitionGroupGuard& guard) const
{
UNUSED(pkey);
guard.set_partition_group(this->get_pg_mgr(), const_cast<MockPartition&>(partition_));
return OB_SUCCESS;
}
int table_scan(common::ObVTableScanParam& param, common::ObNewRowIterator*& result)
int table_scan(
common::ObVTableScanParam &param,
common::ObNewRowIterator *&result)
{
UNUSED(param);
int ret = OB_SUCCESS;
ObObj* value = OB_NEW(ObObj, ObModIds::TEST);
ObObj *value = OB_NEW(ObObj, ObModIds::TEST);
value->set_type(ObIntType);
value->set_int(1);
value->set_collation_type(ObCharset::get_default_collation(ObCharset::get_default_charset()));
ObNewRow* row = OB_NEW(ObNewRow, ObModIds::TEST);
ObNewRow *row = OB_NEW(ObNewRow, ObModIds::TEST);
row->cells_ = value;
row->count_ = 1;
ObSingleRowIteratorWrapper* single = OB_NEW(ObSingleRowIteratorWrapper, ObModIds::TEST);
ObSingleRowIteratorWrapper *single = OB_NEW(ObSingleRowIteratorWrapper, ObModIds::TEST);
single->set_row(row);
result = single;
return ret;
}
int revert_scan_iter(common::ObNewRowIterator* iter)
int revert_scan_iter(common::ObNewRowIterator *iter)
{
UNUSED(iter);
return OB_SUCCESS;
}
int insert_rows(const transaction::ObTransDesc& trans_desc, const storage::ObDMLBaseParam& dml_param,
const common::ObPartitionKey& pkey, const common::ObIArray<uint64_t>& column_ids,
common::ObNewRowIterator* row_iter, int64_t& affected_rows)
int insert_rows(const transaction::ObTransDesc &trans_desc,
const storage::ObDMLBaseParam &dml_param,
const common::ObIArray<uint64_t> &column_ids,
common::ObNewRowIterator *row_iter,
int64_t &affected_rows)
{
UNUSED(trans_desc);
UNUSED(dml_param);
UNUSED(pkey);
UNUSED(column_ids);
int ret = OB_SUCCESS;
common::ObNewRow* row = NULL;
while (OB_SUCCESS == (ret = row_iter->get_next_row(row)))
;
common::ObNewRow *row = NULL;
while (OB_SUCCESS == (ret = row_iter->get_next_row(row)));
affected_rows = 1;
return OB_SUCCESS;
}
int update_rows(const transaction::ObTransDesc& trans_desc, const storage::ObDMLBaseParam& dml_param,
const common::ObPartitionKey& pkey, const common::ObIArray<uint64_t>& column_ids,
const common::ObIArray<uint64_t>& updated_column_ids, common::ObNewRowIterator* row_iter, int64_t& affected_rows)
int update_rows(const transaction::ObTransDesc &trans_desc,
const storage::ObDMLBaseParam &dml_param,
const common::ObIArray<uint64_t> &column_ids,
const common::ObIArray< uint64_t> &updated_column_ids,
common::ObNewRowIterator *row_iter,
int64_t &affected_rows)
{
UNUSED(trans_desc);
UNUSED(dml_param);
UNUSED(pkey);
UNUSED(column_ids);
UNUSED(updated_column_ids);
UNUSED(row_iter);
@ -119,41 +99,25 @@ public:
return OB_SUCCESS;
}
int delete_rows(const transaction::ObTransDesc& trans_desc, const storage::ObDMLBaseParam& dml_param,
const common::ObPartitionKey& pkey, const common::ObIArray<uint64_t>& column_ids,
common::ObNewRowIterator* row_iter, int64_t& affected_rows)
int delete_rows(const transaction::ObTransDesc &trans_desc,
const storage::ObDMLBaseParam &dml_param,
const common::ObIArray<uint64_t> &column_ids,
common::ObNewRowIterator *row_iter,
int64_t &affected_rows)
{
UNUSED(trans_desc);
UNUSED(dml_param);
UNUSED(pkey);
UNUSED(column_ids);
int ret = OB_SUCCESS;
common::ObNewRow* row = NULL;
while (OB_SUCCESS == (ret = row_iter->get_next_row(row)))
;
common::ObNewRow *row = NULL;
while (OB_SUCCESS == (ret = row_iter->get_next_row(row)));
affected_rows = 1;
return OB_SUCCESS;
}
obrpc::ObCommonRpcProxy& get_rs_rpc_proxy()
{
static obrpc::ObCommonRpcProxy rs_proxy;
return rs_proxy;
}
virtual storage::ObIPartitionGroupIterator* alloc_pg_iter()
{
/*
storage::ObPartitionIterator *part_iter = new storage::ObPartitionIterator();
part_iter->set_partition_service(*this);
return part_iter;
*/
return nullptr;
}
MockPartition partition_;
};
} // end namespace test
} // end namespace test
#endif /* _OB_MOCK_PARTITION_SERVICE_H_ */