io.rsocket.core.LoggingDuplexConnection 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.core;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.rsocket.DuplexConnection;
import io.rsocket.RSocketErrorException;
import io.rsocket.frame.FrameUtil;
import java.net.SocketAddress;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

class LoggingDuplexConnection implements DuplexConnection {

  private static final Logger LOGGER = LoggerFactory.getLogger("io.rsocket.FrameLogger");

  final DuplexConnection source;

  LoggingDuplexConnection(DuplexConnection source) {
    this.source = source;
  }

  @Override
  public void dispose() {
    source.dispose();
  }

  @Override
  public Mono onClose() {
    return source.onClose();
  }

  @Override
  public void sendFrame(int streamId, ByteBuf frame) {
    LOGGER.debug("sending -> " + FrameUtil.toString(frame));

    source.sendFrame(streamId, frame);
  }

  @Override
  public void sendErrorAndClose(RSocketErrorException e) {
    LOGGER.debug("sending -> " + e.getClass().getSimpleName() + ": " + e.getMessage());

    source.sendErrorAndClose(e);
  }

  @Override
  public Flux receive() {
    return source
        .receive()
        .doOnNext(frame -> LOGGER.debug("receiving -> " + FrameUtil.toString(frame)));
  }

  @Override
  public ByteBufAllocator alloc() {
    return source.alloc();
  }

  @Override
  public SocketAddress remoteAddress() {
    return source.remoteAddress();
  }

  static DuplexConnection wrapIfEnabled(DuplexConnection source) {
    if (LOGGER.isDebugEnabled()) {
      return new LoggingDuplexConnection(source);
    }

    return source;
  }
}