org.infinispan.distribution.util.ReadOnlySegmentAwareIterator Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of infinispan-embedded

Infinispan Embedded All-in-One module

There is a newer version: 9.1.7.Final

package org.infinispan.distribution.util;

import java.util.Iterator;
import java.util.NoSuchElementException;
import java.util.Set;

import org.infinispan.distribution.ch.ConsistentHash;

/**
 * Iterator implementation that shows a read only view of the provided iterator by only
 * allowing values that map to a given segment using the provided consistent hash.
 * 
 * This iterator is used with the other various SegmentAware Collections such as
 * {@link ReadOnlySegmentAwareSet}
 * 
 * @author wburns
 * @since 7.2
 */
public class ReadOnlySegmentAwareIterator implements Iterator {
   protected final Iterator iter;
   protected final ConsistentHash ch;
   protected final Set allowedSegments;

   protected E next;

   public ReadOnlySegmentAwareIterator(Iterator iter, ConsistentHash ch, Set allowedSegments) {
      super();
      this.iter = iter;
      this.ch = ch;
      this.allowedSegments = allowedSegments;
      next = findNext();
   }

   protected boolean valueAllowed(Object obj) {
      int segment = ch.getSegment(obj);
      return allowedSegments.contains(segment);
   }

   protected E findNext() {
      while (iter.hasNext()) {
         E next = iter.next();
         if (valueAllowed(next)) {
            return next;
         }
      }
      return null;
   }

   @Override
   public boolean hasNext() {
      return next != null;
   }

   @Override
   public E next() {
      E prev = next;
      if (prev == null) {
         throw new NoSuchElementException();
      }
      next = findNext();
      return prev;
   }

   @Override
   public void remove() {
      throw new UnsupportedOperationException("remove");
  }
}