Package io.trino.server.testing

Class TestingTrinoServer

java.lang.Object

io.trino.server.testing.TestingTrinoServer

All Implemented Interfaces:

Closeable, AutoCloseable

public class TestingTrinoServer
extends Object
implements Closeable

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	TestingTrinoServer.Builder
static class	TestingTrinoServer.TestShutdownAction

Field Summary

Fields

Modifier and Type	Field	Description
static final String	SESSION_START_TIME_PROPERTY

Method Summary

Modifier and Type	Method	Description
void	addConnectorEventListeners()	Add the event listeners from connectors.
void	addFinalQueryInfoListener(QueryId queryId, StateMachine.StateChangeListener<QueryInfo> stateChangeListener)
void	addFunctions(FunctionBundle functionBundle)
static TestingTrinoServer.Builder	builder()
void	close()
static TestingTrinoServer	create()
void	createCatalog(String catalogName, String connectorName)
void	createCatalog(String catalogName, String connectorName, Map<String,String> properties)
TestingAccessControlManager	getAccessControl()
com.google.common.net.HostAndPort	getAddress()
Path	getBaseDataDir()
URI	getBaseUrl()
ClusterMemoryManager	getClusterMemoryManager()
Connector	getConnector(String catalogName)
DispatchManager	getDispatchManager()
QueryInfo	getFullQueryInfo(QueryId queryId)
GracefulShutdownHandler	getGracefulShutdownHandler()
TestingGroupProviderManager	getGroupProvider()
com.google.common.net.HostAndPort	getHttpsAddress()
URI	getHttpsBaseUrl()
<T> T	getInstance(com.google.inject.Key<T> key)
LocalMemoryManager	getLocalMemoryManager()
MBeanServer	getMbeanServer()
NodePartitioningManager	getNodePartitioningManager()
PageSourceManager	getPageSourceManager()
PlannerContext	getPlannerContext()
ProcedureTester	getProcedureTester()
QueryExplainer	getQueryExplainer()
QueryManager	getQueryManager()
Optional<Plan>	getQueryPlan(QueryId queryId)
Optional<InternalResourceGroupManager<?>>	getResourceGroupManager()
SessionPropertyDefaults	getSessionPropertyDefaults()
SessionPropertyManager	getSessionPropertyManager()
ShutdownAction	getShutdownAction()
SplitManager	getSplitManager()
StatsCalculator	getStatsCalculator()
TablePropertyManager	getTablePropertyManager()
SqlTaskManager	getTaskManager()
TransactionManager	getTransactionManager()
void	injectTaskFailure(String traceToken, int stageId, int partitionId, int attemptId, FailureInjector.InjectedFailureType injectionType, Optional<ErrorType> errorType)
void	installPlugin(Plugin plugin)
boolean	isCoordinator()
void	loadExchangeManager(String name, Map<String,String> properties)
final AllNodes	refreshNodes()
URI	resolve(String path)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

SESSION_START_TIME_PROPERTY

public static final String SESSION_START_TIME_PROPERTY

See Also:

• Constant Field Values

Method Details

create

public static TestingTrinoServer create()

builder

public static TestingTrinoServer.Builder builder()

close

public void close()
           throws IOException

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable

Throws:

IOException

installPlugin

public void installPlugin(Plugin plugin)

getDispatchManager

public DispatchManager getDispatchManager()

getQueryManager

public QueryManager getQueryManager()

getQueryPlan

public Optional<Plan> getQueryPlan(QueryId queryId)

getFullQueryInfo

public QueryInfo getFullQueryInfo(QueryId queryId)

addFinalQueryInfoListener

public void addFinalQueryInfoListener(QueryId queryId,
 StateMachine.StateChangeListener<QueryInfo> stateChangeListener)

createCatalog

public void createCatalog(String catalogName,
 String connectorName)

createCatalog

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

loadExchangeManager

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

addConnectorEventListeners

public void addConnectorEventListeners()

Add the event listeners from connectors. Connector event listeners are only supported for statically loaded catalogs, and this doesn't match up with the model of the testing Trino server. This method should only be called once after all catalogs are added.

getBaseDataDir

public Path getBaseDataDir()

getBaseUrl

public URI getBaseUrl()

getHttpsBaseUrl

public URI getHttpsBaseUrl()

resolve

public URI resolve(String path)

getAddress

public com.google.common.net.HostAndPort getAddress()

getHttpsAddress

public com.google.common.net.HostAndPort getHttpsAddress()

getTransactionManager

public TransactionManager getTransactionManager()

getTablePropertyManager

public TablePropertyManager getTablePropertyManager()

getPlannerContext

public PlannerContext getPlannerContext()

getQueryExplainer

public QueryExplainer getQueryExplainer()

getSessionPropertyManager

public SessionPropertyManager getSessionPropertyManager()

addFunctions

public void addFunctions(FunctionBundle functionBundle)

getStatsCalculator

public StatsCalculator getStatsCalculator()

getAccessControl

public TestingAccessControlManager getAccessControl()

getGroupProvider

public TestingGroupProviderManager getGroupProvider()

getProcedureTester

public ProcedureTester getProcedureTester()

getSplitManager

public SplitManager getSplitManager()

getPageSourceManager

public PageSourceManager getPageSourceManager()

getResourceGroupManager

public Optional<InternalResourceGroupManager<?>> getResourceGroupManager()

getSessionPropertyDefaults

public SessionPropertyDefaults getSessionPropertyDefaults()

getNodePartitioningManager

public NodePartitioningManager getNodePartitioningManager()

getLocalMemoryManager

public LocalMemoryManager getLocalMemoryManager()

getClusterMemoryManager

public ClusterMemoryManager getClusterMemoryManager()

getMbeanServer

public MBeanServer getMbeanServer()

getGracefulShutdownHandler

public GracefulShutdownHandler getGracefulShutdownHandler()

getTaskManager

public SqlTaskManager getTaskManager()

getShutdownAction

public ShutdownAction getShutdownAction()

getConnector

public Connector getConnector(String catalogName)

isCoordinator

public boolean isCoordinator()

refreshNodes

public final AllNodes refreshNodes()

getInstance

public <T> T getInstance(com.google.inject.Key<T> key)

injectTaskFailure

public void injectTaskFailure(String traceToken,
 int stageId,
 int partitionId,
 int attemptId,
 FailureInjector.InjectedFailureType injectionType,
 Optional<ErrorType> errorType)