io.rsocket.aeron.AeronDuplexConnection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rsocket-transport-aeron Show documentation
Aeron RSocket transport implementation
The newest version!
/*
 * Copyright 2015-2018 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.rsocket.aeron;

import io.netty.buffer.Unpooled;
import io.rsocket.DuplexConnection;
import io.rsocket.Frame;
import io.rsocket.aeron.internal.reactivestreams.AeronChannel;
import org.agrona.concurrent.UnsafeBuffer;
import org.reactivestreams.Publisher;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.core.publisher.MonoProcessor;

/**
 * Implementation of {@link DuplexConnection} over Aeron using an {@link
 * io.rsocket.aeron.internal.reactivestreams.AeronChannel}
 */
public class AeronDuplexConnection implements DuplexConnection {
  private final String name;
  private final AeronChannel channel;
  private final MonoProcessor onClose;

  public AeronDuplexConnection(String name, AeronChannel channel) {
    this.name = name;
    this.channel = channel;
    this.onClose = MonoProcessor.create();
  }

  @Override
  public Mono send(Publisher frame) {
    Flux buffers =
        Flux.from(frame).map(f -> new UnsafeBuffer(f.content().nioBuffer()));

    return channel.send(buffers);
  }

  @Override
  public Flux receive() {
    return channel
        .receive()
        .map(b -> Frame.from(Unpooled.wrappedBuffer(b.byteBuffer())))
        .doOnError(Throwable::printStackTrace);
  }

  @Override
  public void dispose() {
    try {
      channel.dispose();
      onClose.onComplete();
    } catch (Exception e) {
      onClose.onError(e);
    }
  }

  @Override
  public boolean isDisposed() {
    return channel.isDisposed();
  }

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

  @Override
  public String toString() {
    return "AeronDuplexConnection{"
        + "name='"
        + name
        + '\''
        + ", channel="
        + channel
        + ", onClose="
        + onClose
        + '}';
  }
}