From c1b480bedd54ed056cdf5ff2094c3d511ec2700b Mon Sep 17 00:00:00 2001
From: sfc-gh-tclinkenbeard <trevor.clinkenbeard@snowflake.com>
Date: Sun, 12 Dec 2021 20:15:25 -0800
Subject: [PATCH] Add bench_net2 benchmark to flowbench

---
 flowbench/BenchNet2.actor.cpp | 63 +++++++++++++++++++++++++++++++++++
 flowbench/CMakeLists.txt      |  1 +
 2 files changed, 64 insertions(+)
 create mode 100644 flowbench/BenchNet2.actor.cpp

diff --git a/flowbench/BenchNet2.actor.cpp b/flowbench/BenchNet2.actor.cpp
new file mode 100644
index 0000000000..7f788c722c
--- /dev/null
+++ b/flowbench/BenchNet2.actor.cpp
@@ -0,0 +1,63 @@
+/*
+ * BenchNet2.actor.cpp
+ *
+ * This source file is part of the FoundationDB open source project
+ *
+ * Copyright 2013-2020 Apple Inc. and the FoundationDB project authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "benchmark/benchmark.h"
+
+#include "flow/flow.h"
+#include "flow/DeterministicRandom.h"
+#include "flow/network.h"
+#include "flow/ThreadHelper.actor.h"
+
+#include "flow/actorcompiler.h" // This must be the last #include.
+
+ACTOR static Future<Void> increment(TaskPriority priority, uint32_t* sum) {
+  wait(delay(0, priority));
+  ++(*sum);
+  return Void();
+}
+
+static inline TaskPriority getRandomTaskPriority(DeterministicRandom &rand) {
+  return static_cast<TaskPriority>(rand.randomInt(0, 100));
+}
+
+ACTOR static Future<Void> benchNet2Actor(benchmark::State* benchState) {
+  state size_t actorCount = benchState->range(0);
+  state uint32_t sum;
+  state int seed = platform::getRandomSeed();
+  while (benchState->KeepRunning()) {
+    sum = 0;
+    std::vector<Future<Void>> futures;
+    futures.reserve(actorCount);
+    DeterministicRandom rand(seed);
+    for (int i = 0; i < actorCount; ++i) {
+      futures.push_back(increment(getRandomTaskPriority(rand), &sum));
+    }
+    wait(waitForAll(futures));
+    benchmark::DoNotOptimize(sum);
+  }
+  benchState->SetItemsProcessed(actorCount * static_cast<long>(benchState->iterations()));
+  return Void();
+}
+
+static void bench_net2(benchmark::State &benchState) {
+  onMainThread([&benchState]{ return benchNet2Actor(&benchState); }).blockUntilReady();
+}
+
+BENCHMARK(bench_net2)->Range(1, 1 << 16)->ReportAggregatesOnly(true);
diff --git a/flowbench/CMakeLists.txt b/flowbench/CMakeLists.txt
index f8c085616f..f96183d038 100644
--- a/flowbench/CMakeLists.txt
+++ b/flowbench/CMakeLists.txt
@@ -5,6 +5,7 @@ set(FLOWBENCH_SRCS
   BenchIterate.cpp
   BenchMem.cpp
   BenchMetadataCheck.cpp
+  BenchNet2.actor.cpp
   BenchPopulate.cpp
   BenchRandom.cpp
   BenchRef.cpp