perf: RestrictParallel exec opti (#779)

Co-authored-by: Jarvan <[email protected]>
douyu · Mar 14, 2023 · dddbb72 · dddbb72
1 parent 2b8ee1e
commit dddbb72
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/util/xgo/goroutine.go b/pkg/util/xgo/goroutine.go
@@ -53,11 +53,13 @@ func RestrictParallel(restrict int, fns ...func()) func() {
 		var wg sync.WaitGroup
 		for _, fn := range fns {
 			wg.Add(1)
+			channel <- struct{}{}
 			go func(fn func()) {
-				defer wg.Done()
-				channel <- struct{}{}
+				defer func() {
+					wg.Done()
+					<-channel
+				}()
 				_ = try2(fn, nil)
-				<-channel
 			}(fn)
 		}
 		wg.Wait()