backtype.storm.task
Class TopologyContext

java.lang.Object
  backtype.storm.task.GeneralTopologyContext
      backtype.storm.task.WorkerTopologyContext
          backtype.storm.task.TopologyContext

public class TopologyContext
extends WorkerTopologyContext

A TopologyContext is given to bolts and spouts in their "prepare" and "open" methods, respectively. This object provides information about the component's place within the topology, such as task ids, inputs and outputs, etc.

The TopologyContext is also used to declare ISubscribedState objects to synchronize state with StateSpouts this object is subscribed to.

Field Summary

Fields inherited from class backtype.storm.task.WorkerTopologyContext

SHARED_EXECUTOR

Fields inherited from class backtype.storm.task.GeneralTopologyContext

_stormConf

Constructor Summary

TopologyContext(StormTopology topology, java.util.Map stormConf, java.util.Map<java.lang.Integer,java.lang.String> taskToComponent, java.util.Map<java.lang.String,java.util.List<java.lang.Integer>> componentToSortedTasks, java.util.Map<java.lang.String,java.util.Map<java.lang.String,Fields>> componentToStreamToFields, java.lang.String stormId, java.lang.String codeDir, java.lang.String pidDir, java.lang.Integer taskId, java.lang.Integer workerPort, java.util.List<java.lang.Integer> workerTasks, java.util.Map<java.lang.String,java.lang.Object> defaultResources, java.util.Map<java.lang.String,java.lang.Object> userResources, java.util.Map<java.lang.String,java.lang.Object> executorData)

Method Summary

void	addTaskHook(ITaskHook hook)
java.lang.Object	getExecutorData(java.lang.String name)
java.util.Collection<ITaskHook>	getHooks()
java.lang.Object	getTaskData(java.lang.String name)
java.lang.String	getThisComponentId() Gets the component id for this task.
Fields	getThisOutputFields(java.lang.String streamId) Gets the declared output fields for the specified stream id for the component this task is a part of.
java.util.Map<GlobalStreamId,Grouping>	getThisSources() Gets the declared inputs to this component.
java.util.Set<java.lang.String>	getThisStreams() Gets the set of streams declared for the component of this task.
java.util.Map<java.lang.String,java.util.Map<java.lang.String,Grouping>>	getThisTargets() Gets information about who is consuming the outputs of this component, and how.
int	getThisTaskId() Gets the task id of this task.
int	getThisTaskIndex() Gets the index of this task id in getComponentTasks(getThisComponentId()).
<T extends ISubscribedState> T	setAllSubscribedState(T obj) All state from all subscribed state spouts streams will be synced with the provided object.
void	setExecutorData(java.lang.String name, java.lang.Object data)
<T extends ISubscribedState> T	setSubscribedState(java.lang.String componentId, java.lang.String streamId, T obj) Synchronizes the specified stream from the specified state spout component id with the provided ISubscribedState object.
<T extends ISubscribedState> T	setSubscribedState(java.lang.String componentId, T obj) Synchronizes the default stream from the specified state spout component id with the provided ISubscribedState object.
void	setTaskData(java.lang.String name, java.lang.Object data)

Methods inherited from class backtype.storm.task.WorkerTopologyContext

getCodeDir, getPIDDir, getResource, getSharedExecutor, getThisWorkerPort, getThisWorkerTasks

Methods inherited from class backtype.storm.task.GeneralTopologyContext

getComponentCommon, getComponentId, getComponentIds, getComponentOutputFields, getComponentOutputFields, getComponentStreams, getComponentTasks, getRawTopology, getSources, getStormId, getTargets, getTaskToComponent, maxTopologyMessageTimeout, toJSONString

Methods inherited from class java.lang.Object

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

Constructor Detail

TopologyContext

public TopologyContext(StormTopology topology,
                       java.util.Map stormConf,
                       java.util.Map<java.lang.Integer,java.lang.String> taskToComponent,
                       java.util.Map<java.lang.String,java.util.List<java.lang.Integer>> componentToSortedTasks,
                       java.util.Map<java.lang.String,java.util.Map<java.lang.String,Fields>> componentToStreamToFields,
                       java.lang.String stormId,
                       java.lang.String codeDir,
                       java.lang.String pidDir,
                       java.lang.Integer taskId,
                       java.lang.Integer workerPort,
                       java.util.List<java.lang.Integer> workerTasks,
                       java.util.Map<java.lang.String,java.lang.Object> defaultResources,
                       java.util.Map<java.lang.String,java.lang.Object> userResources,
                       java.util.Map<java.lang.String,java.lang.Object> executorData)

Method Detail

setAllSubscribedState

public <T extends ISubscribedState> T setAllSubscribedState(T obj)

All state from all subscribed state spouts streams will be synced with the provided object.

It is recommended that your ISubscribedState object is kept as an instance variable of this object. The recommended usage of this method is as follows:

_myState = context.setAllSubscribedState(new MyState());

Parameters:

obj - Provided ISubscribedState implementation

Returns:

Returns the ISubscribedState object provided

setSubscribedState

public <T extends ISubscribedState> T setSubscribedState(java.lang.String componentId,
                                                         T obj)

Synchronizes the default stream from the specified state spout component id with the provided ISubscribedState object.

The recommended usage of this method is as follows:

_myState = context.setSubscribedState(componentId, new MyState());

Parameters:

componentId - the id of the StateSpout component to subscribe to

obj - Provided ISubscribedState implementation

Returns:

Returns the ISubscribedState object provided

setSubscribedState

public <T extends ISubscribedState> T setSubscribedState(java.lang.String componentId,
                                                         java.lang.String streamId,
                                                         T obj)

Synchronizes the specified stream from the specified state spout component id with the provided ISubscribedState object.

The recommended usage of this method is as follows:

_myState = context.setSubscribedState(componentId, streamId, new MyState());

Parameters:

componentId - the id of the StateSpout component to subscribe to

streamId - the stream to subscribe to

obj - Provided ISubscribedState implementation

Returns:

Returns the ISubscribedState object provided

getThisTaskId

public int getThisTaskId()

Gets the task id of this task.

Returns:

the task id

getThisComponentId

public java.lang.String getThisComponentId()

Gets the component id for this task. The component id maps to a component id specified for a Spout or Bolt in the topology definition.

Returns:

getThisOutputFields

public Fields getThisOutputFields(java.lang.String streamId)

Gets the declared output fields for the specified stream id for the component this task is a part of.

getThisStreams

public java.util.Set<java.lang.String> getThisStreams()

Gets the set of streams declared for the component of this task.

getThisTaskIndex

public int getThisTaskIndex()

Gets the index of this task id in getComponentTasks(getThisComponentId()). An example use case for this method is determining which task accesses which resource in a distributed resource to ensure an even distribution.

getThisSources

public java.util.Map<GlobalStreamId,Grouping> getThisSources()

Gets the declared inputs to this component.

Returns:

A map from subscribed component/stream to the grouping subscribed with.

getThisTargets

public java.util.Map<java.lang.String,java.util.Map<java.lang.String,Grouping>> getThisTargets()

Gets information about who is consuming the outputs of this component, and how.

Returns:

Map from stream id to component id to the Grouping used.

setTaskData

public void setTaskData(java.lang.String name,
                        java.lang.Object data)

getTaskData

public java.lang.Object getTaskData(java.lang.String name)

setExecutorData

public void setExecutorData(java.lang.String name,
                            java.lang.Object data)

getExecutorData

public java.lang.Object getExecutorData(java.lang.String name)

addTaskHook

public void addTaskHook(ITaskHook hook)

getHooks

public java.util.Collection<ITaskHook> getHooks()