Examples with RpcKvStateRegistryListener org.apache.flink.runtime.taskexecutor.rpc.RpcKvStateRegistryListener used on opensource projects

Example 1 with RpcKvStateRegistryListener

use of org.apache.flink.runtime.taskexecutor.rpc.RpcKvStateRegistryListener in project flink by apache.

the class TaskExecutor method registerQueryableState.

private void registerQueryableState(JobID jobId, JobMasterGateway jobMasterGateway) {
    final KvStateServer kvStateServer = kvStateService.getKvStateServer();
    final KvStateRegistry kvStateRegistry = kvStateService.getKvStateRegistry();
    if (kvStateServer != null && kvStateRegistry != null) {
        kvStateRegistry.registerListener(jobId, new RpcKvStateRegistryListener(jobMasterGateway, kvStateServer.getServerAddress()));
    }
    final KvStateClientProxy kvStateProxy = kvStateService.getKvStateClientProxy();
    if (kvStateProxy != null) {
        kvStateProxy.updateKvStateLocationOracle(jobId, jobMasterGateway);
    }
}