go-pkg/lg/metric.go
xuu 3be012e780
All checks were successful
Go Test / build (pull_request) Successful in 1m47s
chore: save sst code
2024-01-15 11:26:54 -07:00

104 lines
2.3 KiB
Go

package lg
import (
"context"
"log"
"net/http"
"os"
"runtime/debug"
"time"
"github.com/prometheus/client_golang/prometheus/promhttp"
"go.opentelemetry.io/contrib/instrumentation/runtime"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/exporters/prometheus"
api "go.opentelemetry.io/otel/metric"
sdk "go.opentelemetry.io/otel/sdk/metric"
"go.opentelemetry.io/otel/sdk/resource"
)
var meterKey = contextKey{"meter"}
var promHTTPKey = contextKey{"promHTTP"}
func Meter(ctx context.Context) api.Meter {
if t := fromContext[contextKey, api.Meter](ctx, meterKey); t != nil {
return t
}
log.Printf("default meter")
return otel.Meter("")
}
func NewHTTP(ctx context.Context) *httpHandle {
t := fromContext[contextKey, *prometheus.Exporter](ctx, promHTTPKey)
return &httpHandle{t}
}
func initMetrics(ctx context.Context, name string) (context.Context, func() error) {
goversion := ""
pkg := ""
host := ""
version := "0.0.1"
if info, ok := debug.ReadBuildInfo(); ok {
goversion = info.GoVersion
pkg = info.Path
version = info.Main.Version
}
if h, err := os.Hostname(); err == nil {
host = h
}
ex, err := prometheus.New()
if err != nil {
log.Println(err)
return ctx, nil
}
ctx = toContext(ctx, promHTTPKey, ex)
provider := sdk.NewMeterProvider(
sdk.WithResource(resource.Default()),
sdk.WithView(sdk.NewView(
sdk.Instrument{Name: "histogram_*"},
sdk.Stream{
Aggregation: sdk.AggregationExplicitBucketHistogram{
Boundaries: []float64{
2 << 6, 2 << 8, 2 << 10, 2 << 12, 2 << 14, 2 << 16, 2 << 18, 2 << 20, 2 << 22, 2 << 24, 2 << 26, 2 << 28,
},
},
},
)),
sdk.WithReader(ex),
)
meter := provider.Meter(name,
api.WithInstrumentationVersion(version),
api.WithInstrumentationAttributes(
attribute.String("app", name),
attribute.String("host", host),
attribute.String("go_version", goversion),
attribute.String("pkg", pkg),
),
)
ctx = toContext(ctx, meterKey, meter)
runtime.Start()
return ctx, func() error {
_, cancel := context.WithTimeout(context.Background(), 1*time.Second)
defer cancel()
defer log.Println("metrics stopped")
return nil
}
}
type httpHandle struct {
exp *prometheus.Exporter
}
func (h *httpHandle) RegisterHTTP(mux *http.ServeMux) {
if h.exp == nil {
return
}
mux.Handle("/metrics", promhttp.Handler())
}