74 lines
2.6 KiB
C++
74 lines
2.6 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.
|
|
*/
|
|
|
|
#define USING_LOG_PREFIX STORAGE
|
|
|
|
#define private public
|
|
|
|
#include "mock_ls_tablet_service.h"
|
|
|
|
#include "lib/oblog/ob_log.h"
|
|
#include "lib/utility/ob_macro_utils.h"
|
|
#include "share/ob_errno.h"
|
|
#include "share/schema/ob_table_schema.h"
|
|
#include "share/schema/ob_table_dml_param.h"
|
|
#include "storage/ob_i_store.h"
|
|
#include "storage/ob_dml_running_ctx.h"
|
|
#include "storage/access/ob_dml_param.h"
|
|
#include "storage/tablet/ob_tablet.h"
|
|
|
|
#undef private
|
|
|
|
using namespace oceanbase::storage;
|
|
|
|
int MockInsertRowsLSTabletService::prepare_dml_running_ctx(
|
|
const common::ObIArray<uint64_t> *column_ids,
|
|
const common::ObIArray<uint64_t> *upd_col_ids,
|
|
ObTabletHandle &tablet_handle,
|
|
ObDMLRunningCtx &run_ctx)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
uint64_t table_id = OB_INVALID_ID;
|
|
ObRelativeTable &relative_table = run_ctx.relative_table_;
|
|
const ObDMLBaseParam &dml_param = run_ctx.dml_param_;
|
|
ObStoreCtx &store_ctx = run_ctx.store_ctx_;
|
|
auto &acc_ctx = store_ctx.mvcc_acc_ctx_;
|
|
memtable::ObIMemtableCtx *mem_ctx = store_ctx.mvcc_acc_ctx_.get_mem_ctx();
|
|
|
|
if (OB_UNLIKELY(!store_ctx.is_valid())
|
|
|| OB_UNLIKELY(!dml_param.is_valid())
|
|
|| OB_ISNULL(dml_param.table_param_)) {
|
|
ret = OB_INVALID_ARGUMENT;
|
|
LOG_WARN("invalid argument", K(ret), KP(mem_ctx), K(dml_param));
|
|
} else {
|
|
const ObTableSchemaParam &schema = dml_param.table_param_->get_data_table();
|
|
if (OB_FAIL(relative_table.init(
|
|
&schema,
|
|
tablet_handle.get_obj()->get_tablet_meta().tablet_id_,
|
|
schema.is_global_index_table() &&
|
|
ObIndexStatus::INDEX_STATUS_UNAVAILABLE == schema.get_index_status()))) {
|
|
LOG_WARN("failed to get relative table", K(ret), K(dml_param));
|
|
} else if (NULL != column_ids && OB_FAIL(run_ctx.prepare_column_info(*column_ids))) {
|
|
LOG_WARN("fail to get column descriptions and column map", K(ret), K(*column_ids));
|
|
} else {
|
|
relative_table.tablet_iter_.set_tablet_handle(tablet_handle);
|
|
store_ctx.table_version_ = dml_param.schema_version_;
|
|
run_ctx.column_ids_ = column_ids;
|
|
}
|
|
}
|
|
|
|
if (OB_FAIL(ret)) {
|
|
relative_table.destroy();
|
|
}
|
|
return ret;
|
|
}
|