main.zig 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. const std = @import("std");
  2. const net = std.net;
  3. const posix = std.posix;
  4. const fmt = std.fmt;
  5. const socket = @import("socket.zig");
  6. const command = @import("command.zig");
  7. const storage = @import("storage.zig");
  8. pub fn main() !void {
  9. const listener = try socket.init(8080);
  10. defer posix.close(listener);
  11. std.debug.print("2025 pizzakv!\n<danilo.fragoso@dev>\n---------\n", .{});
  12. std.debug.print("Commands:\n\nread key\nwrite key|value\ndelete key\nstatus\n", .{});
  13. std.debug.print("---------\n", .{});
  14. while (true) {
  15. var client_address: net.Address = undefined;
  16. var client_address_len: posix.socklen_t = @sizeOf(net.Address);
  17. const conn = posix.accept(listener, &client_address.any, &client_address_len, 0) catch |err| {
  18. std.debug.print("error accept: {any}\n", .{err});
  19. continue;
  20. };
  21. //defer posix.close(conn);
  22. std.debug.print("Connection received\n", .{});
  23. var requestBuffer: [1024 * 1024]u8 = undefined;
  24. var responseBuffer: [1024 * 1024]u8 = undefined;
  25. while (true) {
  26. const n = try socket.readUntilNewLine(conn, &requestBuffer);
  27. if (n == 0) {
  28. break;
  29. }
  30. const cmdResponse = command.parse(requestBuffer[0..n]) orelse {
  31. std.debug.print("Failed to parse message\n", .{});
  32. socket.write(conn, "false\n") catch |err| {
  33. std.debug.print("error writing: {any}\n", .{err});
  34. };
  35. continue;
  36. };
  37. @memcpy(responseBuffer[0..cmdResponse.len], cmdResponse);
  38. responseBuffer[cmdResponse.len] = '\n';
  39. socket.write(conn, responseBuffer[0 .. cmdResponse.len + 1]) catch |err| {
  40. std.debug.print("error writing: {any}\n", .{err});
  41. };
  42. }
  43. }
  44. }