add a multiplexer for the logging.ConnectionTracer

logging/multiplex.go

@@ -0,0 +1,142 @@
+package logging
+
+import (
+	"net"
+	"time"
+)
+
+type connTracerMultiplexer struct {
+	tracers []ConnectionTracer
+}
+
+var _ ConnectionTracer = &connTracerMultiplexer{}
+
+func NewConnectionMultiplexer(tracers ...ConnectionTracer) ConnectionTracer {
+	if len(tracers) == 0 {
+		panic("no tracers")
+	}
+	if len(tracers) == 1 {
+		return tracers[0]
+	}
+	return &connTracerMultiplexer{tracers: tracers}
+}
+
+func (m *connTracerMultiplexer) StartedConnection(local, remote net.Addr, version VersionNumber, srcConnID, destConnID ConnectionID) {
+	for _, t := range m.tracers {
+		t.StartedConnection(local, remote, version, srcConnID, destConnID)
+	}
+}
+
+func (m *connTracerMultiplexer) ClosedConnection(reason CloseReason) {
+	for _, t := range m.tracers {
+		t.ClosedConnection(reason)
+	}
+}
+
+func (m *connTracerMultiplexer) SentTransportParameters(tp *TransportParameters) {
+	for _, t := range m.tracers {
+		t.SentTransportParameters(tp)
+	}
+}
+
+func (m *connTracerMultiplexer) ReceivedTransportParameters(tp *TransportParameters) {
+	for _, t := range m.tracers {
+		t.ReceivedTransportParameters(tp)
+	}
+}
+
+func (m *connTracerMultiplexer) SentPacket(hdr *ExtendedHeader, size ByteCount, ack *AckFrame, frames []Frame) {
+	for _, t := range m.tracers {
+		t.SentPacket(hdr, size, ack, frames)
+	}
+}
+
+func (m *connTracerMultiplexer) ReceivedVersionNegotiationPacket(hdr *Header, versions []VersionNumber) {
+	for _, t := range m.tracers {
+		t.ReceivedVersionNegotiationPacket(hdr, versions)
+	}
+}
+
+func (m *connTracerMultiplexer) ReceivedRetry(hdr *Header) {
+	for _, t := range m.tracers {
+		t.ReceivedRetry(hdr)
+	}
+}
+
+func (m *connTracerMultiplexer) ReceivedPacket(hdr *ExtendedHeader, size ByteCount, frames []Frame) {
+	for _, t := range m.tracers {
+		t.ReceivedPacket(hdr, size, frames)
+	}
+}
+
+func (m *connTracerMultiplexer) BufferedPacket(typ PacketType) {
+	for _, t := range m.tracers {
+		t.BufferedPacket(typ)
+	}
+}
+
+func (m *connTracerMultiplexer) DroppedPacket(typ PacketType, size ByteCount, reason PacketDropReason) {
+	for _, t := range m.tracers {
+		t.DroppedPacket(typ, size, reason)
+	}
+}
+
+func (m *connTracerMultiplexer) UpdatedMetrics(rttStats *RTTStats, cwnd, bytesInFLight ByteCount, packetsInFlight int) {
+	for _, t := range m.tracers {
+		t.UpdatedMetrics(rttStats, cwnd, bytesInFLight, packetsInFlight)
+	}
+}
+
+func (m *connTracerMultiplexer) LostPacket(encLevel EncryptionLevel, pn PacketNumber, reason PacketLossReason) {
+	for _, t := range m.tracers {
+		t.LostPacket(encLevel, pn, reason)
+	}
+}
+
+func (m *connTracerMultiplexer) UpdatedPTOCount(value uint32) {
+	for _, t := range m.tracers {
+		t.UpdatedPTOCount(value)
+	}
+}
+
+func (m *connTracerMultiplexer) UpdatedKeyFromTLS(encLevel EncryptionLevel, perspective Perspective) {
+	for _, t := range m.tracers {
+		t.UpdatedKeyFromTLS(encLevel, perspective)
+	}
+}
+
+func (m *connTracerMultiplexer) UpdatedKey(generation KeyPhase, remote bool) {
+	for _, t := range m.tracers {
+		t.UpdatedKey(generation, remote)
+	}
+}
+
+func (m *connTracerMultiplexer) DroppedEncryptionLevel(encLevel EncryptionLevel) {
+	for _, t := range m.tracers {
+		t.DroppedEncryptionLevel(encLevel)
+	}
+}
+
+func (m *connTracerMultiplexer) SetLossTimer(typ TimerType, encLevel EncryptionLevel, exp time.Time) {
+	for _, t := range m.tracers {
+		t.SetLossTimer(typ, encLevel, exp)
+	}
+}
+
+func (m *connTracerMultiplexer) LossTimerExpired(typ TimerType, encLevel EncryptionLevel) {
+	for _, t := range m.tracers {
+		t.LossTimerExpired(typ, encLevel)
+	}
+}
+
+func (m *connTracerMultiplexer) LossTimerCanceled() {
+	for _, t := range m.tracers {
+		t.LossTimerCanceled()
+	}
+}
+
+func (m *connTracerMultiplexer) Close() {
+	for _, t := range m.tracers {
+		t.Close()
+	}
+}