go get -u github.com/panjf2000/ants/v2

package main

import (
	"fmt"
	"time"

	"github.com/panjf2000/ants/v2"
)

func main() {
	// 創(chuàng)建一個(gè)容量為 10 的 goroutine 池
	pool, _ := ants.NewPool(10)
	defer pool.Release() // 程序結(jié)束釋放資源

	// 提交任務(wù)
	for i := 0; i < 20; i++ {
		n := i
		_ = pool.Submit(func() {
			fmt.Printf("Task %d running\n", n)
			time.Sleep(time.Second)
		})
	}

	fmt.Println("All tasks submitted")
	time.Sleep(3 * time.Second) // 等待任務(wù)執(zhí)行完
}

ants.WithPreAlloc(true)           // 預(yù)先分配內(nèi)存，減少運(yùn)行時(shí)分配
ants.WithNonblocking(true)        // worker 不夠時(shí)立即返回錯(cuò)誤，而不是等待
ants.WithMaxBlockingTasks(1000)   // 最大等待任務(wù)數(shù)
ants.WithPanicHandler(func(err interface{}) {
	fmt.Println("panic:", err)     // 捕獲任務(wù)中的 panic
})
ants.WithExpiryDuration(time.Second * 10) // 空閑 worker 過期時(shí)間

package main

import (
	"fmt"
	"github.com/panjf2000/ants/v2"
	"time"
)

func main() {
	// 定義處理函數(shù)
	handler := func(data interface{}) {
		fmt.Printf("Processing: %v\n", data)
		time.Sleep(time.Millisecond * 500)
	}

	// 創(chuàng)建綁定函數(shù)的池
	pool, _ := ants.NewPoolWithFunc(5, handler)
	defer pool.Release()

	for i := 0; i < 10; i++ {
		_ = pool.Invoke(i) // 直接傳參數(shù)
	}
}

package main

import (
	"fmt"
	"sync"
	"time"

	"github.com/panjf2000/ants/v2"
)

func processTask(id int) {
	fmt.Printf("Task %d start\n", id)
	time.Sleep(time.Second)
	fmt.Printf("Task %d done\n", id)
}

func main() {
	var wg sync.WaitGroup
	pool, _ := ants.NewPool(50) // 控制并發(fā)為 50
	defer pool.Release()

	for i := 0; i < 10000; i++ {
		wg.Add(1)
		taskID := i
		_ = pool.Submit(func() {
			defer wg.Done()
			processTask(taskID)
		})
	}
	wg.Wait()
}

package main

import (
	"fmt"
	"net/http"
	"sync"

	"github.com/panjf2000/ants/v2"
)

func fetch(url string) {
	resp, err := http.Get(url)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer resp.Body.Close()
	fmt.Println(url, resp.StatusCode)
}

func main() {
	urls := []string{
		"https://golang.org",
		"https://github.com",
		"https://google.com",
	}

	var wg sync.WaitGroup
	pool, _ := ants.NewPool(5)
	defer pool.Release()

	for _, url := range urls {
		wg.Add(1)
		u := url
		_ = pool.Submit(func() {
			defer wg.Done()
			fetch(u)
		})
	}

	wg.Wait()
}