com.jauntsdn.rsocket.LeaseRSocketRequester Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rsocket-core Show documentation

RSocket core library

There is a newer version: 0.9.8

package com.jauntsdn.rsocket;

import static com.jauntsdn.rsocket.StreamErrorMappers.*;

import com.jauntsdn.rsocket.frame.decoder.PayloadDecoder;
import com.jauntsdn.rsocket.keepalive.KeepAliveHandler;
import com.jauntsdn.rsocket.lease.RequesterLeaseHandler;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import java.util.function.Consumer;
import java.util.function.LongConsumer;
import javax.annotation.Nullable;

class LeaseRSocketRequester extends RSocketRequester {
  private final RequesterLeaseHandler leaseHandler;

  LeaseRSocketRequester(
      ByteBufAllocator allocator,
      DuplexConnection connection,
      PayloadDecoder payloadDecoder,
      Consumer errorConsumer,
      ErrorFrameMapper errorFrameMapper,
      StreamIdSupplier streamIdSupplier,
      int keepAliveTickPeriod,
      int keepAliveAckTimeout,
      KeepAliveHandler keepAliveHandler,
      RequesterLeaseHandler leaseHandler,
      @Nullable LongConsumer onRtt) {
    super(
        allocator,
        connection,
        payloadDecoder,
        errorConsumer,
        errorFrameMapper,
        streamIdSupplier,
        keepAliveTickPeriod,
        keepAliveAckTimeout,
        keepAliveHandler);
    this.leaseHandler = leaseHandler;
    keepAlive().onRtt(onRtt);
  }

  @Override
  Throwable checkAllowed() {
    Throwable err = super.checkAllowed();
    if (err != null) {
      return err;
    }
    leaseHandler.requestStarted();
    return err;
  }

  @Override
  void handleLeaseFrame(ByteBuf frame) {
    leaseHandler.receive(frame);
  }

  @Override
  public double availability() {
    return Math.min(super.availability(), leaseHandler.availability());
  }
}