GRPC源碼實例解析(二)——UnaryRPC Client 篇

上篇主要介紹了server端的流程移层,這篇的關注點是Client端的流程贞让,同樣只列出核心主流程代碼。

// 調用接口
func (c *greeterClient) SayHello(ctx context.Context, in *HelloRequest, opts ...grpc.CallOption) (*HelloReply, error) {
    out := new(HelloReply)
    err := c.cc.Invoke(ctx, "/helloworld.Greeter/SayHello", in, out, opts...)
    if err != nil {
        return nil, err
    }
    return out, nil
}
// 執(zhí)行接口辽装,通過ClientStream 利用用transport層http2協(xié)議發(fā)送和接收消息被完成序列化和反序列化
func invoke(ctx context.Context, method string, req, reply interface{}, cc *ClientConn, opts ...CallOption) error {
    cs, err := newClientStream(ctx, unaryStreamDesc, cc, method, opts...)
    if err != nil {
        return err
    }
    if err := cs.SendMsg(req); err != nil {
        return err
    }
    return cs.RecvMsg(reply)
}
// 構建核心clientStream對象
func newClientStream(ctx context.Context, desc *StreamDesc, cc *ClientConn, method string, opts ...CallOption) (_ ClientStream, err error){
    var newStream = func(ctx context.Context, done func()) (iresolver.ClientStream, error) {
        return newClientStreamWithParams(ctx, desc, cc, method, mc, onCommit, done, opts...)
    }
    return newStream(ctx, func() {})
}
func newClientStreamWithParams(ctx context.Context, desc *StreamDesc, cc *ClientConn, method string, mc serviceconfig.MethodConfig, onCommit, doneFunc func(), opts ...CallOption) (_ iresolver.ClientStream, err error) {
    callHdr := &transport.CallHdr{
        Host:           cc.authority,
        Method:         method,
        ContentSubtype: c.contentSubtype,
        DoneFunc:       doneFunc,
    }

    cs := &clientStream{
        callHdr:      callHdr,
        ...
        ...

    }
    // 構建attempt吠冤,attempt是在transport層用stream在clientStream中完成實際的sendMsg,并實現(xiàn)retry機制
    cs.newAttemptLocked(sh, trInfo)
    op := func(a *csAttempt) error { return a.newStream() }
    if err := cs.withRetry(op, func() { cs.bufferForRetryLocked(0, op) }); err != nil {
        cs.finish(err)
        return nil, err
    }
}
func (cs *clientStream) SendMsg(m interface{}) (err error) {
    // 預處理數(shù)據(jù)秽梅, 將數(shù)據(jù)encode并壓縮得到hdr抹蚀、data為構建http 2 frame做準備
    hdr, payload, data, err := prepareMsg(m, cs.codec, cs.cp, cs.comp)
    // 利用Retry機制,通過csAttempt實際執(zhí)行向server端發(fā)送消息
    op := func(a *csAttempt) error {
        err := a.sendMsg(m, hdr, payload, data)
        m, data = nil, nil
        return err
    }
    err = cs.withRetry(op, func() { cs.bufferForRetryLocked(len(hdr)+len(payload), op) })
}
// 通過csAttempt將構建好的http2 dataFrame寫入stream中企垦,等于通過http2方式向服務器發(fā)數(shù)據(jù)况鸣。
func (a *csAttempt) sendMsg(m interface{}, hdr, payld, data []byte) error {
    a.t.Write(a.s, hdr, payld, &transport.Options{Last: !cs.desc.ClientStreams})
}

// 將構建好的http2 dataFrame 寫入流中
func (t *http2Client) Write(s *Stream, hdr []byte, data []byte, opts *Options) error {

    df := &dataFrame{
        streamID:  s.id,
        endStream: opts.Last,
        h:         hdr,
        d:         data,
    }
  return t.controlBuf.put(df)
}
// 利用retry機制,調用csAttempt recvMsg
func (cs *clientStream) RecvMsg(m interface{}) error {
    err := cs.withRetry(func(a *csAttempt) error {
        return a.recvMsg(m, recvInfo)
    }, cs.commitAttemptLocked)
}

