YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: disable metrics collector (#580)

Browse Source
This commit is contained in:
boojack
2022-11-26 09:23:38 +08:00
committed by GitHub
parent b511a7b634
commit 85db6721de
2 changed files with 17 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bin/server/main.go
View File

@@ -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)

24
server/metric_collector.go
View File

@@ -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)
} }