starainrt
7ed3dd5b37
- 将 RecordStream 出站路径收敛为单 writer loop - 支持在 batch header 中 piggyback AckSeq，保留独立 ack 作为兼容回退 - 增加 record stream 打开阶段能力协商，支持 mixed-version peer 自动降级 - 补充 RecordSnapshot 与 diagnostics summary 的 record-plane 观测项 - 增加 batch/ack/error frame、piggyback ack、barrier 等待拆分与 apply backlog 指标 - 收紧 TransportConn detach 后的 runtime snapshot 语义 - 补充 README 中的 RecordStream 语义、兼容行为与诊断快照说明 - 补充相关单测与 race 回归验证
118 lines
3.2 KiB
Go
118 lines
3.2 KiB
Go
package notify
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
)
|
|
|
|
func (c *ClientCommon) SetStreamHandler(fn func(StreamAcceptInfo) error) {
|
|
runtime := c.getStreamRuntime()
|
|
if runtime == nil {
|
|
return
|
|
}
|
|
runtime.setHandler(fn)
|
|
}
|
|
|
|
func (c *ClientCommon) OpenStream(ctx context.Context, opt StreamOpenOptions) (Stream, error) {
|
|
if c == nil {
|
|
return nil, errStreamClientNil
|
|
}
|
|
runtime := c.getStreamRuntime()
|
|
if runtime == nil {
|
|
return nil, errStreamRuntimeNil
|
|
}
|
|
req := clientStreamRequest(runtime, opt)
|
|
if req.StreamID == "" {
|
|
return nil, errStreamIDEmpty
|
|
}
|
|
if _, exists := runtime.lookup(clientFileScope(), req.StreamID); exists {
|
|
return nil, errStreamAlreadyExists
|
|
}
|
|
resp, err := sendStreamOpenClient(ctx, c, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.DataID != 0 {
|
|
req.DataID = resp.DataID
|
|
}
|
|
req.Metadata = mergeStreamMetadata(req.Metadata, resp.Metadata)
|
|
stream := newStreamHandle(c.clientStopContextSnapshot(), runtime, clientFileScope(), req, c.currentClientSessionEpoch(), nil, nil, resp.TransportGeneration, clientStreamCloseSender(c), clientStreamResetSender(c), clientStreamDataSender(c, c.currentClientSessionEpoch()), runtime.configSnapshot())
|
|
stream.setClientSnapshotOwner(c)
|
|
stream.setAddrSnapshot(c.clientStreamAddrSnapshot())
|
|
if err := runtime.register(clientFileScope(), stream); err != nil {
|
|
_, _ = sendStreamResetClient(context.Background(), c, StreamResetRequest{
|
|
StreamID: req.StreamID,
|
|
Error: err.Error(),
|
|
})
|
|
return nil, err
|
|
}
|
|
return stream, nil
|
|
}
|
|
|
|
func (c *ClientCommon) clientStreamAddrSnapshot() (net.Addr, net.Addr) {
|
|
if c == nil {
|
|
return nil, nil
|
|
}
|
|
conn := c.clientTransportConnSnapshot()
|
|
if conn == nil {
|
|
return nil, nil
|
|
}
|
|
return conn.LocalAddr(), conn.RemoteAddr()
|
|
}
|
|
|
|
func clientStreamRequest(runtime *streamRuntime, opt StreamOpenOptions) StreamOpenRequest {
|
|
id := opt.ID
|
|
if id == "" && runtime != nil {
|
|
id = runtime.nextID()
|
|
}
|
|
return normalizeStreamOpenRequest(StreamOpenRequest{
|
|
StreamID: id,
|
|
Channel: opt.Channel,
|
|
Metadata: cloneStreamMetadata(opt.Metadata),
|
|
ReadTimeout: opt.ReadTimeout,
|
|
WriteTimeout: opt.WriteTimeout,
|
|
})
|
|
}
|
|
|
|
func clientStreamCloseSender(c *ClientCommon) streamCloseSender {
|
|
return func(ctx context.Context, stream *streamHandle, full bool) error {
|
|
_, err := sendStreamCloseClient(ctx, c, StreamCloseRequest{
|
|
StreamID: stream.ID(),
|
|
Full: full,
|
|
})
|
|
return err
|
|
}
|
|
}
|
|
|
|
func clientStreamResetSender(c *ClientCommon) streamResetSender {
|
|
return func(ctx context.Context, stream *streamHandle, message string) error {
|
|
_, err := sendStreamResetClient(ctx, c, StreamResetRequest{
|
|
StreamID: stream.ID(),
|
|
Error: message,
|
|
})
|
|
return err
|
|
}
|
|
}
|
|
|
|
func clientStreamDataSender(c *ClientCommon, epoch uint64) streamDataSender {
|
|
return func(ctx context.Context, stream *streamHandle, chunk []byte) error {
|
|
if c == nil {
|
|
return errStreamClientNil
|
|
}
|
|
if epoch != 0 && !c.isClientSessionEpochCurrent(epoch) {
|
|
return errTransportDetached
|
|
}
|
|
if ctx != nil {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
default:
|
|
}
|
|
}
|
|
if dataID := stream.dataIDSnapshot(); dataID != 0 {
|
|
return c.sendFastStreamData(dataID, stream.nextOutboundDataSeq(), chunk)
|
|
}
|
|
return c.sendEnvelope(newStreamDataEnvelope(stream.ID(), chunk))
|
|
}
|
|
}
|