private void sendUpdatePartitionInfoRpcCall ( final Iterable < PartitionInfo > partitionInfos ) { final LogicalSlot slot = assignedResource ; if ( slot != null ) { final TaskManagerGateway taskManagerGateway = slot . getTaskManagerGateway ( ) ; final TaskManagerLocation taskManagerLocation = slot . getTaskManagerLocation ( ) ; CompletableFuture < Acknowledge > updatePartitionsResultFuture = taskManagerGateway . updatePartitions ( attemptId , partitionInfos , rpcTimeout ) ; updatePartitionsResultFuture . whenCompleteAsync ( ( ack , failure ) -> { // fail if there was a failure if ( failure != null ) { fail ( new IllegalStateException ( "Update task on TaskManager " + taskManagerLocation + " failed due to:" , failure ) ) ; } } , getVertex ( ) . getExecutionGraph ( ) . getJobMasterMainThreadExecutor ( ) ) ; } }
Update the partition infos on the assigned resource .