Use working cache size when request resolved units in requests.

pkg/analysis_server/lib/src/operation/operation_analysis.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 operation.analysis;
 
 import 'package:analysis_server/src/analysis_server.dart';
 import 'package:analysis_server/src/computer/computer_highlights.dart';
 import 'package:analysis_server/src/computer/computer_navigation.dart';
 import 'package:analysis_server/src/computer/computer_occurrences.dart';
 import 'package:analysis_server/src/computer/computer_outline.dart';
 import 'package:analysis_server/src/computer/computer_overrides.dart';
 import 'package:analysis_server/src/operation/operation.dart';
 import 'package:analysis_server/src/protocol_server.dart' as protocol;
 import 'package:analysis_server/src/services/index/index.dart';
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/error.dart';
 import 'package:analyzer/src/generated/html.dart';
 import 'package:analyzer/src/generated/source.dart';
 
 /**
+ * Runs the given function [f] with the working cache size in [context].
+ * Returns the result of [f] invocation.
+ */
+runWithWorkingCacheSize(AnalysisContext context, f()) {
+  int currentCacheSize = context.analysisOptions.cacheSize;
+  if (currentCacheSize < PerformAnalysisOperation.WORKING_CACHE_SIZE) {
+    setCacheSize(context, PerformAnalysisOperation.WORKING_CACHE_SIZE);
+    try {
+      return f();
+    } finally {
+      setCacheSize(context, currentCacheSize);
+    }
+  } else {
+    return f();
+  }
+}
+
+/**
  * Schedules indexing of the given [file] using the resolved [dartUnit].
  */
 void scheduleIndexOperation(AnalysisServer server, String file,
     AnalysisContext context, CompilationUnit dartUnit) {
   if (server.index != null) {
     server.addOperation(new _DartIndexOperation(context, file, dartUnit));
   }
 }
 
 /**
@@ skipping 112 matching lines @@
 void sendAnalysisNotificationOverrides(
     AnalysisServer server, String file, CompilationUnit dartUnit) {
   _sendNotification(server, () {
     var overrides = new DartUnitOverridesComputer(dartUnit).compute();
     var params = new protocol.AnalysisOverridesParams(file, overrides);
     server.sendNotification(params.toNotification());
   });
 }
 
 /**
+ * Sets the cache size in the given [context] to the given value.
+ */
+void setCacheSize(AnalysisContext context, int cacheSize) {
+  AnalysisOptionsImpl options =
+      new AnalysisOptionsImpl.con1(context.analysisOptions);
+  options.cacheSize = cacheSize;
+  context.analysisOptions = options;
+}
+
+/**
  * Runs the given notification producing function [f], catching exceptions.
  */
 void _sendNotification(AnalysisServer server, f()) {
   ServerPerformanceStatistics.notices.makeCurrentWhile(() {
     try {
       f();
     } catch (exception, stackTrace) {
       server.sendServerErrorNotification(exception, stackTrace);
     }
   });
@@ skipping 35 matching lines @@
   void perform(AnalysisServer server) {
     //
     // TODO(brianwilkerson) Add an optional function-valued parameter to
     // performAnalysisTask that will be called when the task has been computed
     // but before it is performed and send notification in the function:
     //
     // AnalysisResult result = context.performAnalysisTask((taskDescription) {
     //   sendStatusNotification(context.toString(), taskDescription);
     // });
     if (!isContinue) {
-      _setCacheSize(WORKING_CACHE_SIZE);
+      setCacheSize(context, WORKING_CACHE_SIZE);
     }
     // prepare results
     AnalysisResult result = context.performAnalysisTask();
     List<ChangeNotice> notices = result.changeNotices;
     if (notices == null) {
-      _setCacheSize(IDLE_CACHE_SIZE);
+      setCacheSize(context, IDLE_CACHE_SIZE);
       server.sendContextAnalysisDoneNotifications(
           context, AnalysisDoneReason.COMPLETE);
       return;
     }
     // process results
     ServerPerformanceStatistics.notices.makeCurrentWhile(() {
       _sendNotices(server, notices);
       _updateIndex(server, notices);
     });
     // continue analysis
@@ skipping 11 matching lines @@
       // Dart
       CompilationUnit parsedDartUnit = notice.parsedDartUnit;
       CompilationUnit resolvedDartUnit = notice.resolvedDartUnit;
       scheduleNotificationOperations(server, file, notice.lineInfo, context,
           parsedDartUnit, resolvedDartUnit, notice.errors);
       // done
       server.fileAnalyzed(notice);
     }
   }
 
-  void _setCacheSize(int cacheSize) {
-    AnalysisOptionsImpl options =
-        new AnalysisOptionsImpl.con1(context.analysisOptions);
-    options.cacheSize = cacheSize;
-    context.analysisOptions = options;
-  }
-
   void _updateIndex(AnalysisServer server, List<ChangeNotice> notices) {
     if (server.index == null) {
       return;
     }
     for (ChangeNotice notice in notices) {
       String file = notice.source.fullName;
       // Dart
       try {
         CompilationUnit dartUnit = notice.resolvedDartUnit;
         if (dartUnit != null) {
@@ skipping 144 matching lines @@
 abstract class _SingleFileOperation extends SourceSensitiveOperation {
   final String file;
 
   _SingleFileOperation(AnalysisContext context, this.file) : super(context);
 
   @override
   bool shouldBeDiscardedOnSourceChange(Source source) {
     return source.fullName == file;
   }
 }