Fix ssm-api

This commit is contained in:
世界 2025-07-24 12:22:13 +08:00
parent 9095e5763a
commit 66a767d083
No known key found for this signature in database
GPG Key ID: CD109927C34A63C4

View File

@ -50,12 +50,24 @@ func (s *TrafficManager) UpdateUsers(users []string) {
newUserTCPSessions := make(map[string]*atomic.Int64) newUserTCPSessions := make(map[string]*atomic.Int64)
newUserUDPSessions := make(map[string]*atomic.Int64) newUserUDPSessions := make(map[string]*atomic.Int64)
for _, user := range users { for _, user := range users {
newUserUplink[user] = s.userUplinkPackets[user] if counter, loaded := s.userUplink[user]; loaded {
newUserDownlink[user] = s.userDownlinkPackets[user] newUserUplink[user] = counter
newUserUplinkPackets[user] = s.userUplinkPackets[user] }
newUserDownlinkPackets[user] = s.userDownlinkPackets[user] if counter, loaded := s.userDownlink[user]; loaded {
newUserTCPSessions[user] = s.userTCPSessions[user] newUserDownlink[user] = counter
newUserUDPSessions[user] = s.userUDPSessions[user] }
if counter, loaded := s.userUplinkPackets[user]; loaded {
newUserUplinkPackets[user] = counter
}
if counter, loaded := s.userDownlinkPackets[user]; loaded {
newUserDownlinkPackets[user] = counter
}
if counter, loaded := s.userTCPSessions[user]; loaded {
newUserTCPSessions[user] = counter
}
if counter, loaded := s.userUDPSessions[user]; loaded {
newUserUDPSessions[user] = counter
}
} }
s.userUplink = newUserUplink s.userUplink = newUserUplink
s.userDownlink = newUserDownlink s.userDownlink = newUserDownlink