go select編譯期的優(yōu)化處理邏輯使用場(chǎng)景分析
更新時(shí)間：2021年06月28日 11:49:09 作者：yuchenfw
select 是 Go 中的一個(gè)控制結(jié)構(gòu)，類(lèi)似于用于通信的 switch 語(yǔ)句。每個(gè) case 必須是一個(gè)通信操作，要么是發(fā)送要么是接收。接下來(lái)通過(guò)本文給大家介紹go select編譯期的優(yōu)化處理邏輯使用場(chǎng)景分析，感興趣的朋友一起看看吧
前言
select作為Go chan通信的重要監(jiān)聽(tīng)工具，有著很廣泛的使用場(chǎng)景。select的使用主要是搭配通信case使用，表面上看，只是簡(jiǎn)單的select及case搭配，實(shí)際上根據(jù)case的數(shù)量及類(lèi)型，在編譯時(shí)select會(huì)進(jìn)行優(yōu)化處理，根據(jù)不同的情況調(diào)用不同的底層邏輯。
select的編譯處理
select編譯時(shí)的核心處理邏輯如下:
func walkselectcases(cases *Nodes) []*Node {
ncas := cases.Len()
sellineno := lineno
// optimization: zero-case select
// 針對(duì)沒(méi)有case的select優(yōu)化
if ncas == 0 {
return []*Node{mkcall("block", nil, nil)}
}
// optimization: one-case select: single op.
// 針對(duì)1個(gè)case（單個(gè)操作）select的優(yōu)化
if ncas == 1 {
cas := cases.First()
setlineno(cas)
l := cas.Ninit.Slice()
if cas.Left != nil { // not default: 非default case
n := cas.Left // 獲取case表達(dá)式
l = append(l, n.Ninit.Slice()...)
n.Ninit.Set(nil)
switch n.Op {
default:
Fatalf("select %v", n.Op)
case OSEND: // Left <- Right
// already ok
// n中已包含left/right
case OSELRECV, OSELRECV2: // OSELRECV(Left = <-Right.Left) OSELRECV2(List = <-Right.Left)
if n.Op == OSELRECV || n.List.Len() == 0 { // 左側(cè)有0或1個(gè)接收者
if n.Left == nil { // 沒(méi)有接收者
n = n.Right // 只需保留右側(cè)
} else { //
n.Op = OAS // 只有一個(gè)接收者，更新Op為OAS
}
break
}
if n.Left == nil { // 檢查是否表達(dá)式或賦值
nblank = typecheck(nblank, ctxExpr|ctxAssign)
n.Left = nblank
}
n.Op = OAS2 // OSELRECV2多個(gè)接收者
n.List.Prepend(n.Left) // 將left放在前面
n.Rlist.Set1(n.Right)
n.Right = nil
n.Left = nil
n.SetTypecheck(0)
n = typecheck(n, ctxStmt)
}
l = append(l, n)
}
l = append(l, cas.Nbody.Slice()...) // case內(nèi)的處理
l = append(l, nod(OBREAK, nil, nil)) // 添加break
return l
}
// convert case value arguments to addresses.
// this rewrite is used by both the general code and the next optimization.
var dflt *Node
for _, cas := range cases.Slice() {
setlineno(cas)
n := cas.Left
if n == nil {
dflt = cas
continue
}
switch n.Op {
case OSEND:
n.Right = nod(OADDR, n.Right, nil)
n.Right = typecheck(n.Right, ctxExpr)
case OSELRECV, OSELRECV2:
if n.Op == OSELRECV2 && n.List.Len() == 0 {
n.Op = OSELRECV
}
if n.Left != nil {
n.Left = nod(OADDR, n.Left, nil)
n.Left = typecheck(n.Left, ctxExpr)
}
}
}
// optimization: two-case select but one is default: single non-blocking op.
if ncas == 2 && dflt != nil {
cas := cases.First()
if cas == dflt {
cas = cases.Second()
}
n := cas.Left
setlineno(n)
r := nod(OIF, nil, nil)
r.Ninit.Set(cas.Ninit.Slice())
switch n.Op {
default:
Fatalf("select %v", n.Op)
case OSEND:
// if selectnbsend(c, v) { body } else { default body }
ch := n.Left
r.Left = mkcall1(chanfn("selectnbsend", 2, ch.Type), types.Types[TBOOL], &r.Ninit, ch, n.Right)
case OSELRECV:
// if selectnbrecv(&v, c) { body } else { default body }
ch := n.Right.Left
elem := n.Left
if elem == nil {
elem = nodnil()
}
r.Left = mkcall1(chanfn("selectnbrecv", 2, ch.Type), types.Types[TBOOL], &r.Ninit, elem, ch)
case OSELRECV2:
// if selectnbrecv2(&v, &received, c) { body } else { default body }
ch := n.Right.Left
elem := n.Left
if elem == nil {
elem = nodnil()
}
receivedp := nod(OADDR, n.List.First(), nil)
receivedp = typecheck(receivedp, ctxExpr)
r.Left = mkcall1(chanfn("selectnbrecv2", 2, ch.Type), types.Types[TBOOL], &r.Ninit, elem, receivedp, ch)
}
r.Left = typecheck(r.Left, ctxExpr)
r.Nbody.Set(cas.Nbody.Slice())
r.Rlist.Set(append(dflt.Ninit.Slice(), dflt.Nbody.Slice()...))
return []*Node{r, nod(OBREAK, nil, nil)}
}
if dflt != nil {
ncas--
}
casorder := make([]*Node, ncas)
nsends, nrecvs := 0, 0
var init []*Node
// generate sel-struct
lineno = sellineno
selv := temp(types.NewArray(scasetype(), int64(ncas)))
r := nod(OAS, selv, nil)
r = typecheck(r, ctxStmt)
init = append(init, r)
// No initialization for order; runtime.selectgo is responsible for that.
order := temp(types.NewArray(types.Types[TUINT16], 2*int64(ncas)))
var pc0, pcs *Node
if flag_race {
pcs = temp(types.NewArray(types.Types[TUINTPTR], int64(ncas)))
pc0 = typecheck(nod(OADDR, nod(OINDEX, pcs, nodintconst(0)), nil), ctxExpr)
} else {
pc0 = nodnil()
}
// register cases
for _, cas := range cases.Slice() {
setlineno(cas)
init = append(init, cas.Ninit.Slice()...)
cas.Ninit.Set(nil)
n := cas.Left
if n == nil { // default:
continue
}
var i int
var c, elem *Node
switch n.Op {
default:
Fatalf("select %v", n.Op)
case OSEND:
i = nsends
nsends++
c = n.Left
elem = n.Right
case OSELRECV, OSELRECV2:
nrecvs++
i = ncas - nrecvs
c = n.Right.Left
elem = n.Left
}
casorder[i] = cas
setField := func(f string, val *Node) {
r := nod(OAS, nodSym(ODOT, nod(OINDEX, selv, nodintconst(int64(i))), lookup(f)), val)
r = typecheck(r, ctxStmt)
init = append(init, r)
}
c = convnop(c, types.Types[TUNSAFEPTR])
setField("c", c)
if elem != nil {
elem = convnop(elem, types.Types[TUNSAFEPTR])
setField("elem", elem)
}
// TODO(mdempsky): There should be a cleaner way to
// handle this.
if flag_race {
r = mkcall("selectsetpc", nil, nil, nod(OADDR, nod(OINDEX, pcs, nodintconst(int64(i))), nil))
init = append(init, r)
}
}
if nsends+nrecvs != ncas {
Fatalf("walkselectcases: miscount: %v + %v != %v", nsends, nrecvs, ncas)
}
// run the select
lineno = sellineno
chosen := temp(types.Types[TINT])
recvOK := temp(types.Types[TBOOL])
r = nod(OAS2, nil, nil)
r.List.Set2(chosen, recvOK)
fn := syslook("selectgo")
r.Rlist.Set1(mkcall1(fn, fn.Type.Results(), nil, bytePtrToIndex(selv, 0), bytePtrToIndex(order, 0), pc0, nodintconst(int64(nsends)), nodintconst(int64(nrecvs)), nodbool(dflt == nil)))
r = typecheck(r, ctxStmt)
init = append(init, r)
// selv and order are no longer alive after selectgo.
init = append(init, nod(OVARKILL, selv, nil))
init = append(init, nod(OVARKILL, order, nil))
if flag_race {
init = append(init, nod(OVARKILL, pcs, nil))
}
// dispatch cases
dispatch := func(cond, cas *Node) {
cond = typecheck(cond, ctxExpr)
cond = defaultlit(cond, nil)
r := nod(OIF, cond, nil)
if n := cas.Left; n != nil && n.Op == OSELRECV2 {
x := nod(OAS, n.List.First(), recvOK)
x = typecheck(x, ctxStmt)
r.Nbody.Append(x)
}
r.Nbody.AppendNodes(&cas.Nbody)
r.Nbody.Append(nod(OBREAK, nil, nil))
init = append(init, r)
}
if dflt != nil {
setlineno(dflt)
dispatch(nod(OLT, chosen, nodintconst(0)), dflt)
}
for i, cas := range casorder {
setlineno(cas)
dispatch(nod(OEQ, chosen, nodintconst(int64(i))), cas)
}
return init
}
select編譯時(shí)會(huì)根據(jù)case的數(shù)量進(jìn)行優(yōu)化：
1.沒(méi)有case
直接調(diào)用block
2.1個(gè)case
(1)default case，直接執(zhí)行body
(2) send/recv case (block為true)，按照單獨(dú)執(zhí)行的結(jié)果確認(rèn)，可能會(huì)發(fā)生block
(3) send調(diào)用對(duì)應(yīng)的chansend1
(4) recv調(diào)用對(duì)應(yīng)的chanrecv1/chanrecv2
3.2個(gè)case且包含一個(gè)default case
(1) send/recv case (block為false)，按照單獨(dú)執(zhí)行的結(jié)果確認(rèn)case是否ok，!ok則執(zhí)行default case，不會(huì)發(fā)生block
(2) send調(diào)用對(duì)應(yīng)的selectnbsend
(3) recv調(diào)用對(duì)應(yīng)的selectnbrecv/selectnbrecv2
4.一般的case
selectgo
總結(jié)
最后，以一張圖進(jìn)行簡(jiǎn)單總結(jié)。
以上就是go select編譯期的優(yōu)化處理邏輯使用場(chǎng)景分析的詳細(xì)內(nèi)容，更多關(guān)于go select編譯的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
拾遺主要是收集和golang相關(guān)的瑣碎知識(shí)，這篇文章主要是為大家整理了Golang中指針和接口的使用方法，文中的示例代碼講解詳細(xì)，需要的可以參考一下2023-02-02
本文詳細(xì)講解了Golang中的Interface，文中通過(guò)示例代碼介紹的非常詳細(xì)。對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-07-07
這篇文章主要為大家介紹了詳解以go思想去處理js異常拋棄trycatch，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-03-03
這篇文章主要為大家介紹了GO必知必會(huì)的常見(jiàn)面試題匯總2022-08-08
go現(xiàn)在的更新速度是非常的快啊,用著用著網(wǎng)上的教程就不配套了,下面這篇文章主要給大家介紹了關(guān)于快速升級(jí)Go版本的相關(guān)資料,文中介紹的方法幾分鐘就搞定了,需要的朋友可以參考下2024-05-05
本文主要介紹了gORM操作MySQL的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-07-07
Map是一個(gè)強(qiáng)大而又有趣的工具，它可以幫助我們高效地存儲(chǔ)和操作鍵值對(duì)數(shù)據(jù)，本文主要介紹了Go語(yǔ)言中Map的各種操作，包括增加、查找、刪除、遍歷等，具有一定的參考價(jià)值，感興趣的可以了解一下2023-08-08
這篇文章主要介紹了Go 語(yǔ)言中的空接口的實(shí)現(xiàn)方法，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-06-06
Go實(shí)現(xiàn)set類(lèi)型的示例代碼
本文主要介紹了Go實(shí)現(xiàn)set類(lèi)型的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-01-01
Go語(yǔ)言標(biāo)準(zhǔn)庫(kù)中的io包提供了一系列接口,用于處理各種I/O操作,其中Seeker接口在處理大文件或需要隨機(jī)訪(fǎng)問(wèn)的場(chǎng)景中非常有用,本文將結(jié)合具體案例,詳細(xì)介紹Go語(yǔ)言中io包的Seeker接口的用法,需要的朋友可以參考下2024-10-10