Cleanup for deinitializing

1 files changed, 26 insertions, 5 deletions

diff --git a/src/server/main.zig b/src/server/main.zig
index dab0f0a..f3728ea 100644
--- a/src/server/main.zig
+++ b/src/server/main.zig
@@ -25,6 +25,18 @@ fn handleSigInt(sig: std.os.linux.SIG) callconv(.c) void {
     keep_running.store(false, .monotonic);
 }
 
+pub fn deinit(server: *Server, io: std.Io, alloc: std.mem.Allocator) void {
+    server.subs_lock.lockUncancelable(io);
+    defer server.subs_lock.unlock(io);
+    for (server.subscriptions.items) |sub| {
+        alloc.free(sub.sid);
+        alloc.free(sub.subject);
+    }
+    server.subscriptions.shrinkAndFree(alloc, 0);
+
+    server.clients.clearAndFree(alloc);
+}
+
 pub fn main(alloc: std.mem.Allocator, server_config: ServerInfo) !void {
     // Configure the signal action
     const act = std.posix.Sigaction{
@@ -42,14 +54,15 @@ pub fn main(alloc: std.mem.Allocator, server_config: ServerInfo) !void {
         defer _ = dba.deinit();
         const gpa = if (builtin.mode == .Debug or builtin.mode == .ReleaseSafe) dba.allocator() else alloc;
 
-        var server: Server = .{
-            .info = server_config,
-        };
-
         var threaded: std.Io.Threaded = .init(gpa, .{});
         defer threaded.deinit();
         const io = threaded.io();
 
+        var server: Server = .{
+            .info = server_config,
+        };
+        defer server.deinit(io, gpa);
+
         var server_task = try io.concurrent(start, .{ &server, io, gpa });
         defer server_task.cancel(io) catch {};
 
@@ -136,7 +149,15 @@ fn handleConnection(
 
     var qbuf: [8]Message = undefined;
     var queue: std.Io.Queue(Message) = .init(&qbuf);
-    defer queue.close(io);
+    defer {
+        queue.close(io);
+        while (queue.getOne(io)) |msg| {
+            switch (msg) {
+                .msg => |m| m.deinit(server_allocator),
+                else => {},
+            }
+        } else |_| {}
+    }
 
     var client_task = try io.concurrent(Client.start, .{ &client, io, server_allocator, &queue });
     defer client_task.cancel(io) catch {};