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