From 22d9381d9d2b22d690a19abcb96c455f9d6313e1 Mon Sep 17 00:00:00 2001 From: Cecylia Bocovich Date: Tue, 31 Oct 2023 12:20:53 -0400 Subject: [PATCH] Update prometheus metrics to use new EventOnProxyStats --- proxy/lib/pt_event_logger.go | 2 +- 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() } }