1996fanrui commented on code in PR #28856:
URL: https://github.com/apache/flink/pull/28856#discussion_r3693232707
##########
flink-runtime/src/main/java/org/apache/flink/runtime/checkpoint/TaskStateAssignment.java:
##########
@@ -272,10 +266,13 @@ public OperatorSubtaskState
getSubtaskState(OperatorInstanceID instanceID) {
instanceID,
inputOperatorID,
getUpstreamAssignments(),
- (assignment, recompute) -> {
+ (gateIndex, assignment, recompute) -> {
int assignmentIndex =
- getAssignmentIndex(
-
assignment.getDownstreamAssignments(), this);
+
assignment.findResultPartitionIndex(
+ executionJobVertex
+ .getInputs()
+ .get(gateIndex)
+ .getId());
Review Comment:
Hey @rkhachatryan , thanks for the quick review!
I have merged 26be3b10bc8b403ad0994a743e6b9ac27b1fe320 into the second
commit, and only fixed some checkstyle issues.
FYI: and marked you as the co-author.
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
