ISSUES-17 allow sibling binding

jbock-java · Dec 3, 2023 · 837587d · 837587d
1 parent 8e59e4a
commit 837587d
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/compiler/src/main/java/io/jbock/simple/processor/binding/InjectBindingFactory.java b/compiler/src/main/java/io/jbock/simple/processor/binding/InjectBindingFactory.java
@@ -48,14 +48,12 @@ private Map<Key, InjectBinding> injectBindingsMiss(TypeElement typeElement) {
     }
 
     public Optional<Binding> binding(Key key) {
-        return tool.types().asElement(key.type()).flatMap(element -> {
-            TypeElement typeElement = TYPE_ELEMENT_VISITOR.visit(element);
-            if (typeElement == null) {
-                return Optional.empty();
-            }
-            Map<Key, InjectBinding> m = injectBindings(typeElement);
-            return Optional.ofNullable(m.get(key));
-        });
+        return tool.types().asElement(key.type())
+                .map(TYPE_ELEMENT_VISITOR::visit)
+                .flatMap(typeElement -> {
+                    Map<Key, InjectBinding> m = injectBindings(typeElement);
+                    return Optional.ofNullable(m.get(key));
+                });
     }
 
     @Override