93 lines
2.9 KiB
C++
93 lines
2.9 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_EXECUTOR_TASK_SPLITER_
|
|
#define OCEANBASE_SQL_EXECUTOR_TASK_SPLITER_
|
|
|
|
#include "lib/allocator/ob_allocator.h"
|
|
#include "sql/executor/ob_job.h"
|
|
#include "sql/engine/ob_phy_operator_type.h"
|
|
#include "sql/engine/table/ob_table_scan_op.h"
|
|
#include "sql/engine/ob_exec_context.h"
|
|
#include "sql/engine/ob_engine_op_traits.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace common
|
|
{
|
|
class ObIAllocator;
|
|
}
|
|
namespace sql
|
|
{
|
|
class ObJob;
|
|
class ObTaskInfo;
|
|
class ObPhysicalPlanCtx;
|
|
class ObTaskExecutorCtx;
|
|
class ObTableModify;
|
|
|
|
#define ENG_OP typename ObEngineOpTraits<NEW_ENG>
|
|
class ObTaskSpliter
|
|
{
|
|
public:
|
|
enum TaskSplitType {
|
|
INVALID_SPLIT = 0,
|
|
LOCAL_IDENTITY_SPLIT = 1, // root(local) transmit task splitter
|
|
REMOTE_IDENTITY_SPLIT = 2, // remote transmit task splitter
|
|
PARTITION_RANGE_SPLIT = 3, // distributed transmit task splitter
|
|
INTERM_SPLIT = 4,
|
|
INSERT_SPLIT = 5,
|
|
INTRA_PARTITION_SPLIT = 6, // split single partition into multiple individual ranges
|
|
DISTRIBUTED_SPLIT = 7,
|
|
DETERMINATE_TASK_SPLIT = 8 // tasks are splited already
|
|
};
|
|
public:
|
|
ObTaskSpliter();
|
|
virtual ~ObTaskSpliter();
|
|
int init(ObPhysicalPlanCtx *plan_ctx,
|
|
ObExecContext *exec_ctx,
|
|
ObJob &job,
|
|
common::ObIAllocator &allocator);
|
|
// 没有更多task则返回OB_ITER_END
|
|
virtual int get_next_task(ObTaskInfo *&task) = 0;
|
|
virtual TaskSplitType get_type() const = 0;
|
|
VIRTUAL_TO_STRING_KV(K_(server));
|
|
|
|
static int find_scan_ops(common::ObIArray<const ObTableScanSpec*> &scan_ops, const ObOpSpec &op);
|
|
|
|
static int find_insert_ops(common::ObIArray<const ObTableModifySpec *> &insert_ops,
|
|
const ObOpSpec &op);
|
|
bool is_inited() const { return NULL != job_; }
|
|
protected:
|
|
int create_task_info(ObTaskInfo *&task);
|
|
|
|
template <bool NEW_ENG>
|
|
static int find_scan_ops_inner(common::ObIArray<const ENG_OP::TSC *> &scan_ops, const ENG_OP::Root &op);
|
|
|
|
template <bool NEW_ENG>
|
|
static int find_insert_ops_inner(common::ObIArray<const ENG_OP::TableModify *> &insert_ops,
|
|
const ENG_OP::Root &op);
|
|
protected:
|
|
common::ObAddr server_;
|
|
ObPhysicalPlanCtx *plan_ctx_;
|
|
ObExecContext *exec_ctx_;
|
|
common::ObIAllocator *allocator_;
|
|
ObJob *job_;
|
|
common::ObSEArray<ObTaskInfo *, 16> task_store_;
|
|
};
|
|
|
|
#undef ENG_OP
|
|
|
|
}
|
|
}
|
|
#endif /* OCEANBASE_SQL_EXECUTOR_TASK_SPLITER_ */
|
|
//// end of header file
|