gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

metrics: simplify constructor for the default connection tracer (#4555)

Browse Source
This commit is contained in:
Marten Seemann
2024-06-05 18:16:18 +08:00
committed by GitHub
parent 44e0147f2e
commit 91221b4a05
2 changed files with 22 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
metrics/connection_tracer.go
View File

@@ -5,6 +5,7 @@ import (
"errors"
"fmt"
"net"
"sync"
"time"
"github.com/quic-go/quic-go"
@@ -51,25 +52,27 @@ var (
)
)
// DefaultTracer returns a callback that creates a metrics ConnectionTracer.
// The ConnectionTracer returned can be set on the quic.Config for a new connection.
// It should be reused across QUIC connections.
func DefaultTracer() func(_ context.Context, p logging.Perspective, _ logging.ConnectionID) *logging.ConnectionTracer {
return DefaultTracerWithRegisterer(prometheus.DefaultRegisterer)
}
var (
defaultServerConnectionTracerOnce, defaultClientConnectionTracerOnce sync.Once
defaultServerConnectionTracer, defaultClientConnectionTracer *logging.ConnectionTracer
)
// DefaultTracerWithRegisterer returns a callback that creates a metrics ConnectionTracer
// using a given Prometheus registerer.
func DefaultTracerWithRegisterer(registerer prometheus.Registerer) func(_ context.Context, p logging.Perspective, _ logging.ConnectionID) *logging.ConnectionTracer {
return func(_ context.Context, p logging.Perspective, _ logging.ConnectionID) *logging.ConnectionTracer {
switch p {
case logging.PerspectiveClient:
return NewClientConnectionTracerWithRegisterer(registerer)
case logging.PerspectiveServer:
return NewServerConnectionTracerWithRegisterer(registerer)
default:
panic("invalid perspective")
}
// DefaultConnectionTracer returns a callback that creates a metrics ConnectionTracer.
// The ConnectionTracer returned can be set on the quic.Config for a new connection.
func DefaultConnectionTracer(_ context.Context, p logging.Perspective, _ logging.ConnectionID) *logging.ConnectionTracer {
switch p {
case logging.PerspectiveClient:
defaultClientConnectionTracerOnce.Do(func() {
defaultClientConnectionTracer = NewClientConnectionTracerWithRegisterer(prometheus.DefaultRegisterer)
})
return defaultClientConnectionTracer
case logging.PerspectiveServer:
defaultServerConnectionTracerOnce.Do(func() {
defaultServerConnectionTracer = NewServerConnectionTracerWithRegisterer(prometheus.DefaultRegisterer)
})
return defaultServerConnectionTracer
default:
panic("invalid perspective")
}
}