From d700e048b82536b10b99c02ad39d532f1e3752ac Mon Sep 17 00:00:00 2001
From: Quentin Renard <contact@asticode.com>
Date: Fri, 13 Nov 2020 18:06:13 +0100
Subject: [PATCH] Now waiting for first level tasks to be done when using
 Task.Do()

---
 worker.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/worker.go b/worker.go
index 04c4be6..b9a95a3 100644
--- a/worker.go
+++ b/worker.go
@@ -126,6 +126,10 @@ func (t *Task) Do(f func()) {
 
 		// Custom
 		f()
+
+		// Wait for first level subtasks to be done
+		// Wait() can also be called in f() if something needs to be executed just after Wait()
+		t.Wait()
 	}()
 }
 
@@ -136,7 +140,7 @@ func (t *Task) Done() {
 	})
 }
 
-// Wait waits for the task to be finished
+// Wait waits for first level subtasks to be finished
 func (t *Task) Wait() {
 	t.ow.Do(func() {
 		t.wg.Wait()