Optimize top-level element declarations search.

pkg/analysis_server/lib/src/services/index/index_contributor.dart

 // Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library services.src.index.index_contributor;
 
 import 'dart:collection' show Queue;
 
 import 'package:analysis_server/src/services/correction/namespace.dart';
 import 'package:analysis_server/src/services/index/index.dart';
@@ skipping 113 matching lines @@
   visitBinaryExpression(BinaryExpression node) {
     _recordOperatorReference(node.operator, node.bestElement);
     super.visitBinaryExpression(node);
   }
 
   @override
   visitClassDeclaration(ClassDeclaration node) {
     ClassElement element = node.element;
     enterScope(element);
     try {
-      _recordElementDefinition(element);
+      _recordTopLevelElementDefinition(element);
       {
         ExtendsClause extendsClause = node.extendsClause;
         if (extendsClause != null) {
           TypeName superclassNode = extendsClause.superclass;
           _recordSuperType(superclassNode, IndexConstants.IS_EXTENDED_BY);
         } else {
           InterfaceType superType = element.supertype;
           if (superType != null) {
             ClassElement objectElement = superType.element;
             recordRelationship(
@@ skipping 23 matching lines @@
     } finally {
       _exitScope();
     }
   }
 
   @override
   visitClassTypeAlias(ClassTypeAlias node) {
     ClassElement element = node.element;
     enterScope(element);
     try {
-      _recordElementDefinition(element);
+      _recordTopLevelElementDefinition(element);
       {
         TypeName superclassNode = node.superclass;
         if (superclassNode != null) {
           _recordSuperType(superclassNode, IndexConstants.IS_EXTENDED_BY);
         }
       }
       {
         WithClause withClause = node.withClause;
         if (withClause != null) {
           for (TypeName mixinNode in withClause.mixinTypes) {
@@ skipping 90 matching lines @@
     } finally {
       _exitScope();
     }
   }
 
   @override
   visitEnumDeclaration(EnumDeclaration node) {
     ClassElement element = node.element;
     enterScope(element);
     try {
-      _recordElementDefinition(element);
+      _recordTopLevelElementDefinition(element);
       super.visitEnumDeclaration(node);
     } finally {
       _exitScope();
     }
   }
 
   @override
   visitExportDirective(ExportDirective node) {
     ExportElement element = node.element;
     if (element != null) {
@@ skipping 10 matching lines @@
     try {
       super.visitFormalParameter(node);
     } finally {
       _exitScope();
     }
   }
 
   @override
   visitFunctionDeclaration(FunctionDeclaration node) {
     Element element = node.element;
-    _recordElementDefinition(element);
+    _recordTopLevelElementDefinition(element);
     enterScope(element);
     try {
       super.visitFunctionDeclaration(node);
     } finally {
       _exitScope();
     }
   }
 
   @override
   visitFunctionTypeAlias(FunctionTypeAlias node) {
     Element element = node.element;
-    _recordElementDefinition(element);
+    _recordTopLevelElementDefinition(element);
     super.visitFunctionTypeAlias(node);
   }
 
   @override
   visitImportDirective(ImportDirective node) {
     ImportElement element = node.element;
     if (element != null) {
       LibraryElement impLibrary = element.importedLibrary;
       _recordLibraryReference(node, impLibrary);
     }
@@ skipping 183 matching lines @@
     }
     recordRelationship(element, IndexConstants.IS_REFERENCED_BY, location);
     super.visitSuperConstructorInvocation(node);
   }
 
   @override
   visitTopLevelVariableDeclaration(TopLevelVariableDeclaration node) {
     VariableDeclarationList variables = node.variables;
     for (VariableDeclaration variableDeclaration in variables.variables) {
       Element element = variableDeclaration.element;
-      _recordElementDefinition(element);
+      _recordTopLevelElementDefinition(element);
     }
     super.visitTopLevelVariableDeclaration(node);
   }
 
   @override
   visitTypeParameter(TypeParameter node) {
     TypeParameterElement element = node.element;
     enterScope(element);
     try {
       super.visitTypeParameter(node);
@@ skipping 123 matching lines @@
             target.staticElement is PrefixElement) {
           return false;
         }
         return target != null;
       }
     }
     return false;
   }
 
   /**
-   * Records the [Element] definition in the library and universe.
-   */
-  void _recordElementDefinition(Element element) {
-    Location location = createLocation(element);
-    Relationship relationship = IndexConstants.DEFINES;
-    recordRelationship(_libraryElement, relationship, location);
-    recordRelationship(UniverseElement.INSTANCE, relationship, location);
-  }
-
-  /**
    * Records [ImportElement] reference if given [SimpleIdentifier] references some
    * top-level element and not qualified with import prefix.
    */
   void _recordImportElementReferenceWithoutPrefix(SimpleIdentifier node) {
     if (_isIdentifierInImportCombinator(node)) {
       return;
     }
     if (_isIdentifierInPrefixedIdentifier(node)) {
       return;
     }
@@ skipping 77 matching lines @@
         Element superElement = superName.staticElement;
         recordRelationship(
             superElement,
             relationship,
             _createLocationForNode(superNode));
       }
     }
   }
 
   /**
+   * Records the [Element] definition in the library and universe.
+   */
+  void _recordTopLevelElementDefinition(Element element) {
+    Location location = createLocation(element);
+    recordRelationship(_libraryElement, IndexConstants.DEFINES, location);
+    _store.recordTopDeclaration(element);
+  }
+
+  /**
    * Creates a [Location] representing declaration of the [Element].
    */
   static Location createLocation(Element element) {
     if (element != null) {
       int offset = element.nameOffset;
       int length = element.displayName.length;
       return new Location(element, offset, length);
     }
     return null;
   }
@@ skipping 21 matching lines @@
   }
 
   /**
    * @return `true` if given "node" is part of [PrefixedIdentifier] "prefix.node".
    */
   static bool _isIdentifierInPrefixedIdentifier(SimpleIdentifier node) {
     AstNode parent = node.parent;
     return parent is PrefixedIdentifier && parent.identifier == node;
   }
 }