昨天在內(nèi)網(wǎng)上看到一篇講數(shù)據(jù)庫連接的文章，列出了一些 sql 包的一些源碼，我注意到其中取用、歸還連接的方式非常有意思——通過臨時創(chuàng)建的 channel 來傳遞連接。

在 sql.DB 結(jié)構(gòu)體里，使用 freeConn 字段來表示當(dāng)前所有的連接，也就是一個連接池。

type?DB?struct?{freeConn?????[]*driverConn }

當(dāng)需要拿連接的時候，從 freeConn 中取出第一個元素：

conn?:=?db.freeConn[0] copy(db.freeConn,?db.freeConn[1:]) db.freeConn?=?db.freeConn[:numFree-1] conn.inUse?=?true

取 slice 切片的第一個元素，然后將 slice 后面的元素往前挪，最后通過截斷來“釋放”最后一個元素。

當(dāng)然，能進(jìn)行上述操作的前提是切片 db.freeConn 長度大于 0，即有空閑連接存在。如果當(dāng)前沒有空閑連接，那如何處理呢？接下來就是 channel 的妙用的地方。

sql.DB 結(jié)構(gòu)體里還有另一個字段 connRequests，它用來存儲當(dāng)前有哪些“協(xié)程”在申請連接：

type?DB?struct?{freeConn?????[]*driverConnconnRequests?map[uint64]chan?connRequest }

connRequests 的 key 是一個 uint64類型，其實就是一個遞增加 1 的 key；而 connRequest 表示申請一個新連接的請求：

type?connRequest?struct?{conn?*driverConnerr??error }

這里的 conn 正是需要的連接。

當(dāng)連接池中沒有空閑連接的時候：

req?:=?make(chan?connRequest,?1) reqKey?:=?db.nextRequestKeyLocked() db.connRequests[reqKey]?=?req

先是構(gòu)建了一個 chan connRequest，同時拿到了一個 reqKey，將它和 req 綁定到 connRequests 中。

接下來，在 select 中等待超時或者從 req 這個 channel 中拿到空閑連接：

select?{case?<-ctx.Done():case?ret,?ok?:=?<-req:if?!ok?{return?nil,?errDBClosed}return?ret.conn,?ret.err }

可以看到，select 有兩個 case，第一個是通過 context 控制的 <-Done；第二個則是前面構(gòu)造的 <-req，如果從 req 中讀出了元素，那就相當(dāng)于獲得了連接：ret.conn。

那什么時候會向 req 中發(fā)送連接呢？答案是在向連接池歸還連接的時候。

前面提到，空閑連接是一個切片，歸還的時候直接 append 到這個切片就可以了：

func?(db?*DB)?putConnDBLocked(dc?*driverConn,?err?error)?bool?{db.freeConn?=?append(db.freeConn,?dc) }

但其實在 append 之前，還會去檢查當(dāng)前 connRequests 中是否有申請空閑連接的請求：

if?c?:=?len(db.connRequests);?c?>?0?{var?req?chan?connRequestvar?reqKey?uint64for?reqKey,?req?=?range?db.connRequests?{break}delete(db.connRequests,?reqKey)?//?Remove?from?pending?requests.if?err?==?nil?{dc.inUse?=?true}req?<-?connRequest{conn:?dc,err:??err,}return?true }?

如果有請求的話，直接將當(dāng)前連接“塞到” req channel 里去了。另一邊，申請連接的 goroutine 就可以從 req channel 中讀出 conn。

于是，通過 channel 就實現(xiàn)了一次“連接傳輸”的功能。

這讓我想到不久之前芮神寫的一篇《高并發(fā)服務(wù)遇redis瓶頸引發(fā)time-wait事故》，文中提到了將多個 redis command 組裝為一個 pipeline：

調(diào)用方把 redis command 和接收結(jié)果的 chan 推送到任務(wù)隊列中，然后由一個 worker 去消費，worker 組裝多個 redis cmd 為 pipeline，向 redis 發(fā)起請求并拿回結(jié)果，拆解結(jié)果集后，給每個命令對應(yīng)的結(jié)果 chan 推送結(jié)果。調(diào)用方在推送任務(wù)到隊列后，就一直監(jiān)聽傳輸結(jié)果的 chan。

redis commnd 組裝成 pipeline

這里的用法就和本文描述的 channel 用法一致。

細(xì)想一下，以上提到的 channel 用法很神奇嗎？我們平時沒有接觸過嗎？

我用過最多的是“生產(chǎn)者-消費者”模式，先啟動 N 個 goroutine 消費者，讀某個 channel，之后，生產(chǎn)者再在某個時候向 channel 中發(fā)送元素：

for?i?:=?0;?i?<?engine.workerNum;?i++?{go?func()?{for?{work?=?<-engine.workChan}} }

另外，我還會用 channel 充當(dāng)一個 “ready” 的信號，用來指示某個“過程”準(zhǔn)備好了，可以接收結(jié)果了：

func?(j?*Job)?Finished()?<-chan?bool?{return?j.finish }

前面提到的“生產(chǎn)者-消費者”和 “ready” 信號這兩種 channel 用法和本文的 channel 用法并沒有什么本質(zhì)區(qū)別。唯一不同的點是前者的 channel 是事先創(chuàng)建好的，并且是“公用”的；而本文中用到的 channel 實際上是“臨時”創(chuàng)建的，并且只有這一個請求使用。

最后，用曹大最近在讀者群里說的話結(jié)尾：

抄代碼是很好的學(xué)習(xí)方式。

選一兩個感興趣的方向，自己嘗試實現(xiàn)相應(yīng)的 feature list，實現(xiàn)完和標(biāo)準(zhǔn)實現(xiàn)做對比。

先積累再創(chuàng)造，別一上來就想著造輪子，看的多了碰上很多東西就有新思路了。

總結(jié)

以上是生活随笔為你收集整理的Go channel 的妙用的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。