Mirrors/trufflehog
2
0
Fork 0
mirror of https://github.com/trufflesecurity/trufflehog.git synced 2024-11-14 08:57:40 +00:00
Code Issues Projects Releases Packages Wiki Activity

make collector a singleton

Browse source
This commit is contained in:
Ahrav Dutta 2024-11-08 14:32:16 -08:00
parent 9722ad7a41
commit a72e25accf
2 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pkg/sources/s3/metrics.go
View file

@ -29,6 +29,13 @@ type collector struct {
bucketsPerRole *prometheus.GaugeVec bucketsPerRole *prometheus.GaugeVec
} }
var metricsInstance metricsCollector
func init() { metricsInstance = newS3MetricsCollector() }
// getMetricsCollector returns the singleton metrics collector instance..
func getMetricsCollector() metricsCollector { return metricsInstance }
func newS3MetricsCollector() metricsCollector { func newS3MetricsCollector() metricsCollector {
return &collector{ return &collector{
objectsScanned: promauto.NewCounterVec(prometheus.CounterOpts{ objectsScanned: promauto.NewCounterVec(prometheus.CounterOpts{

2
pkg/sources/s3/s3.go
View file

@ -99,7 +99,7 @@ func (s *Source) Init(
if err != nil { if err != nil {
return err return err
} }
s.metricsCollector = newS3MetricsCollector() s.metricsCollector = getMetricsCollector()
s.setMaxObjectSize(conn.GetMaxObjectSize()) s.setMaxObjectSize(conn.GetMaxObjectSize())