針對(一)中所出現(xiàn)的問題荸百，現(xiàn)在使用一個兩級channel系統(tǒng)，一個用來存放任務(wù)隊列株茶，另一個用來控制任務(wù)隊列上執(zhí)行操作的"工人"數(shù)量（[參考文章]：(http://marcio.io/2015/07/handling-1-million-requests-per-minute-with-golang/)）目代。
- 先將具體的業(yè)務(wù)操作抽取出來，里面定義一個具體的業(yè)務(wù)方法
// MyService 業(yè)務(wù)接口
type MyService struct {
}
// WriteInfo 寫日志
func (s *MyService) WriteInfo() {
time.Sleep(1 * time.Second)
t := time.Now()
logFile, err := os.OpenFile("syslog.txt", os.O_RDWR|os.O_CREATE|os.O_APPEND, 0766)
defer logFile.Close()
if err != nil {
panic(err)
}
infoLog := log.New(logFile, "[INFO]", log.LstdFlags)
infoLog.Print("time=" + strconv.FormatInt(t.UTC().UnixNano(), 10))
}
- 定義一個代表工作的結(jié)構(gòu)體Job惕味，和一個代表工人的結(jié)構(gòu)體Worker
// Job 表示要執(zhí)行的作業(yè)
type Job struct {
MyService MyService
}
// Worker 執(zhí)行作業(yè)的工人
type Worker struct {
WorkerPool chan chan Job
JobChannel chan Job
}
// NewWorker 新建一個工人
func NewWorker(workerPool chan chan Job) Worker {
return Worker{
WorkerPool: workerPool,
JobChannel: make(chan Job),
}
}
- 給工人定義一個Start方法，表示監(jiān)聽自己的工作任務(wù)矢劲，有活兒來了就開始工作
func (w Worker) Start() {
go func() {
for {
w.WorkerPool <- w.JobChannel
select {
case job := <-w.JobChannel:
//有工作任務(wù)時赦拘，開始執(zhí)行業(yè)務(wù)接口的方法
job.MyService.WriteInfo()
}
}
}()
}
- 初始化池
var maxWorkers = 20
// JobQueue 作業(yè)隊列
var JobQueue = make(chan Job, maxWorkers)
// InitPool 給池中初始化一定量的工人，以及開啟任務(wù)隊列的監(jiān)聽
func InitPool() {
// 創(chuàng)建工作池
pool := make(chan chan Job, maxWorkers)
// 創(chuàng)建一定數(shù)量的工人（可以看做：創(chuàng)建了N個工人芬沉，每個工人能并發(fā)處理N件工作）
for i := 0; i < maxWorkers; i++ {
worker := NewWorker(pool)
worker.Start()
}
//監(jiān)聽JobQueue上是否有新任務(wù)
go func() {
for {
select {
case job := <-JobQueue:
go func(job Job) {
// 獲取可用的工人channel躺同，若沒有，則阻塞
jobChannel := <-pool
jobChannel <- job
}(job)
}
}
}()
}
- 提供一個http接口丸逸，用于測試
func init() {
InitPool()
}
func main() {
http.HandleFunc("/test/pool/", indexHandler)
http.ListenAndServe(":9000", nil)
}
func indexHandler(w http.ResponseWriter, r *http.Request) {
service := MyService{}
//service.WriteInfo()
work := Job{MyService: service}
JobQueue <- work
w.Header().Set("Content-Type", "application/json; charset=UTF-8")
result := "{\"msg\":\"SUCCESS\",\"code\":0}"
fmt.Fprintln(w, result)
}
-
使用jmeter進行測試蹋艺，和(一)中一樣，100個樣本黄刚，循環(huán)10次執(zhí)行
聚合報告.png
圖形結(jié)果.png
可以看到處理請求的能力還是非常不錯的捎谨。其實這種測試方法并不能和(一)的結(jié)果進行橫向比對，畢竟(一)中只開啟了20個協(xié)程憔维，而在(二)中由于雙隊列的存在涛救，處理請求的協(xié)程數(shù)量肯定是要比(一)中的多的多。
但(二)的模式肯定是要優(yōu)于(一)的业扒，這是毋容置疑的检吆。我只是為了給自己做個筆記，加深對go中channel的理解程储，畢竟蹭沛，好記性不如爛筆頭么 (? ??_??)?
源碼我上傳到了github，地址：https://github.com/wleirock/studygo/tree/master/pool
