shaytan/drip
1
0
Fork 0
mirror of https://github.com/Gouryella/drip.git synced 2026-02-23 21:00:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(tunnel/manager): update tunnel IP metrics statistics

Browse Source 
- Synchronously delete corresponding Prometheus metrics when removing IP records
- Update metric values when the number of IP tunnels changes
- Ensure accuracy and real-time nature of the tunnelsByIP metric
This commit is contained in:
Gouryella
2026-01-05 13:38:08 +08:00
parent d03874b3d3
commit 12bd4115e3
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/server/tunnel/manager.go
View File

@@ -233,6 +233,9 @@ func (m *Manager) RegisterWithIP(conn *websocket.Conn, customSubdomain string, r
m.tunnelsByIP[remoteIP]--
if m.tunnelsByIP[remoteIP] == 0 {
delete(m.tunnelsByIP, remoteIP)
metrics.TunnelsByIP.DeleteLabelValues(remoteIP)
} else {
metrics.TunnelsByIP.WithLabelValues(remoteIP).Set(float64(m.tunnelsByIP[remoteIP]))
}
}
m.ipMu.Unlock()
@@ -410,6 +413,9 @@ func (m *Manager) CleanupStale(timeout time.Duration) int {
m.tunnelsByIP[remoteIP]--
if m.tunnelsByIP[remoteIP] == 0 {
delete(m.tunnelsByIP, remoteIP)
metrics.TunnelsByIP.DeleteLabelValues(remoteIP)
} else {
metrics.TunnelsByIP.WithLabelValues(remoteIP).Set(float64(m.tunnelsByIP[remoteIP]))
}
}
m.ipMu.Unlock()