Go小技巧(二)— 打開已經(jīng)關(guān)閉的channel

概述

有時候我們需要在完全可控的范圍內(nèi)復(fù)用channel扇单,但是關(guān)閉了的channel原生語法并沒有提供方法打開,所以利用指針再次打開赠群。

channel的結(jié)構(gòu)體在chan.go中:

type hchan struct {
    qcount   uint           // total data in the queue
    dataqsiz uint           // size of the circular queue
    buf      unsafe.Pointer // points to an array of dataqsiz elements
    elemsize uint16
    closed   uint32
    //... 以下字段沒有用上褒搔,先省略
}

Channel是否關(guān)閉取決于hchan.closed,0是打開履肃,1是關(guān)閉。
方法:讓指針指向hchan.closed直接修改它的值坐桩。

代碼實現(xiàn)

//go:linkname lock runtime.lock
func lock(l *mutex)

//go:linkname unlock runtime.unlock
func unlock(l *mutex)

func open(c interface{}) error {
    v := reflect.ValueOf(c)
    if v.Type().Kind() != reflect.Chan {
        return errors.New("type must be channel")
    }
    i := (*[2]uintptr)(unsafe.Pointer(&c)) //定位c所在數(shù)據(jù)空間尺棋,這里的c是個指針所以要進行一步取值
    var closedOffset, lockOffset uintptr = 28, 88
    closed := (*uint32)(unsafe.Pointer(i[1] + closedOffset)) //指向closed的地址
    if *closed == 1 {
        lockPtr := (*mutex)(unsafe.Pointer(i[1] + lockOffset)) //指向lock地址
        lock(lockPtr) //上鎖
        if *closed == 1 {
            *closed = 0 //直接修改值
        }
        unlock(lockPtr) //解鎖
    }
    return nil
}

closedOffset為什么是28呢?這個涉及到struct對齊問題绵跷,Go內(nèi)存優(yōu)化(一)— struct對齊
在上面主要用了指針定位closed值膘螟,直接修改標志位。為了保證設(shè)置closed值的安全性所以在給它設(shè)置值的時候使用runtime.lock上鎖抖坪。
關(guān)于go:linkname可以自行百度萍鲸,在目錄下創(chuàng)建一個*.s文件可以躲過編譯時error:missing function body

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末擦俐,一起剝皮案震驚了整個濱河市脊阴,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌蚯瞧,老刑警劉巖嘿期,帶你破解...
    沈念sama閱讀 212,686評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異埋合,居然都是意外死亡备徐,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,668評論 3 385
  • 文/潘曉璐 我一進店門甚颂,熙熙樓的掌柜王于貴愁眉苦臉地迎上來蜜猾,“玉大人秀菱,你說我怎么就攤上這事〔渌” “怎么了衍菱?”我有些...
    開封第一講書人閱讀 158,160評論 0 348
  • 文/不壞的土叔 我叫張陵,是天一觀的道長肩豁。 經(jīng)常有香客問我脊串,道長,這世上最難降的妖魔是什么清钥? 我笑而不...
    開封第一講書人閱讀 56,736評論 1 284
  • 正文 為了忘掉前任琼锋,我火速辦了婚禮,結(jié)果婚禮上祟昭,老公的妹妹穿的比我還像新娘缕坎。我一直安慰自己,他們只是感情好从橘,可當我...
    茶點故事閱讀 65,847評論 6 386
  • 文/花漫 我一把揭開白布念赶。 她就那樣靜靜地躺著础钠,像睡著了一般恰力。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上旗吁,一...
    開封第一講書人閱讀 50,043評論 1 291
  • 那天踩萎,我揣著相機與錄音,去河邊找鬼很钓。 笑死香府,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的码倦。 我是一名探鬼主播企孩,決...
    沈念sama閱讀 39,129評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼袁稽!你這毒婦竟也來了勿璃?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,872評論 0 268
  • 序言:老撾萬榮一對情侶失蹤推汽,失蹤者是張志新(化名)和其女友劉穎补疑,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體歹撒,經(jīng)...
    沈念sama閱讀 44,318評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡莲组,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,645評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了暖夭。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片锹杈。...
    茶點故事閱讀 38,777評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡撵孤,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出竭望,到底是詐尸還是另有隱情早直,我是刑警寧澤,帶...
    沈念sama閱讀 34,470評論 4 333
  • 正文 年R本政府宣布市框,位于F島的核電站霞扬,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏枫振。R本人自食惡果不足惜喻圃,卻給世界環(huán)境...
    茶點故事閱讀 40,126評論 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望粪滤。 院中可真熱鬧斧拍,春花似錦、人聲如沸杖小。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,861評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽予权。三九已至昂勉,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間扫腺,已是汗流浹背岗照。 一陣腳步聲響...
    開封第一講書人閱讀 32,095評論 1 267
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留笆环,地道東北人攒至。 一個月前我還...
    沈念sama閱讀 46,589評論 2 362
  • 正文 我出身青樓,卻偏偏與公主長得像躁劣,于是被迫代替她去往敵國和親迫吐。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 43,687評論 2 351

推薦閱讀更多精彩內(nèi)容