From 53dee08af99dd334b0d227afb5ce2a0f92c35a5d Mon Sep 17 00:00:00 2001
From: Loris Cro <kappaloris@gmail.com>
Date: Fri, 2 Oct 2020 19:15:26 +0200
Subject: add WaitGroup to std.event

Signed-off-by: Loris Cro <kappaloris@gmail.com>
---
 lib/std/event/loop.zig | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'lib/std/event/loop.zig')

diff --git a/lib/std/event/loop.zig b/lib/std/event/loop.zig
index 226d5f1d52..a064f711e2 100644
--- a/lib/std/event/loop.zig
+++ b/lib/std/event/loop.zig
@@ -660,9 +660,11 @@ pub const Loop = struct {
         const Wrapper = struct {
             const Args = @TypeOf(args);
             fn run(func_args: Args, loop: *Loop, allocator: *mem.Allocator) void {
+                loop.beginOneEvent();
                 loop.yield();
                 const result = @call(.{}, func, func_args);
                 suspend {
+                    loop.finishOneEvent();
                     allocator.destroy(@frame());
                 }
             }
-- 
cgit v1.2.3