Package io.trino.testing

Class LocalQueryRunner

java.lang.Object

io.trino.testing.LocalQueryRunner

All Implemented Interfaces:

QueryRunner, Closeable, AutoCloseable

public class LocalQueryRunner
extends Object
implements QueryRunner

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	LocalQueryRunner.Builder
static interface	LocalQueryRunner.PlanOptimizersProvider

Nested classes/interfaces inherited from interface io.trino.testing.QueryRunner

QueryRunner.MaterializedResultWithPlan

Method Summary

Modifier and Type	Method	Description
void	addFunctions(List<? extends SqlFunction> functions)
static LocalQueryRunner.Builder	builder(Session defaultSession)
void	close()
static LocalQueryRunner	create(Session defaultSession)
void	createCatalog(String catalogName, ConnectorFactory connectorFactory, Map<String,String> properties)
void	createCatalog(String catalogName, String connectorName, Map<String,String> properties)
List<Driver>	createDrivers(Session session, String sql, OutputFactory outputFactory, TaskContext taskContext)
List<Driver>	createDrivers(String sql, OutputFactory outputFactory, TaskContext taskContext)
Plan	createPlan(Session session, String sql, WarningCollector warningCollector)
Plan	createPlan(Session session, String sql, LogicalPlanner.Stage stage, boolean forceSingleNode, WarningCollector warningCollector)
Plan	createPlan(Session session, String sql, LogicalPlanner.Stage stage, WarningCollector warningCollector)
Plan	createPlan(Session session, String sql, List<PlanOptimizer> optimizers, WarningCollector warningCollector)
Plan	createPlan(Session session, String sql, List<PlanOptimizer> optimizers, LogicalPlanner.Stage stage, WarningCollector warningCollector)
SubPlan	createSubPlans(Session session, Plan plan, boolean forceSingleNode)
MaterializedResult	execute(Session session, String sql)
MaterializedResult	execute(String sql)
QueryRunner.MaterializedResultWithPlan	executeWithPlan(Session session, String sql, WarningCollector warningCollector)
TestingAccessControlManager	getAccessControl()
BlockTypeOperators	getBlockTypeOperators()
CatalogManager	getCatalogManager()
CostCalculator	getCostCalculator()
Session	getDefaultSession()
CostCalculator	getEstimatedExchangesCostCalculator()
Lock	getExclusiveLock()
ExecutorService	getExecutor()
ExpressionCompiler	getExpressionCompiler()
TestingGroupProvider	getGroupProvider()
Metadata	getMetadata()
int	getNodeCount()
NodePartitioningManager	getNodePartitioningManager()
PageSourceManager	getPageSourceManager()
PlannerContext	getPlannerContext()
List<PlanOptimizer>	getPlanOptimizers(boolean forceSingleNode)
QueryExplainer	getQueryExplainer()
ScheduledExecutorService	getScheduler()
SessionPropertyManager	getSessionPropertyManager()
SplitManager	getSplitManager()
SqlParser	getSqlParser()
StatsCalculator	getStatsCalculator()
TransactionManager	getTransactionManager()
TypeManager	getTypeManager()
TypeOperators	getTypeOperators()
void	injectTaskFailure(String traceToken, int stageId, int partitionId, int attemptId, FailureInjector.InjectedFailureType injectionType, Optional<ErrorType> errorType)
void	installPlugin(Plugin plugin)
<T> T	inTransaction(Session session, Function<Session,T> transactionSessionConsumer)
<T> T	inTransaction(Function<Session,T> transactionSessionConsumer)
List<QualifiedObjectName>	listTables(Session session, String catalog, String schema)
void	loadEventListeners()
LocalQueryRunner	printPlan()
boolean	tableExists(Session session, String table)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

create

public static LocalQueryRunner create(Session defaultSession)

builder

public static LocalQueryRunner.Builder builder(Session defaultSession)

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable

Specified by:

close in interface QueryRunner

loadEventListeners

public void loadEventListeners()

getNodeCount

public int getNodeCount()

Specified by:

getNodeCount in interface QueryRunner

getCatalogManager

public CatalogManager getCatalogManager()

getTransactionManager

public TransactionManager getTransactionManager()

Specified by:

getTransactionManager in interface QueryRunner

getSqlParser

public SqlParser getSqlParser()

getPlannerContext

public PlannerContext getPlannerContext()

getMetadata

public Metadata getMetadata()

Specified by:

getMetadata in interface QueryRunner

getTypeManager

public TypeManager getTypeManager()

Specified by:

getTypeManager in interface QueryRunner

getQueryExplainer

public QueryExplainer getQueryExplainer()

Specified by:

getQueryExplainer in interface QueryRunner

