memory: use a GPA in Debug mode and an Arena (page based) on Release

Signed-off-by: Francis Bouvier <[email protected]>

Author: francisbouvier

src/main.zig

@@ -158,12 +158,31 @@ fn printUsageExit(execname: []const u8, res: u8) void {
 pub fn main() !void {
 
     // allocator
-    var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
-    defer arena.deinit();
-    const alloc = arena.allocator();
+    // - in Debug mode we use the General Purpose Allocator to detect memory leaks
+    // - in Release mode we use an Arena Allocator based on the page allocator
+    var alloc: std.mem.Allocator = undefined;
+    var _gpa: ?std.heap.GeneralPurposeAllocator(.{}) = null;
+    var _arena: ?std.heap.ArenaAllocator = null;
+    if (builtin.mode == .Debug) {
+        _gpa = std.heap.GeneralPurposeAllocator(.{}){};
+        alloc = _gpa.?.allocator();
+    } else {
+        _arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
+        alloc = _arena.?.allocator();
+    }
+    defer {
+        if (_gpa) |*gpa| {
+            switch (gpa.deinit()) {
+                .ok => std.debug.print("No memory leaks\n", .{}),
+                .leak => @panic("Memory leak"),
+            }
+        } else if (_arena) |*arena| {
+            arena.deinit();
+        }
+    }
 
     // args
-    var args = try std.process.argsWithAllocator(arena.allocator());
+    var args = try std.process.argsWithAllocator(alloc);
     defer args.deinit();
 
     const execname = args.next().?;
@@ -263,7 +282,7 @@ pub fn main() !void {
         std.log.info("Listening on: {s}:{d}...", .{ host, port });
 
         // loop
-        var loop = try jsruntime.Loop.init(arena.allocator());
+        var loop = try jsruntime.Loop.init(alloc);
         defer loop.deinit();
 
         // listen
@@ -279,7 +298,7 @@ pub fn main() !void {
         const vm = jsruntime.VM.init();
         defer vm.deinit();
 
-        var loop = try jsruntime.Loop.init(arena.allocator());
+        var loop = try jsruntime.Loop.init(alloc);
         defer loop.deinit();
 
         var browser = Browser{};