main.dart

import 'package:benchmark/benchmark.dart';
import 'package:cbl_flutter/cbl_flutter.dart';
import 'package:cbl_provider/cbl_provider.dart';
import 'package:flutter/material.dart';
import 'package:flutter/services.dart';
import 'package:logging/logging.dart';

Future<void> main() async {
  WidgetsFlutterBinding.ensureInitialized();

  loadDocumentsJson = _loadDocumentsFromAssets;
  await CouchbaseLiteFlutter.init();

  await runBenchmark(
    onProgress: WidgetsBinding.instance.scheduleForcedFrame,
  );
}

Future<String> _loadDocumentsFromAssets() =>
    rootBundle.loadString('packages/benchmark/src/fixture/documents.json');

Future<void> runBenchmark({VoidCallback? onProgress}) {
  final runPlan = BenchmarkPlan(
    warmUpDuration: const FixedTimedDuration(Duration(milliseconds: 200)),
    benchmarkDuration: const FixedTimedDuration(Duration(seconds: 20)),
    runConfigurations: [
      BenchmarkRunConfiguration(
        benchmark: const CreateDocumentBenchmark(),
        databaseProvider: CblProvider(),
        arguments: ParameterArguments((b) {
          b
            ..add(execution, Execution.async)
            ..add(batchSize, 100);
        }),
      )
    ],
  );

  final runner = BenchmarkPlanRunner(
    plan: runPlan,
    logger: Logger.root,
    catchExceptions: true,
    observer: _BenchmarkObserver(onProgress: onProgress),
  )..start();

  return runner.done;
}

class _BenchmarkObserver extends BenchmarkPlanObserver {
  _BenchmarkObserver({this.onProgress});

  final VoidCallback? onProgress;
  int? _lastProgress;

  @override
  void didChangeBenchmarkRunner(
    BenchmarkPlanRunner runner,
    BenchmarkRunConfiguration configuration,
    BenchmarkRunner benchmarkRunner,
  ) {
    final progress = (benchmarkRunner.progress * 100).round();
    if (_lastProgress != progress) {
      print('progress: $progress');
      _lastProgress = progress;
      onProgress?.call();
    }
  }
}