Move getElementAt() to CompilationUnitElement.

pkg/analysis_server/lib/src/services/index/store/codec.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.store.codec;
 
 import 'dart:collection';
 
 import 'package:analysis_server/src/services/index/index.dart';
 import 'package:analyzer/src/generated/element.dart';
@@ skipping 63 matching lines @@
 
   /**
    * Returns an [Element] that corresponds to the given identifiers.
    */
   Element decode(AnalysisContext context, int fileId, int offset, int kindId) {
     String filePath = _stringCodec.decode(fileId);
     List<Source> unitSources = context.getSourcesWithFullName(filePath);
     for (Source unitSource in unitSources) {
       List<Source> libSources = context.getLibrariesContaining(unitSource);
       for (Source libSource in libSources) {
-        LibraryElement libraryElement = context.getLibraryElement(libSource);
-        if (libraryElement == null) {
+        CompilationUnitElement unitElement =
+            context.getCompilationUnitElement(unitSource, libSource);
+        if (unitElement == null) {
           return null;
         }
         if (kindId == ElementKind.LIBRARY.ordinal) {
-          return libraryElement;
+          return unitElement.library;
         } else if (kindId == ElementKind.COMPILATION_UNIT.ordinal) {
-          for (CompilationUnitElement unit in libraryElement.units) {
-            if (unit.source.fullName == filePath) {
-              return unit;
-            }
-          }
-          return null;
+          return unitElement;
         } else {
-          Element element = libraryElement.getElementAt(offset);
+          Element element = unitElement.getElementAt(offset);
           if (element == null) {
             return null;
           }
           if (element is ClassElement && kindId <= _CONSTRUCTOR_KIND_BASE) {
             int constructorIndex = -1 * (kindId - _CONSTRUCTOR_KIND_BASE);
             return element.constructors[constructorIndex];
           }
           if (element is PropertyInducingElement) {
             if (kindId == ElementKind.GETTER.ordinal) {
               return element.getter;
@@ skipping 112 matching lines @@
   int encode(String name) {
     int index = nameToIndex[name];
     if (index == null) {
       index = _indexToName.length;
       nameToIndex[name] = index;
       _indexToName.add(name);
     }
     return index;
   }
 }