com.facebook.presto.testing

Class LocalQueryRunner

java.lang.Object

com.facebook.presto.testing.LocalQueryRunner

All Implemented Interfaces:

QueryRunner, Closeable, AutoCloseable

public class LocalQueryRunner
extends Object
implements QueryRunner

Constructor Summary

Constructors

Constructor and Description
LocalQueryRunner(Session defaultSession)
LocalQueryRunner(Session defaultSession, boolean alwaysRevokeMemory)
LocalQueryRunner(Session defaultSession, FeaturesConfig featuresConfig)
LocalQueryRunner(Session defaultSession, FeaturesConfig featuresConfig, boolean withInitialTransaction, boolean alwaysRevokeMemory)
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)
OperatorFactory	createHashProjectOperator(int operatorId, PlanNodeId planNodeId, List<Type> columnTypes)
Plan	createPlan(Session session, String sql)
Plan	createPlan(Session session, String sql, List<PlanOptimizer> optimizers)
Plan	createPlan(Session session, String sql, List<PlanOptimizer> optimizers, LogicalPlanner.Stage stage)
Plan	createPlan(Session session, String sql, LogicalPlanner.Stage stage)
Plan	createPlan(Session session, String sql, LogicalPlanner.Stage stage, boolean forceSingleNode)
OperatorFactory	createTableScanOperator(Session session, int operatorId, PlanNodeId planNodeId, String tableName, String... columnNames)
MaterializedResult	execute(Session session, String sql)
MaterializedResult	execute(String sql)
TestingAccessControlManager	getAccessControl()
CostCalculator	getCostCalculator()
Session	getDefaultSession()
Lock	getExclusiveLock()
ExecutorService	getExecutor()
ExpressionCompiler	getExpressionCompiler()
Metadata	getMetadata()
int	getNodeCount()
List<PlanOptimizer>	getPlanOptimizers(boolean forceSingleNode)
ScheduledExecutorService	getScheduler()
TransactionManager	getTransactionManager()
TypeRegistry	getTypeManager()
void	installPlugin(Plugin plugin)
<T> T	inTransaction(java.util.function.Function<Session,T> transactionSessionConsumer)
<T> T	inTransaction(Session session, java.util.function.Function<Session,T> transactionSessionConsumer)
List<QualifiedObjectName>	listTables(Session session, String catalog, String schema)
LocalQueryRunner	printPlan()
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,
                        boolean alwaysRevokeMemory)

LocalQueryRunner

public LocalQueryRunner(Session defaultSession,
                        FeaturesConfig featuresConfig)

LocalQueryRunner

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

LocalQueryRunner

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

Method Detail

queryRunnerWithInitialTransaction

public static LocalQueryRunner queryRunnerWithInitialTransaction(Session defaultSession)

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

getMetadata

public Metadata getMetadata()

Specified by:

getMetadata in interface QueryRunner

getCostCalculator

public CostCalculator getCostCalculator()

Specified by:

getCostCalculator 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

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

inTransaction

public <T> T inTransaction(java.util.function.Function<Session,T> transactionSessionConsumer)

inTransaction

public <T> T inTransaction(Session session,
                           java.util.function.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)

createPlan

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

createPlan

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

getPlanOptimizers

public List<PlanOptimizer> getPlanOptimizers(boolean forceSingleNode)

createPlan

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

createPlan

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

createTableScanOperator

public OperatorFactory createTableScanOperator(Session session,
                                               int operatorId,
                                               PlanNodeId planNodeId,
                                               String tableName,
                                               String... columnNames)

createHashProjectOperator

public OperatorFactory createHashProjectOperator(int operatorId,
                                                 PlanNodeId planNodeId,
                                                 List<Type> columnTypes)