Golang并发控制HTTP请求方法

var sem = make(chan struct{}, 10) // 最多允许10个并发
<p>func handler(w http.ResponseWriter, r *http.Request) {
sem <- struct{}{} // 获取令牌
defer func() { <-sem }() // 处理完释放</p><pre class="brush:php;toolbar:false;">// 模拟业务处理
time.Sleep(2 * time.Second)
fmt.Fprintf(w, "Request processed")

}

func multiServiceHandler(w http.ResponseWriter, r *http.Request) {
    var wg sync.WaitGroup
    results := make([]string, 3)
<pre class="brush:php;toolbar:false;">wg.Add(3)
go func() { defer wg.Done(); results[0] = callServiceA() }()
go func() { defer wg.Done(); results[1] = callServiceB() }()
go func() { defer wg.Done(); results[2] = callServiceC() }()

wg.Wait() // 等待全部完成
fmt.Fprintf(w, "Results: %v", results)

}

import "golang.org/x/time/rate"
<p>var limiter = rate.NewLimiter(5, 10) // 每秒5个，最多容纳10个突发</p><p>func limitMiddleware(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
if !limiter.Allow() {
http.StatusText(http.StatusTooManyRequests)
return
}
next(w, r)
}
}</p><p>// 使用
http.HandleFunc("/", limitMiddleware(handler))</p>

srv := &http.Server{
    Addr:              ":8080",
    ReadTimeout:       5 * time.Second,
    WriteTimeout:      10 * time.Second,
    IdleTimeout:       30 * time.Second,
    MaxHeaderBytes:    1