From ba1627ff10c0964b116dd2a4a0722cbe13c10326 Mon Sep 17 00:00:00 2001
From: Andrew Noyes <andrew@weaselab.dev>
Date: Thu, 14 Aug 2025 15:03:30 -0400
Subject: [PATCH] Use same request for one shot and streaming examples

---
 src/main.cpp | 67 ++++++++++++++++++++++++++--------------------------
 1 file changed, 33 insertions(+), 34 deletions(-)

diff --git a/src/main.cpp b/src/main.cpp
index bc99b87..954eb56 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -2,6 +2,30 @@
 #include "config.hpp"
 #include <iostream>
 
+void print_stats(const CommitRequest &request) {
+
+  std::cout << "✓ Successfully parsed commit request:" << std::endl;
+  std::cout << "  Request ID: "
+            << (request.request_id() ? request.request_id().value() : "none")
+            << std::endl;
+  std::cout << "  Leader ID: " << request.leader_id() << std::endl;
+  std::cout << "  Read Version: " << request.read_version() << std::endl;
+  std::cout << "  Preconditions: " << request.preconditions().size()
+            << std::endl;
+  std::cout << "  Operations: " << request.operations().size() << std::endl;
+  std::cout << "  Arena memory used: " << request.used_bytes() << " bytes"
+            << std::endl;
+
+  if (!request.operations().empty()) {
+    const auto &op = request.operations()[0];
+    std::cout << "  First operation: "
+              << (op.type == Operation::Type::Write ? "write" : "other")
+              << " key=" << op.key
+              << " value=" << (op.value ? op.value.value() : "none")
+              << std::endl;
+  }
+}
+
 int main(int argc, char *argv[]) {
   std::string config_file = "config.toml";
 
@@ -40,7 +64,7 @@ int main(int argc, char *argv[]) {
 
   CommitRequest request;
 
-  std::string sample_json = R"({
+  const std::string sample_json = R"({
     "request_id": "demo-12345",
     "leader_id": "leader-abc",
     "read_version": 42,
@@ -59,28 +83,10 @@ int main(int argc, char *argv[]) {
       }
     ]
   })";
+  auto copy = sample_json;
 
-  if (request.parse_json(sample_json)) {
-    std::cout << "✓ Successfully parsed commit request:" << std::endl;
-    std::cout << "  Request ID: "
-              << (request.request_id() ? request.request_id().value() : "none")
-              << std::endl;
-    std::cout << "  Leader ID: " << request.leader_id() << std::endl;
-    std::cout << "  Read Version: " << request.read_version() << std::endl;
-    std::cout << "  Preconditions: " << request.preconditions().size()
-              << std::endl;
-    std::cout << "  Operations: " << request.operations().size() << std::endl;
-    std::cout << "  Arena memory used: " << request.used_bytes() << " bytes"
-              << std::endl;
-
-    if (!request.operations().empty()) {
-      const auto &op = request.operations()[0];
-      std::cout << "  First operation: "
-                << (op.type == Operation::Type::Write ? "write" : "other")
-                << " key=" << op.key
-                << " value=" << (op.value ? op.value.value() : "none")
-                << std::endl;
-    }
+  if (request.parse_json(copy)) {
+    print_stats(request);
   } else {
     std::cout << "✗ Failed to parse commit request" << std::endl;
   }
@@ -94,8 +100,7 @@ int main(int argc, char *argv[]) {
     std::cout << "✓ Initialized streaming parser" << std::endl;
 
     // Simulate receiving data in small chunks like from a network socket
-    std::string streaming_json =
-        R"({"request_id": "stream-456", "leader_id": "stream-leader", "read_version": 999})";
+    std::string copy = sample_json;
 
     size_t chunk_size = 15; // Small chunks to simulate network packets
     size_t offset = 0;
@@ -103,10 +108,10 @@ int main(int argc, char *argv[]) {
 
     CommitRequest::ParseStatus status = CommitRequest::ParseStatus::Incomplete;
 
-    while (offset < streaming_json.size() &&
+    while (offset < copy.size() &&
            status == CommitRequest::ParseStatus::Incomplete) {
-      size_t len = std::min(chunk_size, streaming_json.size() - offset);
-      std::string chunk = streaming_json.substr(offset, len);
+      size_t len = std::min(chunk_size, copy.size() - offset);
+      std::string chunk = copy.substr(offset, len);
 
       std::cout << "  Chunk " << ++chunk_count << " (" << len << " bytes): \""
                 << chunk << "\"" << std::endl;
@@ -125,13 +130,7 @@ int main(int argc, char *argv[]) {
     }
 
     if (status == CommitRequest::ParseStatus::Complete) {
-      std::cout << "✓ Streaming parse successful:" << std::endl;
-      std::cout << "  Request ID: " << streaming_request.request_id().value()
-                << std::endl;
-      std::cout << "  Leader ID: " << streaming_request.leader_id()
-                << std::endl;
-      std::cout << "  Read Version: " << streaming_request.read_version()
-                << std::endl;
+      print_stats(streaming_request);
     } else {
       std::cout << "✗ Streaming parse failed" << std::endl;
     }