Press "Enter" to skip to content

Golang 使用 Channel 做并发限制

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

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

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

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

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
}