Search in sources :

Example 21 with KvStateID

use of org.apache.flink.queryablestate.KvStateID in project flink by apache.

the class TaskKvStateRegistry method registerKvState.

/**
 * Registers the KvState instance at the KvStateRegistry.
 *
 * @param keyGroupRange Key group range the KvState instance belongs to
 * @param registrationName The registration name (not necessarily the same as the KvState name
 *     defined in the state descriptor used to create the KvState instance)
 * @param kvState The
 */
public void registerKvState(KeyGroupRange keyGroupRange, String registrationName, InternalKvState<?, ?, ?> kvState, ClassLoader userClassLoader) {
    KvStateID kvStateId = registry.registerKvState(jobId, jobVertexId, keyGroupRange, registrationName, kvState, userClassLoader);
    registeredKvStates.add(new KvStateInfo(keyGroupRange, registrationName, kvStateId));
}
Also used : KvStateID(org.apache.flink.queryablestate.KvStateID)

Aggregations

KvStateID (org.apache.flink.queryablestate.KvStateID)21 Test (org.junit.Test)20 InetSocketAddress (java.net.InetSocketAddress)11 KvStateInternalRequest (org.apache.flink.queryablestate.messages.KvStateInternalRequest)11 KvStateResponse (org.apache.flink.queryablestate.messages.KvStateResponse)11 MessageSerializer (org.apache.flink.queryablestate.network.messages.MessageSerializer)11 KeyGroupRange (org.apache.flink.runtime.state.KeyGroupRange)11 JobID (org.apache.flink.api.common.JobID)9 AtomicKvStateRequestStats (org.apache.flink.queryablestate.network.stats.AtomicKvStateRequestStats)9 JobVertexID (org.apache.flink.runtime.jobgraph.JobVertexID)9 ByteBuf (org.apache.flink.shaded.netty4.io.netty.buffer.ByteBuf)8 UnknownHostException (java.net.UnknownHostException)7 ExecutionException (java.util.concurrent.ExecutionException)7 ConnectException (java.net.ConnectException)6 ClosedChannelException (java.nio.channels.ClosedChannelException)6 ArrayList (java.util.ArrayList)5 CompletableFuture (java.util.concurrent.CompletableFuture)4 AtomicReference (java.util.concurrent.atomic.AtomicReference)4 KvStateRegistry (org.apache.flink.runtime.query.KvStateRegistry)4 Channel (org.apache.flink.shaded.netty4.io.netty.channel.Channel)4