org.infinispan.util.AbstractDelegatingCacheStream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of infinispan-core Show documentation
Infinispan core module
There is a newer version: 15.1.0.Dev04
package org.infinispan.util;

import java.util.Comparator;
import java.util.Iterator;
import java.util.Optional;
import java.util.Set;
import java.util.Spliterator;
import java.util.concurrent.TimeUnit;
import java.util.function.BiConsumer;
import java.util.function.BiFunction;
import java.util.function.BinaryOperator;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.IntFunction;
import java.util.function.Predicate;
import java.util.function.Supplier;
import java.util.function.ToDoubleFunction;
import java.util.function.ToIntFunction;
import java.util.function.ToLongFunction;
import java.util.stream.Collector;
import java.util.stream.DoubleStream;
import java.util.stream.IntStream;
import java.util.stream.LongStream;
import java.util.stream.Stream;

import org.infinispan.Cache;
import org.infinispan.CacheStream;
import org.infinispan.DoubleCacheStream;
import org.infinispan.IntCacheStream;
import org.infinispan.LongCacheStream;
import org.infinispan.commons.util.IntSet;

/**
 * Delegate that forwards all the of the method calls to the underlying cache stream. This can be useful to intercept
 * a given method call. Note that primitive stream operations are not intercepted (including those from
 * {@link java.util.stream.BaseStream}, however defined {@link org.infinispan.BaseCacheStream} methods would be.
 */
public class AbstractDelegatingCacheStream implements CacheStream {
   protected CacheStream underlyingStream;

   public AbstractDelegatingCacheStream(CacheStream stream) {
      this.underlyingStream = stream;
   }

   private CacheStream castStream(CacheStream stream) {
      return stream;
   }

   // These are methods that convert to a different AbstractDelegating*CacheStream

   @Override
   public IntCacheStream mapToInt(ToIntFunction mapper) {
      return new AbstractDelegatingIntCacheStream(this, castStream(underlyingStream).mapToInt(mapper));
   }

   @Override
   public LongCacheStream mapToLong(ToLongFunction mapper) {
      return new AbstractDelegatingLongCacheStream(this, castStream(underlyingStream).mapToLong(mapper));
   }

   @Override
   public DoubleCacheStream mapToDouble(ToDoubleFunction mapper) {
      return new AbstractDelegatingDoubleCacheStream(this, castStream(underlyingStream).mapToDouble(mapper));
   }

   @Override
   public IntCacheStream flatMapToInt(Function mapper) {
      return new AbstractDelegatingIntCacheStream(this, castStream(underlyingStream).flatMapToInt(mapper));
   }

   @Override
   public LongCacheStream flatMapToLong(Function mapper) {
      return new AbstractDelegatingLongCacheStream(this, castStream(underlyingStream).flatMapToLong(mapper));
   }

   @Override
   public DoubleCacheStream flatMapToDouble(Function mapper) {
      return new AbstractDelegatingDoubleCacheStream(this, castStream(underlyingStream).flatMapToDouble(mapper));
   }

   @Override
   public AbstractDelegatingCacheStream sequentialDistribution() {
      underlyingStream = underlyingStream.sequentialDistribution();
      return this;
   }

   @Override
   public AbstractDelegatingCacheStream parallelDistribution() {
      underlyingStream = underlyingStream.parallelDistribution();
      return this;
   }

   @Override
   public AbstractDelegatingCacheStream filterKeySegments(IntSet segments) {
      underlyingStream = underlyingStream.filterKeySegments(segments);
      return this;
   }

   @Override
   public AbstractDelegatingCacheStream filterKeys(Set keys) {
      underlyingStream = underlyingStream.filterKeys(keys);
      return this;
   }

   @Override
   public AbstractDelegatingCacheStream distributedBatchSize(int batchSize) {
      underlyingStream = underlyingStream.distributedBatchSize(batchSize);
      return this;
   }

   @Override
   public AbstractDelegatingCacheStream segmentCompletionListener(SegmentCompletionListener listener) {
      underlyingStream = underlyingStream.segmentCompletionListener(listener);
      return this;
   }

   @Override
   public AbstractDelegatingCacheStream disableRehashAware() {
      underlyingStream = underlyingStream.disableRehashAware();
      return this;
   }

   @Override
   public AbstractDelegatingCacheStream timeout(long timeout, TimeUnit unit) {
      underlyingStream = underlyingStream.timeout(timeout, unit);
      return this;
   }

