org.infinispan.server.resp.commands.hash.HSCAN Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of infinispan-server-resp Show documentation

Infinispan Resp Protocol Server

There is a newer version: 15.1.4.Final

Show newest version

package org.infinispan.server.resp.commands.hash;

import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletionStage;

import org.infinispan.container.entries.CacheEntry;
import org.infinispan.multimap.impl.EmbeddedMultimapPairCache;
import org.infinispan.server.iteration.IterationInitializationContext;
import org.infinispan.server.iteration.IterationManager;
import org.infinispan.server.iteration.map.MapIterationInitializationContext;
import org.infinispan.server.resp.Resp3Handler;
import org.infinispan.server.resp.commands.iteration.BaseIterationCommand;

/**
 * `HSCAN key cursor [MATCH pattern] [COUNT count]` command.
 * 
 * Scan the key-value properties of the hash stored under key.
 *
 * @since 15.0
 * @see Redis Documentation
 * @author José Bolina
 */
public class HSCAN extends BaseIterationCommand {

   public HSCAN() {
      super(-3, 1, 1, 1);
   }

   @Override
   protected IterationManager retrieveIterationManager(Resp3Handler handler) {
      return handler.respServer().getDataStructureIterationManager();
   }

   @Override
   protected CompletionStage initializeIteration(Resp3Handler handler, List arguments) {
      EmbeddedMultimapPairCache multimap = handler.getHashMapMultimap();
      return multimap.get(arguments.get(0)).thenApply(entry -> {
         if (entry == null) return null;

         return MapIterationInitializationContext.withSource(entry);
      });
   }

   @Override
   protected String cursor(List raw) {
      return new String(raw.get(1), StandardCharsets.US_ASCII);
   }

   @Override
   protected List writeResponse(List response) {
      List output = new ArrayList<>(2 * response.size());
      for (CacheEntry e : response) {
         output.add((byte[]) e.getKey());
         output.add((byte[]) e.getValue());
      }
      return output;
   }
}