hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[FEAT MERGE]4_1_sql_feature

Browse Source 
Co-authored-by: leslieyuchen <leslieyuchen@gmail.com>
Co-authored-by: Charles0429 <xiezhenjiang@gmail.com>
Co-authored-by: raywill <hustos@gmail.com>
This commit is contained in:
obdev
2023-01-28 16:01:26 +08:00
committed by ob-robot
parent 3080f2b66f
commit 2d19a9d8f5
846 changed files with 161957 additions and 116661 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
src/sql/optimizer/ob_log_window_function.h
View File

@ -21,21 +21,31 @@ namespace sql
class ObLogWindowFunction : public ObLogicalOperator
{
public:
enum WindowFunctionRoleType
{
NORMAL = 0,
PARTICIPATOR = 1, // for wf adaptive pushdown
CONSOLIDATOR = 2 // for wf adaptive pushdown
};
ObLogWindowFunction(ObLogPlan &plan)
: ObLogicalOperator(plan),
single_part_parallel_(false),
range_dist_parallel_(false),
rd_pby_sort_cnt_(0)
role_type_(WindowFunctionRoleType::NORMAL),
rd_pby_sort_cnt_(0),
wf_aggr_status_expr_(NULL)
{}
virtual ~ObLogWindowFunction() {}
virtual int print_my_plan_annotation(char *buf,
int64_t &buf_len,
int64_t &pos,
ExplainType type);
virtual int get_explain_name_internal(char *buf,
const int64_t buf_len,
int64_t &pos);
inline void set_aggr_status_expr(ObOpPseudoColumnRawExpr *wf_aggr_status_expr)
{ wf_aggr_status_expr_ = wf_aggr_status_expr; }
inline ObOpPseudoColumnRawExpr *&get_aggr_status_expr() { return wf_aggr_status_expr_; }
inline int add_window_expr(ObWinFunRawExpr *win_expr)
{ return win_exprs_.push_back(win_expr); }
inline ObIArray<ObWinFunRawExpr *> &get_window_exprs() { return win_exprs_; }
virtual uint64_t hash(uint64_t seed) const override;
virtual int est_cost() override;
virtual int est_width() override;
virtual int re_est_cost(EstimateCostInfo &param, double &card, double &cost) override;
@ -47,13 +57,18 @@ namespace sql
virtual int allocate_granule_pre(AllocGIContext &ctx) override;
int get_win_partition_intersect_exprs(ObIArray<ObWinFunRawExpr *> &win_exprs,
ObIArray<ObRawExpr *> &win_part_exprs);
virtual int inner_replace_generated_agg_expr(
virtual int inner_replace_op_exprs(
const common::ObIArray<std::pair<ObRawExpr *, ObRawExpr*>> &to_replace_exprs) override;
void set_single_part_parallel(bool v) { single_part_parallel_ = v; }
bool is_single_part_parallel() const { return single_part_parallel_; }
void set_ragne_dist_parallel(bool v) { range_dist_parallel_ = v; }
bool is_range_dist_parallel() const { return range_dist_parallel_; }
int get_winfunc_output_exprs(ObIArray<ObRawExpr *> &output_exprs);
void set_role_type(WindowFunctionRoleType v) { role_type_ = v; }
WindowFunctionRoleType get_role_type() const { return role_type_; }
bool is_push_down() { return PARTICIPATOR == role_type_|| CONSOLIDATOR == role_type_; }
bool is_participator() { return PARTICIPATOR == role_type_; }
bool is_consolidator() { return CONSOLIDATOR == role_type_; }
int set_rd_sort_keys(const common::ObIArray<OrderItem> &sort_keys)
{
return rd_sort_keys_.assign(sort_keys);
@ -62,7 +77,14 @@ namespace sql
{
return rd_sort_keys_;
}
int set_pushdown_info(const common::ObIArray<bool> &pushdown_info)
{
return pushdown_info_.assign(pushdown_info);
}
const common::ObIArray<bool> &get_pushdown_info() const
{
return pushdown_info_;
}
void set_rd_pby_sort_cnt(const int64_t cnt) { rd_pby_sort_cnt_ = cnt; }
int64_t get_rd_pby_sort_cnt() const { return rd_pby_sort_cnt_; }
@ -70,7 +92,10 @@ namespace sql
{
return dist_hint_.assign(dist_hint);
}
virtual int print_outline(planText &plan_text) override;
virtual int get_plan_item_info(PlanText &plan_text,
ObSqlPlanItem &plan_item) override;
virtual int print_outline_data(PlanText &plan_text) override;
virtual int print_used_hint(PlanText &plan_text) override;
private:
ObSEArray<ObWinFunRawExpr *, 4, common::ModulePageAllocator, true> win_exprs_;
@ -93,6 +118,9 @@ namespace sql
// 3. Only the following functions supported: rank,dense_rank,sum,count,min,max
bool range_dist_parallel_;
// https://yuque.antfin.com/ob/sql/wf_adaptive_parallel_execution#69270e6f
WindowFunctionRoleType role_type_;
// sort keys for range distributed parallel.
common::ObSEArray<OrderItem, 8, common::ModulePageAllocator, true> rd_sort_keys_;
// the first %rd_pby_sort_cnt_ of %rd_sort_keys_ is the partition by of window function.
@ -100,6 +128,10 @@ namespace sql
// for PQ_DISTRIBUTE_WINDOW hint outline
common::ObSEArray<WinDistAlgo, 8, common::ModulePageAllocator, true> dist_hint_;
// for reporting window function adaptive pushdown
ObOpPseudoColumnRawExpr *wf_aggr_status_expr_;
common::ObSEArray<bool, 8, common::ModulePageAllocator, true> pushdown_info_;
};
}
}