StateStoreRDD — RDD for Updating State (in StateStores Across Spark Cluster)

StateStoreRDD is an RDD for executing storeUpdateFunction with StateStore (and data from partitions of the data RDD).

StateStoreRDD is created for the following stateful physical operators (using StateStoreOps.mapPartitionsWithStateStore):

StateStoreRDD SparkPlans LogicalPlans operators.png
Figure 1. StateStoreRDD, Physical and Logical Plans, and operators

StateStoreRDD uses StateStoreCoordinator for the preferred locations of a partition for job scheduling.

StateStoreRDD StateStoreCoordinator.png
Figure 2. StateStoreRDD and StateStoreCoordinator

getPartitions is exactly the partitions of the data RDD.

Computing Partition — compute Method

  partition: Partition,
  ctxt: TaskContext): Iterator[U]
compute is part of the RDD Contract to compute a given partition.

compute computes dataRDD passing the result on to storeUpdateFunction (with a configured StateStore).

Internally, (and similarly to getPreferredLocations) compute creates a StateStoreProviderId with StateStoreId (using checkpointLocation, operatorId and the index of the input partition) and queryRunId.

compute then requests StateStore for the store for the StateStoreProviderId.

In the end, compute computes dataRDD (using the input partition and ctxt) followed by executing storeUpdateFunction (with the store and the result).

Placement Preferences of Partition (Preferred Locations) — getPreferredLocations Method

getPreferredLocations(partition: Partition): Seq[String]
getPreferredLocations is a part of the RDD Contract to specify placement preferences (aka preferred task locations), i.e. where tasks should be executed to be as close to the data as possible.

getPreferredLocations creates a StateStoreProviderId with StateStoreId (using checkpointLocation, operatorId and the index of the input partition) and queryRunId.

checkpointLocation and operatorId are shared across different partitions and so the only difference in StateStoreProviderIds is the partition index.

In the end, getPreferredLocations requests StateStoreCoordinatorRef for the location of the state store for the StateStoreProviderId.

StateStoreCoordinator coordinates instances of StateStores across Spark executors in the cluster, and tracks their locations for job scheduling.

Creating StateStoreRDD Instance

StateStoreRDD takes the following to be created:

  • Data RDD (RDD[T] to update the aggregates in a state store)

  • Store update function ((StateStore, Iterator[T]) ⇒ Iterator[U] where T is the type of rows in the data RDD)

  • Checkpoint directory

  • Run ID of the streaming query

  • Operator ID

  • Version of the store

  • Key schema - schema of the keys

  • Value schema - schema of the values

  • Index

  • SessionState

  • Optional StateStoreCoordinatorRef

StateStoreRDD initializes the internal properties.

Internal Properties

Name Description



Configuration parameters (as StateStoreConf) using the current SQLConf (from SessionState)

results matching ""

    No results matching ""