io.ray.streaming.runtime.worker.context

Class StreamingRuntimeContext

java.lang.Object

io.ray.streaming.runtime.worker.context.StreamingRuntimeContext

All Implemented Interfaces:

io.ray.streaming.api.context.RuntimeContext

public class StreamingRuntimeContext
extends Object
implements io.ray.streaming.api.context.RuntimeContext

Use Ray to implement RuntimeContext.

Field Summary

Fields

Modifier and Type	Field and Description
protected io.ray.streaming.state.backend.KeyStateBackend	keyStateBackend Backend for keyed state.
protected io.ray.streaming.state.backend.OperatorStateBackend	operatorStateBackend Backend for operator state.

Constructor Summary

Constructors

Constructor and Description
StreamingRuntimeContext(ExecutionVertex executionVertex, Map<String,String> config, int parallelism)

Method Summary

Modifier and Type	Method and Description
Long	getCheckpointId()
Map<String,String>	getConfig()
Map<String,String>	getJobConfig()
io.ray.streaming.state.backend.KeyStateBackend	getKeyStateBackend()
<T> io.ray.streaming.state.keystate.state.ListState<T>	getListState(io.ray.streaming.state.keystate.desc.ListStateDescriptor<T> stateDescriptor)
<S,T> io.ray.streaming.state.keystate.state.MapState<S,T>	getMapState(io.ray.streaming.state.keystate.desc.MapStateDescriptor<S,T> stateDescriptor)
int	getParallelism()
int	getTaskId()
int	getTaskIndex()
<T> io.ray.streaming.state.keystate.state.ValueState<T>	getValueState(io.ray.streaming.state.keystate.desc.ValueStateDescriptor<T> stateDescriptor)
void	setCheckpointId(long checkpointId)
void	setCurrentKey(Object key)
void	setKeyStateBackend(io.ray.streaming.state.backend.KeyStateBackend keyStateBackend)
protected void	stateSanityCheck(io.ray.streaming.state.keystate.desc.AbstractStateDescriptor stateDescriptor, io.ray.streaming.state.backend.AbstractKeyStateBackend backend)

Methods inherited from class java.lang.Object

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

Field Detail

keyStateBackend

protected transient io.ray.streaming.state.backend.KeyStateBackend keyStateBackend

Backend for keyed state. This might be empty if we're not on a keyed stream.

operatorStateBackend

protected transient io.ray.streaming.state.backend.OperatorStateBackend operatorStateBackend

Backend for operator state. This might be empty

Constructor Detail

StreamingRuntimeContext

public StreamingRuntimeContext(ExecutionVertex executionVertex,
                               Map<String,String> config,
                               int parallelism)

Method Detail

getTaskId

public int getTaskId()

Specified by:

getTaskId in interface io.ray.streaming.api.context.RuntimeContext

getTaskIndex

public int getTaskIndex()

Specified by:

getTaskIndex in interface io.ray.streaming.api.context.RuntimeContext

getParallelism

public int getParallelism()

Specified by:

getParallelism in interface io.ray.streaming.api.context.RuntimeContext

getConfig

public Map<String,String> getConfig()

Specified by:

getConfig in interface io.ray.streaming.api.context.RuntimeContext

getJobConfig

public Map<String,String> getJobConfig()

Specified by:

getJobConfig in interface io.ray.streaming.api.context.RuntimeContext

getCheckpointId

public Long getCheckpointId()

Specified by:

getCheckpointId in interface io.ray.streaming.api.context.RuntimeContext

setCheckpointId

public void setCheckpointId(long checkpointId)

Specified by:

setCheckpointId in interface io.ray.streaming.api.context.RuntimeContext

setCurrentKey

public void setCurrentKey(Object key)

Specified by:

setCurrentKey in interface io.ray.streaming.api.context.RuntimeContext

getKeyStateBackend

public io.ray.streaming.state.backend.KeyStateBackend getKeyStateBackend()

Specified by:

getKeyStateBackend in interface io.ray.streaming.api.context.RuntimeContext

setKeyStateBackend

public void setKeyStateBackend(io.ray.streaming.state.backend.KeyStateBackend keyStateBackend)

Specified by:

setKeyStateBackend in interface io.ray.streaming.api.context.RuntimeContext

getValueState

public <T> io.ray.streaming.state.keystate.state.ValueState<T> getValueState(io.ray.streaming.state.keystate.desc.ValueStateDescriptor<T> stateDescriptor)

Specified by:

getValueState in interface io.ray.streaming.api.context.RuntimeContext

getListState

public <T> io.ray.streaming.state.keystate.state.ListState<T> getListState(io.ray.streaming.state.keystate.desc.ListStateDescriptor<T> stateDescriptor)

Specified by:

getListState in interface io.ray.streaming.api.context.RuntimeContext

getMapState

public <S,T> io.ray.streaming.state.keystate.state.MapState<S,T> getMapState(io.ray.streaming.state.keystate.desc.MapStateDescriptor<S,T> stateDescriptor)

Specified by:

getMapState in interface io.ray.streaming.api.context.RuntimeContext

stateSanityCheck

protected void stateSanityCheck(io.ray.streaming.state.keystate.desc.AbstractStateDescriptor stateDescriptor,
                                io.ray.streaming.state.backend.AbstractKeyStateBackend backend)