   @Override
   public void forEach(Consumer action) {
      castStream(underlyingStream).forEach(action);
   }

   @Override
   public  void forEach(BiConsumer, ? super R> action) {
      castStream(underlyingStream).forEach(action);
   }

   @Override
   public void forEachOrdered(Consumer action) {
      castStream(underlyingStream).forEachOrdered(action);
   }

   @Override
   public Object[] toArray() {
      return underlyingStream.toArray();
   }

   @Override
   public  A[] toArray(IntFunction generator) {
      return underlyingStream.toArray(generator);
   }

   @Override
   public R reduce(R identity, BinaryOperator accumulator) {
      return castStream(underlyingStream).reduce(identity, accumulator);
   }

   @Override
   public Optional reduce(BinaryOperator accumulator) {
      return castStream(underlyingStream).reduce(accumulator);
   }

   @Override
   public  U reduce(U identity, BiFunction accumulator, BinaryOperator combiner) {
      return castStream(underlyingStream).reduce(identity, accumulator, combiner);
   }

   @Override
   public  R1 collect(Supplier supplier, BiConsumer accumulator, BiConsumer combiner) {
      return castStream(underlyingStream).collect(supplier, accumulator, combiner);
   }

   @Override
   public Iterator iterator() {
      return castStream(underlyingStream).iterator();
   }

   @Override
   public Spliterator spliterator() {
      return castStream(underlyingStream).spliterator();
   }

   @Override
   public boolean isParallel() {
      return underlyingStream.isParallel();
   }

   @Override
   public CacheStream sequential() {
      underlyingStream = underlyingStream.sequential();
      return this;
   }

   @Override
   public CacheStream parallel() {
      underlyingStream = underlyingStream.parallel();
      return this;
   }

   @Override
   public CacheStream unordered() {
      underlyingStream = underlyingStream.unordered();
      return this;
   }

   @Override
   public CacheStream onClose(Runnable closeHandler) {
      underlyingStream = underlyingStream.onClose(closeHandler);
      return this;
   }

   @Override
   public void close() {
      underlyingStream.close();
   }

   @Override
   public CacheStream sorted() {
      underlyingStream = underlyingStream.sorted();
      return this;
   }

   @Override
   public CacheStream sorted(Comparator comparator) {
      underlyingStream = castStream(underlyingStream).sorted(comparator);
      return this;
   }

   @Override
   public CacheStream peek(Consumer action) {
      underlyingStream = castStream(underlyingStream).peek(action);
      return this;
   }

   @Override
   public CacheStream limit(long maxSize) {
      underlyingStream = underlyingStream.limit(maxSize);
      return this;
   }

   @Override
   public CacheStream skip(long n) {
      underlyingStream = underlyingStream.skip(n);
      return this;
   }

   @Override
   public CacheStream filter(Predicate predicate) {
      underlyingStream = castStream(underlyingStream).filter(predicate);
      return this;
   }

   @Override
   public  CacheStream map(Function mapper) {
      underlyingStream = castStream(underlyingStream).map(mapper);
      return (CacheStream) this;
   }

   @Override
   public  CacheStream flatMap(Function> mapper) {
      underlyingStream = castStream(underlyingStream).flatMap(mapper);
      return (CacheStream) this;
   }

   @Override
   public CacheStream distinct() {
      underlyingStream = underlyingStream.distinct();
      return this;
   }

   @Override
   public  R1 collect(Collector collector) {
      return castStream(underlyingStream).collect(collector);
   }

   @Override
   public Optional min(Comparator comparator) {
      return castStream(underlyingStream).min(comparator);
   }

   @Override
   public Optional max(Comparator comparator) {
      return castStream(underlyingStream).max(comparator);
   }

   @Override
   public long count() {
      return underlyingStream.count();
   }

   @Override
   public boolean anyMatch(Predicate predicate) {
      return castStream(underlyingStream).anyMatch(predicate);
   }

   @Override
   public boolean allMatch(Predicate predicate) {
      return castStream(underlyingStream).allMatch(predicate);
   }

   @Override
   public boolean noneMatch(Predicate predicate) {
      return castStream(underlyingStream).noneMatch(predicate);
   }

   @Override
   public Optional findFirst() {
      return castStream(underlyingStream).findFirst();
   }

   @Override
   public Optional findAny() {
      return castStream(underlyingStream).findAny();
   }
}