org.infinispan.server.resp.commands.hash.HMGET 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.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletionStage;

import org.infinispan.commons.marshall.WrappedByteArray;
import org.infinispan.commons.util.Util;
import org.infinispan.multimap.impl.EmbeddedMultimapPairCache;
import org.infinispan.server.resp.Resp3Handler;
import org.infinispan.server.resp.RespCommand;
import org.infinispan.server.resp.RespRequestHandler;
import org.infinispan.server.resp.commands.Resp3Command;
import org.infinispan.server.resp.serialization.Resp3Response;

import io.netty.channel.ChannelHandlerContext;

/**
 * `HMGET key field [field ...]` command.
 * 
 * Returns the values associated with all fields in the hash stored at key, returning a list
 * with the values in the same order of the fields. No error returns from fields or
 * key that do not exist in the hash, and null is returned in the position.
 *
 * @since 15.0
 * @see Redis Documentation
 * @author José Bolina
 */
public class HMGET extends RespCommand implements Resp3Command {

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

   @Override
   public CompletionStage perform(Resp3Handler handler, ChannelHandlerContext ctx, List arguments) {
      EmbeddedMultimapPairCache multimap = handler.getHashMapMultimap();
      CompletionStage> cs = multimap.get(arguments.get(0), arguments.subList(1, arguments.size()).toArray(Util.EMPTY_BYTE_ARRAY_ARRAY))
            .thenApply(this::wrapKeys)
            .thenApply(entries -> {
               List result = new ArrayList<>(arguments.size() - 1);
               for (byte[] argument : arguments.subList(1, arguments.size())) {
                  result.add(entries.get(new WrappedByteArray(argument)));
               }
               return result;
            });
      return handler.stageToReturn(cs, ctx, Resp3Response.ARRAY_BULK_STRING);
   }

   private Map wrapKeys(Map original) {
      return original.entrySet().stream()
            .collect(HashMap::new, (m, v) -> m.put(new WrappedByteArray(v.getKey()), v.getValue()), HashMap::putAll);
   }
}