73 lines
2.7 KiB
C++
73 lines
2.7 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_OB_JOB_CONTROL_
|
|
#define OCEANBASE_SQL_EXECUTOR_OB_JOB_CONTROL_
|
|
|
|
#include "lib/container/ob_se_array.h"
|
|
#include "sql/executor/ob_job.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace sql
|
|
{
|
|
class ObTaskEvent;
|
|
class ObJobControl
|
|
{
|
|
public:
|
|
explicit ObJobControl();
|
|
virtual ~ObJobControl();
|
|
|
|
void reset();
|
|
virtual int sort_job_scan_part_locs(ObExecContext &ctx);
|
|
virtual int init_job_finish_queue(ObExecContext &ctx);
|
|
// int arrange_jobs();
|
|
/*
|
|
* @input ob_execution_id 用于在IRM中区分不同execution的结果，
|
|
* 仅对Distributed模式有效，其余模式均为INVALID_ID
|
|
*/
|
|
int create_job(common::ObIAllocator &alloc,
|
|
const ObExecutionID &ob_execution_id,
|
|
uint64_t root_op_id,
|
|
ObJob *&job) const;
|
|
int find_job_by_job_id(uint64_t job_id, ObJob *&job) const;
|
|
int find_job_by_root_op_id(uint64_t root_op_id, ObJob *&job) const;
|
|
virtual int get_ready_jobs(common::ObIArray<ObJob *> &jobs, bool serial_sched = false) const = 0;
|
|
virtual int get_running_jobs(common::ObIArray<ObJob *> &jobs) const;
|
|
int get_all_jobs(common::ObIArray<ObJob *> &jobs) const;
|
|
int get_all_jobs_except_root_job(common::ObIArray<ObJob *> &jobs) const;
|
|
int all_jobs_finished(bool &is_finished) const;
|
|
int all_jobs_finished_except_root_job(bool &is_finished) const;
|
|
// build op input of the current job
|
|
int print_status(char *buf, int64_t buf_len, bool ignore_normal_state = false) const;
|
|
|
|
inline int add_job(ObJob *job) { return jobs_.push_back(job); }
|
|
inline int64_t get_job_count() const { return jobs_.count(); }
|
|
|
|
DECLARE_TO_STRING;
|
|
private:
|
|
int build_job_ctx(ObExecContext &ctx, ObJob &job);
|
|
int build_job_op_input(ObExecContext &ctx, ObJob &job);
|
|
// int jobs_quick_sort(common::ObIArray<ObJob *> &jobs,
|
|
// int64_t low,
|
|
// int64_t high);
|
|
void print_job_tree(char *buf, const int64_t buf_len, int64_t &pos, ObJob *job) const;
|
|
protected:
|
|
common::ObSEArray<ObJob *, 2> jobs_; // remote plan has two jobs
|
|
private:
|
|
static volatile uint64_t global_job_id_;
|
|
DISALLOW_COPY_AND_ASSIGN(ObJobControl);
|
|
};
|
|
}
|
|
}
|
|
#endif /* OCEANBASE_SQL_EXECUTOR_OB_JOB_CONTROL_ */
|