diff options
Diffstat (limited to 'java/dagger/internal/codegen/binding/InjectionSiteFactory.java')
-rw-r--r-- | java/dagger/internal/codegen/binding/InjectionSiteFactory.java | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/java/dagger/internal/codegen/binding/InjectionSiteFactory.java b/java/dagger/internal/codegen/binding/InjectionSiteFactory.java index 5ede20042..36e831bfd 100644 --- a/java/dagger/internal/codegen/binding/InjectionSiteFactory.java +++ b/java/dagger/internal/codegen/binding/InjectionSiteFactory.java @@ -68,10 +68,13 @@ final class InjectionSiteFactory { XTypeElement typeElement = currentType.get().getTypeElement(); enclosingTypeElementOrder.put(typeElement, enclosingTypeElementOrder.size()); for (XElement enclosedElement : typeElement.getEnclosedElements()) { - enclosedElementOrder.put(enclosedElement, enclosedElementOrder.size()); injectionSiteVisitor .visit(enclosedElement, currentType.get()) - .ifPresent(injectionSites::add); + .ifPresent( + injectionSite -> { + enclosedElementOrder.put(enclosedElement, enclosedElementOrder.size()); + injectionSites.add(injectionSite); + }); } } return ImmutableSortedSet.copyOf( |