org.infinispan.util.MappedCacheStream 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.Set;

import org.infinispan.CacheStream;
import org.infinispan.commons.util.InjectiveFunction;

/**
 * CacheStream that allows for mapping the filtered keys to different keys. Note that the function provided
 * must be a {@link InjectiveFunction} guaranteeing that keys are distinct when mapped.
 * @author wburns
 * @since 9.2
 */
class MappedCacheStream extends AbstractDelegatingCacheStream {
   private final InjectiveFunction keyMapper;
   MappedCacheStream(CacheStream stream, InjectiveFunction keyMapper) {
      super(stream);
      this.keyMapper = keyMapper;
   }

   @Override
   public AbstractDelegatingCacheStream filterKeys(Set keys) {
      return super.filterKeys(keys != null ? new SetMapper<>(keys, keyMapper) : null);
   }
}