current_results_ui/lib/query.dart

// Copyright (c) 2020, 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:async';

import 'package:flutter/material.dart';
import 'package:http/http.dart' as http;
import 'dart:convert';

import 'src/generated/query.pb.dart';
import 'filter.dart';

const String apiHost = 'current-results-rest-zlujsyuhha-uc.a.run.app';
// Current endpoints proxy is limited to 1 MB response size,
// so we limit results fetched to 4000.
// Implement paging on the service in the future to remove this limit.
const int fetchLimit = 4000;
const int maxFetchedResults = 100 * fetchLimit;

class QueryResults extends ChangeNotifier {
  final Filter filter;
  StreamSubscription<GetResultsResponse> fetcher;
  List<String> names = [];
  Map<String, Map<String, int>> counts = {};
  Map<String, Map<ChangeInResult, List<Result>>> grouped = {};
  int fetchedResultsCount = 0;
  bool get noQuery => filter.terms.isEmpty;

  QueryResults(this.filter);

  @override
  void dispose() {
    fetcher?.cancel();
    super.dispose();
  }

  GetResultsResponse resultsObject = GetResultsResponse.create();

  void fetchCurrentResults() async {
    fetcher?.cancel();
    names = [];
    counts = {};
    grouped = {};
    fetchedResultsCount = 0;

    fetcher = fetchResults(filter).listen(onResults, onDone: onDone);
  }

  void onResults(GetResultsResponse response) {
    final results = response.results;
    fetchedResultsCount += results.length;
    if (fetchedResultsCount >= maxFetchedResults) {
      fetcher.cancel();
      fetcher = null;
    }
    for (final result in results) {
      final change = ChangeInResult(result);
      grouped
          .putIfAbsent(result.name, () => <ChangeInResult, List<Result>>{})
          .putIfAbsent(change, () => <Result>[])
          .add(result);
      counts
          .putIfAbsent(result.name, () => <String, int>{})
          .putIfAbsent(change.kind, () => 0);
      ++counts[result.name][change.kind];
    }
    names = grouped.keys.toList()..sort();
    notifyListeners();
  }

  void onDone() {
    fetcher = null;
  }
}

Stream<GetResultsResponse> fetchResults(Filter filter) async* {
  final client = http.Client();
  var pageToken = '';
  do {
    final resultsQuery = Uri.https(apiHost, 'v1/results', {
      'filter': filter.terms.join(','),
      'pageSize': '$fetchLimit',
      'pageToken': pageToken
    });
    final response = await client.get(resultsQuery);
    final results = GetResultsResponse.create()
      ..mergeFromProto3Json(json.decode(response.body));
    yield results;
    pageToken = results.nextPageToken;
  } while (pageToken.isNotEmpty);
}

class ChangeInResult {
  final String result;
  final String expected;
  final bool flaky;
  final String text;
  bool get matches => result == expected;
  String get kind => flaky
      ? 'flaky'
      : matches
          ? 'pass'
          : 'fail';

  ChangeInResult(Result result)
      : this._(result.result, result.expected, result.flaky);

  ChangeInResult._(this.result, this.expected, this.flaky)
      : text = flaky
            ? "flaky (latest result $result expected $expected"
            : "$result (expected $expected)";

  @override
  String toString() => text;
  @override
  bool operator ==(Object other) => text == (other as ChangeInResult)?.text;
  @override
  int get hashCode => text.hashCode;
}

String resultAsCommaSeparated(Result result) => [
      result.name,
      result.configuration,
      result.result,
      result.expected,
      result.flaky,
      result.timeMs
    ].join(',');

String resultTextHeader = "name,configuration,result,expected,flaky,timeMs";