rkhachatryan commented on code in PR #19448: URL: https://github.com/apache/flink/pull/19448#discussion_r873853860
########## flink-runtime/src/main/java/org/apache/flink/runtime/checkpoint/OperatorSubtaskState.java: ########## @@ -227,6 +229,14 @@ private static void registerSharedState( for (KeyedStateHandle stateHandle : stateHandles) { if (stateHandle != null) { stateHandle.registerSharedStates(sharedStateRegistry, checkpointID); + // Registering state handle to the given sharedStateRegistry serves two purposes: + // 1. let sharedStateRegistry be responsible for cleaning the state handle, + // 2. update the status of the checkpoint in sharedStateRegistry to which the state + // handle belongs + sharedStateRegistry.registerReference( + new SharedStateRegistryKey(stateHandle.getStateHandleId().getKeyString()), + new StreamStateHandleWrapper(stateHandle), Review Comment: (replying to https://github.com/apache/flink/pull/19448#discussion_r873445202) As per the [comment](https://github.com/apache/flink/pull/19448#discussion_r873845873) above, I think there should be a wrapper to have an empty `discardState`. So the interface may stay the same (FLINK-25862 not needed here IMO). -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: issues-unsubscr...@flink.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org