Golang 线程池实现http异步响应

参考：https://blog.csdn.net/weixin_44211968/article/details/123048160

最近在用golang改写一个模型服务接口，之前的开发人员用的echo框架，是常见的http方式，不支持多线程的方式，为了提高并发，想用多线程（go routine)。若直接go func()的方式，每次来请求，都开一个线程，会导致线程的数量不受控制。在请求数量未知的情况下，怎么可以控制线程数量，复用线程？这就是线程池的概念。上述链接中的代码很有代表性，总结起来模式就两点：

1. 用for循环开启指定数量的线程，然后让每个线程不停地遍历读取同一个管道，从管道中读取任务信息并进行处理；

2. 每次来请求的时候，把请求信息包装一下，放入管道即可，无需其他操作。

这样，客户端请求一进来就被放进管道，而一放进管道就会被早已等待多时的各个线程处理掉，而线程的数量又是恒定的，在不断地跑，可以重复利用，这样就实现了线程池。

以下代码就是一个带线程池的Golang web服务：

package main
 
import (
    "encoding/json"
    "fmt"
    "net/http"
    "time"
)
func Write_JSON(data map[string]interface{}, w http.ResponseWriter) {
    bytes, _ := json.Marshal(data)
    w.Write(bytes)
}
 
func GenerateResponse(w http.ResponseWriter, r *http.Request) {
    //处理业务请求的方法
    fmt.Printf("handle request:%s\n", r.RequestURI)
    data := make(map[string]interface{})
    data["id"] = r.URL.Query().Get("id")
    data["host"] = r.Host
    data["tm"] = time.Now().UnixMilli() % 10000
    //把结果写入response
    Write_JSON(http.StatusOK, data, w)
}
 
 
type Context struct {
    response http.ResponseWriter
    request  *http.Request
}
 
func HandleAllRequest(w http.ResponseWriter, r *http.Request) {
    //处理请求的方法只需把请求放入管道进行排队
    contextChan <- &Context{response: w,request: r}
}
 
//启动多个worker线程
func CreateWorkers(num_workers int, contextChan chan *Context) {
    for i := 0; i < num_workers; i++ {
        go func(contextChan chan *Context) {
            //每个线程读取管道中的请求任务
            for context := range contextChan {
                //用指定方法处理获取到的上述请求
                GenerateResponse(context.response, context.request)
            }
        }(contextChan)
    }
}
 
//任务(请求)管道
var contextChan chan *Context
 
func main() {
    num_workers := 32 //指定线程个数
    contextChan = make(chan *Context, 10) //初始化任务管道
    CreateWorkers(num_workers, contextChan) //启动多个处理请求的线程
 
    http.HandleFunc("/sayhi", HandleAllRequest) //指定处理请求的方法
    fmt.Printf("Server is running.\n")
    http.ListenAndServe(":8888", nil) //指定Http服务的运行端口
}