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

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

QueryRunner.MaterializedResultWithPlan

Method Summary

Modifier and Type	Method	Description
void	addFunctions(FunctionBundle functionBundle)
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)
Plan	createPlan(Session session, String sql, List<PlanOptimizer> optimizers, LogicalPlanner.Stage stage, WarningCollector warningCollector, PlanOptimizersStatsCollector planOptimizersStatsCollector)
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()
AnalyzePropertyManager	getAnalyzePropertyManager()
BlockTypeOperators	getBlockTypeOperators()
CatalogHandle	getCatalogHandle(String catalogName)
CatalogManager	getCatalogManager()
ColumnPropertyManager	getColumnPropertyManager()
Connector	getConnector(String catalogName)
CostCalculator	getCostCalculator()
Session	getDefaultSession()
CostCalculator	getEstimatedExchangesCostCalculator()
ExchangeManager	getExchangeManager()
Lock	getExclusiveLock()
ExecutorService	getExecutor()
ExpressionCompiler	getExpressionCompiler()
FunctionManager	getFunctionManager()
GlobalFunctionCatalog	getGlobalFunctionCatalog()
TestingGroupProviderManager	getGroupProvider()
LanguageFunctionManager	getLanguageFunctionManager()
MaterializedViewPropertyManager	getMaterializedViewPropertyManager()
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()
TableHandle	getTableHandle(String catalogName, String schemaName, String tableName)
TablePropertyManager	getTablePropertyManager()
TaskCountEstimator	getTaskCountEstimator()
TransactionManager	getTransactionManager()
TypeManager	getTypeManager()
TypeOperators	getTypeOperators()
TypeRegistry	getTypeRegistry()
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()
void	loadExchangeManager(String name, Map<String,String> properties)
LocalQueryRunner	printPlan()
void	registerCatalogFactory(ConnectorFactory connectorFactory)
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

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

getTablePropertyManager

public TablePropertyManager getTablePropertyManager()

getColumnPropertyManager

public ColumnPropertyManager getColumnPropertyManager()

getMaterializedViewPropertyManager

public MaterializedViewPropertyManager getMaterializedViewPropertyManager()

getAnalyzePropertyManager

public AnalyzePropertyManager getAnalyzePropertyManager()

getTypeRegistry

public TypeRegistry getTypeRegistry()

getTypeManager

public TypeManager getTypeManager()

Specified by:

getTypeManager in interface QueryRunner

getGlobalFunctionCatalog

public GlobalFunctionCatalog getGlobalFunctionCatalog()

getQueryExplainer

public QueryExplainer getQueryExplainer()

Specified by:

getQueryExplainer in interface QueryRunner

getSessionPropertyManager

public SessionPropertyManager getSessionPropertyManager()

Specified by:

getSessionPropertyManager in interface QueryRunner

getFunctionManager

public FunctionManager getFunctionManager()

Specified by:

getFunctionManager in interface QueryRunner

getLanguageFunctionManager

public LanguageFunctionManager getLanguageFunctionManager()

Specified by:

getLanguageFunctionManager 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

getExchangeManager

public ExchangeManager getExchangeManager()

Specified by:

getExchangeManager in interface QueryRunner

getStatsCalculator

public StatsCalculator getStatsCalculator()

Specified by:

getStatsCalculator in interface QueryRunner

getCostCalculator

public CostCalculator getCostCalculator()

getEstimatedExchangesCostCalculator

public CostCalculator getEstimatedExchangesCostCalculator()

getTaskCountEstimator

public TaskCountEstimator getTaskCountEstimator()

getGroupProvider

public TestingGroupProviderManager 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)

registerCatalogFactory

public void registerCatalogFactory(ConnectorFactory connectorFactory)

installPlugin

public void installPlugin(Plugin plugin)

Specified by:

installPlugin in interface QueryRunner

addFunctions

public void addFunctions(FunctionBundle functionBundle)

Specified by:

addFunctions in interface QueryRunner

createCatalog

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

Specified by:

createCatalog in interface QueryRunner

getCatalogManager

public CatalogManager getCatalogManager()

getConnector

public Connector getConnector(String catalogName)

printPlan

public LocalQueryRunner printPlan()

getCatalogHandle

public CatalogHandle getCatalogHandle(String catalogName)

getTableHandle

public TableHandle getTableHandle(String catalogName,
 String schemaName,
 String tableName)

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(@Language("SQL")
 String sql)

Specified by:

execute in interface QueryRunner

execute

public MaterializedResult execute(Session session,
 @Language("SQL")
 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(@Language("SQL")
 String sql,
 OutputFactory outputFactory,
 TaskContext taskContext)

loadExchangeManager

public void loadExchangeManager(String name,
 Map<String,String> properties)

Specified by:

loadExchangeManager in interface QueryRunner

createDrivers

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

createSubPlans

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

createPlan

public Plan createPlan(Session session,
 @Language("SQL")
 String sql)

Specified by:

createPlan in interface QueryRunner

getPlanOptimizers

public List<PlanOptimizer> getPlanOptimizers(boolean forceSingleNode)

createPlan

public Plan createPlan(Session session,
 @Language("SQL")
 String sql,
 List<PlanOptimizer> optimizers,
 LogicalPlanner.Stage stage,
 WarningCollector warningCollector,
 PlanOptimizersStatsCollector planOptimizersStatsCollector)