func (a *csAttempt) recvMsg(m interface{}, payInfo *payloadInfo) (err error) {
      err = recv(a.p, cs.codec, a.s, a.dc, m, *cs.callInfo.maxReceiveMessageSize, payInfo, a.decomp)
}

// 實現(xiàn)數(shù)據(jù)解壓竹观,利用預定義Codec Unmarshal數(shù)據(jù)得到reply镐捧,在本例中就是HelloReply結構體
func recv(p *parser, c baseCodec, s *transport.Stream, dc Decompressor, m interface{}, maxReceiveMessageSize int, payInfo *payloadInfo, compressor encoding.Compressor) error {
    d, err := recvAndDecompress(p, s, dc, maxReceiveMessageSize, payInfo, compressor)
    c.Unmarshal(d, m);
}
``

GRPC源碼實例解析(一)
http://www.reibang.com/p/8bbc6dc36859
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末潜索,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子懂酱,更是在濱河造成了極大的恐慌竹习,老刑警劉巖,帶你破解...
    沈念sama閱讀 212,454評論 6 493
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件列牺,死亡現(xiàn)場離奇詭異整陌,居然都是意外死亡,警方通過查閱死者的電腦和手機瞎领,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,553評論 3 385
  • 文/潘曉璐 我一進店門泌辫,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人九默,你說我怎么就攤上這事震放。” “怎么了驼修?”我有些...
    開封第一講書人閱讀 157,921評論 0 348
  • 文/不壞的土叔 我叫張陵殿遂,是天一觀的道長。 經(jīng)常有香客問我乙各,道長墨礁,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,648評論 1 284
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上谆焊,老公的妹妹穿的比我還像新娘。我一直安慰自己驶乾,他們只是感情好,可當我...
    茶點故事閱讀 65,770評論 6 386
  • 文/花漫 我一把揭開白布冠句。 她就那樣靜靜地躺著轻掩,像睡著了一般幸乒。 火紅的嫁衣襯著肌膚如雪懦底。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,950評論 1 291
  • 那天罕扎,我揣著相機與錄音聚唐,去河邊找鬼。 笑死腔召,一個胖子當著我的面吹牛杆查,可吹牛的內容都是我干的。 我是一名探鬼主播臀蛛,決...
    沈念sama閱讀 39,090評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼亲桦,長吁一口氣:“原來是場噩夢啊……” “哼崖蜜!你這毒婦竟也來了?” 一聲冷哼從身側響起客峭,我...
    開封第一講書人閱讀 37,817評論 0 268
  • 序言:老撾萬榮一對情侶失蹤豫领,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后舔琅,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體等恐,經(jīng)...
    沈念sama閱讀 44,275評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,592評論 2 327
  • 正文 我和宋清朗相戀三年备蚓,在試婚紗的時候發(fā)現(xiàn)自己被綠了课蔬。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,724評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡郊尝,死狀恐怖二跋,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情虚循,我是刑警寧澤同欠,帶...
    沈念sama閱讀 34,409評論 4 333
  • 正文 年R本政府宣布,位于F島的核電站横缔,受9級特大地震影響铺遂,放射性物質發(fā)生泄漏。R本人自食惡果不足惜茎刚,卻給世界環(huán)境...
    茶點故事閱讀 40,052評論 3 316
  • 文/蒙蒙 一襟锐、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧膛锭,春花似錦粮坞、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,815評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至奢入,卻和暖如春筝闹,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背腥光。 一陣腳步聲響...
    開封第一講書人閱讀 32,043評論 1 266
  • 我被黑心中介騙來泰國打工关顷, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人武福。 一個月前我還...
    沈念sama閱讀 46,503評論 2 361
  • 正文 我出身青樓议双,卻偏偏與公主長得像,于是被迫代替她去往敵國和親捉片。 傳聞我的和親對象是個殘疾皇子平痰,可洞房花燭夜當晚...
    茶點故事閱讀 43,627評論 2 350

推薦閱讀更多精彩內容