70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
|
package metrics
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
stdout "go.opentelemetry.io/otel/exporters/stdout/stdoutmetric"
|
||
|
|
"go.opentelemetry.io/otel/metric"
|
||
|
|
"go.opentelemetry.io/otel/metric/global"
|
||
|
|
"go.opentelemetry.io/otel/metric/unit"
|
||
|
|
controller "go.opentelemetry.io/otel/sdk/metric/controller/basic"
|
||
|
|
processor "go.opentelemetry.io/otel/sdk/metric/processor/basic"
|
||
|
|
"go.opentelemetry.io/otel/sdk/metric/selector/simple"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
FrameAge metric.Int64Histogram
|
||
|
|
InferenceDuration metric.Int64Histogram
|
||
|
|
)
|
||
|
|
|
||
|
|
func initMeter(ctx context.Context) func() {
|
||
|
|
zap.S().Info("init telemetry")
|
||
|
|
exporter, err := stdout.New(
|
||
|
|
stdout.WithPrettyPrint(),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
zap.S().Panicf("failed to initialize prometheus exporter %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
pusher := controller.New(
|
||
|
|
processor.NewFactory(
|
||
|
|
simple.NewWithInexpensiveDistribution(),
|
||
|
|
//simple.NewWithHistogramDistribution(
|
||
|
|
// histogram.WithExplicitBoundaries(
|
||
|
|
// []float64{.005, .5, 1, 2.5, 5, 10, 20, 50, 100},
|
||
|
|
// ),
|
||
|
|
//),
|
||
|
|
exporter,
|
||
|
|
),
|
||
|
|
controller.WithExporter(exporter),
|
||
|
|
)
|
||
|
|
|
||
|
|
if err = pusher.Start(ctx); err != nil {
|
||
|
|
zap.S().Fatalf("starting push controller: %v", err)
|
||
|
|
}
|
||
|
|
global.SetMeterProvider(pusher)
|
||
|
|
return func() {
|
||
|
|
if err := pusher.Stop(ctx); err != nil {
|
||
|
|
zap.S().Fatalf("stopping push controller: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Init(ctx context.Context) func() {
|
||
|
|
cleaner := initMeter(ctx)
|
||
|
|
|
||
|
|
meter := global.Meter("robocar/rc-steering")
|
||
|
|
|
||
|
|
FrameAge = metric.Must(meter).NewInt64Histogram(
|
||
|
|
"robocar.frame_age",
|
||
|
|
metric.WithUnit(unit.Milliseconds),
|
||
|
|
metric.WithDescription("time before frame processing"))
|
||
|
|
InferenceDuration = metric.Must(meter).NewInt64Histogram(
|
||
|
|
"robocar.inference_duration",
|
||
|
|
metric.WithUnit(unit.Milliseconds),
|
||
|
|
metric.WithDescription("tensorflow inference duration"))
|
||
|
|
|
||
|
|
return cleaner
|
||
|
|
}
|