You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-11-03 17:13:17 +03:00
14fe4401bc
* fix builds * MCOL-6143: rbo as a separate library * Move get_unstable_optimizer out of rbo, move findStatisticsForATable to header bo be (dirty fix, but cuts the corner) * Add some helpers to a headerfile for unittesting * Simple unittests with some mocks
83 lines
2.5 KiB
C++
83 lines
2.5 KiB
C++
/* Copyright (C) 2025 MariaDB Corporation
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; version 2 of
|
|
the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
MA 02110-1301, USA. */
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
|
|
#define PREFER_MY_CONFIG_H
|
|
#include <my_config.h>
|
|
// #include "idb_mysql.h"
|
|
|
|
#include "ha_mcs_impl_if.h"
|
|
|
|
#include "execplan/calpontselectexecutionplan.h"
|
|
|
|
namespace optimizer
|
|
{
|
|
|
|
class RBOptimizerContext
|
|
{
|
|
public:
|
|
RBOptimizerContext() = delete;
|
|
RBOptimizerContext(cal_impl_if::gp_walk_info& walk_info, THD& thd, bool logRules)
|
|
: gwi(walk_info), thd(thd), logRules(logRules)
|
|
{
|
|
}
|
|
// gwi lifetime should be longer than optimizer context.
|
|
// In plugin runtime this is always true.
|
|
cal_impl_if::gp_walk_info& gwi;
|
|
THD& thd;
|
|
uint64_t uniqueId{0};
|
|
bool logRules{false};
|
|
};
|
|
|
|
struct Rule
|
|
{
|
|
// returns true if rule may be applied
|
|
using RuleApplierFilter = bool (*)(execplan::CalpontSelectExecutionPlan&, RBOptimizerContext&);
|
|
// returns true if rule was applied
|
|
using RuleApplier = bool (*)(execplan::CalpontSelectExecutionPlan&, RBOptimizerContext&);
|
|
|
|
Rule(std::string&& name, RuleApplierFilter mayApply, RuleApplier applyRule)
|
|
: name(name), mayApply(mayApply), applyRule(applyRule) {};
|
|
|
|
std::string name;
|
|
RuleApplierFilter mayApply;
|
|
RuleApplier applyRule;
|
|
// TODO Wrap CSEP into Nodes to be able to navigate up and down the tree and remove this flag
|
|
bool applyOnlyOnce = true;
|
|
|
|
Rule() = default;
|
|
Rule(const Rule&) = default;
|
|
Rule(Rule&&) = default;
|
|
|
|
std::string getName() const
|
|
{
|
|
return name;
|
|
}
|
|
|
|
Rule& operator=(const Rule&) = default;
|
|
Rule& operator=(Rule&&) = default;
|
|
|
|
bool apply(execplan::CalpontSelectExecutionPlan& csep, RBOptimizerContext& ctx) const;
|
|
bool walk(execplan::CalpontSelectExecutionPlan& csep, RBOptimizerContext& ctx) const;
|
|
};
|
|
|
|
bool optimizeCSEP(execplan::CalpontSelectExecutionPlan& root, RBOptimizerContext& ctx,
|
|
bool useUnstableOptimizer);
|
|
} // namespace optimizer
|