From 091d4510dbc1d0e147f5692b03992ee51697052a Mon Sep 17 00:00:00 2001
From: "Maxim.Mossienko" <Maxim.Mossienko@jetbrains.com>
Date: Fri, 3 Mar 2017 19:41:30 +0100
Subject: [PATCH] check java's should build stub earlier, in input filter

---
 .../psi/impl/search/JavaNullMethodArgumentIndex.java  | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/java/java-indexing-impl/src/com/intellij/psi/impl/search/JavaNullMethodArgumentIndex.java b/java/java-indexing-impl/src/com/intellij/psi/impl/search/JavaNullMethodArgumentIndex.java
index fa630b924a6a..054fc73f086c 100644
--- a/java/java-indexing-impl/src/com/intellij/psi/impl/search/JavaNullMethodArgumentIndex.java
+++ b/java/java-indexing-impl/src/com/intellij/psi/impl/search/JavaNullMethodArgumentIndex.java
@@ -20,6 +20,7 @@ import com.intellij.lang.LighterAST;
 import com.intellij.lang.LighterASTNode;
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.JavaTokenType;
 import com.intellij.psi.PsiKeyword;
 import com.intellij.psi.impl.java.stubs.JavaStubElementTypes;
@@ -64,9 +65,6 @@ public class JavaNullMethodArgumentIndex extends ScalarIndexExtension<JavaNullMe
   @Override
   public DataIndexer<MethodCallData, Void, FileContent> getIndexer() {
     return inputData -> {
-      if (!JavaStubElementTypes.JAVA_FILE.shouldBuildStubFor(inputData.getFile())) {
-        return Collections.emptyMap();
-      }
       if (myOfflineMode) {
         return Collections.emptyMap();
       }
@@ -182,7 +180,12 @@ public class JavaNullMethodArgumentIndex extends ScalarIndexExtension<JavaNullMe
   @NotNull
   @Override
   public FileBasedIndex.InputFilter getInputFilter() {
-    return new DefaultFileTypeSpecificInputFilter(JavaFileType.INSTANCE);
+    return new DefaultFileTypeSpecificInputFilter(JavaFileType.INSTANCE) {
+      @Override
+      public boolean acceptInput(@NotNull VirtualFile file) {
+        return JavaStubElementTypes.JAVA_FILE.shouldBuildStubFor(file);
+      }
+    };
   }
 
   @Override