101 lines
3.2 KiB
C++
101 lines
3.2 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_SQL_OB_TABLE_REPLACE_OP_
|
|
#define OCEANBASE_SQL_OB_TABLE_REPLACE_OP_
|
|
|
|
#include "sql/engine/dml/ob_table_modify_op.h"
|
|
#include "sql/engine/dml/ob_table_delete_op.h"
|
|
#include "storage/ob_partition_service.h"
|
|
#include "sql/session/ob_sql_session_info.h"
|
|
#include "sql/executor/ob_task_executor_ctx.h"
|
|
|
|
namespace oceanbase {
|
|
namespace sql {
|
|
|
|
class ObTableReplaceSpec : public ObTableModifySpec {
|
|
OB_UNIS_VERSION_V(1);
|
|
|
|
public:
|
|
ObTableReplaceSpec(common::ObIAllocator& alloc, const ObPhyOperatorType type)
|
|
: ObTableModifySpec(alloc, type), only_one_unique_key_(false), table_column_exprs_(alloc)
|
|
{}
|
|
|
|
bool only_one_unique_key_;
|
|
ExprFixedArray table_column_exprs_; // table column exprs
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(ObTableReplaceSpec);
|
|
};
|
|
|
|
class ObTableReplaceOp : public ObTableModifyOp {
|
|
public:
|
|
ObTableReplaceOp(ObExecContext& ctx, const ObOpSpec& spec, ObOpInput* input)
|
|
: ObTableModifyOp(ctx, spec, input),
|
|
row2exprs_projector_(ctx.get_allocator()),
|
|
record_(0),
|
|
affected_rows_(0),
|
|
delete_count_(0),
|
|
dml_param_(),
|
|
part_infos_()
|
|
{}
|
|
virtual ~ObTableReplaceOp()
|
|
{}
|
|
|
|
virtual int inner_open() override;
|
|
virtual int rescan() override;
|
|
virtual int inner_get_next_row() override;
|
|
virtual void destroy() override;
|
|
int do_table_replace();
|
|
|
|
protected:
|
|
int check_values(bool& is_equal) const;
|
|
|
|
private:
|
|
int try_insert(ObSQLSessionInfo& my_session, const ObPartitionKey& part_key,
|
|
storage::ObPartitionService& partition_service, const storage::ObDMLBaseParam& dml_param,
|
|
ObNewRowIterator*& dup_rows_iter);
|
|
int do_replace(ObSQLSessionInfo& my_session, const ObPartitionKey& part_key,
|
|
storage::ObPartitionService& partition_service, const ObNewRowIterator& dup_rows_iter,
|
|
const storage::ObDMLBaseParam& dml_param);
|
|
int scan_row(ObSQLSessionInfo& my_session, const ObPartitionKey& part_key,
|
|
storage::ObPartitionService& partition_service, storage::ObTableScanParam& scan_param,
|
|
const common::ObNewRowIterator& dup_rows_iter, common::ObNewRowIterator** result);
|
|
|
|
protected:
|
|
storage::ObRow2ExprsProjector row2exprs_projector_;
|
|
int64_t record_;
|
|
int64_t affected_rows_;
|
|
int64_t delete_count_;
|
|
storage::ObDMLBaseParam dml_param_;
|
|
common::ObSEArray<DMLPartInfo, 1> part_infos_;
|
|
};
|
|
|
|
class ObTableReplaceOpInput : public ObTableModifyOpInput {
|
|
OB_UNIS_VERSION_V(1);
|
|
|
|
public:
|
|
ObTableReplaceOpInput(ObExecContext& ctx, const ObOpSpec& spec) : ObTableModifyOpInput(ctx, spec)
|
|
{}
|
|
int init(ObTaskInfo& task_info) override
|
|
{
|
|
return ObTableModifyOpInput::init(task_info);
|
|
}
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(ObTableReplaceOpInput);
|
|
};
|
|
|
|
} // end namespace sql
|
|
} // end namespace oceanbase
|
|
|
|
#endif
|