fix: close stream adaptive gaps and switch notify to stario v0.1.1
- make stream fast path honor adaptive soft payload limits end-to-end - split oversized fast-stream payloads into sequential frames before batching - use adaptive soft cap when encoding stream batch payloads - move timeout-like error detection into production code for adaptive tx - tune notify FrameReader read size explicitly to avoid throughput regression - drop local stario replace and depend on released b612.me/stario v0.1.1
This commit is contained in:
+105
-4
@@ -12,6 +12,10 @@ import (
|
||||
const bulkDispatchRejectTimeout = 300 * time.Millisecond
|
||||
|
||||
func (c *ClientCommon) dispatchFastBulkFrame(frame bulkFastFrame) {
|
||||
c.dispatchFastBulkFrameWithOwner(frame, nil)
|
||||
}
|
||||
|
||||
func (c *ClientCommon) dispatchFastBulkFrameWithOwner(frame bulkFastFrame, owner *bulkReadPayloadOwner) {
|
||||
if frame.DataID == 0 {
|
||||
return
|
||||
}
|
||||
@@ -38,7 +42,13 @@ func (c *ClientCommon) dispatchFastBulkFrame(frame bulkFastFrame) {
|
||||
}
|
||||
switch frame.Type {
|
||||
case bulkFastPayloadTypeData:
|
||||
if err := bulk.pushOwnedChunk(frame.Payload); err != nil {
|
||||
var err error
|
||||
if owner != nil {
|
||||
err = bulk.pushChunkWithOwnershipOptionsAndRelease(frame.Payload, true, true, owner.retainChunk())
|
||||
} else {
|
||||
err = bulk.pushOwnedChunk(frame.Payload)
|
||||
}
|
||||
if err != nil {
|
||||
if c.showError || c.debugMode {
|
||||
fmt.Println("client bulk push chunk error", err)
|
||||
}
|
||||
@@ -58,14 +68,28 @@ func (c *ClientCommon) dispatchFastBulkFrame(frame bulkFastFrame) {
|
||||
resetErr = bulkRemoteResetError(string(frame.Payload))
|
||||
}
|
||||
bulk.markReset(bulkResetError(resetErr))
|
||||
case bulkFastPayloadTypeRelease:
|
||||
bytes, chunks, err := decodeBulkDedicatedReleasePayload(frame.Payload)
|
||||
if err != nil {
|
||||
if c.showError || c.debugMode {
|
||||
fmt.Println("client bulk release decode error", err)
|
||||
}
|
||||
c.bestEffortRejectInboundBulkData(bulk.ID(), frame.DataID, err.Error())
|
||||
return
|
||||
}
|
||||
bulk.releaseOutboundWindow(bytes, chunks)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *ClientCommon) dispatchFastBulkData(frame bulkFastDataFrame) {
|
||||
c.dispatchFastBulkFrame(frame)
|
||||
c.dispatchFastBulkFrameWithOwner(frame, nil)
|
||||
}
|
||||
|
||||
func (s *ServerCommon) dispatchFastBulkFrame(logical *LogicalConn, transport *TransportConn, conn net.Conn, frame bulkFastFrame) {
|
||||
s.dispatchFastBulkFrameWithOwner(logical, transport, conn, frame, nil)
|
||||
}
|
||||
|
||||
func (s *ServerCommon) dispatchFastBulkFrameWithOwner(logical *LogicalConn, transport *TransportConn, conn net.Conn, frame bulkFastFrame, owner *bulkReadPayloadOwner) {
|
||||
if logical == nil || frame.DataID == 0 {
|
||||
return
|
||||
}
|
||||
@@ -91,7 +115,13 @@ func (s *ServerCommon) dispatchFastBulkFrame(logical *LogicalConn, transport *Tr
|
||||
}
|
||||
switch frame.Type {
|
||||
case bulkFastPayloadTypeData:
|
||||
if err := bulk.pushOwnedChunk(frame.Payload); err != nil {
|
||||
var err error
|
||||
if owner != nil {
|
||||
err = bulk.pushChunkWithOwnershipOptionsAndRelease(frame.Payload, true, true, owner.retainChunk())
|
||||
} else {
|
||||
err = bulk.pushOwnedChunk(frame.Payload)
|
||||
}
|
||||
if err != nil {
|
||||
if s.showError || s.debugMode {
|
||||
fmt.Println("server bulk push chunk error", err)
|
||||
}
|
||||
@@ -111,11 +141,82 @@ func (s *ServerCommon) dispatchFastBulkFrame(logical *LogicalConn, transport *Tr
|
||||
resetErr = bulkRemoteResetError(string(frame.Payload))
|
||||
}
|
||||
bulk.markReset(bulkResetError(resetErr))
|
||||
case bulkFastPayloadTypeRelease:
|
||||
bytes, chunks, err := decodeBulkDedicatedReleasePayload(frame.Payload)
|
||||
if err != nil {
|
||||
if s.showError || s.debugMode {
|
||||
fmt.Println("server bulk release decode error", err)
|
||||
}
|
||||
s.bestEffortRejectInboundBulkData(logical, transport, conn, bulk.ID(), frame.DataID, err.Error())
|
||||
return
|
||||
}
|
||||
bulk.releaseOutboundWindow(bytes, chunks)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *ServerCommon) dispatchFastBulkData(logical *LogicalConn, transport *TransportConn, conn net.Conn, frame bulkFastDataFrame) {
|
||||
s.dispatchFastBulkFrame(logical, transport, conn, frame)
|
||||
s.dispatchFastBulkFrameWithOwner(logical, transport, conn, frame, nil)
|
||||
}
|
||||
|
||||
func (c *ClientCommon) tryDispatchBorrowedBulkTransportPayload(payload []byte) bool {
|
||||
if c == nil || len(payload) == 0 {
|
||||
return false
|
||||
}
|
||||
plain, plainRelease, err := decryptTransportPayloadCodecOwnedPooled(c.modernPSKRuntime, c.msgDe, c.SecretKey, payload)
|
||||
if err != nil {
|
||||
if c.showError || c.debugMode {
|
||||
fmt.Println("client decode transport payload error", err)
|
||||
}
|
||||
return true
|
||||
}
|
||||
owner := newBulkReadPayloadOwner(plainRelease)
|
||||
matched, walkErr := walkBulkFastFrames(plain, func(frame bulkFastFrame) error {
|
||||
c.dispatchFastBulkFrameWithOwner(frame, owner)
|
||||
return nil
|
||||
})
|
||||
if owner != nil {
|
||||
owner.done()
|
||||
}
|
||||
if !matched {
|
||||
return false
|
||||
}
|
||||
if walkErr != nil && (c.showError || c.debugMode) {
|
||||
fmt.Println("client decode bulk fast payload error", walkErr)
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func (s *ServerCommon) tryDispatchBorrowedBulkTransportPayload(source interface{}, payload []byte) bool {
|
||||
if s == nil || len(payload) == 0 {
|
||||
return false
|
||||
}
|
||||
logical, transport := s.resolveInboundSource(source)
|
||||
if logical == nil {
|
||||
return false
|
||||
}
|
||||
plain, plainRelease, err := decryptTransportPayloadCodecOwnedPooled(logical.modernPSKRuntimeSnapshot(), logical.msgDeSnapshot(), logical.secretKeySnapshot(), payload)
|
||||
if err != nil {
|
||||
if s.showError || s.debugMode {
|
||||
fmt.Println("server decode transport payload error", err)
|
||||
}
|
||||
return true
|
||||
}
|
||||
conn := serverInboundConn(source)
|
||||
owner := newBulkReadPayloadOwner(plainRelease)
|
||||
matched, walkErr := walkBulkFastFrames(plain, func(frame bulkFastFrame) error {
|
||||
s.dispatchFastBulkFrameWithOwner(logical, transport, conn, frame, owner)
|
||||
return nil
|
||||
})
|
||||
if owner != nil {
|
||||
owner.done()
|
||||
}
|
||||
if !matched {
|
||||
return false
|
||||
}
|
||||
if walkErr != nil && (s.showError || s.debugMode) {
|
||||
fmt.Println("server decode bulk fast payload error", walkErr)
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func (c *ClientCommon) bestEffortRejectInboundBulkData(bulkID string, dataID uint64, message string) {
|
||||
|
||||
Reference in New Issue
Block a user