Index: pkg/analysis_server/lib/src/services/index/store/split_store.dart |
diff --git a/pkg/analysis_server/lib/src/services/index/store/split_store.dart b/pkg/analysis_server/lib/src/services/index/store/split_store.dart |
index 6e3956c32cc3c4f2d4bd98214477e92e9715476d..9c51d6ac0131ff06a7edac36a533e5160cb34e42 100644 |
--- a/pkg/analysis_server/lib/src/services/index/store/split_store.dart |
+++ b/pkg/analysis_server/lib/src/services/index/store/split_store.dart |
@@ -8,7 +8,7 @@ import 'dart:async'; |
import 'dart:collection'; |
import 'dart:typed_data'; |
-import 'package:analysis_server/analysis/index/index_core.dart'; |
+import 'package:analysis_server/analysis/index_core.dart'; |
import 'package:analysis_server/src/analysis_server.dart'; |
import 'package:analysis_server/src/services/index/index.dart'; |
import 'package:analysis_server/src/services/index/index_store.dart'; |
@@ -20,6 +20,7 @@ import 'package:analyzer/src/generated/engine.dart'; |
import 'package:analyzer/src/generated/java_engine.dart'; |
import 'package:analyzer/src/generated/source.dart'; |
import 'package:analyzer/src/generated/utilities_general.dart'; |
+import 'package:analyzer/src/generated/ast.dart' show CompilationUnit; |
/** |
* The implementation of [IndexObjectManager] for indexing |
@@ -40,10 +41,12 @@ class DartUnitIndexObjectManager extends IndexObjectManager { |
@override |
String aboutToIndex(AnalysisContext context, Object object) { |
- if (object is! CompilationUnitElement) { |
- return null; |
+ CompilationUnitElement unitElement; |
+ if (object is CompilationUnit) { |
+ unitElement = object.element; |
+ } else if (object is CompilationUnitElement) { |
+ unitElement = object; |
} |
- CompilationUnitElement unitElement = object; |
// validate unit |
if (unitElement == null) { |
return null; |