14.11 限制同时处理的请求数

使用带缓冲区的通道很容易实现这一点(参见 14.2.5),其缓冲区容量就是同时处理请求的最大数量。程序 max_tasks.go 虽然没有做什么有用的事但是却包含了这个技巧:超过 MAXREQS 的请求将不会被同时处理,因为当信号通道表示缓冲区已满时 handle() 函数会阻塞且不再处理其他请求,直到某个请求从 sem 中被移除。sem 就像一个信号量,这一专业术语用于在程序中表示特定条件的标志变量。

示例:14.16-max_tasks.go

  1. package main
  2. const MAXREQS = 50
  3. var sem = make(chan int, MAXREQS)
  4. type Request struct {
  5. a, b int
  6. replyc chan int
  7. }
  8. func process(r *Request) {
  9. // do something
  10. }
  11. func handle(r *Request) {
  12. sem <- 1 // doesn't matter what we put in it
  13. process(r)
  14. <-sem // one empty place in the buffer: the next request can start
  15. }
  16. func server(service chan *Request) {
  17. for {
  18. request := <-service
  19. go handle(request)
  20. }
  21. }
  22. func main() {
  23. service := make(chan *Request)
  24. go server(service)
  25. }

通过这种方式,应用程序可以通过使用缓冲通道(通道被用作信号量)使协程同步其对该资源的使用,从而充分利用有限的资源(如内存)。

链接