feat: Add handling time histogram with custom buckets

ethpandaops · Jun 28, 2024 · 595e8fe · 595e8fe
1 parent fcbe764
commit 595e8fe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pkg/output/xatu/exporter.go b/pkg/output/xatu/exporter.go
@@ -33,6 +33,7 @@ func NewItemExporter(name string, config *Config, log logrus.FieldLogger) (ItemE
 	opts := []grpc.DialOption{
 		grpc.WithChainUnaryInterceptor(grpc_prometheus.UnaryClientInterceptor, retry.UnaryClientInterceptor()),
 		grpc.WithChainStreamInterceptor(grpc_prometheus.StreamClientInterceptor, retry.StreamClientInterceptor()),
+		grpc.WithTimeout(config.ExportTimeout),
 	}
 
 	if config.TLS {

diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -22,6 +22,7 @@ import (
 	"github.com/go-co-op/gocron"
 	grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
 	"github.com/pkg/errors"
+	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/sirupsen/logrus"
 	"go.opentelemetry.io/otel/sdk/trace"
@@ -269,7 +270,7 @@ func (x *Xatu) startGrpcServer(ctx context.Context) error {
 
 	mb100 := 1024 * 1024 * 100
 
-	grpc_prometheus.EnableHandlingTimeHistogram()
+	grpc_prometheus.EnableHandlingTimeHistogram(grpc_prometheus.WithHistogramBuckets(prometheus.ExponentialBucketsRange(0.1, 30, 10)))
 
 	opts := []grpc.ServerOption{
 		grpc.MaxRecvMsgSize(mb100),