ISPN-11541 Add internal metadata to write commands

infinispan · Apr 14, 2020 · f10d8b5 · f10d8b5
1 parent 01f13b0
commit f10d8b5
Show file tree

Hide file tree

Showing 45 changed files with 621 additions and 144 deletions.
diff --git a/commons/all/src/main/java/org/infinispan/commons/marshall/Ids.java b/commons/all/src/main/java/org/infinispan/commons/marshall/Ids.java
@@ -220,6 +220,7 @@ public interface Ids {
    int CACHE_BI_CONSUMERS = 150;
 
    int PUBLISHER_TRANSFORMERS = 151;
+   int PREPARE_RESPONSE = 152;
 
    int COUNTER_CONFIGURATION = 2000; //from counter
    int COUNTER_STATE = 2001; //from counter

diff --git a/core/src/main/java/org/infinispan/commands/functional/AbstractWriteKeyCommand.java b/core/src/main/java/org/infinispan/commands/functional/AbstractWriteKeyCommand.java
@@ -7,6 +7,7 @@
 import org.infinispan.commands.write.ValueMatcher;
 import org.infinispan.encoding.DataConversion;
 import org.infinispan.factories.ComponentRegistry;
+import org.infinispan.functional.impl.MetaParamsInternalMetadata;
 import org.infinispan.functional.impl.Params;
 
 public abstract class AbstractWriteKeyCommand<K, V> extends AbstractDataWriteCommand implements FunctionalCommand<K, V> {
@@ -16,6 +17,7 @@ public abstract class AbstractWriteKeyCommand<K, V> extends AbstractDataWriteCom
    boolean successful = true;
    DataConversion keyDataConversion;
    DataConversion valueDataConversion;
+   MetaParamsInternalMetadata internalMetadata;
 
    public AbstractWriteKeyCommand(Object key, ValueMatcher valueMatcher, int segment,
                                   CommandInvocationId id, Params params,
@@ -84,4 +86,14 @@ public DataConversion getKeyDataConversion() {
    public DataConversion getValueDataConversion() {
       return valueDataConversion;
    }
+
+   @Override
+   public MetaParamsInternalMetadata getInternalMetadata() {
+      return internalMetadata;
+   }
+
+   @Override
+   public void setInternalMetadata(MetaParamsInternalMetadata internalMetadata) {
+      this.internalMetadata = internalMetadata;
+   }
 }
diff --git a/core/src/main/java/org/infinispan/commands/functional/AbstractWriteManyCommand.java b/core/src/main/java/org/infinispan/commands/functional/AbstractWriteManyCommand.java
@@ -1,11 +1,15 @@
 package org.infinispan.commands.functional;
 
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
 import org.infinispan.commands.CommandInvocationId;
 import org.infinispan.commands.write.ValueMatcher;
 import org.infinispan.commands.write.WriteCommand;
 import org.infinispan.context.impl.FlagBitSets;
 import org.infinispan.encoding.DataConversion;
 import org.infinispan.factories.ComponentRegistry;
+import org.infinispan.functional.impl.MetaParamsInternalMetadata;
 import org.infinispan.functional.impl.Params;
 import org.infinispan.util.concurrent.locks.RemoteLockCommand;
 
@@ -20,6 +24,7 @@ public abstract class AbstractWriteManyCommand<K, V> implements WriteCommand, Fu
    long flags;
    DataConversion keyDataConversion;
    DataConversion valueDataConversion;
+   Map<Object, MetaParamsInternalMetadata> internalMetadataMap;
 
    protected AbstractWriteManyCommand(CommandInvocationId commandInvocationId,
                                       Params params,
@@ -30,13 +35,15 @@ protected AbstractWriteManyCommand(CommandInvocationId commandInvocationId,
       this.flags = params.toFlagsBitSet();
       this.keyDataConversion = keyDataConversion;
       this.valueDataConversion = valueDataConversion;
+      this.internalMetadataMap = new ConcurrentHashMap<>();
    }
 
    protected <K, V> AbstractWriteManyCommand(AbstractWriteManyCommand<K, V> command) {
       this.commandInvocationId = command.commandInvocationId;
       this.topologyId = command.topologyId;
       this.params = command.params;
       this.flags = command.flags;
+      this.internalMetadataMap = new ConcurrentHashMap<>(command.internalMetadataMap);
    }
 
    protected AbstractWriteManyCommand() {
@@ -139,4 +146,14 @@ public DataConversion getKeyDataConversion() {
    public DataConversion getValueDataConversion() {
       return valueDataConversion;
    }
+
+   @Override
+   public MetaParamsInternalMetadata getInternalMetadata(Object key) {
+      return internalMetadataMap.get(key);
+   }
+
+   @Override
+   public void setInternalMetadata(Object key, MetaParamsInternalMetadata internalMetadata) {
+      this.internalMetadataMap.put(key, internalMetadata);
+   }
 }
diff --git a/core/src/main/java/org/infinispan/commands/functional/ReadWriteKeyCommand.java b/core/src/main/java/org/infinispan/commands/functional/ReadWriteKeyCommand.java
@@ -18,6 +18,7 @@
 import org.infinispan.encoding.DataConversion;
 import org.infinispan.factories.ComponentRegistry;
 import org.infinispan.functional.EntryView.ReadWriteEntryView;
+import org.infinispan.functional.impl.MetaParamsInternalMetadata;
 import org.infinispan.functional.impl.Params;
 
 // TODO: the command does not carry previous values to backup, so it can cause
@@ -63,6 +64,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       CommandInvocationId.writeTo(output, commandInvocationId);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      output.writeObject(internalMetadata);
    }
 
    @Override
@@ -76,6 +78,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       commandInvocationId = CommandInvocationId.readFrom(input);
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      internalMetadata = (MetaParamsInternalMetadata) input.readObject();
    }
 
    @Override

diff --git a/core/src/main/java/org/infinispan/commands/functional/ReadWriteKeyValueCommand.java b/core/src/main/java/org/infinispan/commands/functional/ReadWriteKeyValueCommand.java
@@ -18,6 +18,7 @@
 import org.infinispan.encoding.DataConversion;
 import org.infinispan.factories.ComponentRegistry;
 import org.infinispan.functional.EntryView.ReadWriteEntryView;
+import org.infinispan.functional.impl.MetaParamsInternalMetadata;
 import org.infinispan.functional.impl.Params;
 import org.infinispan.metadata.Metadata;
 
@@ -71,6 +72,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       output.writeObject(prevMetadata);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      output.writeObject(internalMetadata);
    }
 
    @Override
@@ -87,6 +89,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       prevMetadata = (Metadata) input.readObject();
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      internalMetadata = (MetaParamsInternalMetadata) input.readObject();
    }
 
    @Override

diff --git a/core/src/main/java/org/infinispan/commands/functional/ReadWriteManyCommand.java b/core/src/main/java/org/infinispan/commands/functional/ReadWriteManyCommand.java
@@ -5,6 +5,7 @@
 import java.io.ObjectOutput;
 import java.util.ArrayList;
 import java.util.Collection;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.Function;
 
 import org.infinispan.commands.CommandInvocationId;
@@ -62,6 +63,7 @@ public Function<ReadWriteEntryView<K, V>, R> getFunction() {
 
    public void setKeys(Collection<?> keys) {
       this.keys = keys;
+      this.internalMetadataMap.keySet().retainAll(keys);
    }
 
    public final ReadWriteManyCommand<K, V, R> withKeys(Collection<?> keys) {
@@ -85,6 +87,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       output.writeLong(flags);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      MarshallUtil.marshallMap(internalMetadataMap, output);
    }
 
    @Override
@@ -98,6 +101,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       flags = input.readLong();
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      this.internalMetadataMap = MarshallUtil.unmarshallMap(input, ConcurrentHashMap::new);
    }
 
    public boolean isForwarded() {

diff --git a/core/src/main/java/org/infinispan/commands/functional/ReadWriteManyEntriesCommand.java b/core/src/main/java/org/infinispan/commands/functional/ReadWriteManyEntriesCommand.java
@@ -6,6 +6,7 @@
 import java.util.Collection;
 import java.util.LinkedHashMap;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.BiFunction;
 
 import org.infinispan.commands.CommandInvocationId;
@@ -68,6 +69,7 @@ public BiFunction<T, ReadWriteEntryView<K, V>, R> getBiFunction() {
 
    public void setArguments(Map<?, ?> arguments) {
       this.arguments = arguments;
+      this.internalMetadataMap.keySet().retainAll(arguments.keySet());
    }
 
    public final ReadWriteManyEntriesCommand<K, V, T, R> withArguments(Map<?, ?> entries) {
@@ -91,6 +93,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       output.writeLong(flags);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      MarshallUtil.marshallMap(internalMetadataMap, output);
    }
 
    @Override
@@ -105,6 +108,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       flags = input.readLong();
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      this.internalMetadataMap = MarshallUtil.unmarshallMap(input, ConcurrentHashMap::new);
    }
 
    public boolean isForwarded() {

diff --git a/core/src/main/java/org/infinispan/commands/functional/WriteOnlyKeyCommand.java b/core/src/main/java/org/infinispan/commands/functional/WriteOnlyKeyCommand.java
@@ -16,6 +16,7 @@
 import org.infinispan.encoding.DataConversion;
 import org.infinispan.factories.ComponentRegistry;
 import org.infinispan.functional.EntryView.WriteEntryView;
+import org.infinispan.functional.impl.MetaParamsInternalMetadata;
 import org.infinispan.functional.impl.Params;
 
 public final class WriteOnlyKeyCommand<K, V> extends AbstractWriteKeyCommand<K, V> {
@@ -61,6 +62,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       CommandInvocationId.writeTo(output, commandInvocationId);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      output.writeObject(internalMetadata);
    }
 
    @Override
@@ -74,6 +76,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       commandInvocationId = CommandInvocationId.readFrom(input);
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      internalMetadata = (MetaParamsInternalMetadata) input.readObject();
    }
 
    @Override

diff --git a/core/src/main/java/org/infinispan/commands/functional/WriteOnlyKeyValueCommand.java b/core/src/main/java/org/infinispan/commands/functional/WriteOnlyKeyValueCommand.java
@@ -16,6 +16,7 @@
 import org.infinispan.encoding.DataConversion;
 import org.infinispan.factories.ComponentRegistry;
 import org.infinispan.functional.EntryView.WriteEntryView;
+import org.infinispan.functional.impl.MetaParamsInternalMetadata;
 import org.infinispan.functional.impl.Params;
 
 public final class WriteOnlyKeyValueCommand<K, V, T> extends AbstractWriteKeyCommand<K, V> {
@@ -65,6 +66,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       CommandInvocationId.writeTo(output, commandInvocationId);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      output.writeObject(internalMetadata);
    }
 
    @Override
@@ -79,6 +81,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       commandInvocationId = CommandInvocationId.readFrom(input);
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      internalMetadata = (MetaParamsInternalMetadata) input.readObject();
    }
 
    @Override

diff --git a/core/src/main/java/org/infinispan/commands/functional/WriteOnlyManyCommand.java b/core/src/main/java/org/infinispan/commands/functional/WriteOnlyManyCommand.java
@@ -5,6 +5,7 @@
 import java.io.ObjectOutput;
 import java.util.ArrayList;
 import java.util.Collection;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.Consumer;
 
 import org.infinispan.commands.CommandInvocationId;
@@ -59,6 +60,7 @@ public Consumer<WriteEntryView<K, V>> getConsumer() {
 
    public void setKeys(Collection<?> keys) {
       this.keys = keys;
+      this.internalMetadataMap.keySet().retainAll(keys);
    }
 
    public final WriteOnlyManyCommand<K, V> withKeys(Collection<?> keys) {
@@ -82,6 +84,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       output.writeLong(flags);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      MarshallUtil.marshallMap(internalMetadataMap, output);
    }
 
    @Override
@@ -95,6 +98,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       flags = input.readLong();
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      this.internalMetadataMap = MarshallUtil.unmarshallMap(input, ConcurrentHashMap::new);
    }
 
    @Override

diff --git a/core/src/main/java/org/infinispan/commands/functional/WriteOnlyManyEntriesCommand.java b/core/src/main/java/org/infinispan/commands/functional/WriteOnlyManyEntriesCommand.java
@@ -6,6 +6,7 @@
 import java.util.Collection;
 import java.util.LinkedHashMap;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.BiConsumer;
 
 import org.infinispan.commands.CommandInvocationId;
@@ -64,6 +65,7 @@ public BiConsumer<T, WriteEntryView<K, V>> getBiConsumer() {
 
    public void setArguments(Map<?, ?> arguments) {
       this.arguments = arguments;
+      this.internalMetadataMap.keySet().retainAll(arguments.keySet());
    }
 
    public final WriteOnlyManyEntriesCommand<K, V, T> withArguments(Map<?, ?> entries) {
@@ -87,6 +89,7 @@ public void writeTo(ObjectOutput output) throws IOException {
       output.writeLong(flags);
       DataConversion.writeTo(output, keyDataConversion);
       DataConversion.writeTo(output, valueDataConversion);
+      MarshallUtil.marshallMap(internalMetadataMap, output);
    }
 
    @Override
@@ -101,6 +104,7 @@ public void readFrom(ObjectInput input) throws IOException, ClassNotFoundExcepti
       flags = input.readLong();
       keyDataConversion = DataConversion.readFrom(input);
       valueDataConversion = DataConversion.readFrom(input);
+      this.internalMetadataMap = MarshallUtil.unmarshallMap(input, ConcurrentHashMap::new);
    }
 
    @Override

diff --git a/core/src/main/java/org/infinispan/commands/triangle/PutMapBackupWriteCommand.java b/core/src/main/java/org/infinispan/commands/triangle/PutMapBackupWriteCommand.java
@@ -10,6 +10,7 @@
 import org.infinispan.commands.write.PutMapCommand;
 import org.infinispan.commands.write.WriteCommand;
 import org.infinispan.commons.marshall.MarshallUtil;
+import org.infinispan.functional.impl.MetaParamsInternalMetadata;
 import org.infinispan.metadata.Metadata;
 import org.infinispan.util.ByteString;
 import org.infinispan.util.TriangleFunctionsUtil;
@@ -26,6 +27,7 @@ public class PutMapBackupWriteCommand extends BackupWriteCommand {
 
    private Map<Object, Object> map;
    private Metadata metadata;
+   private Map<Object, MetaParamsInternalMetadata> internalMetadataMap;
 
    //for testing
    @SuppressWarnings("unused")
@@ -47,19 +49,25 @@ public void writeTo(ObjectOutput output) throws IOException {
       writeBase(output);
       MarshallUtil.marshallMap(map, output);
       output.writeObject(metadata);
+      MarshallUtil.marshallMap(internalMetadataMap, output);
    }
 
    @Override
    public void readFrom(ObjectInput input) throws IOException, ClassNotFoundException {
       readBase(input);
       map = MarshallUtil.unmarshallMap(input, HashMap::new);
       metadata = (Metadata) input.readObject();
+      internalMetadataMap = MarshallUtil.unmarshallMap(input, HashMap::new);
    }
 
    public void setPutMapCommand(PutMapCommand command, Collection<Object> keys) {
       setCommonAttributesFromCommand(command);
       this.map = TriangleFunctionsUtil.filterEntries(command.getMap(), keys);
       this.metadata = command.getMetadata();
+      this.internalMetadataMap = new HashMap<>();
+      for (Object key : map.keySet()) {
+         internalMetadataMap.put(key, command.getInternalMetadata(key));
+      }
    }
 
    @Override
@@ -71,13 +79,15 @@ public String toString() {
    WriteCommand createWriteCommand() {
       PutMapCommand cmd = new PutMapCommand(map, metadata, getFlags(), getCommandInvocationId());
       cmd.setForwarded(true);
+      internalMetadataMap.forEach(cmd::setInternalMetadata);
       return cmd;
    }
 
    @Override
    String toStringFields() {
       return super.toStringFields() +
             ", map=" + map +
-            ", metadata=" + metadata;
+            ", metadata=" + metadata +
+            ", internalMetadata=" + internalMetadataMap;
    }
 }