org.nd4j.parameterserver.status.play

Class BaseStatusStorage

java.lang.Object

org.nd4j.parameterserver.status.play.BaseStatusStorage

All Implemented Interfaces:

StatusStorage

Direct Known Subclasses:

InMemoryStatusStorage, MapDbStatusStorage

public abstract class BaseStatusStorage
extends Object
implements StatusStorage

Base status storage for storage logic and scheduling of ejection of instances indicating failure

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
protected Map<Integer,org.nd4j.parameterserver.model.SubscriberState>	statusStorageMap
protected Map<Integer,Long>	updated

Constructor Summary

Constructors

Constructor and Description
BaseStatusStorage()
BaseStatusStorage(long heartBeatEjectionMilliSeconds, long checkInterval)

Method Summary

Modifier and Type	Method and Description
abstract Map<Integer,org.nd4j.parameterserver.model.SubscriberState>	createMap() Create the storage map
abstract Map<Integer,Long>	createUpdatedMap() Create the storage map
org.nd4j.parameterserver.model.SubscriberState	getState(int id) Get the state given an id.
List<Integer>	ids() The list of state ids for the given SubscriberState
int	numStates() Returns the number of states held by this storage
void	updateState(org.nd4j.parameterserver.model.SubscriberState subscriberState) Update the state for storage

Methods inherited from class java.lang.Object

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

Field Detail

statusStorageMap

protected Map<Integer,org.nd4j.parameterserver.model.SubscriberState> statusStorageMap

updated

protected Map<Integer,Long> updated

Constructor Detail

BaseStatusStorage

public BaseStatusStorage()

BaseStatusStorage

public BaseStatusStorage(long heartBeatEjectionMilliSeconds,
                         long checkInterval)

Parameters:

heartBeatEjectionMilliSeconds - the amount of time before ejecting a given subscriber as failed

checkInterval - the interval to check for

Method Detail

ids

public List<Integer> ids()

The list of state ids for the given SubscriberState

Specified by:

ids in interface StatusStorage

Returns:

the list of ids for the given state

numStates

public int numStates()

Returns the number of states held by this storage

Specified by:

numStates in interface StatusStorage

Returns:

createUpdatedMap

public abstract Map<Integer,Long> createUpdatedMap()

Create the storage map

Returns:

createMap

public abstract Map<Integer,org.nd4j.parameterserver.model.SubscriberState> createMap()

Create the storage map

Returns:

getState

public org.nd4j.parameterserver.model.SubscriberState getState(int id)

Get the state given an id. The integer represents a stream id for a given ParameterServerSubscriber.

A SubscriberState is supposed to be 1 to 1 mapping for a stream and a MediaDriver.

Specified by:

getState in interface StatusStorage

Parameters:

id - the id of the state to get

Returns:

the subscriber state for the given id or none if it doesn't exist

updateState

public void updateState(org.nd4j.parameterserver.model.SubscriberState subscriberState)

Update the state for storage

Specified by:

updateState in interface StatusStorage

Parameters:

subscriberState - the subscriber state to update