gRPC源码/transport/流控

流控

代码

https://github.com/messixukej…
在liangzhiyang/annotate-grpc-go基础上补充了部分注释

发送流控

HTTP/2 流量控制的目标,在流量窗口初始值的约束下,给予接收端以全权,控制当下想要接受的流量大小。

  • 算法:

    • 两端(收发)保有一个流量控制窗口(window)初始值。
    • 发送端每发送一个DATA帧,就把window递减,递减量为这个帧的大小,要是window小于帧大小,那么这个帧就必须被拆分。如果window等于0,就不能发送任何帧。
    • 接收端可以发送 WINDOW_UPDATE帧给发送端,发送端以帧内指定的Window Size Increment作为增量,加到window上。
// 用于发送流控。 将当前可用的quota(字节数)写入c,有数据发送需要时,从c中获取。acquire到的大小即为可支持的最大发送量。
// acquire 将quota全部获取出来,根据实际使用量,将未使用的重新add回pool。
type quotaPool struct{
    c chan int
    mu    sync.Mutex
    quota int
}

http2Client.Write消耗quota,client跟stream有各自的控制。
handleWindowUpdate补充quota。

接收流控

//用于接收流控。onData通过判断pendingData、pendingUpdate之和是否超过limit来进行流控。
type inFlow struct{
    /// The inbound flow control limit for pending data./    
    limit uint32

    mu sync.Mutex
    // pendingData is the overall data which have been received but not been consumed by applications.
    //接收但是未被应用消费的数据,对应onData。
    pendingData uint32

    // The amount of data the application has consumed but grpc has not sent window update for them. Used to reduce window update frequency.
    //对应onRead 
    pendingUpdate uint32
}

pendingData:handleData->onData增加pendingData->s.write(可供io.ReadFull读取)

pendingUpdate:io.ReadFull(s1, p)->Stream.Read读取数据->windowHandler->updateWindow->onRead减少pendingData,按照1/4limit量还清空pendingUpdate->windowUpdate->framer.writeWindowUpdate更新发送端窗口大小->发送端处理handleWindowUpdate(id=0更新client,非0更新对应stream)
->进而增加发送端quota
    原文作者:蓝白狂想
    原文地址: https://segmentfault.com/a/1190000015232494
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