io.prestosql.testing

Class LocalQueryRunner

java.lang.Object

io.prestosql.testing.LocalQueryRunner

All Implemented Interfaces:

QueryRunner, Closeable, AutoCloseable

public class LocalQueryRunner
extends Object
implements QueryRunner

Nested Class Summary

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

QueryRunner.MaterializedResultWithPlan

Constructor Summary

Constructors

Constructor and Description
LocalQueryRunner(Session defaultSession)
LocalQueryRunner(Session defaultSession, FeaturesConfig featuresConfig)
LocalQueryRunner(Session defaultSession, FeaturesConfig featuresConfig, NodeSpillConfig nodeSpillConfig, boolean withInitialTransaction, boolean alwaysRevokeMemory)

Method Summary

Modifier and Type	Method and Description
void	close()
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, List<PlanOptimizer> optimizers, LogicalPlanner.Stage stage, WarningCollector warningCollector)
Plan	createPlan(Session session, String sql, List<PlanOptimizer> optimizers, 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, WarningCollector warningCollector)
MaterializedResult	execute(Session session, String sql)
MaterializedResult	execute(String sql)
QueryRunner.MaterializedResultWithPlan	executeWithPlan(Session session, String sql, WarningCollector warningCollector)
TestingAccessControlManager	getAccessControl()
CostCalculator	getCostCalculator()
Session	getDefaultSession()
CostCalculator	getEstimatedExchangesCostCalculator()
Lock	getExclusiveLock()
ExecutorService	getExecutor()
ExpressionCompiler	getExpressionCompiler()
Metadata	getMetadata()
int	getNodeCount()
NodePartitioningManager	getNodePartitioningManager()
PageSourceManager	getPageSourceManager()
List<PlanOptimizer>	getPlanOptimizers(boolean forceSingleNode)
ScheduledExecutorService	getScheduler()
SplitManager	getSplitManager()
SqlParser	getSqlParser()
StatsCalculator	getStatsCalculator()
TransactionManager	getTransactionManager()
TypeRegistry	getTypeManager()
void	installPlugin(Plugin plugin)
<T> T	inTransaction(Function<Session,T> transactionSessionConsumer)
<T> T	inTransaction(Session session, Function<Session,T> transactionSessionConsumer)
List<QualifiedObjectName>	listTables(Session session, String catalog, String schema)
LocalQueryRunner	printPlan()
static LocalQueryRunner	queryRunnerWithFakeNodeCountForStats(Session defaultSession, int nodeCount)
static LocalQueryRunner	queryRunnerWithInitialTransaction(Session defaultSession)
boolean	tableExists(Session session, String table)

Methods inherited from class java.lang.Object

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

Constructor Detail

LocalQueryRunner

public LocalQueryRunner(Session defaultSession)

LocalQueryRunner

public LocalQueryRunner(Session defaultSession,
                        FeaturesConfig featuresConfig)

LocalQueryRunner

public LocalQueryRunner(Session defaultSession,
                        FeaturesConfig featuresConfig,
                        NodeSpillConfig nodeSpillConfig,
                        boolean withInitialTransaction,
                        boolean alwaysRevokeMemory)

Method Detail

queryRunnerWithInitialTransaction

public static LocalQueryRunner queryRunnerWithInitialTransaction(Session defaultSession)

queryRunnerWithFakeNodeCountForStats

public static LocalQueryRunner queryRunnerWithFakeNodeCountForStats(Session defaultSession,
                                                                    int nodeCount)

close

public void close()

Specified by:

close in interface QueryRunner

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

getNodeCount

public int getNodeCount()

Specified by:

getNodeCount in interface QueryRunner

getTypeManager

public TypeRegistry getTypeManager()

getTransactionManager

public TransactionManager getTransactionManager()

Specified by:

getTransactionManager in interface QueryRunner

getSqlParser

public SqlParser getSqlParser()

getMetadata

public Metadata getMetadata()

Specified by:

getMetadata in interface QueryRunner

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()

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

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

createDrivers

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

createDrivers

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

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)