root
/
kafka
mirror of https://github.com/apache/kafka.git
1
0
Fork 0
Code Issues Actions 2 Packages Projects Releases Wiki Activity

KAFKA-12958: add an invariant that notified leaders are never asked to load snapshot (#10932) 

Track handleSnapshot calls and make sure it is never triggered on the leader node.

Reviewers: Luke Chen <showuon@gmail.com>, José Armando García Sancio <jsancio@users.noreply.github.com>, Boyang Chen <bchen11@outlook.com>
This commit is contained in:
zhaohaidao 2021-07-04 23:32:12 +08:00 committed by GitHub
parent fa685fa152
commit 10b1f73cd4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
raft/src/main/java/org/apache/kafka/raft/ReplicatedCounter.java
View File

@ -38,6 +38,8 @@ public class ReplicatedCounter implements RaftClient.Listener<Integer> {
private OptionalInt claimedEpoch = OptionalInt.empty();
private long lastOffsetSnapshotted = -1;
private int handleSnapshotCalls = 0;
public ReplicatedCounter(
int nodeId,
RaftClient<Integer> client,
@ -152,6 +154,7 @@ public class ReplicatedCounter implements RaftClient.Listener<Integer> {
}
}
lastOffsetSnapshotted = reader.lastContainedLogOffset();
handleSnapshotCalls += 1;
log.debug("Finished loading snapshot. Set value: {}", committed);
} finally {
reader.close();
@ -170,5 +173,11 @@ public class ReplicatedCounter implements RaftClient.Listener<Integer> {
uncommitted = -1;
claimedEpoch = OptionalInt.empty();
}
handleSnapshotCalls = 0;
}
/** Use handleSnapshotCalls to verify leader is never asked to load snapshot */
public int handleSnapshotCalls() {
return handleSnapshotCalls;
}
}

18
raft/src/test/java/org/apache/kafka/raft/RaftEventSimulationTest.java
View File

@ -352,6 +352,7 @@ public class RaftEventSimulationTest {
scheduler.addInvariant(new MajorityReachedHighWatermark(cluster));
scheduler.addInvariant(new SingleLeader(cluster));
scheduler.addInvariant(new SnapshotAtLogStart(cluster));
scheduler.addInvariant(new LeaderNeverLoadSnapshot(cluster));
scheduler.addValidation(new ConsistentCommittedData(cluster));
return scheduler;
}
@ -1014,6 +1015,23 @@ public class RaftEventSimulationTest {
}
}
private static class LeaderNeverLoadSnapshot implements Invariant {
final Cluster cluster;
private LeaderNeverLoadSnapshot(Cluster cluster) {
this.cluster = cluster;
}
@Override
public void verify() {
for (RaftNode raftNode : cluster.running()) {
if (raftNode.counter.isWritable()) {
assertEquals(0, raftNode.counter.handleSnapshotCalls());
}
}
}
}
/**
* Validating the committed data is expensive, so we do this as a {@link Validation}. We depend
* on the following external invariants: