BindingContextUtils.getNotNull()

2012-08-22 14:33:27 +04:00
parent 763e15a567
commit 1c2e8ad550
1 changed files with 23 additions and 0 deletions
@@ -194,4 +194,27 @@ public class BindingContextUtils {

        trace.record(BindingContext.FUNCTION, psiElement, function);
    }
+
+    @NotNull
+    public static <K, V> V getNotNull(
+        @NotNull BindingContext bindingContext,
+        @NotNull ReadOnlySlice<K, V> slice,
+        @NotNull K key
+    ) {
+        return getNotNull(bindingContext, slice, key, "Value at " + slice + " must not be null for " + key);
+    }
+
+    @NotNull
+    public static <K, V> V getNotNull(
+            @NotNull BindingContext bindingContext,
+            @NotNull ReadOnlySlice<K, V> slice,
+            @NotNull K key,
+            @NotNull String messageIfNull
+    ) {
+        V value = bindingContext.get(slice, key);
+        if (value == null) {
+            throw new IllegalStateException(messageIfNull);
+        }
+        return value;
+    }
 }