@Override public KeyedStateInputSplit[] createInputSplits(int minNumSplits) throws IOException { return operatorState.getSubtaskStates().entrySet().stream() .map(entry -> new KeyedStateInputSplit(entry.getKey(), entry.getValue())) .toArray(KeyedStateInputSplit[]::new); }
/** * Restores the OperatorStateBackends corresponding to the different * subtasks. The backends are completely restored in-memory. */ public Map<Integer, OperatorStateBackend> createOperatorStateBackendsFromSnapshot() throws Exception { return Maps.transformValues(opState.getSubtaskStates(), sst -> { try { return restoreOperatorStateBackend(sst.getManagedOperatorState()); } catch (Exception e) { throw new RuntimeException(e); } }); }
Map<Integer, OperatorSubtaskState> subtaskStateMap = operatorState.getSubtaskStates(); dos.writeInt(subtaskStateMap.size()); for (Map.Entry<Integer, OperatorSubtaskState> entry : subtaskStateMap.entrySet()) {
Map<Integer, OperatorSubtaskState> subtaskStateMap = operatorState.getSubtaskStates(); dos.writeInt(subtaskStateMap.size()); for (Map.Entry<Integer, OperatorSubtaskState> entry : subtaskStateMap.entrySet()) {
Map<Integer, OperatorSubtaskState> subtaskStateMap = operatorState.getSubtaskStates(); dos.writeInt(subtaskStateMap.size()); for (Map.Entry<Integer, OperatorSubtaskState> entry : subtaskStateMap.entrySet()) {
Map<Integer, OperatorSubtaskState> subtaskStateMap = operatorState.getSubtaskStates(); dos.writeInt(subtaskStateMap.size()); for (Map.Entry<Integer, OperatorSubtaskState> entry : subtaskStateMap.entrySet()) {
Map<Integer, OperatorSubtaskState> subtaskStateMap = operatorState.getSubtaskStates(); dos.writeInt(subtaskStateMap.size()); for (Map.Entry<Integer, OperatorSubtaskState> entry : subtaskStateMap.entrySet()) {
baseOpState.getSubtaskStates().forEach((subtaskId, subtaskState) -> { KeyedStateHandle newKeyedHandle = handleMap.get(subtaskId); StateObjectCollection<OperatorStateHandle> opHandle = transformSubtaskOpState(outDir, subtaskId,