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

83
src/sql/executor/ob_task_spliter.h
View File

@ -16,92 +16,77 @@
#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.h"
#include "sql/engine/table/ob_table_scan_op.h"
#include "sql/engine/table/ob_mv_table_scan.h"
#include "sql/engine/ob_exec_context.h"
#include "sql/engine/ob_engine_op_traits.h"
namespace oceanbase {
namespace common {
namespace oceanbase
{
namespace common
{
class ObIAllocator;
}
namespace sql {
namespace sql
{
class ObJob;
class ObJobConf;
class ObTransmit;
class ObTaskInfo;
class ObPhysicalPlanCtx;
class ObTaskExecutorCtx;
class ObTableScan;
class ObPhyOperator;
class ObTableModify;
#define ENG_OP typename ObEngineOpTraits<NEW_ENG>
class ObTaskSpliter {
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
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
INTRA_PARTITION_SPLIT = 6, // split single partition into multiple individual ranges
DISTRIBUTED_SPLIT = 7,
DETERMINATE_TASK_SPLIT = 8 // tasks are splited already
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);
virtual int get_next_task(ObTaskInfo*& task) = 0;
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 ObTableScan*>& scan_ops, const ObPhyOperator& op);
static int find_scan_ops(common::ObIArray<const ObTableScanSpec*>& scan_ops, const ObOpSpec& op);
static int find_insert_ops(common::ObIArray<const ObTableModify*>& insert_ops, const ObPhyOperator& op);
static int find_insert_ops(common::ObIArray<const ObTableModifySpec*>& insert_ops, const ObOpSpec& op);
static int find_all_table_location_keys(
common::ObIArray<TableLocationKey>& table_location_keys, const ObPhyOperator& op);
static int find_all_table_location_keys(common::ObIArray<TableLocationKey>& table_location_keys, const ObOpSpec& op);
static int check_dml_exist(bool& exist, const ObPhyOperator& op);
bool is_inited() const
{
return NULL != job_;
}
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);
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);
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);
template <bool NEW_ENG>
static int find_all_table_location_keys_inner(
common::ObIArray<TableLocationKey>& table_location_keys, const ENG_OP::Root& op);
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_;
ObJobConf* job_conf_;
common::ObSEArray<ObTaskInfo*, 16> task_store_;
ObPhysicalPlanCtx *plan_ctx_;
ObExecContext *exec_ctx_;
common::ObIAllocator *allocator_;
ObJob *job_;
common::ObSEArray<ObTaskInfo *, 16> task_store_;
};
#undef ENG_OP
} // namespace sql
} // namespace oceanbase
}
}
#endif /* OCEANBASE_SQL_EXECUTOR_TASK_SPLITER_ */
//// end of header file