Skip to content

Commit 3cd7091

Browse files
committed
fix prometheus metric reset for loganalytics and resourcegraph queries
Signed-off-by: Markus Blaschke <mblaschke82@gmail.com>
1 parent b68e205 commit 3cd7091

File tree

1 file changed

+4
-12
lines changed

1 file changed

+4
-12
lines changed

auditor/auditor.go

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -169,18 +169,14 @@ func (auditor *AzureAuditor) Run() {
169169
fmt.Sprintf(ReportResourceGraph, queryName),
170170
auditor.Opts.Cronjobs.ResourceGraph,
171171
func(ctx context.Context, logger *log.Entry) {
172-
for _, queryConfig := range auditor.config.ResourceGraph.Queries {
173-
queryConfig.Reset()
174-
}
172+
auditor.config.ResourceGraph.Queries[queryName].Reset()
175173
},
176174
func(ctx context.Context, logger *log.Entry, subscription *subscriptions.Subscription, report *AzureAuditorReport, callback chan<- func()) {
177175
contextLogger := log.WithField("configQueryName", queryName)
178176
auditor.auditResourceGraph(ctx, contextLogger, subscription, queryName, resourceGraphConfig, report, callback)
179177
},
180178
func(ctx context.Context, logger *log.Entry) {
181-
for _, gauge := range auditor.prometheus.resourceGraph {
182-
gauge.Reset()
183-
}
179+
auditor.prometheus.resourceGraph[queryName].Reset()
184180
},
185181
)
186182
}
@@ -193,18 +189,14 @@ func (auditor *AzureAuditor) Run() {
193189
fmt.Sprintf(ReportLogAnalytics, queryName),
194190
auditor.Opts.Cronjobs.LogAnalytics,
195191
func(ctx context.Context, logger *log.Entry) {
196-
for _, queryConfig := range auditor.config.LogAnalytics.Queries {
197-
queryConfig.Reset()
198-
}
192+
auditor.config.LogAnalytics.Queries[queryName].Reset()
199193
},
200194
func(ctx context.Context, logger *log.Entry, report *AzureAuditorReport, callback chan<- func()) {
201195
contextLogger := log.WithField("configQueryName", queryName)
202196
auditor.auditLogAnalytics(ctx, contextLogger, queryName, logAnalyticsConfig, report, callback)
203197
},
204198
func(ctx context.Context, logger *log.Entry) {
205-
for _, gauge := range auditor.prometheus.logAnalytics {
206-
gauge.Reset()
207-
}
199+
auditor.prometheus.logAnalytics[queryName].Reset()
208200
},
209201
)
210202
}

0 commit comments

Comments
 (0)