| // Copyright (c) 2017, 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. |
| |
| import 'dart:io'; |
| import 'dart:async'; |
| import 'dart:collection'; |
| |
| class LogdogException implements Exception { |
| final int errorCode; |
| final String stdout; |
| final String stderr; |
| |
| LogdogException(this.errorCode, this.stdout, this.stderr); |
| LogdogException.fromProcessResult(ProcessResult result) |
| : this(result.exitCode, result.stdout, result.stderr); |
| |
| toString() => "Error during logdog execution:\n$stderr"; |
| } |
| |
| bool logdogCheckDone = false; |
| |
| void checkLogdog({bool tryToInstall: true}) { |
| if (logdogCheckDone) return; |
| var result = Process.runSync("cit", []); |
| if (result.exitCode != 0) { |
| print("cit (from depot_tools) must be in the path."); |
| throw new StateError("cit not accessible"); |
| } |
| String stdout = result.stdout; |
| if (stdout.contains("logdog")) { |
| logdogCheckDone = true; |
| return; |
| } |
| if (tryToInstall) { |
| print("logdog isn't yet installed. Installation might take some time"); |
| result = Process.runSync("cit", ["logdog"]); |
| checkLogdog(tryToInstall: false); |
| } else { |
| print("Couldn't install logdog"); |
| throw new StateError("logdog not accessible"); |
| } |
| } |
| |
| String logdog(List<String> args) { |
| checkLogdog(); |
| args = args.toList()..insert(0, "logdog"); |
| var result = Process.runSync("cit", args); |
| if (result.exitCode == 0) return result.stdout; |
| throw new LogdogException.fromProcessResult(result); |
| } |
| |
| String cat(String log) { |
| return logdog(["cat", "-raw", log]); |
| } |
| |
| class LogResult<T> { |
| final String log; |
| final T result; |
| |
| LogResult(this.log, this.result); |
| } |
| |
| const int maxConcurrentLogdogs = 20; |
| |
| /// Fetches the given [logs] concurrently using [logdog]. |
| /// |
| /// At most [maxConcurrentLogdogs] connections are opened at the same time. |
| /// |
| /// The resulting [LogResult] has a [LogResult.result] equal to `null` if |
| /// the log didn't exist. |
| Stream<LogResult<String>> catN(Iterable<String> logs) async* { |
| var queue = new Queue<Future<LogResult<ProcessResult>>>(); |
| var it = logs.iterator; |
| |
| // Launches a new logdog to fetch the next log. |
| // Returns false when nothing was left to enqueue. |
| bool enqueueNext() { |
| if (!it.moveNext()) return false; |
| var log = it.current; |
| queue.add(new Future.sync(() async { |
| var logPath = log.substring(0, log.lastIndexOf("/")); |
| var lsResult = await Process.run("cit", ["logdog", "ls", logPath]); |
| if (lsResult.exitCode != 0) return new LogResult(log, lsResult); |
| if (lsResult.stdout == "") return new LogResult(log, null); |
| return new LogResult( |
| log, await Process.run("cit", ["logdog", "cat", "-raw", log])); |
| })); |
| return true; |
| } |
| |
| for (int i = 0; i < maxConcurrentLogdogs; i++) { |
| enqueueNext(); |
| } |
| |
| while (queue.isNotEmpty) { |
| var logResult = |
| await queue.removeFirst().timeout(const Duration(seconds: 15)); |
| enqueueNext(); |
| if (logResult.result == null) { |
| yield new LogResult(logResult.log, null); |
| } else if (logResult.result.exitCode != 0) { |
| throw new LogdogException.fromProcessResult(logResult.result); |
| } else { |
| yield new LogResult(logResult.log, logResult.result.stdout); |
| } |
| } |
| } |
| |
| /* |
| main() async { |
| // print(cat( |
| // "chromium/bb/client.dart/dart2js-win7-ie11ff-4-4-be/4215/+/recipes/steps/dart2js_ie11_tests/0/stdout")); |
| catN(new Iterable.generate(10, (i) { |
| return "chromium/bb/client.dart/dart2js-win7-ie11ff-4-4-be/" |
| "${4200 + i}" |
| "/+/recipes/steps/dart2js_ie11_tests/0/stdout"; |
| })).listen((logResult) { |
| print("--------------------------"); |
| if (logResult.result == null) { |
| print("${logResult.log} - empty"); |
| } else { |
| print(logResult.result.substring(0, 200)); |
| } |
| }); |
| } |
| */ |