archives/snowflake
2
0
Fork 0
mirror of https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git synced 2025-10-13 20:11:19 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add distinct IP counter to metrics

Browse source
This commit is contained in:
Shelikhoo 2022-05-27 16:21:28 +01:00
parent 211254fa98
commit fa7d1e2bb7
No known key found for this signature in database
GPG key ID: C4D5E79D22B25316
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
broker/metrics.go
View file

@ -14,6 +14,7 @@ import (
"sync"
"time"
"git.torproject.org/pluggable-transports/snowflake.git/v2/common/ipsetsink/sinkcluster"
"git.torproject.org/pluggable-transports/snowflake.git/v2/common/messages"
"github.com/prometheus/client_golang/prometheus"
"gitlab.torproject.org/tpo/anti-censorship/geoip"
@ -41,6 +42,8 @@ type Metrics struct {
logger *log.Logger
geoipdb *geoip.Geoip
distinctIPWriter *sinkcluster.ClusterWriter
countryStats CountryStats
clientRoundtripEstimate time.Duration
proxyIdleCount uint
@ -324,3 +327,13 @@ func initPrometheus() *PromMetrics {
return promMetrics
}
func (m *Metrics) RecordIPAddress(ip string) {
if m.distinctIPWriter != nil {
m.distinctIPWriter.AddIPToSet(ip)
}
}
func (m *Metrics) SetIPAddressRecorder(recorder *sinkcluster.ClusterWriter) {
m.distinctIPWriter = recorder
}