com.jauntsdn.rsocket.keepalive.KeepAliveHandler Maven / Gradle / Ivy

Go to download
package com.jauntsdn.rsocket.keepalive;

import com.jauntsdn.rsocket.Closeable;
import com.jauntsdn.rsocket.resume.ResumableDuplexConnection;
import io.netty.buffer.ByteBuf;
import java.util.function.Consumer;

public interface KeepAliveHandler {

  Consumer start(KeepAlive keepAlive, Runnable onTimeout);

  class DefaultKeepAliveHandler implements KeepAliveHandler {
    private final Closeable duplexConnection;

    public DefaultKeepAliveHandler(Closeable duplexConnection) {
      this.duplexConnection = duplexConnection;
    }

    @Override
    public Consumer start(KeepAlive keepAlive, Runnable onTimeout) {
      duplexConnection.onClose().doFinally(s -> keepAlive.stop()).subscribe();
      return keepAlive.onTimeout(onTimeout).start();
    }
  }

  class ResumableKeepAliveHandler implements KeepAliveHandler {
    private final ResumableDuplexConnection resumableDuplexConnection;

    public ResumableKeepAliveHandler(ResumableDuplexConnection resumableDuplexConnection) {
      this.resumableDuplexConnection = resumableDuplexConnection;
    }

    @Override
    public Consumer start(KeepAlive keepAlive, Runnable onTimeout) {
      resumableDuplexConnection.onResume(keepAlive::start);
      resumableDuplexConnection.onDisconnect(keepAlive::stop);
      return keepAlive
          .resumeState(resumableDuplexConnection)
          .onTimeout(resumableDuplexConnection::disconnect)
          .start();
    }
  }
}