aboutsummaryrefslogtreecommitdiff
path: root/java/dagger/internal/codegen/binding/InjectionSiteFactory.java
diff options
context:
space:
mode:
Diffstat (limited to 'java/dagger/internal/codegen/binding/InjectionSiteFactory.java')
-rw-r--r--java/dagger/internal/codegen/binding/InjectionSiteFactory.java7
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(