gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

add methods needed for connection-level FlowController

Browse Source 
ref #39
This commit is contained in:
Marten Seemann
2016-05-18 12:28:00 +07:00
parent e0801262bc
commit 9ecbfa65ac
3 changed files with 40 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
flow_controller.go
View File

@@ -23,13 +23,21 @@ type flowController struct {
mutex sync.RWMutex
}
func newFlowController(connectionParametersManager *handshake.ConnectionParametersManager) *flowController {
return &flowController{
sendFlowControlWindow: connectionParametersManager.GetSendStreamFlowControlWindow(),
receiveFlowControlWindow: connectionParametersManager.GetReceiveStreamFlowControlWindow(),
func newFlowController(streamID protocol.StreamID, connectionParametersManager *handshake.ConnectionParametersManager) *flowController {
fc := flowController{
receiveWindowUpdateThreshold: protocol.WindowUpdateThreshold,
receiveFlowControlWindowIncrement: protocol.ReceiveStreamFlowControlWindowIncrement,
}
if streamID == 0 {
fc.sendFlowControlWindow = connectionParametersManager.GetSendConnectionFlowControlWindow()
fc.receiveFlowControlWindow = connectionParametersManager.GetReceiveConnectionFlowControlWindow()
} else {
fc.sendFlowControlWindow = connectionParametersManager.GetSendStreamFlowControlWindow()
fc.receiveFlowControlWindow = connectionParametersManager.GetReceiveStreamFlowControlWindow()
}
return &fc
}
func (c *flowController) AddBytesSent(n protocol.ByteCount) {
@@ -62,13 +70,27 @@ func (c *flowController) SendWindowSize() protocol.ByteCount {
return c.sendFlowControlWindow - c.bytesSent
}
func (c *flowController) UpdateHighestReceived(n protocol.ByteCount) {
// UpdateHighestReceived updates the highestReceived value, if the byteOffset is higher
// Should **only** be used for the stream-level FlowController
func (c *flowController) UpdateHighestReceived(byteOffset protocol.ByteCount) protocol.ByteCount {
c.mutex.Lock()
defer c.mutex.Unlock()
if n > c.highestReceived {
c.highestReceived = n
if byteOffset > c.highestReceived {
increment := byteOffset - c.highestReceived
c.highestReceived = byteOffset
return increment
}
return 0
}
// IncrementHighestReceived adds an increment to the highestReceived value
// Should **only** be used for the connection-level FlowController
func (c *flowController) IncrementHighestReceived(increment protocol.ByteCount) {
c.mutex.Lock()
defer c.mutex.Unlock()
c.highestReceived += increment
}
func (c *flowController) AddBytesRead(n protocol.ByteCount) {