Support for analysis.setPriorityFiles for files in SDK.

pkg/analysis_server/lib/src/analysis_server.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 analysis.server;
 
 import 'dart:async';
 import 'dart:collection';
 import 'dart:core' hide Resource;
 import 'dart:math' show max;
@@ skipping 165 matching lines @@
    * A table mapping [AnalysisService]s to the file paths for which these
    * notifications should be sent.
    */
   Map<AnalysisService, Set<String>> analysisServices =
       new HashMap<AnalysisService, Set<String>>();
 
   /**
    * A table mapping [AnalysisContext]s to the completers that should be
    * completed when analysis of this context is finished.
    */
-  Map<AnalysisContext, Completer<AnalysisDoneReason>> contextAnalysisDoneCompleters =
+  Map<AnalysisContext,
+          Completer<AnalysisDoneReason>> contextAnalysisDoneCompleters =
       new HashMap<AnalysisContext, Completer<AnalysisDoneReason>>();
 
   /**
    * Performance information before initial analysis is complete.
    */
   ServerPerformance performanceDuringStartup = new ServerPerformance();
 
   /**
    * Performance information after initial analysis is complete
    * or `null` if the initial analysis is not yet complete
@@ skipping 87 matching lines @@
    * responses to the given [channel].
    *
    * If a [contextManager] is provided, then the [packageResolverProvider] will
    * be ignored.
    *
    * If [rethrowExceptions] is true, then any exceptions thrown by analysis are
    * propagated up the call stack.  The default is true to allow analysis
    * exceptions to show up in unit tests, but it should be set to false when
    * running a full analysis server.
    */
-  AnalysisServer(this.channel, this.resourceProvider,
-      PubPackageMapProvider packageMapProvider, Index _index, this.serverPlugin,
-      this.options, this.defaultSdk, this.instrumentationService,
+  AnalysisServer(
+      this.channel,
+      this.resourceProvider,
+      PubPackageMapProvider packageMapProvider,
+      Index _index,
+      this.serverPlugin,
+      this.options,
+      this.defaultSdk,
+      this.instrumentationService,
       {ContextManager contextManager: null,
       ResolverProvider packageResolverProvider: null,
       this.rethrowExceptions: true})
       : index = _index,
         searchEngine = _index != null ? createSearchEngine(_index) : null {
     _performance = performanceDuringStartup;
     operationQueue = new ServerOperationQueue();
     if (contextManager == null) {
       contextManager = new ContextManagerImpl(resourceProvider,
           packageResolverProvider, packageMapProvider, instrumentationService);
@@ skipping 176 matching lines @@
    * context and `null` [Source] is returned.
    *
    * Does not return `null`.
    */
   ContextSourcePair getContextSourcePair(String path) {
     // try SDK
     {
       Uri uri = resourceProvider.pathContext.toUri(path);
       Source sdkSource = defaultSdk.fromFileUri(uri);
       if (sdkSource != null) {
-        AnalysisContext anyContext = folderMap.values.first;
-        return new ContextSourcePair(anyContext, sdkSource);
+        AnalysisContext sdkContext = defaultSdk.context;
+        return new ContextSourcePair(sdkContext, sdkSource);
       }
     }
     // try to find the deep-most containing context
     Resource resource = resourceProvider.getResource(path);
     if (resource is! File) {
       return new ContextSourcePair(null, null);
     }
     File file = resource;
     {
       AnalysisContext containingContext = getContainingContext(path);
@@ skipping 527 matching lines @@
               preferredContext = context;
               source =
                   ContextManagerImpl.createSourceInContext(context, resource);
               break;
             }
           }
         }
       }
       // Fill the source map.
       bool contextFound = false;
+      if (preferredContext != null) {
+        sourceMap.putIfAbsent(preferredContext, () => <Source>[]).add(source);
+        contextFound = true;
+      }
       for (AnalysisContext context in folderMap.values) {
-        if (context == preferredContext ||
-            context.getKindOf(source) != SourceKind.UNKNOWN) {
+        if (context.getKindOf(source) != SourceKind.UNKNOWN) {
           sourceMap.putIfAbsent(context, () => <Source>[]).add(source);
           contextFound = true;
         }
       }
       if (firstSource == null) {
         firstSource = source;
       }
       if (!contextFound) {
         unanalyzed.add(file);
       }
     });
     if (unanalyzed.isNotEmpty) {
       StringBuffer buffer = new StringBuffer();
       buffer.writeAll(unanalyzed, ', ');
       throw new RequestFailure(
           new Response.unanalyzedPriorityFiles(requestId, buffer.toString()));
     }
-    folderMap.forEach((Folder folder, AnalysisContext context) {
-      List<Source> sourceList = sourceMap[context];
-      if (sourceList == null) {
-        sourceList = Source.EMPTY_LIST;
-      }
+    sourceMap.forEach((context, List<Source> sourceList) {
       context.analysisPriorityOrder = sourceList;
       // Schedule the context for analysis so that it has the opportunity to
       // cache the AST's for the priority sources as soon as possible.
       schedulePerformAnalysisOperation(context);
     });
     operationQueue.reschedule();
     _onPriorityChangeController.add(new PriorityChangeEvent(firstSource));
   }
 
   /**
@@ skipping 378 matching lines @@
     resolvers.add(resourceResolver);
     return new SourceFactory(resolvers, disposition.packages);
   }
 }
 
 /**
  * A class used by [AnalysisServer] to record performance information
  * such as request latency.
  */
 class ServerPerformance {
-
   /**
    * The creation time and the time when performance information
    * started to be recorded here.
    */
   int startTime = new DateTime.now().millisecondsSinceEpoch;
 
   /**
    * The number of requests.
    */
   int requestCount = 0;
@@ skipping 76 matching lines @@
   /**
    * The [PerformanceTag] for time spent in server request handlers.
    */
   static PerformanceTag serverRequests = new PerformanceTag('serverRequests');
 
   /**
    * The [PerformanceTag] for time spent in split store microtasks.
    */
   static PerformanceTag splitStore = new PerformanceTag('splitStore');
 }