Handle network blocking well

1 files changed, 4 insertions, 3 deletions

diff --git a/src/main.zig b/src/main.zig
index d1a0165..6c3dd09 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -32,9 +32,9 @@ pub fn main() !void {
 
     // _ = try Saprus.sendInitialConnection(if (message.items.len > 0) message.items else "Hello darkness my old friend", 6868, gpa);
 
-    const res = Saprus.connect(if (message.items.len > 0) message.items else "Hello darkness my old friend", gpa) catch |err| {
-        std.debug.print("Error: {s}", .{@errorName(err)});
-        return;
+    const res: ?SaprusMessage = Saprus.connect(if (message.items.len > 0) message.items else "Hello darkness my old friend", gpa) catch |err| switch (err) {
+        error.WouldBlock => null,
+        else => return err,
     };
     defer if (res) |r| r.deinit(gpa);
     if (res) |r| {
@@ -52,3 +52,4 @@ const DebugAllocator = std.heap.DebugAllocator(.{});
 const ArrayList = std.ArrayList;
 
 const Saprus = @import("./saprus.zig");
+const SaprusMessage = Saprus.SaprusMessage;