【原創】請避免GO語言中的攜程空跑（CPU突然激增）

其實GO語言從1.6版本開始非常不錯了，GC性能優化非常到位，並且各種並行設計比從新實現一套C++版本的確是方便不少。

語言包也很多，庫也相對穩定，完全可以適用於生產環境。

 

本文主要是給剛剛入門新手注意一個攜程空跑的問題，因為這種問題可能在C++中也遇到過，只是一些代碼書寫習慣導致。

 

首先來看一段代碼：

func (c *WSConn) processHandler() {
	for {
		select {
		case message, ok := <-c.processMsg: // 處理數據包
			if !ok {
				break
			}
			Call(message.MsgHead.Id, c, message.MsgContext, int(message.MsgHead.Msglen))
		}
	}
}

以上代碼是用於處理一個WEBSOCKET的二進制消息后轉換為指定處理信息的行為。

但是有沒有同學發現有什么問題？但是這段代碼的確有問題，因為當連接銷毀后會導致processHandler這個攜程空跑，CPU完全占滿，當你有多個連接出現這種問題后整台服務器就會爆掉。

首先processMsg是一個channel，這里如果連接關閉了會同時關閉掉這個channel，首先我們知道select本身會等待channel，這樣是不會消耗CPU的，就像C中的select函數一樣，本身是不消耗的（使用不當的略過）。

但是當channel關閉后，整個攜程本因直接銷毀，但是代碼中的break導致select無限循環跑，程序出現空跑現象，這里的break是相對於select而言的，所以看上去沒毛病可跑起來毛病很大。

所以如果當出現空跑或GO語言某個攜程CPU激增，可以去查看是不是哪個channel和select在無限循環。

所以正確的代碼是：

func (c *WSConn) processHandler() {
	for {
		select {
		case message, ok := <-c.processMsg: // 處理數據包
			if !ok {
				return // 這里必須強制結束攜程
			}
			Call(message.MsgHead.Id, c, message.MsgContext, int(message.MsgHead.Msglen))
		}
	}
}

我的排錯方法是使用http的一種性能分析方式

下面是詳細代碼：

main.go

package main

import (
	"log"
	"runtime"

	"net/http" // http包引入
	_ "net/http/pprof" // 性能分析包引入

)

func main() {
	// 設置並行運行
	runtime.GOMAXPROCS(2)

	logger.SetLogName("testserver.log")

    
	go func() {
		log.Println(http.ListenAndServe("localhost:6060", nil))
	}() // 此處建立http專用的性能分析端口

	webSock := knlWebsocket.Create(":88", "null")
	webSock.Listen()

}

　　以上代碼僅供拋磚引玉，無法通過編譯，注意注釋內的代碼。

看代碼很簡單，import 2個包：

import (
	"net/http" // http包引入
	_ "net/http/pprof" // 性能分析包引入

)

然后main函數中加入代碼：

go func() {
		log.Println(http.ListenAndServe("localhost:6060", nil))
	}() // 此處建立http專用的性能分析端口

　我在這里加入了一個攜程來做性能分析，是因為我本身有自己的主處理邏輯，所以必須使用攜程。

   完成以上代碼添加后，直接編譯啟動程序，訪問地址：http://localhost:6060/debug/pprof/，然后就可以進行愉快的性能分析了