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, Map<String,String> properties)
Plan	createPlan(Session session, String sql)
MaterializedResult	execute(Session session, String sql)
QueryRunner.MaterializedResultWithPlan	executeWithPlan(Session session, String sql)
TestingAccessControlManager	getAccessControl()
TestingTrinoClient	getClient()
TestingTrinoServer	getCoordinator()
Session	getDefaultSession()
Lock	getExclusiveLock()
TestingGroupProviderManager	getGroupProvider()
int	getNodeCount()
NodePartitioningManager	getNodePartitioningManager()
PageSourceManager	getPageSourceManager()
PlannerContext	getPlannerContext()
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()
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

Methods inherited from interface io.trino.testing.QueryRunner

createCatalog, execute, inTransaction, inTransaction

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

Specified by:

getSpans in interface QueryRunner

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

getPlannerContext

public PlannerContext getPlannerContext()

Specified by:

getPlannerContext 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 TestingGroupProviderManager getGroupProvider()

Specified by:

getGroupProvider in interface QueryRunner

getSessionPropertyDefaults

public SessionPropertyDefaults getSessionPropertyDefaults()

getCoordinator

public TestingTrinoServer getCoordinator()

Specified by:

getCoordinator in interface QueryRunner

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,
 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(Session session,
 @Language("SQL")
 String sql)

Specified by:

execute in interface QueryRunner

executeWithPlan

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

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