文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

golang线程池怎么实现

2023-10-27 05:31

关注

在Go语言中,可以使用`goroutine`和`channel`来实现线程池。下面是一个简单的线程池实现示例:

package main

import (
    "fmt"
    "sync"
)

type ThreadPool struct {
    workerNum int           // 工作线程数量
    taskQueue chan func()  // 任务队列
    wg        sync.WaitGroup // 等待所有任务完成
}

func NewThreadPool(workerNum int) *ThreadPool {
    return &ThreadPool{
        workerNum: workerNum,
        taskQueue: make(chan func()),
    }
}

func (tp *ThreadPool) Start() {
    for i := 0; i < tp.workerNum; i++ {
        go tp.worker()
    }
}

func (tp *ThreadPool) worker() {
    for task := range tp.taskQueue {
        task()
        tp.wg.Done()
    }
}

func (tp *ThreadPool) AddTask(task func()) {
    tp.wg.Add(1)
    tp.taskQueue <- task
}

func (tp *ThreadPool) Wait() {
    tp.wg.Wait()
}

func main() {
    // 创建线程池
    pool := NewThreadPool(3)
    // 启动线程池
    pool.Start()

    // 添加任务到线程池
    for i := 0; i < 10; i++ {
        taskNum := i
        pool.AddTask(func() {
            fmt.Printf("Task %d is running\n", taskNum)
        })
    }

    // 等待所有任务完成
    pool.Wait()
}

在上面的示例中,`ThreadPool`结构体表示线程池,包含`workerNum`表示工作线程的数量,`taskQueue`表示任务队列,`wg`为`sync.WaitGroup`用于等待所有任务完成。

`NewThreadPool`函数用于创建一个新的线程池,`Start`方法启动线程池中的工作线程,`worker`方法用于执行任务,`AddTask`方法用于往线程池中添加任务,`Wait`方法用于等待所有任务完成。

在`main`函数中,创建一个线程池并启动,然后循环添加任务到线程池中,最后调用`Wait`方法等待所有任务完成。

当运行上述代码时,你会看到输出结果中的任务是并发执行的,最多同时执行3个任务,这就是线程池的作用。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     801人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     348人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     311人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     432人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     220人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