b612
0826e17063
- 为控制消息增加优先级、公平调度、队列字节预算和自适应批处理 - 支持可取消的写门等待，收紧 shared/dedicated bulk、stream 和 Reply 写入边界 - 修复 bulk reset/close、连接 handoff 和安全 profile 切换时序 - 保留旧取消与超时错误契约，新增阶段化 TransportSendError - 增加 ReplyCtx、ReplyObjCtx、写超时配置及黑洞连接和竞态回归测试
118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
package notify
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net"
|
|
)
|
|
|
|
func (c *ClientConn) startClientConnSession(tuConn net.Conn, stopCtx context.Context, stopFn context.CancelFunc) (context.Context, context.CancelFunc) {
|
|
if c == nil {
|
|
return stopCtx, stopFn
|
|
}
|
|
return c.LogicalConn().startSession(tuConn, stopCtx, stopFn)
|
|
}
|
|
|
|
func (c *ClientConn) startClientConnSessionTransport(tuConn net.Conn, stopCtx context.Context, stopFn context.CancelFunc) (context.Context, context.CancelFunc) {
|
|
if c == nil {
|
|
return stopCtx, stopFn
|
|
}
|
|
return c.LogicalConn().startSessionTransport(tuConn, stopCtx, stopFn)
|
|
}
|
|
|
|
func (c *ClientConn) attachClientConnSessionTransport(tuConn net.Conn) error {
|
|
if c == nil {
|
|
return errors.New("client conn is nil")
|
|
}
|
|
return c.LogicalConn().attachSessionTransport(tuConn)
|
|
}
|
|
|
|
func (c *ClientConn) detachClientConnTransportForTransfer() (net.Conn, error) {
|
|
if c == nil {
|
|
return nil, errors.New("client conn is nil")
|
|
}
|
|
return c.LogicalConn().detachTransportForTransfer()
|
|
}
|
|
|
|
func (c *ClientConn) stopServerOwnedSession(reason string, err error) {
|
|
c.stopServerOwnedSessionWith(nil, reason, err)
|
|
}
|
|
|
|
func (c *LogicalConn) stopServerOwnedSession(reason string, err error) {
|
|
c.stopServerOwnedSessionWith(nil, reason, err)
|
|
}
|
|
|
|
func (c *ClientConn) stopServerOwnedSessionWith(removeFn func(*ClientConn), reason string, err error) {
|
|
if c == nil {
|
|
return
|
|
}
|
|
c.markSessionStopped(reason, err)
|
|
c.detachServerOwnedSessionWith(removeFn)
|
|
}
|
|
|
|
func (c *LogicalConn) stopServerOwnedSessionWith(removeFn func(*LogicalConn), reason string, err error) {
|
|
client := c.compatClientConn()
|
|
if client == nil {
|
|
return
|
|
}
|
|
client.markSessionStopped(reason, err)
|
|
c.detachServerOwnedSessionWith(removeFn)
|
|
}
|
|
|
|
func (c *ClientConn) detachServerOwnedSession() {
|
|
c.detachServerOwnedSessionWith(nil)
|
|
}
|
|
|
|
func (c *LogicalConn) detachServerOwnedSession() {
|
|
c.detachServerOwnedSessionWith(nil)
|
|
}
|
|
|
|
func (c *ClientConn) detachServerOwnedSessionWith(removeFn func(*ClientConn)) {
|
|
if c == nil {
|
|
return
|
|
}
|
|
c.detachServerOwnedTransport()
|
|
if removeFn != nil {
|
|
removeFn(c)
|
|
return
|
|
}
|
|
if c.server != nil {
|
|
c.server.removeClient(c)
|
|
}
|
|
}
|
|
|
|
func (c *LogicalConn) detachServerOwnedSessionWith(removeFn func(*LogicalConn)) {
|
|
client := c.compatClientConn()
|
|
if client == nil {
|
|
return
|
|
}
|
|
c.detachServerOwnedTransport()
|
|
if removeFn != nil {
|
|
removeFn(c)
|
|
return
|
|
}
|
|
if client.server != nil {
|
|
client.server.removeLogical(c)
|
|
}
|
|
}
|
|
|
|
func (c *ClientConn) detachServerOwnedTransport() {
|
|
if c == nil {
|
|
return
|
|
}
|
|
c.LogicalConn().detachServerOwnedTransport()
|
|
}
|
|
|
|
func (c *LogicalConn) detachServerOwnedTransport() {
|
|
if c == nil {
|
|
return
|
|
}
|
|
conn := c.transportSnapshot()
|
|
// Revoke read-loop ownership before closing the socket. Otherwise the close
|
|
// error can race with detach and incorrectly stop the logical session.
|
|
c.clearSessionRuntimeTransport()
|
|
if conn != nil {
|
|
_ = conn.Close()
|
|
}
|
|
}
|