Go 通过 goroutines 实现类似线程池的模式

Go 语言自身不提供传统意义上的线程池(thread pool)概念,因为它有一套自己的并发模型,基于 goroutines 和 channels。goroutine 是 Go 语言的轻量级线程,调度由 Go 运行时管理,不需要像操作系统线程那样显式创建和管理线程池。

goroutines 在多核处理器上可以并行运行,Go 运行时会自动在可用的逻辑核心之间分配 goroutines。这意味着通常不需要自己管理一个线程池;只需启动所需数量的 goroutines,剩下的工作交给 Go 的调度器来处理。

如果需要限制同时运行的任务数量,或者想要复用一定数量的工作单元以减少创建和销毁 goroutines 的开销,可以实现一个类似线程池的模式。一个简单的方式是使用 buffered channel 来限制并发。

下面是一个实现了固定数量工作者的 goroutines 池的简单例子:

package main

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

func worker(tasksCh <-chan int, wg *sync.WaitGroup) {
   
	defer wg.Done()
	for task := range tasksCh {
   
		process(task)
	}
}

func process(task int) {
   
	fmt.Printf("处理任务 %d\n", task)
	time.Sleep(time.Second) // 模拟任务处理时间
}

func main() {
   
	tasks := []int{
   1, 2, 3, 4, 5, 6, 7, 8, 9, 10} // 任务列表
	nWorkers := 3 // 工作者数量(类似线程池大小)

	tasksCh := make(chan int, len(tasks))
	wg := sync.WaitGroup{
   }

	// 启动指定数量的工作者 goroutines
	for i := 0; i < nWorkers; i++ {
   
		wg.Add(1)
		go worker(tasksCh, &wg)
	}

	// 将任务发送到任务通道
	for _, task := range tasks {
   
		tasksCh <- task
	}
	close(tasksCh) // 发送完毕后关闭通道

	// 等待所有工作者完成
	wg.Wait()
}

其中worker 函数是每个工作者 goroutine 执行的函数,它从 tasksCh 中接收任务。main 函数创建了一个有限大小的任务通道(buffered channel),并启动了固定数量的工作者 goroutines。然后它将所有任务发送到通道中,工作者会并发地处理这些任务,但同时运行的工作者数量不会超过 nWorkers。这样,就可以控制并发执行的任务数量,从而实现类似线程池的行为。

相关推荐

  1. Go 通过 goroutines 实现类似线模式

    2024-01-27 08:08:03       50 阅读
  2. Go实现简单通过channel实现

    2024-01-27 08:08:03       41 阅读
  3. Go 线实现案例

    2024-01-27 08:08:03       43 阅读
  4. 简易线实现

    2024-01-27 08:08:03       49 阅读
  5. 线参数 以及实现

    2024-01-27 08:08:03       42 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-27 08:08:03       106 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-27 08:08:03       116 阅读
  3. 在Django里面运行非项目文件

    2024-01-27 08:08:03       95 阅读
  4. Python语言-面向对象

    2024-01-27 08:08:03       103 阅读

热门阅读

  1. css flex布局详解

    2024-01-27 08:08:03       54 阅读
  2. trino-435: trino接入TIDB数据源

    2024-01-27 08:08:03       59 阅读
  3. WPF中的选项卡布局TabControl控件

    2024-01-27 08:08:03       56 阅读
  4. [AIGC大数据基础] Flink: 大数据流处理的未来

    2024-01-27 08:08:03       62 阅读
  5. ARM Cortex-A 内核的运行模式切换

    2024-01-27 08:08:03       53 阅读