如何用Golang优化goroutine池使用_Golang goroutine池高效管理实践

type Task func()
<p>type Pool struct {
tasks chan Task
workers int
}</p><p>func NewPool(workers, queueSize int) *Pool {
return &Pool{
tasks: make(chan Task, queueSize),
workers: workers,
}
}</p><p>func (p *Pool) Start() {
for i := 0; i < p.workers; i++ {
go func() {
for task := range p.tasks {
if task != nil {
task()
}
}
}()
}
}</p><p>func (p *Pool) Submit(task Task) {
p.tasks <- task
}</p><p>func (p *Pool) Close() {
close(p.tasks)
}

type Result struct {
    Data interface{}
    Err  error
}
<p>func (p *Pool) SubmitWithResult(fn func() (interface{}, error)) chan Result {
resultCh := make(chan Result, 1)
p.tasks <- func() {
defer func() {
if r := recover(); r != nil {
resultCh <- Result{nil, fmt.Errorf("panic: %v", r)}
}
}()
data, err := fn()
resultCh <- Result{data, err}
}
return resultCh
}

func (p *Pool) TrySubmit(task Task, timeout time.Duration) bool {
    select {
    case p.tasks <- task:
        return true
    case <-time.After(timeout):
        return false // 超时丢弃或降级处理
    }
}

func (p *Pool) Shutdown(timeout time.Duration) {
    close(p.tasks)
    timer := time.NewTimer(timeout)
    done := make(chan struct{})
    go func() {
        for range p.tasks { } // 消费剩余任务
        done <- struct{}{}
    }()
    select {
    case <-done:
    case <-timer.C:
    }
}

import "github.com/panjf2000/ants/v2"
<p>pool, _ := ants.NewPool(100)
defer pool.Release()</p><p>pool.Submit(func() {
// 执行任务
})