oceanbase/unittest/storage/mock_ls_tablet_service.h
wangzelin.wzl 93a1074b0c patch 4.0
2022-10-24 17:57:12 +08:00

48 lines
1.5 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_STORAGE_MOCK_LS_TABLET_SERVICE
#define OCEANBASE_STORAGE_MOCK_LS_TABLET_SERVICE
#include "storage/ls/ob_ls_tablet_service.h"
namespace oceanbase
{
namespace storage
{
// mock different ls tablet service for dml interfaces(table_scan, insert_rows...)
// just override prepare_dml_running_ctx
class MockInsertRowsLSTabletService : public ObLSTabletService
{
public:
MockInsertRowsLSTabletService() = default;
virtual ~MockInsertRowsLSTabletService() = default;
protected:
virtual int 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) override;
private:
int prepare_relative_tables(
const int64_t schema_version,
const common::ObIArray<uint64_t> *upd_col_ids,
common::ObIArray<ObTabletHandle> &tablet_handles,
ObDMLRunningCtx &run_ctx);
};
} // namespace storage
} // namespace oceanbase
#endif // OCEANBASE_STORAGE_MOCK_LS_TABLET_SERVICE