Expand bench-api-headers benchmarks

Author: jasnell

src/workerd/tests/bench-api-headers.c++

@@ -14,6 +14,12 @@ namespace {
 struct ApiHeaders: public benchmark::Fixture {
   virtual ~ApiHeaders() noexcept(true) {}
 
+  struct Header {
+    bool append;
+    jsg::ByteString name;
+    jsg::ByteString value;
+  };
+
   void SetUp(benchmark::State& state) noexcept(true) override {
     fixture = kj::heap<TestFixture>();
 
@@ -45,6 +51,22 @@ struct ApiHeaders: public benchmark::Fixture {
   kj::Own<TestFixture> fixture;
   kj::Own<kj::HttpHeaderTable> table;
   kj::Own<kj::HttpHeaders> kjHeaders;
+  Header kHeaders[13] = {Header{false, "Host"_bs, "example.com"_bs},
+    Header{false, "User-Agent"_bs,
+      "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"_bs},
+    Header{false, "Accept"_bs,
+      "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8"_bs},
+    Header{false, "Accept-Language"_bs, "en-US,en;q=0.9"_bs},
+    Header{false, "Accept-Encoding"_bs, "gzip, deflate, br"_bs},
+    Header{false, "Content-Type"_bs, "application/json; charset=utf-8"_bs},
+    Header{false, "Authorization"_bs,
+      "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIn0"_bs},
+    Header{false, "Cache-Control"_bs, "no-cache, no-store, must-revalidate"_bs},
+    Header{false, "Content-Length"_bs, "1234"_bs},
+    Header{false, "Referer"_bs, "https://www.example.com/page?query=value&other=param"_bs},
+    Header{false, "X-Forwarded-For"_bs, "203.0.113.1, 198.51.100.17"_bs},
+    Header{true, "Set-Cookie"_bs, "new_session=token123; Path=/; Secure; HttpOnly"_bs},
+    Header{true, "Set-Cookie"_bs, "new_session=token124; Path=/abc; Secure; HttpOnly"_bs}};
 };
 
 // initialization performs a lot of copying, benchmark it
@@ -61,5 +83,27 @@ BENCHMARK_F(ApiHeaders, constructor)(benchmark::State& state) {
   });
 }
 
+BENCHMARK_F(ApiHeaders, set_append)(benchmark::State& state) {
+  fixture->runInIoContext([&](const TestFixture::Environment& env) {
+    auto& js = env.js;
+    for (auto _: state) {
+      for (size_t i = 0; i < 1000; ++i) {
+        auto headers = js.alloc<api::Headers>();
+        // Set common headers with various representative lengths
+        for (int n = 0; n < 13; n++) {
+          auto& h = kHeaders[n];
+          if (h.append) {
+            headers->append(
+                env.js, jsg::ByteString(kj::str(h.name)), jsg::ByteString(kj::str(h.value)));
+          } else {
+            headers->set(
+                env.js, jsg::ByteString(kj::str(h.name)), jsg::ByteString(kj::str(h.value)));
+          }
+        }
+        benchmark::DoNotOptimize(i);
+      }
+    }
+  });
+}
 }  // namespace
 }  // namespace workerd