starainrt
09d972c7b7
- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
96 lines
3.6 KiB
Go
96 lines
3.6 KiB
Go
package notify
|
|
|
|
import (
|
|
"b612.me/stario"
|
|
"context"
|
|
"net"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
type ServerCommon struct {
|
|
msgID uint64
|
|
alive atomic.Value
|
|
status Status
|
|
sessionOwnerState atomic.Int32
|
|
sessionRuntime atomic.Pointer[serverSessionRuntime]
|
|
listener net.Listener
|
|
udpListener *net.UDPConn
|
|
queue *stario.StarQueue
|
|
stopFn context.CancelFunc
|
|
stopCtx context.Context
|
|
maxReadTimeout time.Duration
|
|
maxWriteTimeout time.Duration
|
|
parallelNum int
|
|
wg stario.WaitGroup
|
|
peerRegistry *serverPeerRegistry
|
|
mu sync.RWMutex
|
|
handshakeRsaKey []byte
|
|
SecretKey []byte
|
|
defaultMsgEn func([]byte, []byte) []byte
|
|
defaultMsgDe func([]byte, []byte) []byte
|
|
defaultFastStreamEncode transportFastStreamEncoder
|
|
defaultFastBulkEncode transportFastBulkEncoder
|
|
defaultFastPlainEncode transportFastPlainEncoder
|
|
linkFns map[string]func(message *Message)
|
|
defaultFns func(message *Message)
|
|
noFinSyncMsgMaxKeepSeconds int64
|
|
maxHeartbeatLostSeconds int64
|
|
sequenceDe func([]byte) (interface{}, error)
|
|
sequenceEn func(interface{}) ([]byte, error)
|
|
logicalSession *logicalSessionState
|
|
onFileEvent func(FileEvent)
|
|
fileEventObserver func(FileEvent)
|
|
fileTransferCfg fileTransferConfig
|
|
signalReliableCfg signalReliabilityConfig
|
|
streamRuntime *streamRuntime
|
|
recordRuntime *recordRuntime
|
|
bulkRuntime *bulkRuntime
|
|
connectionRetryState *connectionRetryState
|
|
detachedClientKeepSeconds int64
|
|
securityReadyCheck bool
|
|
showError bool
|
|
debugMode bool
|
|
}
|
|
|
|
func NewServer() Server {
|
|
transport := defaultModernPSKTransportBundle()
|
|
var server ServerCommon
|
|
server.wg = stario.NewWaitGroup(0)
|
|
server.parallelNum = 0
|
|
server.noFinSyncMsgMaxKeepSeconds = 0
|
|
server.maxHeartbeatLostSeconds = 300
|
|
server.stopCtx, server.stopFn = context.WithCancel(context.Background())
|
|
server.SecretKey = nil
|
|
server.handshakeRsaKey = defaultRsaKey
|
|
server.peerRegistry = newServerPeerRegistry()
|
|
server.defaultMsgEn = transport.msgEn
|
|
server.defaultMsgDe = transport.msgDe
|
|
server.defaultFastStreamEncode = transport.fastStreamEncode
|
|
server.defaultFastBulkEncode = transport.fastBulkEncode
|
|
server.defaultFastPlainEncode = transport.fastPlainEncode
|
|
server.securityReadyCheck = true
|
|
server.sequenceEn = encode
|
|
server.sequenceDe = Decode
|
|
server.alive.Store(false)
|
|
server.linkFns = make(map[string]func(*Message))
|
|
server.fileTransferCfg = defaultFileTransferConfig()
|
|
server.signalReliableCfg = defaultSignalReliabilityConfig()
|
|
server.logicalSession = newLogicalSessionState(server.fileTransferCfg, server.signalReliableCfg)
|
|
server.streamRuntime = newStreamRuntime("sstrm")
|
|
server.recordRuntime = newRecordRuntime()
|
|
server.bulkRuntime = newBulkRuntime("sblk")
|
|
server.connectionRetryState = newConnectionRetryState()
|
|
server.onFileEvent = normalizeFileEventCallback(nil)
|
|
server.fileEventObserver = normalizeFileEventCallback(nil)
|
|
server.defaultFns = func(message *Message) {
|
|
return
|
|
}
|
|
server.sessionRuntime.Store(newServerSessionRuntimeBase(server.stopCtx, server.stopFn))
|
|
bindServerStreamControl(&server)
|
|
bindServerBulkControl(&server)
|
|
server.getTransferState().setBuiltinHandler(server.builtinFileTransferHandler)
|
|
return &server
|
|
}
|