org.apache.flink.runtime.query

Class KvStateRegistry

java.lang.Object

org.apache.flink.runtime.query.KvStateRegistry

public class KvStateRegistry
extends Object

A registry for InternalKvState instances per task manager.

This is currently only used for KvState queries: KvState instances, which are marked as queryable in their state descriptor are registered here and can be queried by the KvStateServer.

KvState is registered when it is created/restored and unregistered when the owning operator stops running.

Constructor Summary

Constructors

Constructor and Description
KvStateRegistry()

Method Summary

Methods

Modifier and Type	Method and Description
TaskKvStateRegistry	createTaskRegistry(JobID jobId, JobVertexID jobVertexId) Creates a TaskKvStateRegistry facade for the Task identified by the given JobID and JobVertexID instance.
InternalKvState<?>	getKvState(KvStateID kvStateId) Returns the KvState instance identified by the given KvStateID or null if none is registered.
KvStateID	registerKvState(JobID jobId, JobVertexID jobVertexId, KeyGroupRange keyGroupRange, String registrationName, InternalKvState<?> kvState) Registers the KvState instance and returns the assigned ID.
void	registerListener(KvStateRegistryListener listener) Registers a listener with the registry.
void	unregisterKvState(JobID jobId, JobVertexID jobVertexId, KeyGroupRange keyGroupRange, String registrationName, KvStateID kvStateId) Unregisters the KvState instance identified by the given KvStateID.
void	unregisterListener() Unregisters the listener with the registry

Methods inherited from class java.lang.Object

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

Constructor Detail

KvStateRegistry

public KvStateRegistry()

Method Detail

registerListener

public void registerListener(KvStateRegistryListener listener)

Registers a listener with the registry.

Parameters:

listener - The registry listener.

Throws:

IllegalStateException - If there is a registered listener

unregisterListener

public void unregisterListener()

Unregisters the listener with the registry

registerKvState

public KvStateID registerKvState(JobID jobId,
                        JobVertexID jobVertexId,
                        KeyGroupRange keyGroupRange,
                        String registrationName,
                        InternalKvState<?> kvState)

Registers the KvState instance and returns the assigned ID.

Parameters:

jobId - JobId the KvState instance belongs to

jobVertexId - JobVertexID the KvState instance belongs to

keyGroupRange - Key group range the KvState instance belongs to

registrationName - Name under which the KvState is registered

kvState - KvState instance to be registered

Returns:

Assigned KvStateID

unregisterKvState

public void unregisterKvState(JobID jobId,
                     JobVertexID jobVertexId,
                     KeyGroupRange keyGroupRange,
                     String registrationName,
                     KvStateID kvStateId)

Unregisters the KvState instance identified by the given KvStateID.

Parameters:

jobId - JobId the KvState instance belongs to

kvStateId - KvStateID to identify the KvState instance

keyGroupRange - Key group range the KvState instance belongs to

getKvState

public InternalKvState<?> getKvState(KvStateID kvStateId)

Returns the KvState instance identified by the given KvStateID or null if none is registered.

Parameters:

kvStateId - KvStateID to identify the KvState instance

Returns:

KvState instance identified by the KvStateID or null

createTaskRegistry

public TaskKvStateRegistry createTaskRegistry(JobID jobId,
                                     JobVertexID jobVertexId)

Creates a TaskKvStateRegistry facade for the Task identified by the given JobID and JobVertexID instance.

Parameters:

jobId - JobID of the task

jobVertexId - JobVertexID of the task

Returns:

A TaskKvStateRegistry facade for the task