[bench] streams findbyte

glozow · LarryRuane · commit b576994bea62 · 2020-12-04T11:57:09.000-07:00
diff --git a/src/Makefile.bench.include b/src/Makefile.bench.include
@@ -43,7 +43,8 @@ bench_bench_bitcoin_SOURCES = \
   bench/bech32.cpp \
   bench/lockedpool.cpp \
   bench/poly1305.cpp \
-  bench/prevector.cpp
+  bench/prevector.cpp \
+  bench/streams_findbyte.cpp
 
 nodist_bench_bench_bitcoin_SOURCES = $(GENERATED_BENCH_FILES)
 
diff --git a/src/bench/streams_findbyte.cpp b/src/bench/streams_findbyte.cpp
@@ -0,0 +1,33 @@
+// Copyright (c) 2020 The Bitcoin Core developers
+// Distributed under the MIT software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+#include <bench/bench.h>
+#include <streams.h>
+#include <test/util/setup_common.h>
+
+static void FindByte(benchmark::Bench& bench)
+{
+    // Setup
+    FILE* file = fsbridge::fopen("streams_tmp", "w+b");
+    const size_t fileSize = 200;
+    uint8_t b = 0;
+    for (size_t i = 0; i < fileSize; ++i) {
+        fwrite(&b, 1, 1, file);
+    }
+    b = 1;
+    fwrite(&b, 1, 1, file);
+    rewind(file);
+    CBufferedFile bf(file, fileSize * 2, fileSize, 0, 0);
+
+    bench.minEpochIterations(1e7).run([&] {
+        bf.SetPos(0);
+        bf.FindByte(1);
+    });
+
+    // Cleanup
+    bf.fclose();
+    fs::remove("streams_tmp");
+}
+
+BENCHMARK(FindByte);
