golang倒腾一款简配的具有请求排队功能的并发受限服务器

golang官方指南给了一些代码片段来，层层递进演示了信道的能力:

1>. 信号量

2>. 限流能力

go 复制代码

var sem = make(chan int, MaxOutstanding) 
 
func Serve(queue chan *Request) {
    for req := range queue {
        req:= req
        sem <- 1   
        go func() {   // 只会开启MaxOutstanding个并发协程
            process(req)
            <-sem
        }()
    }
}

上面出现了两个信道：

① sem 提供了限制服务端并发处理请求的信号量

② queue 提供了一个客户端请求队列，起媒介/解耦的作用

进一步指南给出了信道的另一个用法： 3>. 解多路复用

多路复用是网络编程中一个耳熟能详的概念，nginx redis等高性能web、内存kv都用到了这个技术。

这个解多路复用是怎么理解呢？

离散/独立/并发的客户端请求被服务端Serve收敛之后， Serve就起到了多路复用的概念，在Request定义resultChan信道，就给每个客户端请求提供了独立获取请求结果的能力，这便是一种解多路复用。

从实际效果看这就是常见的互联网web服务器：一款具备请求排队功能的并发限流服务器。

官方指南并没有完整实现客户端和服务器端工程。

下面是我的工程化实现，记录下实践中遇到的问题。

并发受限服务器

信道queue接收客户端请求,解耦客户端和服务器，天然具备排队能力
信号量信道sem提供了并发受限的能力
服务器处理完，向解多路复用信道req.resultChan写入响应结果。

server.go 复制代码

/* 实现一个有请求队列功能的并发请求受限服务器*/

package main

import (
	"fmt"
	"sync"
	"time"
)

var sem = make(chan int, Maxoutstanding)

var wg2 sync.WaitGroup

func server(queue chan *Request) {
	fmt.Printf("Server is already, listen req \n")

	for req := range queue {
		req := req
		sem <- 1

		wg2.Add(1)
		go func() {
			defer wg2.Done()
			process(req)
			<-sem
		}()
	}
}

func process(req *Request) {
	s := sum(req.args)
	req.resultChan <- s
}
func sum(a []int) (s int) {
	for i := 1; i <= a[0]; i++ {
		s += i
	}
	time.Sleep(time.Millisecond * 20)
	return s
}

time.Sleep模拟服务器处理请求单次耗时20ms，输出数字的累加，

eg： input： 100；

output: 1+100/2*100 =5050

wg2 sync.WaitGroup是一个动态活跃的Goroutine计数器，注意用法和位置，wg2的作用是：等待所有请求处理完成。

并发客户端请求

for循环开启并发客户端请求，

每个请求入驻一个独立的Goroutine，独立向信道queue投递请求和接收响应

clients.go 复制代码

package main

import (
	"fmt"
	"sync"
)

type Request struct {
	args       []int
	resultChan chan int
}

var wg1 sync.WaitGroup

func clients() {
	fmt.Printf("start %d concurrency client request\n ", concurrencyClients)
	for i := 1; i <= concurrencyClients; i++ {
		r := &Request{
			args:       []int{i},
			resultChan: make(chan int),
		}
		wg1.Add(1)
		go ClientReq(r)
	}
	wg1.Wait() 

}

func ClientReq(r *Request) {
	defer wg1.Done()
	queue <- r
	go func() {
		res := <-r.resultChan
		fmt.Printf("current args is %d, the result is %d \n", r.args[0], res)
	}()
}

wg1 WaitGroup的目的是确保所有的客户端请求都已经发出，之后客户端任务结束，所以此处我们新开Goroutine处理响应结果（这里又有闭包的参与）。

工程化

工程化代码的先后顺序，决定了代码是否死锁。

server需要处于监听状态，故先启动。

本处clients在主协程整体上是同步发送，如果放在clients()的后面，clients内的wg1可能会有部分请求Goroutine阻塞在信道queue且没法唤醒运行时会检测到报死锁。

main.go 复制代码

package main

import (
	"fmt"
	"time"
)

var concurrencyClients = 1000
var queueLength = 100
var queue = make(chan *Request, queueLength) // 请求队列长度
var Maxoutstanding int = 10                  // 服务器并发受限10

func main() {

	go server(queue)
	var start = time.Now()

	clients() // 确保所有的请求都已经发出去

	wg2.Wait() // 确保服务器处理完所有的请求
	fmt.Printf("客户端并发%d请求，服务器请求队列长度%d，服务器限流%d，总共耗时%d ms \n", concurrencyClients, queueLength, Maxoutstanding, time.Since(start).Milliseconds())
}

