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(List<? extends SqlFunction> functions)
void	addServers(int nodeCount)
static DistributedQueryRunner.Builder<?>	builder(Session defaultSession)
final 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()
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)
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

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

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(List<? extends SqlFunction> functions)

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

close

public final void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable

Specified by:

close in interface QueryRunner