Golang 使用 Channel 做并发限制

最近写一个前端日志统一入口简单web server

对于请求并发有一些要求，由于磁盘IO限制，故设置请求并发上限，保护磁盘。

大概思路：定义缓存队列10个空格子，当请求多于10，请求将阻塞，直到处理完最后一个，释放格子。

以下代码只是这个思路的简单示例：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30


package main
import (
    "net/http"
    "log"
    "fmt"
    "time"
)
func main() {
    http.HandleFunc("/", handler)
    log.Fatal(http.ListenAndServe("localhost:8000", nil))
}
// 定义缓存队列10个空格子，当请求多于10，请求将阻塞，直到处理完最后一个，释放格子
var ch = make(chan string, 10)
func handler(w http.ResponseWriter, r *http.Request) {
    // 请求路径放到channel里
    ch <- r.URL.Path
    log.Println(cap(ch))
    log.Println(len(ch))
    flagCh := make(chan bool)
    go doHandler(ch, flagCh)
    if <- flagCh {
        fmt.Fprintf(w, "URL.Path = %q\n", r.URL.Path)
    }
}
func doHandler(ch chan string, flagCh chan bool)  {
    time.Sleep(2 * time.Second)
    // 从channerl里拿出
    log.Println(<- ch)
    flagCh <- true
}

Golang 使用 Channel 做并发限制

See Also: