Package io.trino.testing

Class DistributedQueryRunner

java.lang.Object

io.trino.testing.DistributedQueryRunner

All Implemented Interfaces:

QueryRunner, Closeable, AutoCloseable

public class DistributedQueryRunner
extends Object
implements QueryRunner

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	DistributedQueryRunner.Builder<SELF extends DistributedQueryRunner.Builder<?>>

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

QueryRunner.MaterializedResultWithPlan

Method Summary

Modifier and Type	Method	Description
void	addFunctions(FunctionBundle functionBundle)
void	addServers(int nodeCount)
static DistributedQueryRunner.Builder<?>	builder(Session defaultSession)
void	close()
void	createCatalog(String catalogName, String connectorName)
void	createCatalog(String catalogName, String connectorName, Map<String,String> properties)
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)
ResultWithQueryId<MaterializedResult>	executeWithQueryId(Session session, String sql)
TestingAccessControlManager	getAccessControl()
Optional<TestingTrinoServer>	getBackupCoordinator()
TestingTrinoClient	getClient()
TestingTrinoServer	getCoordinator()
Session	getDefaultSession()
Lock	getExclusiveLock()
FunctionManager	getFunctionManager()
TestingGroupProvider	getGroupProvider()
Metadata	getMetadata()
int	getNodeCount()
NodePartitioningManager	getNodePartitioningManager()
PageSourceManager	getPageSourceManager()
QueryExplainer	getQueryExplainer()
Plan	getQueryPlan(QueryId queryId)
List<TestingTrinoServer>	getServers()
SessionPropertyDefaults	getSessionPropertyDefaults()
SessionPropertyManager	getSessionPropertyManager()
SplitManager	getSplitManager()
StatsCalculator	getStatsCalculator()
TransactionManager	getTransactionManager()
TypeManager	getTypeManager()
void	injectTaskFailure(String traceToken, int stageId, int partitionId, int attemptId, FailureInjector.InjectedFailureType injectionType, Optional<ErrorType> errorType)
void	installPlugin(Plugin plugin)
List<QualifiedObjectName>	listTables(Session session, String catalog, String schema)
void	loadExchangeManager(String name, Map<String,String> properties)
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 Detail

builder

public static DistributedQueryRunner.Builder<?> builder(Session defaultSession)

addServers

public void addServers(int nodeCount)
                throws Exception

Throws:

Exception

getClient

public TestingTrinoClient getClient()

getNodeCount

public int getNodeCount()

Specified by:

getNodeCount in interface QueryRunner

getDefaultSession

public Session getDefaultSession()

Specified by:

getDefaultSession in interface QueryRunner

getTransactionManager

public TransactionManager getTransactionManager()

Specified by:

getTransactionManager in interface QueryRunner

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

getFunctionManager

public FunctionManager getFunctionManager()

Specified by:

getFunctionManager in interface QueryRunner

getSplitManager

public SplitManager getSplitManager()

Specified by:

getSplitManager in interface QueryRunner

getPageSourceManager

public PageSourceManager getPageSourceManager()

Specified by:

getPageSourceManager in interface QueryRunner

getNodePartitioningManager

public NodePartitioningManager getNodePartitioningManager()

Specified by:

getNodePartitioningManager in interface QueryRunner

getStatsCalculator

public StatsCalculator getStatsCalculator()

Specified by:

getStatsCalculator in interface QueryRunner

getAccessControl

public TestingAccessControlManager getAccessControl()

Specified by:

getAccessControl in interface QueryRunner

getGroupProvider

public TestingGroupProvider getGroupProvider()

Specified by:

getGroupProvider in interface QueryRunner

getSessionPropertyDefaults

public SessionPropertyDefaults getSessionPropertyDefaults()

getCoordinator

public TestingTrinoServer getCoordinator()

getBackupCoordinator

public Optional<TestingTrinoServer> getBackupCoordinator()

getServers

public List<TestingTrinoServer> getServers()

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)

createCatalog

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

Specified by:

createCatalog in interface QueryRunner

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

executeWithQueryId

public ResultWithQueryId<MaterializedResult> executeWithQueryId(Session session,
                                                                String sql)

executeWithPlan

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

Specified by:

executeWithPlan in interface QueryRunner

createPlan

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

Specified by:

createPlan in interface QueryRunner

getQueryPlan

public Plan getQueryPlan(QueryId queryId)

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

loadExchangeManager

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

Specified by:

loadExchangeManager in interface QueryRunner

close

public final void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable

Specified by:

close in interface QueryRunner