Package org.apache.solr.cloud.overseer

Class ZkStateWriter

java.lang.Object

org.apache.solr.cloud.overseer.ZkStateWriter

public class ZkStateWriter
extends Object

ZkStateWriter is responsible for writing updates to the cluster state stored in ZooKeeper for collections each of which gets their own individual state.json in ZK.

Updates to the cluster state are specified using the enqueueUpdate(ClusterState, List, ZkWriteCallback) method. The class buffers updates to reduce the number of writes to ZK. The buffered updates are flushed during enqueueUpdate automatically if necessary. The writePendingUpdates() can be used to force flush any pending updates.

If either enqueueUpdate(ClusterState, List, ZkWriteCallback) or writePendingUpdates() throws a KeeperException.BadVersionException then the internal buffered state of the class is suspect and the current instance of the class should be discarded and a new instance should be created and used for any future updates.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	ZkStateWriter.ZkWriteCallback

Field Summary

Fields

Modifier and Type	Field	Description
protected ClusterState	clusterState
protected boolean	invalidState	Set to true if we ever get a BadVersionException so that we can disallow future operations with this instance
protected long	lastUpdatedTime
static ZkWriteCommand	NO_OP	Represents a no-op ZkWriteCommand which will result in no modification to cluster state
protected ZkStateReader	reader
protected Stats	stats
protected Map<String,ZkWriteCommand>	updates

Constructor Summary

Constructors

Constructor	Description
ZkStateWriter(ZkStateReader zkStateReader, Stats stats)

Method Summary

Modifier and Type	Method	Description
ClusterState	enqueueUpdate(ClusterState prevState, List<ZkWriteCommand> cmds, ZkStateWriter.ZkWriteCallback callback)	Applies the given ZkWriteCommand on the prevState.
ClusterState	getClusterState()
boolean	hasPendingUpdates()
ClusterState	writePendingUpdates()
ClusterState	writePendingUpdates(Map<String,ZkWriteCommand> updates)	Writes all pending updates to ZooKeeper and returns the modified cluster state
ClusterState	writeUpdate(ZkWriteCommand command)

Methods inherited from class java.lang.Object

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

Field Detail

NO_OP

public static ZkWriteCommand NO_OP

Represents a no-op ZkWriteCommand which will result in no modification to cluster state

reader

protected final ZkStateReader reader

stats

protected final Stats stats

updates

protected Map<String,ZkWriteCommand> updates

clusterState

protected ClusterState clusterState

lastUpdatedTime

protected long lastUpdatedTime

invalidState

protected boolean invalidState

Set to true if we ever get a BadVersionException so that we can disallow future operations with this instance

Constructor Detail

ZkStateWriter

public ZkStateWriter(ZkStateReader zkStateReader,
                     Stats stats)

Method Detail

enqueueUpdate

public ClusterState enqueueUpdate(ClusterState prevState,
                                  List<ZkWriteCommand> cmds,
                                  ZkStateWriter.ZkWriteCallback callback)
                           throws IllegalStateException,
                                  Exception

Applies the given ZkWriteCommand on the prevState. The modified ClusterState is returned and it is expected that the caller will use the returned cluster state for the subsequent invocation of this method.

The modified state may be buffered or flushed to ZooKeeper depending on the internal buffering logic of this class. The hasPendingUpdates() method may be used to determine if the last enqueue operation resulted in buffered state. The method writePendingUpdates() can be used to force an immediate flush of pending cluster state changes.

Parameters:

prevState - the cluster state information on which the given cmd is applied

cmds - the list of ZkWriteCommand which specifies the change to be applied to cluster state in atomic

callback - a ZkStateWriter.ZkWriteCallback object to be used for any callbacks

Returns:

modified cluster state created after applying cmd to prevState . If cmd is a no-op (NO_OP) then the prevState is returned unmodified.

Throws:

IllegalStateException - if the current instance is no longer usable. The current instance must be discarded.

Exception - on an error in ZK operations or callback. If a flush to ZooKeeper results in a KeeperException.BadVersionException this instance becomes unusable and must be discarded

hasPendingUpdates

public boolean hasPendingUpdates()

writeUpdate

public ClusterState writeUpdate(ZkWriteCommand command)
                         throws IllegalStateException,
                                org.apache.zookeeper.KeeperException,
                                InterruptedException

Throws:

IllegalStateException

org.apache.zookeeper.KeeperException

InterruptedException

writePendingUpdates

public ClusterState writePendingUpdates()
                                 throws org.apache.zookeeper.KeeperException,
                                        InterruptedException

Throws:

org.apache.zookeeper.KeeperException

InterruptedException

writePendingUpdates

public ClusterState writePendingUpdates(Map<String,ZkWriteCommand> updates)
                                 throws IllegalStateException,
                                        org.apache.zookeeper.KeeperException,
                                        InterruptedException

Writes all pending updates to ZooKeeper and returns the modified cluster state

Returns:

the modified cluster state

Throws:

IllegalStateException - if the current instance is no longer usable and must be discarded

org.apache.zookeeper.KeeperException - if any ZooKeeper operation results in an error

InterruptedException - if the current thread is interrupted

getClusterState

public ClusterState getClusterState()

Returns:

the most up-to-date cluster state until the last enqueueUpdate operation