notify/server_dispatcher.go
starainrt 09d972c7b7
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

67 lines
1.5 KiB
Go

package notify
func (s *ServerCommon) dispatchMsg(message Message) {
message = hydrateServerMessagePeerFields(message)
logical := messageLogicalConnSnapshot(&message)
transport := messageTransportConnSnapshot(&message)
switch message.TransferMsg.Type {
case MSG_SYS_WAIT:
fallthrough
case MSG_SYS:
s.sysMsg(message)
return
case MSG_KEY_CHANGE:
if logical != nil {
logical.rsaDecode(message)
}
return
case MSG_SYS_REPLY:
fallthrough
case MSG_SYNC_REPLY:
scopes := serverTransportDeliveryScopes(logical)
if transport != nil {
scopes = serverTransportDeliveryScopesForTransport(transport)
}
if s.getPendingWaitPool().deliverWithScopes(message.TransferMsg.ID, scopes, message) {
return
}
fallthrough
default:
}
if s.dispatchInternalTransferControl(message) {
return
}
callFn := func(fn func(*Message)) {
fn(&message)
}
fn, ok := s.linkFns[message.TransferMsg.Key]
if ok {
callFn(fn)
}
if s.defaultFns != nil {
callFn(s.defaultFns)
}
}
func (s *ServerCommon) sysMsg(message Message) {
if s.handleBulkAttachSystemMessage(message) {
return
}
if s.handlePeerAttachSystemMessage(message) {
return
}
logical := messageLogicalConnSnapshot(&message)
switch message.Key {
case "bye":
if message.TransferMsg.Type == MSG_SYS_WAIT {
message.Reply(nil)
}
s.stopLogicalSession(logical, "recv stop signal from client", nil)
case "heartbeat":
if logical != nil {
logical.markHeartbeatNow()
}
message.Reply(nil)
}
}