io.rsocket.lease.Leases 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

There is a newer version: 1.1.4

/*
 * Copyright 2015-2019 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.lease;

import java.util.Objects;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Function;
import reactor.core.publisher.Flux;

public class Leases {
  private static final Function> noopLeaseSender = leaseStats -> Flux.never();
  private static final Consumer> noopLeaseReceiver = leases -> {};

  private Function> leaseSender = noopLeaseSender;
  private Consumer> leaseReceiver = noopLeaseReceiver;
  private Optional stats = Optional.empty();

  public static  Leases create() {
    return new Leases<>();
  }

  public Leases sender(Function, Flux> leaseSender) {
    this.leaseSender = leaseSender;
    return this;
  }

  public Leases receiver(Consumer> leaseReceiver) {
    this.leaseReceiver = leaseReceiver;
    return this;
  }

  public Leases stats(T stats) {
    this.stats = Optional.of(Objects.requireNonNull(stats));
    return this;
  }

  @SuppressWarnings("unchecked")
  public Function, Flux> sender() {
    return (Function, Flux>) leaseSender;
  }

  public Consumer> receiver() {
    return leaseReceiver;
  }

  @SuppressWarnings("unchecked")
  public Optional stats() {
    return (Optional) stats;
  }
}