Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.trino.sql.analyzer.QueryExplainer Maven / Gradle / Ivy
/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.trino.sql.analyzer;
import io.trino.Session;
import io.trino.client.NodeVersion;
import io.trino.cost.CachingTableStatsProvider;
import io.trino.cost.CostCalculator;
import io.trino.cost.StatsCalculator;
import io.trino.execution.querystats.PlanOptimizersStatsCollector;
import io.trino.execution.warnings.WarningCollector;
import io.trino.spi.TrinoException;
import io.trino.sql.PlannerContext;
import io.trino.sql.SqlFormatter;
import io.trino.sql.planner.LogicalPlanner;
import io.trino.sql.planner.Plan;
import io.trino.sql.planner.PlanFragmenter;
import io.trino.sql.planner.PlanNodeIdAllocator;
import io.trino.sql.planner.PlanOptimizersFactory;
import io.trino.sql.planner.SubPlan;
import io.trino.sql.planner.optimizations.PlanOptimizer;
import io.trino.sql.planner.planprinter.PlanPrinter;
import io.trino.sql.tree.CreateCatalog;
import io.trino.sql.tree.CreateMaterializedView;
import io.trino.sql.tree.CreateSchema;
import io.trino.sql.tree.CreateTable;
import io.trino.sql.tree.CreateView;
import io.trino.sql.tree.DropCatalog;
import io.trino.sql.tree.DropSchema;
import io.trino.sql.tree.ExplainType.Type;
import io.trino.sql.tree.Expression;
import io.trino.sql.tree.Prepare;
import io.trino.sql.tree.Statement;
import java.util.List;
import java.util.Optional;
import static io.trino.execution.ParameterExtractor.bindParameters;
import static io.trino.spi.StandardErrorCode.NOT_SUPPORTED;
import static io.trino.sql.analyzer.QueryType.EXPLAIN;
import static io.trino.sql.planner.LogicalPlanner.Stage.OPTIMIZED_AND_VALIDATED;
import static io.trino.sql.planner.planprinter.IoPlanPrinter.textIoPlan;
import static io.trino.sql.planner.planprinter.PlanPrinter.jsonDistributedPlan;
import static io.trino.sql.planner.planprinter.PlanPrinter.jsonLogicalPlan;
import static io.trino.util.StatementUtils.isDataDefinitionStatement;
import static java.lang.String.format;
import static java.util.Objects.requireNonNull;
public class QueryExplainer
{
private final List planOptimizers;
private final PlanFragmenter planFragmenter;
private final PlannerContext plannerContext;
private final AnalyzerFactory analyzerFactory;
private final StatsCalculator statsCalculator;
private final CostCalculator costCalculator;
private final NodeVersion version;
QueryExplainer(
PlanOptimizersFactory planOptimizersFactory,
PlanFragmenter planFragmenter,
PlannerContext plannerContext,
AnalyzerFactory analyzerFactory,
StatsCalculator statsCalculator,
CostCalculator costCalculator,
NodeVersion version)
{
this.planOptimizers = requireNonNull(planOptimizersFactory.getPlanOptimizers(), "planOptimizers is null");
this.planFragmenter = requireNonNull(planFragmenter, "planFragmenter is null");
this.plannerContext = requireNonNull(plannerContext, "plannerContext is null");
this.analyzerFactory = requireNonNull(analyzerFactory, "analyzerFactory is null");
this.statsCalculator = requireNonNull(statsCalculator, "statsCalculator is null");
this.costCalculator = requireNonNull(costCalculator, "costCalculator is null");
this.version = requireNonNull(version, "version is null");
}
public void validate(Session session, Statement statement, List parameters, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
{
analyze(session, statement, parameters, warningCollector, planOptimizersStatsCollector);
}
public String getPlan(Session session, Statement statement, Type planType, List parameters, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
{
Optional explain = explainDataDefinition(statement, parameters);
if (explain.isPresent()) {
return explain.get();
}
return switch (planType) {
case LOGICAL -> {
Plan plan = getLogicalPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector);
yield PlanPrinter.textLogicalPlan(plan.getRoot(), plannerContext.getMetadata(), plannerContext.getFunctionManager(), plan.getStatsAndCosts(), session, 0, false, Optional.of(version));
}
case DISTRIBUTED -> PlanPrinter.textDistributedPlan(
getDistributedPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector),
plannerContext.getMetadata(),
plannerContext.getFunctionManager(),
session,
false,
version);
case IO -> textIoPlan(getLogicalPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector), plannerContext, session);
default -> throw new IllegalArgumentException("Unhandled plan type: " + planType);
};
}
public String getGraphvizPlan(Session session, Statement statement, Type planType, List parameters, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
{
Optional explain = explainDataDefinition(statement, parameters);
if (explain.isPresent()) {
// todo format as graphviz
return explain.get();
}
return switch (planType) {
case LOGICAL -> {
Plan plan = getLogicalPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector);
yield PlanPrinter.graphvizLogicalPlan(plan.getRoot());
}
case DISTRIBUTED -> PlanPrinter.graphvizDistributedPlan(getDistributedPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector));
default -> throw new IllegalArgumentException("Unhandled plan type: " + planType);
};
}
public String getJsonPlan(Session session, Statement statement, Type planType, List parameters, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
{
Optional explain = explainDataDefinition(statement, parameters);
if (explain.isPresent()) {
// todo format as json
return explain.get();
}
return switch (planType) {
case IO -> textIoPlan(getLogicalPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector), plannerContext, session);
case LOGICAL -> {
Plan plan = getLogicalPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector);
yield jsonLogicalPlan(plan.getRoot(), session, plannerContext.getMetadata(), plannerContext.getFunctionManager(), plan.getStatsAndCosts());
}
case DISTRIBUTED -> jsonDistributedPlan(
getDistributedPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector),
plannerContext.getMetadata(),
plannerContext.getFunctionManager(),
session);
default -> throw new TrinoException(NOT_SUPPORTED, format("Unsupported explain plan type %s for JSON format", planType));
};
}
public Plan getLogicalPlan(Session session, Statement statement, List parameters, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
{
// analyze statement
Analysis analysis = analyze(session, statement, parameters, warningCollector, planOptimizersStatsCollector);
PlanNodeIdAllocator idAllocator = new PlanNodeIdAllocator();
// plan statement
LogicalPlanner logicalPlanner = new LogicalPlanner(
session,
planOptimizers,
idAllocator,
plannerContext,
statsCalculator,
costCalculator,
warningCollector,
planOptimizersStatsCollector,
new CachingTableStatsProvider(plannerContext.getMetadata(), session));
return logicalPlanner.plan(analysis, OPTIMIZED_AND_VALIDATED, true);
}
private Analysis analyze(Session session, Statement statement, List parameters, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
{
Analyzer analyzer = analyzerFactory.createAnalyzer(session, parameters, bindParameters(statement, parameters), warningCollector, planOptimizersStatsCollector);
return analyzer.analyze(statement, EXPLAIN);
}
private SubPlan getDistributedPlan(Session session, Statement statement, List parameters, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
{
Plan plan = getLogicalPlan(session, statement, parameters, warningCollector, planOptimizersStatsCollector);
return planFragmenter.createSubPlans(session, plan, false, warningCollector);
}
private static Optional explainDataDefinition(T statement, List parameters)
{
if (!isDataDefinitionStatement(statement.getClass())) {
return Optional.empty();
}
return Optional.of(switch (statement) {
case CreateCatalog createCatalog -> "CREATE CATALOG " + createCatalog.getCatalogName();
case DropCatalog dropCatalog -> "DROP CATALOG " + dropCatalog.getCatalogName();
case CreateSchema createSchema -> "CREATE SCHEMA " + createSchema.getSchemaName();
case DropSchema dropSchema -> "DROP SCHEMA " + dropSchema.getSchemaName();
case CreateTable createTable -> "CREATE TABLE " + createTable.getName();
case CreateView createView -> "CREATE VIEW " + createView.getName();
case CreateMaterializedView createMaterializedView -> "CREATE MATERIALIZED VIEW " + createMaterializedView.getName();
case Prepare prepare -> "PREPARE " + prepare.getName();
default -> {
StringBuilder builder = new StringBuilder();
builder.append(SqlFormatter.formatSql(statement));
if (!parameters.isEmpty()) {
builder.append("\n")
.append("Parameters: ")
.append(parameters);
}
yield builder.toString();
}
});
}
}
