src/tests/benchmarks/fidl/rust/gen_benchmark_suite_lib.py

#!/usr/bin/env fuchsia-vendored-python
# Copyright 2018 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# Generates a rust file that combines multiple GIDL-generated benchmarks that export a
# BENCHMARK array of benchmark functions into an ALL_BENCHMARKS array that is used by
# src/tests/benchmarks/fidl/rust/src/main.rs.

# Example output:
# use fuchsia_criterion::criterion::Bencher;
#
# use benchmarkfidltable;
# use benchmarkfidlunion;
#
# pub const ALL_BENCHMARKS: [&\'static[(&\'static str, fn(&mut Bencher))]; %d] = [
#  &benchmarkfidltable::BENCHMARKS,
#  &benchmarkfidlunion::BENCHMARKS,
# ];

import sys


def main():
    if len(sys.argv) < 2:
        print(
            "expect at least 1 argument: [output file] ([package name])*",
            file=sys.stderr,
        )
        return 2

    output_file = sys.argv[1]
    packages = sys.argv[2:]

    with open(output_file, "w") as fo:
        # TODO(https://fxbug.dev/326330182): clean up unused imports.
        fo.write("#![allow(unused_imports)]")
        fo.write("use fuchsia_criterion::criterion::Bencher;\n")
        for package in packages:
            fo.write("use %s;\n" % package)
        fo.write(
            "\npub const ALL_BENCHMARKS: [&'static[(&'static str, fn(&mut Bencher))]; %d] = [\n"
            % len(packages)
        )
        for package in packages:
            fo.write("\t&%s::BENCHMARKS,\n" % package)
        fo.write("];\n")


if __name__ == "__main__":
    sys.exit(main())