Skip to content

Commit a20b3e8

Browse files
Merge branch 'master' into jira2328
2 parents ddd16eb + fb866d1 commit a20b3e8

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

ratis-server/src/main/java/org/apache/ratis/server/impl/PendingRequests.java

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ private static class RequestMap {
117117
raftServerMetrics.addNumPendingRequestsMegaByteSize(resource::getMegaByteSize);
118118
}
119119

120-
Permit tryAcquire(Message message) {
120+
synchronized Permit tryAcquire(Message message) {
121121
final int messageSize = Message.getSize(message);
122122
final int messageSizeMb = roundUpMb(messageSize );
123123
final Acquired acquired = resource.tryAcquire(messageSizeMb);
@@ -139,13 +139,7 @@ Permit tryAcquire(Message message) {
139139
if (messageSizeMb > diffMb) {
140140
resource.releaseExtraMb(messageSizeMb - diffMb);
141141
}
142-
return putPermit();
143-
}
144142

145-
private synchronized Permit putPermit() {
146-
if (resource.isClosed()) {
147-
return null;
148-
}
149143
final Permit permit = new Permit();
150144
permits.put(permit, permit);
151145
return permit;
@@ -157,9 +151,9 @@ synchronized PendingRequest put(Permit permit, PendingRequest p) {
157151
if (removed == null) {
158152
return null;
159153
}
160-
Preconditions.assertTrue(removed == permit);
154+
Preconditions.assertSame(permit, removed, "permit");
161155
final PendingRequest previous = map.put(p.getTermIndex(), p);
162-
Preconditions.assertTrue(previous == null);
156+
Preconditions.assertNull(previous, "previous");
163157
return p;
164158
}
165159

0 commit comments

Comments
 (0)