Handle network blocking well

1 files changed, 3 insertions, 3 deletions

diff --git a/src/saprus.zig b/src/saprus.zig
index e89998a..b8a84b2 100644
--- a/src/saprus.zig
+++ b/src/saprus.zig
@@ -98,13 +98,13 @@ pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusMessage {
     };
 
     // timeout 1s
-    try sock.setReadTimeout(1_000_000);
+    try sock.setReadTimeout(1 * std.time.us_per_s);
     try sock.bind(bind_addr);
 
     const msg = try sendInitialConnection(payload, initial_port, allocator);
 
     var response_buf: [4096]u8 = undefined;
-    _ = try sock.receive(&response_buf);
+    _ = try sock.receive(&response_buf); // Ignore message that I sent.
     const len = try sock.receive(&response_buf);
 
     initial_conn_res = try SaprusMessage.fromBytes(response_buf[0..len], allocator);
@@ -115,7 +115,7 @@ pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusMessage {
     return initial_conn_res;
 }
 
-const SaprusMessage = @import("./saprus_message.zig").SaprusMessage;
+pub const SaprusMessage = @import("./saprus_message.zig").SaprusMessage;
 
 const std = @import("std");
 const Random = std.Random;