Github user StefanRRichter commented on a diff in the pull request: https://github.com/apache/flink/pull/5239#discussion_r169896224 --- Diff: flink-runtime/src/main/java/org/apache/flink/runtime/state/CheckpointStreamWithResultProvider.java --- @@ -172,4 +167,30 @@ public CheckpointStreamWithResultProvider create( return new CheckpointStreamWithResultProvider.PrimaryStreamOnly(primaryOut); } } + + /** + * Helper method that takes a {@link SnapshotResult<StreamStateHandle>} and a {@link KeyGroupRangeOffsets} and + * creates a {@link SnapshotResult<KeyGroupsStateHandle>} by combining the key groups offsets with all the + * present stream state handles. + */ + static SnapshotResult<KeyedStateHandle> toKeyedStateHandleSnapshotResult( + @Nullable SnapshotResult<StreamStateHandle> snapshotResult, --- End diff -- Already changed in my updated branch.
---