cache pub list results

pkg/analysis_server/lib/src/source/caching_pub_package_map_provider.dart

+// Copyright (c) 2015, 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 source.caching_pub_package_map_provider;
+
+import 'dart:convert';
+import 'dart:io' as io;
+
+import 'package:analyzer/file_system/file_system.dart';
+import 'package:analyzer/source/package_map_provider.dart';
+import 'package:analyzer/source/pub_package_map_provider.dart';
+import 'package:analyzer/src/generated/engine.dart';
+import 'package:analyzer/src/generated/sdk_io.dart';
+import 'package:analyzer/src/generated/source.dart';
+
+/**
+ * The function used to write the cache file.
+ * Returns the modification stamp for the newly written file.
+ */
+typedef int WriteFile(File file, String content);
+
+/**
+ * [PubPackageMapProvider] extension which caches pub list results.
+ * These results are cached in memory and in a single place on disk that is
+ * shared cross session and between different simultaneous sessions.
+ */
+class CachingPubPackageMapProvider extends PubPackageMapProvider {
+  static const cacheKey = 'pub_list_cache';
+  static const cacheVersion = 1;
+  static const cacheVersionKey = 'pub_list_cache_version';
+  static const pubListResultKey = 'pub_list_result';
+  static const modificationStampsKey = 'modification_stamps';
+
+  /**
+   * A cache of folder path to pub list information as shown below
+   * or `null` if the cache has not yet been initialized.
+   *
+   *     {
+   *       "path/to/folder": {
+   *         "pub_list_result": {
+   *           "packages": {
+   *             "foo": "path/to/foo",
+   *             "bar": ["path/to/bar1", "path/to/bar2"],
+   *             "myapp": "path/to/myapp",  // self link is included
+   *           },
+   *           "input_files": [
+   *             "path/to/myapp/pubspec.lock"
+   *           ]
+   *         },
+   *         "modification_stamps": {
+   *           "path/to/myapp/pubspec.lock": 1424305309
+   *         }
+   *       }
+   *       "path/to/another/folder": {
+   *         ...
+   *       }
+   *       ...
+   *     }
+   */
+  Map<String, Map> _cache;
+
+  /**
+   * The modification time of the cache file
+   * or `null` if it has not yet been read.
+   */
+  int _cacheModificationTime;
+
+  /**
+   * The function used to write the cache file.
+   */
+  WriteFile _writeFile;
+
+  /**
+   * Construct a new instance.
+   * [RunPubList] and [WriteFile] implementations may be injected for testing
+   */
+  CachingPubPackageMapProvider(ResourceProvider resourceProvider,
+      DirectoryBasedDartSdk sdk, [RunPubList runPubList, this._writeFile])
+      : super(resourceProvider, sdk, runPubList) {
+    if (_writeFile == null) {
+      _writeFile = _writeFileDefault;
+    }
+  }
+
+  File get cacheFile => _cacheDir.getChild('cache');
+  Folder get _cacheDir => resourceProvider.getStateLocation('.pub-list');
+  File get _touchFile => _cacheDir.getChild('touch');
+
+  @override
+  PackageMapInfo computePackageMap(Folder folder) {
+    //
+    // Return error if folder does not exist, but don't remove previously
+    // cached result because folder may be only temporarily inaccessible
+    //
+    if (!folder.exists) {
+      return computePackageMapError(folder);
+    }
+    // Ensure cache is up to date
+    _readCache();
+    // Check for cached entry
+    Map entry = _cache[folder.path];
+    if (entry != null) {
+      Map<String, int> modificationStamps = entry[modificationStampsKey];
+      if (modificationStamps != null) {
+        //
+        // Check to see if any dependencies have changed
+        // before returning cached result
+        //
+        bool dependencyChanged = false;
+        modificationStamps.forEach((String path, int modificationStamp) {
+          Resource res = resourceProvider.getResource(path);
+          if (res is File) {
+            if (!res.exists ||
+                res.createSource().modificationStamp != modificationStamp) {
+              dependencyChanged = true;
+            }
+          } else {
+            dependencyChanged = true;
+          }
+        });
+        if (!dependencyChanged) {
+          return parsePackageMap(entry[pubListResultKey], folder);
+        }
+      }
+    }
+    int runCount = 0;
+    PackageMapInfo info;
+    while (true) {
+      // Capture the current time so that we can tell if an input file
+      // has changed while running pub list. This is done
+      // by writing to a file rather than getting millisecondsSinceEpoch
+      // because file modification time has different granularity
+      // on diferent systems.
+      int startStamp;
+      try {
+        startStamp = _writeFile(_touchFile, 'touch');
+      } catch (exception, stackTrace) {
+        AnalysisEngine.instance.logger.logInformation(
+            'Exception writing $_touchFile\n$exception\n$stackTrace');
+        startStamp = new DateTime.now().millisecondsSinceEpoch;
+      }
+      // computePackageMap calls parsePackageMap which caches the result
+      info = super.computePackageMap(folder);
+      ++runCount;
+      if (!_haveDependenciesChangedSince(info, startStamp)) {
+        // If no dependencies have changed while running pub then finished
+        break;
+      }
+      if (runCount == 4) {
+        // Don't run forever
+        AnalysisEngine.instance.logger.logInformation(
+            'pub list called $runCount times: $folder');
+        break;
+      }
+    }
+    _writeCache();
+    return info;
+  }
+
+  @override
+  PackageMapInfo parsePackageMap(Map obj, Folder folder) {
+    PackageMapInfo info = super.parsePackageMap(obj, folder);
+    Map<String, int> modificationStamps = new Map<String, int>();
+    for (String path in info.dependencies) {
+      Resource res = resourceProvider.getResource(path);
+      if (res is File && res.exists) {
+        modificationStamps[path] = res.createSource().modificationStamp;
+      }
+    }
+    // Assumes entry has been initialized by computePackageMap
+    _cache[folder.path] = <String, Map>{
+      pubListResultKey: obj,
+      modificationStampsKey: modificationStamps
+    };
+    return info;
+  }
+
+  /**
+   * Determine if any of the dependencies have changed since the given time.
+   */
+  bool _haveDependenciesChangedSince(PackageMapInfo info, int startStamp) {
+    for (String path in info.dependencies) {
+      Resource res = resourceProvider.getResource(path);
+      if (res is File) {
+        int modStamp = res.createSource().modificationStamp;
+        if (modStamp != null && modStamp >= startStamp) {
+          return true;
+        }
+      }
+    }
+    return false;
+  }
+
+  /**
+   * Read the cache from disk if it has not been read before.
+   */
+  void _readCache() {
+    // TODO(danrubel) This implementation assumes that
+    // two separate processes are not accessing the cache file at the same time
+    Source source = cacheFile.createSource();
+    if (source.exists() &&
+        (_cache == null || _cacheModificationTime != source.modificationStamp)) {
+      try {
+        TimestampedData<String> data = source.contents;
+        Map map = JSON.decode(data.data);
+        if (map[cacheVersionKey] == cacheVersion) {
+          _cache = map[cacheKey];
+          _cacheModificationTime = data.modificationTime;
+        }
+      } catch (exception, stackTrace) {
+        AnalysisEngine.instance.logger.logInformation(
+            'Exception reading $cacheFile\n$exception\n$stackTrace');
+      }
+    }
+    if (_cache == null) {
+      _cache = new Map<String, Map>();
+    }
+  }
+
+  /**
+   * Write the cache to disk.
+   */
+  void _writeCache() {
+    try {
+      _cacheModificationTime = _writeFile(cacheFile, JSON.encode({
+        cacheVersionKey: cacheVersion,
+        cacheKey: _cache
+      }));
+    } catch (exception, stackTrace) {
+      AnalysisEngine.instance.logger.logInformation(
+          'Exception writing $cacheFile\n$exception\n$stackTrace');
+    }
+  }
+
+  /**
+   * Update the given file with the specified content.
+   */
+  int _writeFileDefault(File cacheFile, String content) {
+    // TODO(danrubel) This implementation assumes that
+    // two separate processes are not accessing the cache file at the same time
+    io.File file = new io.File(cacheFile.path);
+    if (!file.parent.existsSync()) {
+      file.parent.createSync(recursive: true);
+    }
+    file.writeAsStringSync(content, flush: true);
+    return file.lastModifiedSync().millisecondsSinceEpoch;
+  }
+}