上面出现了3个配置变量

1>. 客户端并发请求数量concurrencyClients=100

2>. 服务器排队队列长度queueLength，会作用到信道queue=50

3>. 服务器并发受限阈值Maxoutstanding=10

yaml 复制代码

start 1000 concurrency client request
 Server is already, listen req 
current args is 14, the result is 105 
current args is 2, the result is 3 
current args is 3, the result is 6 
current args is 1, the result is 1 
current args is 4, the result is 10 
current args is 8, the result is 36 
current args is 6, the result is 21 
current args is 12, the result is 78 
current args is 5, the result is 15 
current args is 7, the result is 28 
current args is 18, the result is 171 
current args is 16, the result is 136 
current args is 15, the result is 120 
current args is 20, the result is 210 
current args is 19, the result is 190 
current args is 13, the result is 91 
current args is 21, the result is 231 
current args is 10, the result is 55 
current args is 17, the result is 153 
current args is 9, the result is 45 
current args is 22, the result is 253 
current args is 28, the result is 406 
current args is 27, the result is 378 
current args is 11, the result is 66 
current args is 26, the result is 351 
current args is 30, the result is 465 
current args is 23, the result is 276 
current args is 25, the result is 325 
current args is 29, the result is 435 
current args is 24, the result is 300 
current args is 31, the result is 496 
current args is 34, the result is 595 
current args is 38, the result is 741 
current args is 36, the result is 666 
current args is 41, the result is 861 
current args is 32, the result is 528 
current args is 35, the result is 630 
current args is 33, the result is 561 
current args is 37, the result is 703 
current args is 39, the result is 780 
current args is 52, the result is 1378 
current args is 46, the result is 1081 
current args is 47, the result is 1128 
current args is 49, the result is 1225 
current args is 45, the result is 1035 
current args is 43, the result is 946 
current args is 48, the result is 1176 
current args is 40, the result is 820 
current args is 42, the result is 903 
current args is 44, the result is 990 
current args is 59, the result is 1770 
current args is 55, the result is 1540 
current args is 53, the result is 1431 
current args is 57, the result is 1653 
current args is 51, the result is 1326 
current args is 54, the result is 1485 
current args is 50, the result is 1275 
current args is 56, the result is 1596 
current args is 58, the result is 1711 
current args is 60, the result is 1830 
current args is 66, the result is 2211 
current args is 63, the result is 2016 
current args is 70, the result is 2485 
current args is 62, the result is 1953 
current args is 61, the result is 1891 
current args is 65, the result is 2145 
current args is 67, the result is 2278 
current args is 64, the result is 2080 
current args is 68, the result is 2346 
current args is 69, the result is 2415 
current args is 76, the result is 2926 
current args is 77, the result is 3003 
current args is 71, the result is 2556 
current args is 80, the result is 3240 
current args is 75, the result is 2850 
current args is 74, the result is 2775 
current args is 73, the result is 2701 
current args is 72, the result is 2628 
current args is 78, the result is 3081 
current args is 81, the result is 3321 
current args is 89, the result is 4005 
current args is 83, the result is 3486 
current args is 88, the result is 3916 
current args is 82, the result is 3403 
current args is 79, the result is 3160 
current args is 86, the result is 3741 
current args is 84, the result is 3570 
current args is 90, the result is 4095 
current args is 85, the result is 3655 
current args is 87, the result is 3828 
current args is 101, the result is 5151 
current args is 92, the result is 4278 
current args is 94, the result is 4465 
current args is 93, the result is 4371 
current args is 98, the result is 4851 
current args is 91, the result is 4186 
current args is 99, the result is 4950 
current args is 100, the result is 5050 
current args is 95, the result is 4560 
current args is 96, the result is 4656 
current args is 109, the result is 5995 
current args is 107, the result is 5778 
current args is 108, the result is 5886 
current args is 102, the result is 5253 
current args is 103, the result is 5356 
current args is 106, the result is 5671 
current args is 105, the result is 5565 
current args is 104, the result is 5460 
current args is 111, the result is 6216 
current args is 97, the result is 4753 
current args is 120, the result is 7260 
current args is 112, the result is 6328 
current args is 113, the result is 6441 
current args is 114, the result is 6555 
current args is 110, the result is 6105 
current args is 119, the result is 7140 
current args is 115, the result is 6670 
current args is 117, the result is 6903 
current args is 116, the result is 6786 
current args is 118, the result is 7021 
current args is 123, the result is 7626 
current args is 122, the result is 7503 
current args is 130, the result is 8515 
current args is 121, the result is 7381 
current args is 126, the result is 8001 
current args is 129, the result is 8385 
current args is 128, the result is 8256 
current args is 124, the result is 7750 
current args is 125, the result is 7875 
current args is 127, the result is 8128 
current args is 133, the result is 8911 
current args is 135, the result is 9180 
current args is 139, the result is 9730 
current args is 137, the result is 9453 
current args is 141, the result is 10011 
current args is 131, the result is 8646 
current args is 134, the result is 9045 
current args is 138, the result is 9591 
current args is 132, the result is 8778 
current args is 136, the result is 9316 
current args is 148, the result is 11026 
current args is 142, the result is 10153 
current args is 149, the result is 11175 
current args is 147, the result is 10878 
current args is 140, the result is 9870 
current args is 150, the result is 11325 
current args is 145, the result is 10585 
current args is 146, the result is 10731 
current args is 144, the result is 10440 
current args is 143, the result is 10296 
current args is 156, the result is 12246 
current args is 155, the result is 12090 
current args is 161, the result is 13041 
current args is 157, the result is 12403 
current args is 153, the result is 11781 
current args is 151, the result is 11476 
current args is 159, the result is 12720 
current args is 154, the result is 11935 
current args is 152, the result is 11628 
current args is 158, the result is 12561 
current args is 170, the result is 14535 
current args is 164, the result is 13530 
current args is 166, the result is 13861 
current args is 160, the result is 12880 
current args is 162, the result is 13203 
current args is 167, the result is 14028 
current args is 163, the result is 13366 
current args is 169, the result is 14365 
current args is 168, the result is 14196 
current args is 165, the result is 13695 
current args is 176, the result is 15576 
current args is 179, the result is 16110 
current args is 178, the result is 15931 
current args is 174, the result is 15225 
current args is 173, the result is 15051 
current args is 172, the result is 14878 
current args is 177, the result is 15753 
current args is 171, the result is 14706 
current args is 180, the result is 16290 
current args is 175, the result is 15400 
current args is 181, the result is 16471 
current args is 193, the result is 18721 
current args is 182, the result is 16653 
current args is 184, the result is 17020 
current args is 187, the result is 17578 
current args is 186, the result is 17391 
current args is 189, the result is 17955 
current args is 183, the result is 16836 
current args is 188, the result is 17766 
current args is 185, the result is 17205 
current args is 198, the result is 19701 
current args is 195, the result is 19110 
current args is 199, the result is 19900 
current args is 190, the result is 18145 
current args is 196, the result is 19306 
current args is 192, the result is 18528 
current args is 191, the result is 18336 
current args is 205, the result is 21115 
current args is 194, the result is 18915 
current args is 197, the result is 19503 
current args is 207, the result is 21528 
current args is 210, the result is 22155 
current args is 200, the result is 20100 
current args is 213, the result is 22791 
current args is 202, the result is 20503 

