io.rsocket.keepalive.KeepAliveHandler 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

Core functionality for the RSocket library

The newest version!

package io.rsocket.keepalive;

import io.netty.buffer.ByteBuf;
import io.rsocket.keepalive.KeepAliveSupport.KeepAlive;
import io.rsocket.resume.RSocketSession;
import io.rsocket.resume.ResumableDuplexConnection;
import io.rsocket.resume.ResumeStateHolder;
import java.util.function.Consumer;

public interface KeepAliveHandler {

  KeepAliveFramesAcceptor start(
      KeepAliveSupport keepAliveSupport,
      Consumer onFrameSent,
      Consumer onTimeout);

  class DefaultKeepAliveHandler implements KeepAliveHandler {
    @Override
    public KeepAliveFramesAcceptor start(
        KeepAliveSupport keepAliveSupport,
        Consumer onSendKeepAliveFrame,
        Consumer onTimeout) {
      return keepAliveSupport
          .onSendKeepAliveFrame(onSendKeepAliveFrame)
          .onTimeout(onTimeout)
          .start();
    }
  }

  class ResumableKeepAliveHandler implements KeepAliveHandler {

    private final ResumableDuplexConnection resumableDuplexConnection;
    private final RSocketSession rSocketSession;
    private final ResumeStateHolder resumeStateHolder;

    public ResumableKeepAliveHandler(
        ResumableDuplexConnection resumableDuplexConnection,
        RSocketSession rSocketSession,
        ResumeStateHolder resumeStateHolder) {
      this.resumableDuplexConnection = resumableDuplexConnection;
      this.rSocketSession = rSocketSession;
      this.resumeStateHolder = resumeStateHolder;
    }

    @Override
    public KeepAliveFramesAcceptor start(
        KeepAliveSupport keepAliveSupport,
        Consumer onSendKeepAliveFrame,
        Consumer onTimeout) {

      rSocketSession.setKeepAliveSupport(keepAliveSupport);

      return keepAliveSupport
          .resumeState(resumeStateHolder)
          .onSendKeepAliveFrame(onSendKeepAliveFrame)
          .onTimeout(keepAlive -> resumableDuplexConnection.disconnect())
          .start();
    }
  }
}