log metrics in tests

livekit · Jan 9, 2024 · 0acefd9 · 0acefd9
1 parent 702269d
commit 0acefd9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/ioserver.go b/test/ioserver.go
@@ -22,6 +22,7 @@ import (
 	"google.golang.org/protobuf/types/known/emptypb"
 
 	"github.com/livekit/protocol/livekit"
+	"github.com/livekit/protocol/logger"
 	"github.com/livekit/protocol/rpc"
 	"github.com/livekit/psrpc"
 )
@@ -52,3 +53,8 @@ func (s *ioTestServer) UpdateEgress(_ context.Context, info *livekit.EgressInfo)
 	s.updates <- info
 	return &emptypb.Empty{}, nil
 }
+
+func (s *ioTestServer) UpdateMetrics(_ context.Context, req *rpc.UpdateMetricsRequest) (*emptypb.Empty, error) {
+	logger.Infow("egress metrics", "egressID", req.Info.EgressId, "avgCPU", req.AvgCpuUsage, "maxCPU", req.MaxCpuUsage)
+	return &emptypb.Empty{}, nil
+}