大学网 > php中文网 > 后端开发Golang 函数优化:如何使用 goroutine 池提升效率正文

Golang 函数优化:如何使用 goroutine 池提升效率

中国大学网 2024-10-17

go语言中,goroutine 池可提升效率,具体操作包括:创建 goroutine 池在池中运行一定数量的 goroutine向池中提交任务,任务将由空闲 goroutine 处理等待所有任务完成,关闭池

Go 函数优化:使用 goroutine 池提升效率

Go 语言中的 goroutine 是一种轻量级的并发原语,可以轻松地执行并发任务。但是,创建和销毁 goroutine 可能会很昂贵,尤其是当频繁创建和销毁 goroutine 时。

goroutine 池提供了一种解决方案,它可以复用已创建的 goroutine,从而避免了创建和销毁 goroutine 的开销。

立即学习“go语言免费学习笔记(深入)”;

创建 goroutine 池

type Worker struct {
    W           sync.WaitGroup
    Pool        chan func()
    MaxPoolSize int
}

func NewWorker(maxPoolSize int) *Worker {
    return &Worker{
        Pool:        make(chan func()),
        MaxPoolSize: maxPoolSize,
    }
}

使用 goroutine 池

func (w *Worker) Run() {
    for i := 0; i < w.MaxPoolSize; i++ {
        go func() {
            for f := range w.Pool {
                f()
            }
            w.W.Done()
        }()
    }
}

func (w *Worker) Submit(task func()) {
    w.Pool <- task
    w.W.Add(1)
}

func (w *Worker) Stop() {
    close(w.Pool)
    w.W.Wait()
}

实战案例

考虑一个使用 goroutine 对大列表进行排序的场景。使用 goroutine 池的优化版本如下:

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    // 创建列表
    list := []int{1, 5, 2, 9, 4, 7, 3, 6, 8}

    // 创建 goroutine 池
    worker := NewWorker(4)
    worker.Run()

    // Submit 任务
    for _, v := range list {
        worker.Submit(func() {
            time.Sleep(time.Second)
            fmt.Println(v)
        })
    }

    // 等待所有任务完成
    worker.Stop()
}

优化后的版本通过使用 goroutine 池避免了频繁地创建和销毁 goroutine,从而显著提升了性能。

以上就是Golang 函数优化:如何使用 goroutine 池提升效率的详细内容,更多请关注中国大学网其它相关文章!