Go实战准备工作---创建协程池和定时任务

    科技2025-04-06  25

    Go实战准备工作---创建协程池和定时任务

    创建携程池创建定时任务Gin的简单使用

    我们的业务可能用不上协程池,很多同事也不是很懂,之前和同事们沟通了一下,很多人对单例都理解不了,任务没有必要,甚至用PHP写单例都不会写,也从没想过使用,那可想而知,其他的设计模式一概不知了。这就是一直做PHP,没做过其他语言的弊端,在此也建议其他人多多接触下其他语言的概念,尤其是一些通用型的,业务上用不上,我们可以给在自己增加难度的来使用,不能总是守着一门脚本做个五六年不放吧。虽然用不上,还是写上,比如一些耗时的操作,我这边会当作优化来处理。目前IOC(控制反转) 、DI(依赖注入),等都没用上过,这次的改写都用上。之前做的Go开发工作都是改改bug,做些边边角料的活,这次除了系统的熟悉Go代码之外,也对一些常用的设计思想都接触一下,一句话:没难度给自己增加难度也要上,不然工作得多无趣,这也是一个成长的过程嘛。

    创建携程池

    鉴于我们用不上,先写个Demo用下,后续优化的时候具体的再补上,但是对于Go而言,这一块也是重点所在吧。昨天文章有大佬说滥用了代码块,我也不是什么老师,所以,这次我就我代码一次性贴出来,顺便加上一个对外的函数接口,代码上我加上尽量详细的注释。参考网上的,不知道是谁写的,但是 比我写的要清楚多了,在基础上我修改了部分代码。

    //定义携程池类型 type Pool struct { //对外接收Task的入口 CommonTaskChan chan *CommonTask //协程池最大工作数量,限定Goroutine的个数 ChannelNum int //协程池内部的任务就绪队列 JobChannels chan *CommonTask } //定义任务CommonTask类型,每一个任务CommonTask都可以抽象成一个函数 type CommonTask struct { thread func() error //一个无参的函数类型 } //通过CreateTask来创建一个CommonTask func CreateTask(f func() error) *CommonTask { t := CommonTask{ thread: f, } return &t } //执行CommonTask任务的方法 func (t *CommonTask) Execute() { err:= t.thread() //调用任务所绑定的函数 if err !=nil { fmt.Println("Execute err", err.Error()) } } //创建一个协程池 func NewGoPool(cap int) *Pool { p := Pool{ CommonTaskChan: make(chan *CommonTask), ChannelNum: cap, JobChannels: make(chan *CommonTask), } return &p } //协程池创建一个协程并且开始工作 func (p *Pool) start(taskId int) { //worker不断的从JobChannels内部任务队列中拿任务 for task := range p.JobChannels { //如果拿到任务,则执行任务 task.Execute() fmt.Println("taskId ID ", taskId, " 执行完毕任务") } } //让协程池Pool开始工作 func (p *Pool) Run() { //1,首先根据协程池的协程数量限定,开启固定数量的协程, for i := 0; i < p.ChannelNum; i++ { go p.start(i) } // 2、从协程池入口取外界传递过来的任务并且将任务送进JobChannels中 for task := range p.CommonTaskChan { p.JobChannels <- task } //3, 执行完毕需要关闭JobChannels close(p.JobChannels) //4, 执行完毕需要关闭 close(p.CommonTaskChan) } func CallDemo() { //创建一个Task t := CreateTask(func() error { fmt.Println(time.Now()) return nil }) //创建一个协程池,最大开启3个协程worker p := NewGoPool(3) //开一个协程 不断的向 Pool 输送打印一条时间的task任务 go func() { for { p.CommonTaskChan <- t time.Sleep(10 * time.Second) } }() //启动协程池p p.Run() }

    其中,CallDemo就是调用的方式。这里,实际上行就是一个10秒中的定时任务。结果显示是:

    创建定时任务

    对于PHP而言,做一步操作,或者定时任务,只能借助第三方的工具,重启一个进程来定时执行接口,或者swoole的work进程来实现,借助的是C扩展。上面创建协程池的时候,其实就可以看出,这是一个10秒钟的定时任务,时间可以修改。go里面其实有内置的time包来处理定时任务,直接上示例:

    ticker := time.NewTicker(time.Minute * 1) go func() { for _ = range ticker.C { fmt.Printf("ticked at %v", time.Now()) } }()

    这个可以因人而异,根据业务需求来做,其实第三方的就比较稳定,设置的定时格式也是相当丰富的,重点是不需要做保活或者其他的确认机制等。比如:supervisor,我们目前用的就是这个玩意,没有守护进程之类的开发,水平不高,基本属于写业务的能手。这个有时间可以来研究一波,毕竟对于多进程的概念,我要不是做Android开发的,也是很少接触这个概念的。当然,移动端不一定都会接触多进程,笔者刚好有这个业务需求,所以研究过一段时间。装逼了,不好意思。

    Gin的简单使用

    我们目前讨论的并没有决定使用,因为我们之前的项目已经使用了,所以就简单的写上,后续决定使用的框架再来重新封装。使用起来比我们想象的要简单的多。

    engine := gin.Default() //loginGroup := engine.Group("/index.php/Login") loginGroup := engine.Group("/Login") loginGroup.POST("/login", controller.Login) err := engine.Run(":8001") if err != nil { log.Fatal(err) }

    注意接口的映射,我们有时候会加上index.php,这个是可以设置处理的哈。Group是设置请求组,方便统一处理,也可以对组编写过滤器和验证器。端口最好放ymal配置文件,上篇已经解释过了。没错, 上面简短的代码就可以直接使用了。简单看下login的逻辑代码,没什么用处,只是做个演示。

    func Login(context *gin.Context) { var login LoginInfo var loginParams LoginParams header := context.Request.Header accessToken := header.Get("accessToken") fmt.Println("---header/---" + accessToken) err := context.ShouldBind(&loginParams) if err != nil { utils.ReturnError(context, "登录失败: "+err.Error()) return } sqlStr := "select id,LoginId,Pwd,FullName,DepartmentId,deleted ,session_id from c_login where LoginId=?" rowObj := utils.DBClientPool.GetDB("callout").QueryRow(sqlStr, loginParams.Name) err = rowObj.Scan(&login.id, &login.LoginId, &login.Pwd, &login.FullName, &login.DepartmentId, &login.deleted, &login.sessionId) if err != nil { utils.ReturnError(context, "登录失败: "+err.Error()) return } fmt.Println(login) if login.deleted == 1 { panic("deleted") } utils.ReturnSuccess(context, "登录成功") }

    目前准备好这几种工具就可以准备开始写业务了,因为也写业务的过程中会有很多细节上的问题需要时间解决。当然,也有点迫不及待改写了,目测400+ 的接口数量,年前写完时间上怕是不够,主要是和各系统之间的交互问题需要重新改写,优化的点也是个头痛的问题,毕竟数据库的优化除了加索引之外就剩下sql语句的优化了,而这次换成Go,我想的全程不涉及原始sql语句,使用模型和预编译方式处理,这也是规范的必要性。后续写代码的过程中需要添加的工具或者第三方的,再来详细描述。

    Processed: 0.008, SQL: 8