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<?>>
static interface	DistributedQueryRunner.TestingTrinoClientFactory

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)
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)
MaterializedResult	execute(Session session, String sql)
MaterializedResult	execute(String sql)
QueryRunner.MaterializedResultWithPlan	executeWithPlan(Session session, String sql, WarningCollector warningCollector)
MaterializedResultWithQueryId	executeWithQueryId(Session session, String sql)
TestingAccessControlManager	getAccessControl()
Optional<TestingTrinoServer>	getBackupCoordinator()
TestingTrinoClient	getClient()
TestingTrinoServer	getCoordinator()
Session	getDefaultSession()
ExchangeManager	getExchangeManager()
Lock	getExclusiveLock()
FunctionManager	getFunctionManager()
TestingGroupProviderManager	getGroupProvider()
LanguageFunctionManager	getLanguageFunctionManager()
Metadata	getMetadata()
int	getNodeCount()
NodePartitioningManager	getNodePartitioningManager()
PageSourceManager	getPageSourceManager()
QueryExplainer	getQueryExplainer()
Plan	getQueryPlan(QueryId queryId)
List<TestingTrinoServer>	getServers()
SessionPropertyDefaults	getSessionPropertyDefaults()
SessionPropertyManager	getSessionPropertyManager()
List<io.opentelemetry.sdk.trace.data.SpanData>	getSpans()
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 Details

builder

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

addServers

public void addServers(int nodeCount)

getClient

public TestingTrinoClient getClient()

getSpans

public List<io.opentelemetry.sdk.trace.data.SpanData> getSpans()

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

getLanguageFunctionManager

public LanguageFunctionManager getLanguageFunctionManager()

Specified by:

getLanguageFunctionManager in interface QueryRunner

getSplitManager

public SplitManager getSplitManager()

Specified by:

getSplitManager in interface QueryRunner

getExchangeManager

public ExchangeManager getExchangeManager()

Specified by:

getExchangeManager 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 TestingGroupProviderManager 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(@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

executeWithQueryId

public MaterializedResultWithQueryId executeWithQueryId(Session session,
 @Language("SQL")
 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)

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