mcp: modernize I/O processing and reuse message buffer

This commit is contained in:
Adrià Arrufat
2026-03-01 22:35:28 +09:00
parent e9c36fd6f8
commit 42b5e32473

View File

@@ -18,13 +18,22 @@ pub fn processRequests(server: *Server) !void {
var arena: std.heap.ArenaAllocator = .init(server.allocator); var arena: std.heap.ArenaAllocator = .init(server.allocator);
defer arena.deinit(); defer arena.deinit();
while (server.is_running.load(.acquire)) { var msg_buf = std.Io.Writer.Allocating.init(server.allocator);
const msg = stdin.interface.adaptToOldInterface().readUntilDelimiterAlloc(server.allocator, '\n', 1024 * 1024 * 10) catch |err| { defer msg_buf.deinit();
if (err == error.EndOfStream) break;
return err;
};
defer server.allocator.free(msg);
while (server.is_running.load(.acquire)) {
msg_buf.clearRetainingCapacity();
const n = try stdin.interface.streamDelimiterLimit(&msg_buf.writer, '\n', .limited(1024 * 1024 * 10));
var found_newline = true;
_ = stdin.interface.discardDelimiterInclusive('\n') catch |err| switch (err) {
error.EndOfStream => found_newline = false,
else => return err,
};
if (n == 0 and !found_newline) break;
const msg = msg_buf.written();
if (msg.len == 0) continue; if (msg.len == 0) continue;
handleMessage(server, arena.allocator(), msg) catch |err| { handleMessage(server, arena.allocator(), msg) catch |err| {