Merge pull request #2375 from ItsDoot/api8/datatrans

Add shorthand value fetchers to DataTransactionResult

Author: gabizou

src/main/java/org/spongepowered/api/data/DataTransactionResult.java

@@ -35,6 +35,7 @@
 import java.util.Collection;
 import java.util.List;
 import java.util.Objects;
+import java.util.Optional;
 import java.util.Set;
 import java.util.StringJoiner;
 import java.util.function.BiConsumer;
@@ -332,6 +333,24 @@ public List<Value.Immutable<?>> successfulData() {
         return this.success;
     }
 
+    /**
+     * Gets the successfully applied {@link Value} based on the provided {@link Key}.
+     *
+     * @param key The key
+     * @param <T> The data type
+     * @param <V> The value type
+     * @return The value, if available
+     */
+    @SuppressWarnings("unchecked")
+    public <T, V extends Value<T>> Optional<Value.Immutable<T>> successfulValue(final Key<V> key) {
+        for (final Value.Immutable<?> value : this.successfulData()) {
+            if (value.key() == key) {
+                return Optional.of((Value.Immutable<T>) value);
+            }
+        }
+        return Optional.empty();
+    }
+
     /**
      * If {@link Value.Mutable}s were supplied to the operation, this
      * collection will return any {@link Value.Immutable}s which were rejected
@@ -343,6 +362,24 @@ public List<Value.Immutable<?>> rejectedData() {
         return this.rejected;
     }
 
+    /**
+     * Gets the rejected {@link Value} based on the provided {@link Key}.
+     *
+     * @param key The key
+     * @param <T> The data type
+     * @param <V> The value type
+     * @return The value, if available
+     */
+    @SuppressWarnings("unchecked")
+    public <T, V extends Value<T>> Optional<Value.Immutable<T>> rejectedValue(final Key<V> key) {
+        for (final Value.Immutable<?> value : this.rejectedData()) {
+            if (value.key() == key) {
+                return Optional.of((Value.Immutable<T>) value);
+            }
+        }
+        return Optional.empty();
+    }
+
     /**
      * If the operation replaced any {@link Value.Mutable}s, this returns a collection
      * of the replaced {@link Value.Immutable}s.
@@ -353,6 +390,24 @@ public List<Value.Immutable<?>> replacedData() {
         return this.replaced;
     }
 
+    /**
+     * Gets the replaced {@link Value} based on the provided {@link Key}.
+     *
+     * @param key The key
+     * @param <T> The data type
+     * @param <V> The value type
+     * @return The value, if available
+     */
+    @SuppressWarnings("unchecked")
+    public <T, V extends Value<T>> Optional<Value.Immutable<T>> replacedValue(final Key<V> key) {
+        for (final Value.Immutable<?> value : this.replacedData()) {
+            if (value.key() == key) {
+                return Optional.of((Value.Immutable<T>) value);
+            }
+        }
+        return Optional.empty();
+    }
+
     /**
      * If this result of {@link #isSuccessful()} returns {@code true},
      * the provided {@link Consumer} is called provided a list of all