All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.infinispan.commands.write.PrimaryAckCommand Maven / Gradle / Ivy

    There is a newer version: 9.1.7.Final
    Show newest version
    package org.infinispan.commands.write;

import org.infinispan.commands.CommandInvocationId;
import org.infinispan.commands.remote.BaseRpcCommand;
import org.infinispan.commons.marshall.MarshallUtil;
import org.infinispan.util.ByteString;
import org.infinispan.util.concurrent.CommandAckCollector;
import org.infinispan.util.concurrent.CompletableFutures;

import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.util.concurrent.CompletableFuture;

/**
 * A command that represents an acknowledge sent by the primary owner to the originator.
 * 
    
 * The acknowledge signals a successful or unsuccessful execution of the operation and it contains the return value of
 * the operation.
 *
 * @author Pedro Ruivo
 * @since 9.0
 */
public class PrimaryAckCommand extends BaseRpcCommand {

   public static final byte COMMAND_ID = 30;
   private static final Type[] CACHED_TYPE = Type.values();
   private CommandInvocationId commandInvocationId;
   private Object returnValue;
   private Type type;
   private CommandAckCollector commandAckCollector;
   private int topologyId;

   public PrimaryAckCommand() {
      super(null);
   }

   public PrimaryAckCommand(ByteString cacheName) {
      super(cacheName);
   }

   private static Type valueOf(int index) {
      return CACHED_TYPE[index];
   }

   public void initCommandInvocationIdAndTopologyId(CommandInvocationId id, int topologyId) {
      this.commandInvocationId = id;
      this.topologyId = topologyId;
   }

   @Override
   public CompletableFuture invokeAsync() throws Throwable {
      switch (type) {
         case SUCCESS_WITH_BOOL_RETURN_VALUE:
         case SUCCESS_WITH_RETURN_VALUE:
         case SUCCESS_WITHOUT_RETURN_VALUE:
            commandAckCollector.primaryAck(commandInvocationId, returnValue, true, getOrigin(), topologyId);
            break;
         case UNSUCCESSFUL_WITH_BOOL_RETURN_VALUE:
         case UNSUCCESSFUL_WITH_RETURN_VALUE:
         case UNSUCCESSFUL_WITHOUT_RETURN_VALUE:
            commandAckCollector.primaryAck(commandInvocationId, returnValue, false, getOrigin(), topologyId);
            break;
      }
      return CompletableFutures.completedNull();
   }

   @Override
   public byte getCommandId() {
      return COMMAND_ID;
   }

   @Override
   public boolean isReturnValueExpected() {
      return false;
   }

   @Override
   public void writeTo(ObjectOutput output) throws IOException {
      output.writeInt(topologyId);
      CommandInvocationId.writeTo(output, commandInvocationId);
      MarshallUtil.marshallEnum(type, output);
      switch (type) {
         case SUCCESS_WITH_RETURN_VALUE:
         case UNSUCCESSFUL_WITH_RETURN_VALUE:
            output.writeObject(returnValue);
            break;
         default:
      }
   }

   @Override
   public void readFrom(ObjectInput input) throws IOException, ClassNotFoundException {
      topologyId = input.readInt();
      commandInvocationId = CommandInvocationId.readFrom(input);
      type = MarshallUtil.unmarshallEnum(input, PrimaryAckCommand::valueOf);
      assert type != null;
      switch (type) {
         case SUCCESS_WITH_RETURN_VALUE:
         case UNSUCCESSFUL_WITH_RETURN_VALUE:
            returnValue = input.readObject();
            break;
         case SUCCESS_WITH_BOOL_RETURN_VALUE:
            returnValue = true;
            break;
         case UNSUCCESSFUL_WITH_BOOL_RETURN_VALUE:
            returnValue = false;
            break;
         default:
      }
   }

   public void initWithReturnValue(boolean success, Object returnValue) {
      this.returnValue = returnValue;
      if (success) {
         type = Type.SUCCESS_WITH_RETURN_VALUE;
      } else {
         type = Type.UNSUCCESSFUL_WITH_RETURN_VALUE;
      }
   }

   public void initWithBoolReturnValue(boolean success) {
      this.returnValue = success;
      if (success) {
         type = Type.SUCCESS_WITH_BOOL_RETURN_VALUE;
      } else {
         type = Type.UNSUCCESSFUL_WITH_BOOL_RETURN_VALUE;
      }
   }

   public void initWithoutReturnValue(boolean success) {
      if (success) {
         type = Type.SUCCESS_WITHOUT_RETURN_VALUE;
      } else {
         type = Type.UNSUCCESSFUL_WITHOUT_RETURN_VALUE;
      }
   }

   public void setCommandAckCollector(CommandAckCollector commandAckCollector) {
      this.commandAckCollector = commandAckCollector;
   }

   @Override
   public String toString() {
      return "PrimaryAckCommand{" +
            "commandInvocationId=" + commandInvocationId +
            ", returnValue=" + returnValue +
            ", type=" + type +
            ", topologyId=" + topologyId +
            '}';
   }

   private enum Type {
      SUCCESS_WITH_RETURN_VALUE,
      SUCCESS_WITH_BOOL_RETURN_VALUE,
      SUCCESS_WITHOUT_RETURN_VALUE,
      UNSUCCESSFUL_WITH_RETURN_VALUE,
      UNSUCCESSFUL_WITH_BOOL_RETURN_VALUE,
      UNSUCCESSFUL_WITHOUT_RETURN_VALUE
   }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy