@@ -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