feat 初次提交

utils/workerpool.go

@@ -0,0 +1,41 @@
+package utils
+
+import (
+	"sync"
+)
+
+// DefaultMaxWorkers 默认并发数
+const DefaultMaxWorkers = 32
+
+// WorkerPool 并发执行一批任务。
+func WorkerPool(tasks []interface{}, taskFunc func(interface{}), maxWorkers ...int) {
+	if len(tasks) == 0 {
+		return
+	}
+
+	workers := DefaultMaxWorkers
+	if len(maxWorkers) > 0 && maxWorkers[0] > 0 {
+		workers = maxWorkers[0]
+	}
+	if workers > len(tasks) {
+		workers = len(tasks)
+	}
+
+	jobs := make(chan interface{}, len(tasks))
+	var wg sync.WaitGroup
+	for i := 0; i < workers; i++ {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			for task := range jobs {
+				taskFunc(task)
+			}
+		}()
+	}
+	for _, task := range tasks {
+		jobs <- task
+	}
+
+	close(jobs)
+	wg.Wait()
+}