org.infinispan.server.resp.commands.string.SETRANGE 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.string;

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

import org.infinispan.commons.util.Util;
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;

/**
 * SETRANGE Resp Command
 *
 * @link https://redis.io/commands/setrange/
 * @since 15.0
 */
public class SETRANGE extends RespCommand implements Resp3Command {
   public SETRANGE() {
      super(4, 1, 1, 1);
   }

   @Override
   public CompletionStage perform(Resp3Handler handler,
         ChannelHandlerContext ctx,
         List arguments) {
      byte[] keyBytes = arguments.get(0);
      int offset = ArgumentUtils.toInt(arguments.get(1));
      byte[] patch = arguments.get(2);
      CompletionStage objectCompletableFuture = handler.cache().computeAsync(keyBytes, (ignoredKey, value) -> setrange(value, patch, offset)).thenApply(newValue -> newValue == null ? 0L : (long)newValue.length);
      return handler.stageToReturn(objectCompletableFuture, ctx, Resp3Response.INTEGER);
   }

   private static byte[] setrange(byte[] value, byte[] patch, int offset) {
      if (patch.length == 0) {
         return value;
      }

      if (value == null) {
         value = Util.EMPTY_BYTE_ARRAY;
      }

      if (patch.length+offset < value.length) {
         var buf = ByteBuffer.wrap(value);
         buf.put(offset, patch);
         return buf.array();
      }

      byte[] output = new byte[offset + patch.length];
      var buf = ByteBuffer.wrap(output);
      buf.put(value, 0, Math.min(value.length, offset));
      buf.put(offset, patch);
      return output;
   }
}