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 回归验证
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package notify
|
|
|
|
import "context"
|
|
|
|
func (s *ServerCommon) SetRecordStreamHandler(fn func(RecordAcceptInfo) error) {
|
|
runtime := s.getRecordRuntime()
|
|
if runtime == nil {
|
|
return
|
|
}
|
|
runtime.setHandler(fn)
|
|
}
|
|
|
|
func (s *ServerCommon) OpenRecordStreamLogical(ctx context.Context, logical *LogicalConn, opt RecordOpenOptions) (RecordStream, error) {
|
|
if s == nil {
|
|
return nil, errStreamServerNil
|
|
}
|
|
opt = normalizeRecordOpenOptions(opt)
|
|
stream, err := s.OpenStreamLogical(ctx, logical, opt.Stream)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
record, err := WrapStreamAsRecord(stream, opt)
|
|
if err != nil {
|
|
_ = stream.Reset(err)
|
|
return nil, err
|
|
}
|
|
bindRecordRuntime(record, s.getRecordRuntime())
|
|
return record, nil
|
|
}
|
|
|
|
func (s *ServerCommon) OpenRecordStreamTransport(ctx context.Context, transport *TransportConn, opt RecordOpenOptions) (RecordStream, error) {
|
|
if s == nil {
|
|
return nil, errStreamServerNil
|
|
}
|
|
opt = normalizeRecordOpenOptions(opt)
|
|
stream, err := s.OpenStreamTransport(ctx, transport, opt.Stream)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
record, err := WrapStreamAsRecord(stream, opt)
|
|
if err != nil {
|
|
_ = stream.Reset(err)
|
|
return nil, err
|
|
}
|
|
bindRecordRuntime(record, s.getRecordRuntime())
|
|
return record, nil
|
|
}
|
|
|
|
func (s *ServerCommon) claimInboundRecordStream(logical *LogicalConn, transport *TransportConn, stream *streamHandle) (bool, error) {
|
|
if stream == nil || stream.Channel() != StreamRecordChannel {
|
|
return false, nil
|
|
}
|
|
runtime := s.getRecordRuntime()
|
|
if runtime == nil {
|
|
return true, errRecordRuntimeNil
|
|
}
|
|
handler := runtime.handlerSnapshot()
|
|
if handler == nil {
|
|
return true, errRecordHandlerNotConfigured
|
|
}
|
|
record, err := WrapStreamAsRecord(stream, RecordOpenOptions{
|
|
Stream: StreamOpenOptions{
|
|
ID: stream.ID(),
|
|
Channel: stream.Channel(),
|
|
Metadata: stream.Metadata(),
|
|
ReadTimeout: stream.readTimeoutSnapshot(),
|
|
WriteTimeout: stream.writeTimeoutSnapshot(),
|
|
},
|
|
})
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
bindRecordRuntime(record, runtime)
|
|
info := RecordAcceptInfo{
|
|
ID: stream.ID(),
|
|
Metadata: stream.Metadata(),
|
|
LogicalConn: logical,
|
|
TransportConn: transport,
|
|
TransportGeneration: stream.TransportGeneration(),
|
|
RecordStream: record,
|
|
}
|
|
go func() {
|
|
if err := handler(info); err != nil {
|
|
_ = record.Reset(err)
|
|
}
|
|
}()
|
|
return true, nil
|
|
}
|