Add: upper limit of concurrency (#70)

sanposhiho · Sep 25, 2021 · 469236b · 469236b
1 parent 9d2cfab
commit 469236b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/internal/command/check.go b/internal/command/check.go
@@ -6,8 +6,11 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"runtime"
 	"strings"
 
+	"golang.org/x/sync/semaphore"
+
 	"golang.org/x/sync/errgroup"
 
 	"github.com/sanposhiho/gomockhandler/internal/mockgen"
@@ -38,11 +41,14 @@ func (r Runner) Check() {
 
 	isFail := false
 	g, _ := errgroup.WithContext(ctx)
+	sem := semaphore.NewWeighted(int64(runtime.GOMAXPROCS(0)))
 	for _, m := range ch.Mocks {
 		m := m
+		sem.Acquire(ctx, 1)
 
 		var runner mockgen.Runner
 		g.Go(func() error {
+			defer sem.Release(1)
 			switch m.Mode {
 			case model.ReflectMode:
 				runner = m.ReflectModeRunner

diff --git a/internal/command/mockgen.go b/internal/command/mockgen.go
@@ -6,8 +6,11 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"runtime"
 	"strings"
 
+	"golang.org/x/sync/semaphore"
+
 	"github.com/sanposhiho/gomockhandler/internal/mockgen"
 
 	"github.com/sanposhiho/gomockhandler/internal/model"
@@ -31,9 +34,13 @@ func (r Runner) Mockgen() {
 	}
 
 	g, _ := errgroup.WithContext(context.Background())
+	sem := semaphore.NewWeighted(int64(runtime.GOMAXPROCS(0)))
 	for _, m := range ch.Mocks {
 		m := m
+		sem.Acquire(context.Background(), 1)
+
 		g.Go(func() error {
+			defer sem.Release(1)
 			var runner mockgen.Runner
 			var sourceChecksum string
 			switch m.Mode {