[hotfix][checkpointing] Add VisibleForTesting annotation for related …

…methods
spotify · Jun 9, 2020 · bb0ebca · bb0ebca
1 parent 44acd50
commit bb0ebca
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/...-java/src/main/java/org/apache/flink/streaming/runtime/io/CheckpointBarrierUnaligner.java b/...-java/src/main/java/org/apache/flink/streaming/runtime/io/CheckpointBarrierUnaligner.java
@@ -348,10 +348,6 @@ public synchronized void close() throws IOException {
  allBarriersReceivedFuture.cancel(false);
  }
 
- boolean isCheckpointPending() {
- return numBarriersReceived > 0;
- }
-
  private synchronized void handleNewCheckpoint(CheckpointBarrier barrier) throws IOException {
  long barrierId = barrier.getId();
  if (!allBarriersReceivedFuture.isDone() && isCheckpointPending()) {
@@ -410,8 +406,14 @@ synchronized int getNumOpenChannels() {
  return numOpenChannels;
  }
 
+ @VisibleForTesting
  synchronized long getCurrentCheckpointId() {
  return currentReceivedCheckpointId;
  }
+
+ @VisibleForTesting
+ boolean isCheckpointPending() {
+ return numBarriersReceived > 0;
+ }
  }
 }
diff --git a/...aming-java/src/main/java/org/apache/flink/streaming/runtime/io/CheckpointedInputGate.java b/...aming-java/src/main/java/org/apache/flink/streaming/runtime/io/CheckpointedInputGate.java
@@ -185,7 +185,8 @@ public void close() throws IOException {
  *
  * @return The ID of the pending of completed checkpoint.
  */
- public long getLatestCheckpointId() {
+ @VisibleForTesting
+ long getLatestCheckpointId() {
  return barrierHandler.getLatestCheckpointId();
  }