org.infinispan.server.resp.commands.set.SRANDMEMBER 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.set;

import java.util.List;
import java.util.concurrent.CompletionStage;

import org.infinispan.multimap.impl.EmbeddedSetCache;
import org.infinispan.server.resp.Resp3Handler;
import org.infinispan.server.resp.RespCommand;
import org.infinispan.server.resp.RespRequestHandler;
import org.infinispan.server.resp.commands.ArgumentUtils;
import org.infinispan.server.resp.commands.Resp3Command;
import org.infinispan.server.resp.serialization.Resp3Response;

import io.netty.channel.ChannelHandlerContext;

/**
 * SRANDMEMBER implementation, see:
 * @link https://redis.io/commands/srandmember/
 * @since 15.0
 */
public class SRANDMEMBER extends RespCommand implements Resp3Command {
   public SRANDMEMBER() {
      super(-2, 1, 1, 1);
   }

   @Override
   public CompletionStage perform(Resp3Handler handler,
         ChannelHandlerContext ctx,
         List arguments) {
      final Long count = (arguments.size() <= 1) ? 1 : ArgumentUtils.toLong(arguments.get(1));
      final byte[] key = arguments.get(0);
      EmbeddedSetCache esc = handler.getEmbeddedSetCache();
      return handler.stageToReturn(esc.pop(key, count, false), ctx, Resp3Response.ARRAY_BULK_STRING);
   }
}