Use working cache size when request resolved units in requests.

pkg/analysis_server/lib/src/analysis_server.dart

Index: pkg/analysis_server/lib/src/analysis_server.dart
diff --git a/pkg/analysis_server/lib/src/analysis_server.dart b/pkg/analysis_server/lib/src/analysis_server.dart
index 706b38290e015944473b69b95c1927bd260c19ed..77a5aad95326685998d83f6d95308c5f72b96f06 100644
--- a/pkg/analysis_server/lib/src/analysis_server.dart
+++ b/pkg/analysis_server/lib/src/analysis_server.dart
@@ -585,15 +585,17 @@ class AnalysisServer {
       return units;
     }
     // add a unit for each unit/library combination
-    Source unitSource = contextSource.source;
-    List<Source> librarySources = context.getLibrariesContaining(unitSource);
-    for (Source librarySource in librarySources) {
-      CompilationUnit unit =
-          context.resolveCompilationUnit2(unitSource, librarySource);
-      if (unit != null) {
-        units.add(unit);
+    runWithWorkingCacheSize(context, () {
+      Source unitSource = contextSource.source;
+      List<Source> librarySources = context.getLibrariesContaining(unitSource);
+      for (Source librarySource in librarySources) {
+        CompilationUnit unit =
+            context.resolveCompilationUnit2(unitSource, librarySource);
+        if (unit != null) {
+          units.add(unit);
+        }
       }
-    }
+    });
     // done
     return units;
   }
@@ -1183,7 +1185,9 @@ class AnalysisServer {
       return null;
     }
     // if library has been already resolved, resolve unit
-    return context.resolveCompilationUnit2(source, librarySource);
+    return runWithWorkingCacheSize(context, () {
+      return context.resolveCompilationUnit2(source, librarySource);
+    });
   }
 
   /**