Use same request for one shot and streaming examples

This commit is contained in:
2025-08-14 15:03:30 -04:00
parent 035bd8b068
commit ba1627ff10

View File

@@ -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;
}