7999d24f81
this commit updates our monkit dependency to the v3 version where it outputs in an influx style. this makes discovery much easier as many tools are built to look at it this way. graphite and rothko will suffer some due to no longer being a tree based on dots. hopefully time will exist to update rothko to index based on the new metric format. it adds an influx output for the statreceiver so that we can write to influxdb v1 or v2 directly. Change-Id: Iae9f9494a6d29cfbd1f932a5e71a891b490415ff
93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package process
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
hw "github.com/jtolds/monkit-hw/v2"
|
|
"github.com/spacemonkeygo/monkit/v3"
|
|
"github.com/spacemonkeygo/monkit/v3/environment"
|
|
"github.com/zeebo/admission/v2/admproto"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/common/identity"
|
|
"storj.io/storj/pkg/cfgstruct"
|
|
"storj.io/storj/pkg/telemetry"
|
|
"storj.io/storj/private/version"
|
|
)
|
|
|
|
var (
|
|
metricInterval = flag.Duration("metrics.interval", telemetry.DefaultInterval, "how frequently to send up telemetry")
|
|
metricCollector = flag.String("metrics.addr", flagDefault("", "collectora.storj.io:9000"), "address to send telemetry to")
|
|
metricApp = flag.String("metrics.app", filepath.Base(os.Args[0]), "application name for telemetry identification")
|
|
metricAppSuffix = flag.String("metrics.app-suffix", flagDefault("-dev", "-release"), "application suffix")
|
|
metricInstancePrefix = flag.String("metrics.instance-prefix", "", "instance id prefix")
|
|
)
|
|
|
|
const (
|
|
maxInstanceLength = 52
|
|
)
|
|
|
|
func flagDefault(dev, release string) string {
|
|
if cfgstruct.DefaultsType() == "release" {
|
|
return release
|
|
}
|
|
return dev
|
|
}
|
|
|
|
// InitMetrics initializes telemetry reporting. Makes a telemetry.Client and calls
|
|
// its Run() method in a goroutine.
|
|
func InitMetrics(ctx context.Context, log *zap.Logger, r *monkit.Registry, instanceID string) (err error) {
|
|
if r == nil {
|
|
r = monkit.Default
|
|
}
|
|
environment.Register(r)
|
|
hw.Register(r)
|
|
r.ScopeNamed("env").Chain(monkit.StatSourceFunc(version.Build.Stats))
|
|
|
|
log = log.Named("telemetry")
|
|
if *metricCollector == "" || *metricInterval == 0 {
|
|
log.Info("disabled")
|
|
return nil
|
|
}
|
|
|
|
if instanceID == "" {
|
|
instanceID = telemetry.DefaultInstanceID()
|
|
}
|
|
instanceID = *metricInstancePrefix + instanceID
|
|
if len(instanceID) > maxInstanceLength {
|
|
instanceID = instanceID[:maxInstanceLength]
|
|
}
|
|
c, err := telemetry.NewClient(log, *metricCollector, telemetry.ClientOpts{
|
|
Interval: *metricInterval,
|
|
Application: *metricApp + *metricAppSuffix,
|
|
Instance: instanceID,
|
|
Registry: r,
|
|
FloatEncoding: admproto.Float32Encoding,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
go c.Run(ctx)
|
|
return nil
|
|
}
|
|
|
|
// InitMetricsWithCertPath initializes telemetry reporting, using the node ID
|
|
// corresponding to the given certificate as the telemetry instance ID.
|
|
func InitMetricsWithCertPath(ctx context.Context, log *zap.Logger, r *monkit.Registry, certPath string) error {
|
|
var metricsID string
|
|
nodeID, err := identity.NodeIDFromCertPath(certPath)
|
|
if err != nil {
|
|
log.Sugar().Errorf("Could not read identity for telemetry setup: %v", err)
|
|
metricsID = "" // InitMetrics() will fill in a default value
|
|
} else {
|
|
metricsID = nodeID.String()
|
|
}
|
|
return InitMetrics(ctx, log, r, metricsID)
|
|
}
|