......

current args is 981, the result is 481671 
current args is 978, the result is 478731 
current args is 982, the result is 482653 
current args is 970, the result is 470935 
current args is 979, the result is 479710 
current args is 980, the result is 480690 
current args is 983, the result is 483636 
current args is 989, the result is 489555 
current args is 986, the result is 486591 
current args is 987, the result is 487578 
current args is 985, the result is 485605 
current args is 977, the result is 477753 
current args is 988, the result is 488566 
current args is 992, the result is 492528 
current args is 976, the result is 476776 
current args is 984, the result is 484620 
current args is 995, the result is 495510 
current args is 999, the result is 499500 
current args is 1000, the result is 500500 
current args is 990, the result is 490545 
客户端并发1000请求，服务器请求队列长度100，服务器限流10，总共耗时2099 ms

读者可以随意调整3个参数的大小，来感受服务器调参的魅力。

并发客户端请求数concurrencyClients	服务器请求队列queueLength	服务器限流阈值 Maxoutstanding	耗时ms
1000	100	10	2067
1000	100	50	454
1000	100	100	210
1000	300	10	2082
1000	500	10	2071
3000	100	10	6259
5000	500	10	10516

完整代码传送门

That's All,本文根据golang有关信道的指南，实现了一个带有请求队列功能的首先服务器，巩固了信道、WaitGroup的用法。