test(storage): benchmark option to disable retry loop (#9597)

Author: coryan

google/cloud/storage/benchmarks/storage_throughput_vs_cpu_benchmark.cc

@@ -297,6 +297,10 @@ gcs_bm::ClientProvider BaseProvider(ThroughputOptions const& options) {
                         options.download_stall_timeout)
                     .set<gcs::TransferStallTimeoutOption>(
                         options.transfer_stall_timeout);
+    if (!options.enable_retry_loop) {
+      opts.set<gcs::RetryPolicyOption>(
+          gcs::LimitedErrorCountRetryPolicy(0).clone());
+    }
     if (options.target_api_version_path.has_value()) {
       opts.set<google::cloud::storage::internal::TargetApiVersionOption>(
           *options.target_api_version_path);

google/cloud/storage/benchmarks/throughput_options.cc

@@ -396,6 +396,10 @@ google::cloud::StatusOr<ThroughputOptions> ParseThroughputOptions(
        [&options](std::string const& val) {
          options.grpc_background_threads = std::stoi(val);
        }},
+      {"--enable-retry-loop", "use the client library retry loop",
+       [&options](std::string const& val) {
+         options.enable_retry_loop = ParseBoolean(val).value_or(true);
+       }},
   };
   auto usage = BuildUsage(desc, argv[0]);
 

google/cloud/storage/benchmarks/throughput_options.h

@@ -71,6 +71,7 @@ struct ThroughputOptions {
   std::int64_t read_size_quantum = 128 * kKiB;
   absl::optional<std::string> target_api_version_path;
   absl::optional<int> grpc_background_threads;
+  bool enable_retry_loop = true;
 };
 
 google::cloud::StatusOr<ThroughputOptions> ParseThroughputOptions(

google/cloud/storage/benchmarks/throughput_options_test.cc

@@ -66,6 +66,7 @@ TEST(ThroughputOptions, Basic) {
       "--read-size-quantum=16KiB",
       "--target-api-version-path=vN",
       "--grpc-background-threads=16",
+      "--enable-retry-loop=false",
   });
   ASSERT_STATUS_OK(options);
   EXPECT_EQ("test-project", options->project_id);
@@ -103,6 +104,7 @@ TEST(ThroughputOptions, Basic) {
   EXPECT_EQ(std::chrono::milliseconds(250), options->minimum_sample_delay);
   EXPECT_EQ("vN", options->target_api_version_path.value_or(""));
   EXPECT_EQ(16, options->grpc_background_threads.value_or(0));
+  EXPECT_FALSE(options->enable_retry_loop);
 }
 
 TEST(ThroughputOptions, Description) {