cyonjan/frp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: a negetive number in statistic

Browse Source
This commit is contained in:
blizard863 2021-10-14 19:58:25 +08:00
parent 998e678a7f
commit d0bfbf2ca7
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/metrics/mem/server.go
View File

@ -115,6 +115,9 @@ func (m *serverMetrics) CloseProxy(name string, proxyType string) {
defer m.mu.Unlock() defer m.mu.Unlock()
if counter, ok := m.info.ProxyTypeCounts[proxyType]; ok { if counter, ok := m.info.ProxyTypeCounts[proxyType]; ok {
counter.Dec(1) counter.Dec(1)
if counter.Count() < 0 {
counter.Clear()
}
} }
if proxyStats, ok := m.info.ProxyStatistics[name]; ok { if proxyStats, ok := m.info.ProxyStatistics[name]; ok {
proxyStats.LastCloseTime = time.Now() proxyStats.LastCloseTime = time.Now()
@ -135,12 +138,18 @@ func (m *serverMetrics) OpenConnection(name string, proxyType string) {
func (m *serverMetrics) CloseConnection(name string, proxyType string) { func (m *serverMetrics) CloseConnection(name string, proxyType string) {
m.info.CurConns.Dec(1) m.info.CurConns.Dec(1)
if m.info.CurConns.Count() < 0 {
m.info.CurConns.Clear()
}
m.mu.Lock() m.mu.Lock()
defer m.mu.Unlock() defer m.mu.Unlock()
proxyStats, ok := m.info.ProxyStatistics[name] proxyStats, ok := m.info.ProxyStatistics[name]
if ok { if ok {
proxyStats.CurConns.Dec(1) proxyStats.CurConns.Dec(1)
if proxyStats.CurConns.Count() < 0 {
proxyStats.CurConns.Clear()
}
m.info.ProxyStatistics[name] = proxyStats m.info.ProxyStatistics[name] = proxyStats
} }
} }