feat(notify): 重构通信内核并补齐 stream/bulk/record/transfer 能力

- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
2026-04-15 15:24:36 +08:00
parent d14d13c393
commit 09d972c7b7
216 changed files with 51374 additions and 1715 deletions
@@ -0,0 +1,183 @@
+package notify
+
+import (
+	itransfer "b612.me/notify/internal/transfer"
+	"errors"
+)
+
+type transferControlRuntimeAccessor interface {
+	getTransferRuntime() *transferRuntime
+}
+
+func transferControlRuntimeFromClient(c Client) *transferRuntime {
+	if c == nil {
+		return nil
+	}
+	accessor, ok := any(c).(transferControlRuntimeAccessor)
+	if !ok {
+		return nil
+	}
+	return accessor.getTransferRuntime()
+}
+
+func transferControlRuntimeFromServer(s Server) *transferRuntime {
+	if s == nil {
+		return nil
+	}
+	accessor, ok := any(s).(transferControlRuntimeAccessor)
+	if !ok {
+		return nil
+	}
+	return accessor.getTransferRuntime()
+}
+
+func transferControlPrepareBegin(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, publicScope string, transportGeneration uint64, req TransferBeginRequest) {
+	if runtime == nil || req.TransferID == "" {
+		return
+	}
+	runtime.ensureTransferDescriptor(direction, runtimeScope, publicScope, transportGeneration, itransfer.Descriptor{
+		ID:       req.TransferID,
+		Channel:  itransfer.Channel(req.Channel),
+		Size:     req.Size,
+		Checksum: req.Checksum,
+		Metadata: itransfer.Metadata(cloneTransferMetadata(req.Metadata)),
+	})
+	runtime.recordStage(direction, runtimeScope, req.TransferID, "begin")
+}
+
+func transferControlFinishBegin(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, transferID string, resp TransferBeginResponse, err error) {
+	if runtime == nil {
+		return
+	}
+	transferID = transferControlTransferID(transferID, resp.TransferID)
+	if transferID == "" {
+		return
+	}
+	if err != nil {
+		runtime.recordFailureStage(direction, runtimeScope, transferID, "begin")
+		runtime.fail(direction, runtimeScope, transferID, err)
+		return
+	}
+	runtime.resume(direction, runtimeScope, transferID, transferControlOffset(resp.NextOffset))
+}
+
+func transferControlPrepareResume(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, publicScope string, transportGeneration uint64, req TransferResumeRequest) {
+	if runtime == nil || req.TransferID == "" {
+		return
+	}
+	runtime.ensureTransferDescriptor(direction, runtimeScope, publicScope, transportGeneration, itransfer.Descriptor{
+		ID:      req.TransferID,
+		Channel: itransfer.DataChannel,
+	})
+	runtime.recordStage(direction, runtimeScope, req.TransferID, "resume")
+}
+
+func transferControlFinishResume(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, transferID string, resp TransferResumeResponse, err error) {
+	if runtime == nil {
+		return
+	}
+	transferID = transferControlTransferID(transferID, resp.TransferID)
+	if transferID == "" {
+		return
+	}
+	if err != nil {
+		runtime.recordFailureStage(direction, runtimeScope, transferID, "resume")
+		runtime.fail(direction, runtimeScope, transferID, err)
+		return
+	}
+	runtime.resume(direction, runtimeScope, transferID, transferControlOffset(resp.NextOffset))
+}
+
+func transferControlPrepareCommit(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, publicScope string, transportGeneration uint64, req TransferCommitRequest) {
+	if runtime == nil || req.TransferID == "" {
+		return
+	}
+	runtime.ensureTransferDescriptor(direction, runtimeScope, publicScope, transportGeneration, itransfer.Descriptor{
+		ID:       req.TransferID,
+		Channel:  itransfer.DataChannel,
+		Size:     req.Size,
+		Checksum: req.Checksum,
+	})
+	runtime.recordStage(direction, runtimeScope, req.TransferID, "commit")
+	if direction == fileTransferDirectionSend {
+		runtime.beginCommit(direction, runtimeScope, req.TransferID)
+	}
+}
+
+func transferControlFinishCommit(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, transferID string, resp TransferCommitResponse, err error) {
+	if runtime == nil {
+		return
+	}
+	transferID = transferControlTransferID(transferID, resp.TransferID)
+	if transferID == "" {
+		return
+	}
+	if err != nil {
+		runtime.recordFailureStage(direction, runtimeScope, transferID, "commit")
+		runtime.fail(direction, runtimeScope, transferID, err)
+		return
+	}
+	if direction == fileTransferDirectionReceive {
+		runtime.beginVerify(direction, runtimeScope, transferID)
+	}
+	runtime.complete(direction, runtimeScope, transferID)
+}
+
+func transferControlPrepareAbort(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, publicScope string, transportGeneration uint64, req TransferAbortRequest) {
+	if runtime == nil || req.TransferID == "" {
+		return
+	}
+	stage := transferControlAbortStage(req.Stage)
+	runtime.ensureTransferDescriptor(direction, runtimeScope, publicScope, transportGeneration, itransfer.Descriptor{
+		ID:      req.TransferID,
+		Channel: itransfer.DataChannel,
+	})
+	runtime.recordStage(direction, runtimeScope, req.TransferID, stage)
+	runtime.recordFailureStage(direction, runtimeScope, req.TransferID, stage)
+	runtime.abort(direction, runtimeScope, req.TransferID, transferControlAbortError(req.Error))
+}
+
+func transferControlFinishAbort(runtime *transferRuntime, direction fileTransferDirection, runtimeScope string, transferID string, req TransferAbortRequest, resp TransferAbortResponse, err error) {
+	if runtime == nil {
+		return
+	}
+	transferID = transferControlTransferID(transferID, resp.TransferID)
+	if transferID == "" {
+		return
+	}
+	if err == nil {
+		return
+	}
+	stage := transferControlAbortStage(req.Stage)
+	runtime.recordStage(direction, runtimeScope, transferID, stage)
+	runtime.recordFailureStage(direction, runtimeScope, transferID, stage)
+	runtime.abort(direction, runtimeScope, transferID, err)
+}
+
+func transferControlAbortStage(stage string) string {
+	if stage == "" {
+		return "abort"
+	}
+	return stage
+}
+
+func transferControlAbortError(message string) error {
+	if message == "" {
+		return nil
+	}
+	return errors.New(message)
+}
+
+func transferControlTransferID(primary string, fallback string) string {
+	if primary != "" {
+		return primary
+	}
+	return fallback
+}
+
+func transferControlOffset(offset int64) int64 {
+	if offset < 0 {
+		return 0
+	}
+	return offset
+}