feat(notify): 重构通信内核并补齐 stream/bulk/record/transfer 能力
- 引入 LogicalConn/TransportConn 分层,ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
This commit is contained in:
@@ -0,0 +1,117 @@
|
||||
package notify
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
||||
const defaultReceivedSignalCacheLimit = 256
|
||||
|
||||
type receivedSignalCache struct {
|
||||
mu sync.Mutex
|
||||
entries map[string]struct{}
|
||||
order []string
|
||||
limit int
|
||||
}
|
||||
|
||||
func newReceivedSignalCache(limit int) *receivedSignalCache {
|
||||
if limit <= 0 {
|
||||
limit = defaultReceivedSignalCacheLimit
|
||||
}
|
||||
return &receivedSignalCache{
|
||||
entries: make(map[string]struct{}),
|
||||
limit: limit,
|
||||
}
|
||||
}
|
||||
|
||||
func receivedSignalCacheKey(scope string, signalID uint64) string {
|
||||
return normalizeFileScope(scope) + "|" + strconv.FormatUint(signalID, 10)
|
||||
}
|
||||
|
||||
func (c *receivedSignalCache) seenOrRemember(scope string, signalID uint64) bool {
|
||||
if c == nil || signalID == 0 {
|
||||
return false
|
||||
}
|
||||
key := receivedSignalCacheKey(scope, signalID)
|
||||
c.mu.Lock()
|
||||
defer c.mu.Unlock()
|
||||
if _, ok := c.entries[key]; ok {
|
||||
return true
|
||||
}
|
||||
c.entries[key] = struct{}{}
|
||||
c.order = append(c.order, key)
|
||||
c.trimLocked()
|
||||
return false
|
||||
}
|
||||
|
||||
func (c *receivedSignalCache) closeAll() {
|
||||
if c == nil {
|
||||
return
|
||||
}
|
||||
c.mu.Lock()
|
||||
c.entries = make(map[string]struct{})
|
||||
c.order = nil
|
||||
c.mu.Unlock()
|
||||
}
|
||||
|
||||
func (c *receivedSignalCache) closeScope(scope string) {
|
||||
if c == nil {
|
||||
return
|
||||
}
|
||||
scope = normalizeFileScope(scope)
|
||||
prefix := scope + "|"
|
||||
c.mu.Lock()
|
||||
defer c.mu.Unlock()
|
||||
for key := range c.entries {
|
||||
if strings.HasPrefix(key, prefix) {
|
||||
delete(c.entries, key)
|
||||
}
|
||||
}
|
||||
if len(c.order) == 0 {
|
||||
return
|
||||
}
|
||||
filtered := c.order[:0]
|
||||
for _, key := range c.order {
|
||||
if _, ok := c.entries[key]; !ok {
|
||||
continue
|
||||
}
|
||||
filtered = append(filtered, key)
|
||||
}
|
||||
c.order = filtered
|
||||
}
|
||||
|
||||
func (c *receivedSignalCache) trimLocked() {
|
||||
if c.limit <= 0 || len(c.entries) <= c.limit {
|
||||
return
|
||||
}
|
||||
for len(c.entries) > c.limit && len(c.order) > 0 {
|
||||
key := c.order[0]
|
||||
c.order = c.order[1:]
|
||||
if _, ok := c.entries[key]; !ok {
|
||||
continue
|
||||
}
|
||||
delete(c.entries, key)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *receivedSignalCache) applyLimit(limit int) {
|
||||
if c == nil {
|
||||
return
|
||||
}
|
||||
if limit <= 0 {
|
||||
limit = defaultReceivedSignalCacheLimit
|
||||
}
|
||||
c.mu.Lock()
|
||||
c.limit = limit
|
||||
c.trimLocked()
|
||||
c.mu.Unlock()
|
||||
}
|
||||
|
||||
func (c *ClientCommon) getReceivedSignalCache() *receivedSignalCache {
|
||||
return c.getLogicalSessionState().receivedSignals
|
||||
}
|
||||
|
||||
func (s *ServerCommon) getReceivedSignalCache() *receivedSignalCache {
|
||||
return s.getLogicalSessionState().receivedSignals
|
||||
}
|
||||
Reference in New Issue
Block a user