chore: disable metrics collector (#580)

pull/581/head^2
boojack 2 years ago committed by GitHub
parent b511a7b634
commit 85db6721de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -40,6 +40,8 @@ func run(profile *profile.Profile) error {
serverInstance.Store = storeInstance serverInstance.Store = storeInstance
metricCollector := server.NewMetricCollector(profile, storeInstance) metricCollector := server.NewMetricCollector(profile, storeInstance)
// Disable metrics collector.
metricCollector.Enabled = false
serverInstance.Collector = &metricCollector serverInstance.Collector = &metricCollector
println(greetingBanner) println(greetingBanner)

@ -13,9 +13,10 @@ import (
// MetricCollector is the metric collector. // MetricCollector is the metric collector.
type MetricCollector struct { type MetricCollector struct {
collector metric.Collector Collector metric.Collector
profile *profile.Profile Enabled bool
store *store.Store Profile *profile.Profile
Store *store.Store
} }
const ( const (
@ -26,23 +27,28 @@ func NewMetricCollector(profile *profile.Profile, store *store.Store) MetricColl
c := segment.NewCollector(segmentMetricWriteKey) c := segment.NewCollector(segmentMetricWriteKey)
return MetricCollector{ return MetricCollector{
collector: c, Collector: c,
profile: profile, Enabled: true,
store: store, Profile: profile,
Store: store,
} }
} }
func (mc *MetricCollector) Collect(_ context.Context, metric *metric.Metric) { func (mc *MetricCollector) Collect(_ context.Context, metric *metric.Metric) {
if mc.profile.Mode == "dev" { if !mc.Enabled {
return
}
if mc.Profile.Mode == "dev" {
return return
} }
if metric.Labels == nil { if metric.Labels == nil {
metric.Labels = map[string]string{} metric.Labels = map[string]string{}
} }
metric.Labels["version"] = version.GetCurrentVersion(mc.profile.Mode) metric.Labels["version"] = version.GetCurrentVersion(mc.Profile.Mode)
err := mc.collector.Collect(metric) err := mc.Collector.Collect(metric)
if err != nil { if err != nil {
fmt.Printf("Failed to request segment, error: %+v\n", err) fmt.Printf("Failed to request segment, error: %+v\n", err)
} }

Loading…
Cancel
Save