Use lookup instead of Map.get

1 files changed, 5 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/util/MammothMaps.java b/src/main/java/org/zwobble/mammoth/internal/util/MammothMaps.java
index 912a454..2f098ce 100644
--- a/src/main/java/org/zwobble/mammoth/internal/util/MammothMaps.java
+++ b/src/main/java/org/zwobble/mammoth/internal/util/MammothMaps.java
@@ -3,6 +3,7 @@ package org.zwobble.mammoth.internal.util;
 import com.google.common.collect.ImmutableMap;
 
 import java.util.Map;
+import java.util.Optional;
 import java.util.function.Function;
 
 public class MammothMaps {
@@ -22,6 +23,10 @@ public class MammothMaps {
         return ImmutableMap.of(key1, value1, key2, value2, key3, value3);
     }
 
+    public static <K, V> Optional<V> lookup(Map<K, V> map, K key) {
+        return Optional.ofNullable(map.get(key));
+    }
+
     public static <T, K, V> Map<K, V> toMap(Iterable<T> iterable, Function<T, Map.Entry<K, V>> function) {
         ImmutableMap.Builder<K, V> builder = ImmutableMap.builder();
         for (T element : iterable) {