mirror of
https://github.com/imgproxy/imgproxy.git
synced 2024-11-24 08:12:38 +02:00
128 lines
2.7 KiB
Go
128 lines
2.7 KiB
Go
package metrics
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/imgproxy/imgproxy/v3/metrics/datadog"
|
|
"github.com/imgproxy/imgproxy/v3/metrics/newrelic"
|
|
"github.com/imgproxy/imgproxy/v3/metrics/prometheus"
|
|
)
|
|
|
|
func Init() error {
|
|
prometheus.Init()
|
|
|
|
if err := newrelic.Init(); err != nil {
|
|
return nil
|
|
}
|
|
|
|
datadog.Init()
|
|
|
|
return nil
|
|
}
|
|
|
|
func Stop() {
|
|
newrelic.Stop()
|
|
datadog.Stop()
|
|
}
|
|
|
|
func Enabled() bool {
|
|
return prometheus.Enabled() ||
|
|
newrelic.Enabled() ||
|
|
datadog.Enabled()
|
|
}
|
|
|
|
func StartRequest(ctx context.Context, rw http.ResponseWriter, r *http.Request) (context.Context, context.CancelFunc, http.ResponseWriter) {
|
|
promCancel := prometheus.StartRequest()
|
|
ctx, nrCancel, rw := newrelic.StartTransaction(ctx, rw, r)
|
|
ctx, ddCancel, rw := datadog.StartRootSpan(ctx, rw, r)
|
|
|
|
cancel := func() {
|
|
promCancel()
|
|
nrCancel()
|
|
ddCancel()
|
|
}
|
|
|
|
return ctx, cancel, rw
|
|
}
|
|
|
|
func StartQueueSegment(ctx context.Context) context.CancelFunc {
|
|
promCancel := prometheus.StartQueueSegment()
|
|
nrCancel := newrelic.StartSegment(ctx, "Queue")
|
|
ddCancel := datadog.StartSpan(ctx, "queue")
|
|
|
|
cancel := func() {
|
|
promCancel()
|
|
nrCancel()
|
|
ddCancel()
|
|
}
|
|
|
|
return cancel
|
|
}
|
|
|
|
func StartDownloadingSegment(ctx context.Context) context.CancelFunc {
|
|
promCancel := prometheus.StartDownloadingSegment()
|
|
nrCancel := newrelic.StartSegment(ctx, "Downloading image")
|
|
ddCancel := datadog.StartSpan(ctx, "downloading_image")
|
|
|
|
cancel := func() {
|
|
promCancel()
|
|
nrCancel()
|
|
ddCancel()
|
|
}
|
|
|
|
return cancel
|
|
}
|
|
|
|
func StartProcessingSegment(ctx context.Context) context.CancelFunc {
|
|
promCancel := prometheus.StartProcessingSegment()
|
|
nrCancel := newrelic.StartSegment(ctx, "Processing image")
|
|
ddCancel := datadog.StartSpan(ctx, "processing_image")
|
|
|
|
cancel := func() {
|
|
promCancel()
|
|
nrCancel()
|
|
ddCancel()
|
|
}
|
|
|
|
return cancel
|
|
}
|
|
|
|
func StartStreamingSegment(ctx context.Context) context.CancelFunc {
|
|
promCancel := prometheus.StartStreamingSegment()
|
|
nrCancel := newrelic.StartSegment(ctx, "Streaming image")
|
|
ddCancel := datadog.StartSpan(ctx, "streaming_image")
|
|
|
|
cancel := func() {
|
|
promCancel()
|
|
nrCancel()
|
|
ddCancel()
|
|
}
|
|
|
|
return cancel
|
|
}
|
|
|
|
func SendError(ctx context.Context, errType string, err error) {
|
|
prometheus.IncrementErrorsTotal(errType)
|
|
newrelic.SendError(ctx, errType, err)
|
|
datadog.SendError(ctx, errType, err)
|
|
}
|
|
|
|
func ObserveBufferSize(t string, size int) {
|
|
prometheus.ObserveBufferSize(t, size)
|
|
newrelic.ObserveBufferSize(t, size)
|
|
datadog.ObserveBufferSize(t, size)
|
|
}
|
|
|
|
func SetBufferDefaultSize(t string, size int) {
|
|
prometheus.SetBufferDefaultSize(t, size)
|
|
newrelic.SetBufferDefaultSize(t, size)
|
|
datadog.SetBufferDefaultSize(t, size)
|
|
}
|
|
|
|
func SetBufferMaxSize(t string, size int) {
|
|
prometheus.SetBufferMaxSize(t, size)
|
|
newrelic.SetBufferMaxSize(t, size)
|
|
datadog.SetBufferMaxSize(t, size)
|
|
}
|