Golang 函数优化：如何使用 goroutine 池提升效率-Golang-PHP中文网

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

WBOY

发布： 2024-10-02 10:00:03

原创

879人浏览过

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

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

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

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

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

立即学习“go语言免费学习笔记（深入）”；

创建 goroutine 池

MOKI

MOKI是美图推出的一款AI短片创作工具，旨在通过AI技术自动生成分镜图并转为视频素材。

375

查看详情

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 池提升效率的详细内容，更多请关注php中文网其它相关文章！