getSessionPropertyManager

public SessionPropertyManager getSessionPropertyManager()

Specified by:

getSessionPropertyManager in interface QueryRunner

getTypeOperators

public TypeOperators getTypeOperators()

getBlockTypeOperators

public BlockTypeOperators getBlockTypeOperators()

getNodePartitioningManager

public NodePartitioningManager getNodePartitioningManager()

Specified by:

getNodePartitioningManager in interface QueryRunner

getPageSourceManager

public PageSourceManager getPageSourceManager()

Specified by:

getPageSourceManager in interface QueryRunner

getSplitManager

public SplitManager getSplitManager()

Specified by:

getSplitManager in interface QueryRunner

getStatsCalculator

public StatsCalculator getStatsCalculator()

Specified by:

getStatsCalculator in interface QueryRunner

getCostCalculator

public CostCalculator getCostCalculator()

getEstimatedExchangesCostCalculator

public CostCalculator getEstimatedExchangesCostCalculator()

getGroupProvider

public TestingGroupProvider getGroupProvider()

Specified by:

getGroupProvider in interface QueryRunner

getAccessControl

public TestingAccessControlManager getAccessControl()

Specified by:

getAccessControl in interface QueryRunner

getExecutor

public ExecutorService getExecutor()

getScheduler

public ScheduledExecutorService getScheduler()

getDefaultSession

public Session getDefaultSession()

Specified by:

getDefaultSession in interface QueryRunner

getExpressionCompiler

public ExpressionCompiler getExpressionCompiler()

createCatalog

public void createCatalog(String catalogName,
 ConnectorFactory connectorFactory,
 Map<String,String> properties)

installPlugin

public void installPlugin(Plugin plugin)

Specified by:

installPlugin in interface QueryRunner

addFunctions

public void addFunctions(List<? extends SqlFunction> functions)

Specified by:

addFunctions in interface QueryRunner

createCatalog

public void createCatalog(String catalogName,
 String connectorName,
 Map<String,String> properties)

Specified by:

createCatalog in interface QueryRunner

printPlan

public LocalQueryRunner printPlan()

listTables

public List<QualifiedObjectName> listTables(Session session,
 String catalog,
 String schema)

Specified by:

listTables in interface QueryRunner

tableExists

public boolean tableExists(Session session,
 String table)

Specified by:

tableExists in interface QueryRunner

execute

public MaterializedResult execute(String sql)

Specified by:

execute in interface QueryRunner

execute

public MaterializedResult execute(Session session,
 String sql)

Specified by:

execute in interface QueryRunner

executeWithPlan

public QueryRunner.MaterializedResultWithPlan executeWithPlan(Session session,
 String sql,
 WarningCollector warningCollector)

Specified by:

executeWithPlan in interface QueryRunner

inTransaction

public <T> T inTransaction(Function<Session,T> transactionSessionConsumer)

inTransaction

public <T> T inTransaction(Session session,
 Function<Session,T> transactionSessionConsumer)

getExclusiveLock

public Lock getExclusiveLock()

Specified by:

getExclusiveLock in interface QueryRunner

injectTaskFailure

public void injectTaskFailure(String traceToken,
 int stageId,
 int partitionId,
 int attemptId,
 FailureInjector.InjectedFailureType injectionType,
 Optional<ErrorType> errorType)

Specified by:

injectTaskFailure in interface QueryRunner

createDrivers

public List<Driver> createDrivers(String sql,
 OutputFactory outputFactory,
 TaskContext taskContext)

createDrivers

public List<Driver> createDrivers(Session session,
 String sql,
 OutputFactory outputFactory,
 TaskContext taskContext)

createSubPlans

public SubPlan createSubPlans(Session session,
 Plan plan,
 boolean forceSingleNode)

createPlan

public Plan createPlan(Session session,
 String sql,
 WarningCollector warningCollector)

Specified by:

createPlan in interface QueryRunner

createPlan

public Plan createPlan(Session session,
 String sql,
 LogicalPlanner.Stage stage,
 WarningCollector warningCollector)

createPlan

public Plan createPlan(Session session,
 String sql,
 LogicalPlanner.Stage stage,
 boolean forceSingleNode,
 WarningCollector warningCollector)

getPlanOptimizers

public List<PlanOptimizer> getPlanOptimizers(boolean forceSingleNode)

createPlan

public Plan createPlan(Session session,
 String sql,
 List<PlanOptimizer> optimizers,
 WarningCollector warningCollector)

createPlan

public Plan createPlan(Session session,
 String sql,
 List<PlanOptimizer> optimizers,
 LogicalPlanner.Stage stage,
 WarningCollector warningCollector)