Function CompressedInputStreamBenchmark

cpp/src/arrow/io/compressed_benchmark.cc:101–143

template <typename BufReader, BufferReadMode Mode>
static void CompressedInputStreamBenchmark(::benchmark::State& state,
                                           Compression::type compression) {
  const int64_t input_size = state.range(0);
  const int64_t batch_size = state.range(1);

  const std::vector<uint8_t> data = MakeCompressibleData(static_cast<int>(input_size));
  auto codec = ::arrow::util::Codec::Create(compression).ValueOrDie();
  int64_t max_compress_len =
      codec->MaxCompressedLen(static_cast<int64_t>(data.size()), data.data());
  std::shared_ptr<::arrow::ResizableBuffer> buf =
      ::arrow::AllocateResizableBuffer(max_compress_len).ValueOrDie();
  const int64_t compressed_length =
      codec
          ->Compress(static_cast<int64_t>(data.size()), data.data(), max_compress_len,
                     buf->mutable_data())
          .ValueOrDie();
  ABORT_NOT_OK(buf->Resize(compressed_length));
  for (auto _ : state) {
    state.PauseTiming();
    auto reader = std::make_shared<BufReader>(buf);
    [[maybe_unused]] std::unique_ptr<Buffer> read_buffer;
    if constexpr (Mode == BufferReadMode::ProvidedByCaller) {
      read_buffer = ::arrow::AllocateBuffer(batch_size).ValueOrDie();
    }
    state.ResumeTiming();
    // Put `CompressedInputStream::Make` in timing.
    auto input_stream =
        ::arrow::io::CompressedInputStream::Make(codec.get(), reader).ValueOrDie();
    auto remaining_size = input_size;
    while (remaining_size > 0) {
      if constexpr (Mode == BufferReadMode::ProvidedByCaller) {
        auto value = input_stream->Read(batch_size, read_buffer->mutable_data());
        ABORT_NOT_OK(value);
        remaining_size -= value.ValueOrDie();
      } else {
        auto value = input_stream->Read(batch_size);
        ABORT_NOT_OK(value);
        remaining_size -= value.ValueOrDie()->size();
      }
    }
  }
  state.SetBytesProcessed(input_size * state.iterations());
}

template <Compression::type kCompression>
static void CompressedInputStreamZeroCopyBufferProvidedByCaller(