diff options
author | Cecylia Bocovich <cohosh@torproject.org> | 2023-10-31 12:20:53 -0400 |
---|---|---|
committer | Cecylia Bocovich <cohosh@torproject.org> | 2023-10-31 13:11:38 -0400 |
commit | 22d9381d9d2b22d690a19abcb96c455f9d6313e1 (patch) | |
tree | f6e0a980eeb42620e392934b6995450358ffdeab | |
parent | caa2b3646329b9342d3346a9814e653285c0a8d0 (diff) | |
download | snowflake-22d9381d9d2b22d690a19abcb96c455f9d6313e1.tar.gz snowflake-22d9381d9d2b22d690a19abcb96c455f9d6313e1.zip |
Update prometheus metrics to use new EventOnProxyStats
-rw-r--r-- | proxy/lib/pt_event_logger.go | 2 | ||||
-rw-r--r-- | proxy/lib/pt_event_metrics.go | 7 |
2 files changed, 5 insertions, 4 deletions
diff --git a/proxy/lib/pt_event_logger.go b/proxy/lib/pt_event_logger.go index 3d9a770..4ce4b68 100644 --- a/proxy/lib/pt_event_logger.go +++ b/proxy/lib/pt_event_logger.go @@ -46,7 +46,7 @@ func (p *periodicProxyStats) OnNewSnowflakeEvent(e event.SnowflakeEvent) { func (p *periodicProxyStats) logTick() error { inboundSum, outboundSum := p.bytesLogger.GetStat() - e := &event.EventOnProxyStats{ + e := event.EventOnProxyStats{ SummaryInterval: p.logPeriod, ConnectionCount: p.connectionCount, } diff --git a/proxy/lib/pt_event_metrics.go b/proxy/lib/pt_event_metrics.go index 64a371a..e1c1d70 100644 --- a/proxy/lib/pt_event_metrics.go +++ b/proxy/lib/pt_event_metrics.go @@ -20,10 +20,11 @@ func NewEventMetrics(collector EventCollector) *EventMetrics { func (em *EventMetrics) OnNewSnowflakeEvent(e event.SnowflakeEvent) { switch e.(type) { + case event.EventOnProxyStats: + e := e.(event.EventOnProxyStats) + em.collector.TrackInBoundTraffic(e.InboundBytes) + em.collector.TrackOutBoundTraffic(e.OutboundBytes) case event.EventOnProxyConnectionOver: - e := e.(event.EventOnProxyConnectionOver) - em.collector.TrackInBoundTraffic(e.InboundTraffic) - em.collector.TrackOutBoundTraffic(e.OutboundTraffic) em.collector.TrackNewConnection() } } |