From 0c18565cea2b31df688ff89c09a815acfd4be273 Mon Sep 17 00:00:00 2001
From: Rene Schallner <rene@renerocks.ai>
Date: Fri, 19 May 2023 15:37:58 +0200
Subject: [PATCH] fixed wrk_zigstd. It compiles but can't handle wrk :(

---
 targets.txt         |  1 +
 wrk/zigstd/main.zig | 23 ++++++++++++++++-------
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/targets.txt b/targets.txt
index a55c08e..42fe186 100644
--- a/targets.txt
+++ b/targets.txt
@@ -5,6 +5,7 @@ serve
 hello_json
 endpoint
 wrk
+wrk_zigstd
 mustache
 endpoint_auth
 http_params
diff --git a/wrk/zigstd/main.zig b/wrk/zigstd/main.zig
index 41a3397..30e0636 100644
--- a/wrk/zigstd/main.zig
+++ b/wrk/zigstd/main.zig
@@ -1,20 +1,29 @@
 const std = @import("std");
 
 pub fn main() !void {
-    const allocator = std.heap.page_allocator;
+    var gpa = std.heap.GeneralPurposeAllocator(.{
+        .thread_safe = true,
+    }){};
+    var allocator = gpa.allocator();
+
     var server = std.http.Server.init(allocator, .{
         .reuse_address = true,
     });
     defer server.deinit();
+
     const address = try std.net.Address.parseIp("127.0.0.1", 3000);
     try server.listen(address);
+
     const max_header_size = 8192;
 
     while (true) {
-        var res = try server.accept(.{ .dynamic = max_header_size });
-        const start_time = std.time.nanoTimestamp();
+        var res = try server.accept(.{
+            .allocator = allocator,
+            .header_strategy = .{ .dynamic = max_header_size },
+        });
+        // const start_time = std.time.nanoTimestamp();
         defer res.deinit();
-        defer res.reset();
+        defer _ = res.reset();
         try res.wait();
 
         const server_body: []const u8 = "HI FROM ZIG STD!\n";
@@ -27,8 +36,8 @@ pub fn main() !void {
         _ = try res.readAll(&buf);
         _ = try res.writer().writeAll(server_body);
         try res.finish();
-        const end_time = std.time.nanoTimestamp();
-        const diff = end_time - start_time;
-        std.debug.print("{d}\n", .{diff});
+        // const end_time = std.time.nanoTimestamp();
+        // const diff = end_time - start_time;
+        // std.debug.print("{d}\n", .{diff});
     }
 }