2727import io .rsocket .exceptions .Exceptions ;
2828import io .rsocket .frame .*;
2929import io .rsocket .frame .decoder .PayloadDecoder ;
30- import io .rsocket .internal .RateLimitableRequestPublisher ;
30+ import io .rsocket .internal .LimitableRequestPublisher ;
3131import io .rsocket .internal .SynchronizedIntObjectHashMap ;
3232import io .rsocket .internal .UnboundedProcessor ;
3333import io .rsocket .internal .UnicastMonoProcessor ;
4747import org .reactivestreams .Subscriber ;
4848import org .reactivestreams .Subscription ;
4949import reactor .core .publisher .*;
50- import reactor .util .concurrent .Queues ;
5150
5251/**
5352 * Requester Side of a RSocket socket. Sends {@link ByteBuf}s to a {@link RSocketResponder} of peer
@@ -61,7 +60,7 @@ class RSocketRequester implements RSocket {
6160 private final PayloadDecoder payloadDecoder ;
6261 private final Consumer <Throwable > errorConsumer ;
6362 private final StreamIdSupplier streamIdSupplier ;
64- private final IntObjectMap <RateLimitableRequestPublisher > senders ;
63+ private final IntObjectMap <LimitableRequestPublisher > senders ;
6564 private final IntObjectMap <Processor <Payload , Payload >> receivers ;
6665 private final UnboundedProcessor <ByteBuf > sendProcessor ;
6766 private final RequesterLeaseHandler leaseHandler ;
@@ -132,7 +131,7 @@ private void handleSendProcessorError(Throwable t) {
132131 }
133132 });
134133
135- senders .values ().forEach (RateLimitableRequestPublisher ::cancel );
134+ senders .values ().forEach (LimitableRequestPublisher ::cancel );
136135 }
137136
138137 private void handleSendProcessorCancel (SignalType t ) {
@@ -151,7 +150,7 @@ private void handleSendProcessorCancel(SignalType t) {
151150 }
152151 });
153152
154- senders .values ().forEach (RateLimitableRequestPublisher ::cancel );
153+ senders .values ().forEach (LimitableRequestPublisher ::cancel );
155154 }
156155
157156 @ Override
@@ -344,8 +343,8 @@ public void accept(long n) {
344343 request
345344 .transform (
346345 f -> {
347- RateLimitableRequestPublisher <Payload > wrapped =
348- RateLimitableRequestPublisher .wrap (f , Queues . SMALL_BUFFER_SIZE );
346+ LimitableRequestPublisher <Payload > wrapped =
347+ LimitableRequestPublisher .wrap (f );
349348 // Need to set this to one for first the frame
350349 wrapped .request (1 );
351350 senders .put (streamId , wrapped );
@@ -422,7 +421,7 @@ protected void hookOnError(Throwable t) {
422421 .doFinally (
423422 s -> {
424423 receivers .remove (streamId );
425- RateLimitableRequestPublisher sender = senders .remove (streamId );
424+ LimitableRequestPublisher sender = senders .remove (streamId );
426425 if (sender != null ) {
427426 sender .cancel ();
428427 }
@@ -490,7 +489,7 @@ private void setTerminationError(Throwable error) {
490489 }
491490
492491 private synchronized void cleanUpLimitableRequestPublisher (
493- RateLimitableRequestPublisher <?> limitableRequestPublisher ) {
492+ LimitableRequestPublisher <?> limitableRequestPublisher ) {
494493 try {
495494 limitableRequestPublisher .cancel ();
496495 } catch (Throwable t ) {
@@ -562,7 +561,7 @@ private void handleFrame(int streamId, FrameType type, ByteBuf frame) {
562561 break ;
563562 case CANCEL :
564563 {
565- RateLimitableRequestPublisher sender = senders .remove (streamId );
564+ LimitableRequestPublisher sender = senders .remove (streamId );
566565 if (sender != null ) {
567566 sender .cancel ();
568567 }
@@ -573,7 +572,7 @@ private void handleFrame(int streamId, FrameType type, ByteBuf frame) {
573572 break ;
574573 case REQUEST_N :
575574 {
576- RateLimitableRequestPublisher sender = senders .get (streamId );
575+ LimitableRequestPublisher sender = senders .get (streamId );
577576 if (sender != null ) {
578577 int n = RequestNFrameFlyweight .requestN (frame );
579578 sender .request (n >= Integer .MAX_VALUE ? Long .MAX_VALUE : n );
0 commit comments