io.rsocket.SocketAcceptor 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!
/*
 * Copyright 2015-2020 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;

import io.rsocket.exceptions.SetupException;
import java.util.function.Function;
import org.reactivestreams.Publisher;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

/**
 * RSocket is a full duplex protocol where a client and server are identical in terms of both having
 * the capability to initiate requests to their peer. This interface provides the contract where a
 * client or server handles the {@code setup} for a new connection and creates a responder {@code
 * RSocket} for accepting requests from the remote peer.
 */
public interface SocketAcceptor {

  /**
   * Handle the {@code SETUP} frame for a new connection and create a responder {@code RSocket} for
   * handling requests from the remote peer.
   *
   * @param setup the {@code setup} received from a client in a server scenario, or in a client
   *     scenario this is the setup about to be sent to the server.
   * @param sendingSocket socket for sending requests to the remote peer.
   * @return {@code RSocket} to accept requests with.
   * @throws SetupException If the acceptor needs to reject the setup of this socket.
   */
  Mono accept(ConnectionSetupPayload setup, RSocket sendingSocket);

  /** Create a {@code SocketAcceptor} that handles requests with the given {@code RSocket}. */
  static SocketAcceptor with(RSocket rsocket) {
    return (setup, sendingRSocket) -> Mono.just(rsocket);
  }

  /** Create a {@code SocketAcceptor} for fire-and-forget interactions with the given handler. */
  static SocketAcceptor forFireAndForget(Function> handler) {
    return with(
        new RSocket() {
          @Override
          public Mono fireAndForget(Payload payload) {
            return handler.apply(payload);
          }
        });
  }

  /** Create a {@code SocketAcceptor} for request-response interactions with the given handler. */
  static SocketAcceptor forRequestResponse(Function> handler) {
    return with(
        new RSocket() {
          @Override
          public Mono requestResponse(Payload payload) {
            return handler.apply(payload);
          }
        });
  }

  /** Create a {@code SocketAcceptor} for request-stream interactions with the given handler. */
  static SocketAcceptor forRequestStream(Function> handler) {
    return with(
        new RSocket() {
          @Override
          public Flux requestStream(Payload payload) {
            return handler.apply(payload);
          }
        });
  }

  /** Create a {@code SocketAcceptor} for request-channel interactions with the given handler. */
  static SocketAcceptor forRequestChannel(Function, Flux> handler) {
    return with(
        new RSocket() {
          @Override
          public Flux requestChannel(Publisher payloads) {
            return handler.apply(payloads);
          }
        });
  }
}