如何为Google Benchmark传递输入文件？相关技术疑问

核心实现方案：分离自定义参数与Benchmark内置参数 #

Google Benchmark支持保留自定义命令行参数，无需修改原始argc/argv的核心逻辑，只需在初始化Benchmark前提前提取自己的文件路径参数，剩余参数留给Benchmark处理即可，步骤如下：

1. 提前解析自定义参数：在调用benchmark::Initialize前，遍历argv提取所有文件路径，将非Benchmark内置参数（以--开头的参数属于Benchmark内置）单独存储，调整argc和argv后再传给Benchmark初始化。
2. 注册带参数的基准测试：利用benchmark::RegisterBenchmark的重载版本，将提取到的文件路径关联到测试函数中。

示例代码 #

#include <benchmark/benchmark.h>
#include <vector>
#include <string>

// 存储从命令行提取的文件路径
std::vector<std::string> g_input_files;

// 基准测试函数，通过state索引获取对应文件路径
void BM_ProcessFile(benchmark::State& state) {
    const std::string& file_path = g_input_files[state.range(0)];
    // 这里编写你的文件处理逻辑：读取文件、执行被测代码
    for (auto _ : state) {
        // 被测代码：处理file_path对应的文件内容
    }
}

int main(int argc, char** argv) {
    // 第一步：提取自定义文件路径参数
    int i = 1;
    for (; i < argc; ++i) {
        std::string arg = argv[i];
        // 遇到Benchmark内置参数（--开头）则停止提取
        if (arg.starts_with("--")) {
            // 调整argc和argv，将剩余参数交给Benchmark处理
            argc -= (i - 1);
            argv += (i - 1);
            break;
        }
        g_input_files.push_back(arg);
    }
    // 处理所有参数都是自定义文件的情况
    if (i == argc) {
        argc = 1;
        argv += (i - 1);
    }

    // 初始化Google Benchmark
    benchmark::Initialize(&argc, argv);
    if (argc != 1) {
        benchmark::ReportUnrecognizedArguments(argc, argv);
        return 1;
    }

    // 注册基准测试，range范围对应文件索引
    benchmark::RegisterBenchmark("ProcessFile", BM_ProcessFile)
        ->ArgNames({"FileIndex"})
        ->Range(0, static_cast<int>(g_input_files.size()) - 1);

    benchmark::RunSpecifiedBenchmarks();
    benchmark::Shutdown();
    return 0;
}

运行时直接执行：

./benchmark file1.txt file2.txt file3.txt

也可以叠加Benchmark内置参数：

./benchmark file1.txt file2.txt --benchmark_repetitions=5 --benchmark_time_unit=ms

如果担心文件名与Benchmark参数冲突，可约定自定义参数前缀（比如-f），修改解析逻辑：

int i = 1;
for (; i < argc; ++i) {
    std::string arg = argv[i];
    if (arg == "-f" && i + 1 < argc) {
        g_input_files.push_back(argv[i+1]);
        i++; // 跳过已读取的文件名参数
    } else if (arg.starts_with("--")) {
        argc -= (i - 1);
        argv += (i - 1);
        break;
    }
}

对应运行命令：

./benchmark -f file1.txt -f file2.txt --benchmark_repetitions=5

你的疑问解答 #

• 是否属于禁止/非预期场景？
  完全不是。Google Benchmark的设计允许用户传入外部数据，动态加载测试数据集属于合理且常见的使用场景。
• 是否必须硬编码文件路径？
  不需要硬编码。上述方案可动态接收命令行传入的文件路径，大幅提升测试程序的复用性。
• 传递外部数据是否不推荐？
  传递外部数据是推荐做法，尤其是针对不同数据集做对比测试时，硬编码会导致测试程序灵活性极差，动态传入数据更符合基准测试的设计初衷。

内容的提问来源于stack exchange，提问作者user13840624