relay stats

connet-dev · Jan 18, 2025 · 33dbd1c · 33dbd1c
1 parent dbe804f
commit 33dbd1c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/client/peer_relay.go b/client/peer_relay.go
@@ -13,6 +13,7 @@ import (
 	"github.com/connet-dev/connet/netc"
 	"github.com/connet-dev/connet/pb"
 	"github.com/connet-dev/connet/pbc"
+	"github.com/connet-dev/connet/quicc"
 	"github.com/quic-go/quic-go"
 	"golang.org/x/sync/errgroup"
 	"google.golang.org/protobuf/types/known/timestamppb"
@@ -88,14 +89,15 @@ func (r *relayPeer) connect(ctx context.Context) (quic.Connection, error) {
 
 	cfg := r.serverConf.Load()
 	r.logger.Debug("dialing relay", "relay", r.serverHostport, "addr", addr, "server", cfg.name, "cert", cfg.key)
-	return r.local.direct.transport.Dial(ctx, addr, &tls.Config{
+	return r.local.direct.transport.Dial(quicc.RTTContext(ctx), addr, &tls.Config{
 		Certificates: []tls.Certificate{r.local.clientCert},
 		RootCAs:      cfg.cas,
 		ServerName:   cfg.name,
 		NextProtos:   []string{"connet-relay"},
 	}, &quic.Config{
 		MaxIdleTimeout:  20 * time.Second,
 		KeepAlivePeriod: 10 * time.Second,
+		Tracer:          quicc.RTTTracer,
 	})
 }
 
@@ -120,6 +122,11 @@ func (r *relayPeer) keepalive(ctx context.Context, conn quic.Connection) error {
 		if err := r.heartbeat(ctx, stream); err != nil {
 			return err
 		}
+		if rttStats := quicc.RTTStats(conn); rttStats != nil {
+			r.logger.Debug("rtt-pro", "last", rttStats.LatestRTT(), "smoothed", rttStats.SmoothedRTT())
+		} else {
+			r.logger.Debug("no rtt")
+		}
 	}
 }
 

diff --git a/relay/clients.go b/relay/clients.go
@@ -15,6 +15,7 @@ import (
 	"github.com/connet-dev/connet/netc"
 	"github.com/connet-dev/connet/pb"
 	"github.com/connet-dev/connet/pbc"
+	"github.com/connet-dev/connet/quicc"
 	"github.com/klev-dev/kleverr"
 	"github.com/quic-go/quic-go"
 	"golang.org/x/sync/errgroup"
@@ -169,6 +170,7 @@ func (s *clientsServer) run(ctx context.Context, transport *quic.Transport) erro
 	l, err := transport.Listen(s.tlsConf, &quic.Config{
 		MaxIdleTimeout:  20 * time.Second,
 		KeepAlivePeriod: 10 * time.Second,
+		Tracer:          quicc.RTTTracer,
 	})
 	if err != nil {
 		return kleverr.Ret(err)
@@ -369,6 +371,9 @@ func (c *clientConn) heartbeat(ctx context.Context, stream quic.Stream, hbt *pbc
 			if err := pb.Write(stream, &pbc.Response{Heartbeat: req.Heartbeat}); err != nil {
 				return err
 			}
+			if rttStats := quicc.RTTStats(c.conn); rttStats != nil {
+				c.logger.Debug("rtt", "last", rttStats.LatestRTT(), "smoothed", rttStats.SmoothedRTT())
+			}
 		}
 	})
 

diff --git a/relay/server.go b/relay/server.go
@@ -11,6 +11,7 @@ import (
 	"slices"
 
 	"github.com/connet-dev/connet/model"
+	"github.com/connet-dev/connet/quicc"
 	"github.com/connet-dev/connet/statusc"
 	"github.com/klev-dev/kleverr"
 	"github.com/quic-go/quic-go"
@@ -94,6 +95,7 @@ func (s *Server) Run(ctx context.Context) error {
 		Conn:               udpConn,
 		ConnectionIDLength: 8,
 		StatelessResetKey:  s.statelessResetKey,
+		ConnContext:        quicc.RTTContext,
 		// TODO review other options
 	}
 	defer transport.Close()