TUN-3848: Use transport logger for h2mux

This commit is contained in:
Adam Chalmers
2021-02-03 12:32:54 -06:00
parent 88b53eb886
commit 0d22106416
9 changed files with 21 additions and 14 deletions

View File

@@ -27,7 +27,7 @@ var (
Scheme: "https",
Host: "connectiontest.argotunnel.com",
}
testObserver = NewObserver(&log, false)
testObserver = NewObserver(&log, &log, false)
testLargeResp = make([]byte, largeFileSize)
)

View File

@@ -77,7 +77,7 @@ func NewH2muxConnection(
// Establish a muxed connection with the edge
// Client mux handshake with agent server
muxer, err := h2mux.Handshake(edgeConn, edgeConn, *muxerConfig.H2MuxerConfig(h, observer.log), h2mux.ActiveStreams)
muxer, err := h2mux.Handshake(edgeConn, edgeConn, *muxerConfig.H2MuxerConfig(h, observer.logTransport), h2mux.ActiveStreams)
if err != nil {
recoverable := isHandshakeErrRecoverable(err, connIndex, observer)
return nil, err, recoverable

View File

@@ -17,6 +17,7 @@ const (
type Observer struct {
log *zerolog.Logger
logTransport *zerolog.Logger
metrics *tunnelMetrics
tunnelEventChan chan Event
uiEnabled bool
@@ -27,9 +28,10 @@ type EventSink interface {
OnTunnelEvent(event Event)
}
func NewObserver(log *zerolog.Logger, uiEnabled bool) *Observer {
func NewObserver(log, logTransport *zerolog.Logger, uiEnabled bool) *Observer {
o := &Observer{
log: log,
logTransport: logTransport,
metrics: newTunnelMetrics(),
uiEnabled: uiEnabled,
tunnelEventChan: make(chan Event, observerChannelBufferSize),

View File

@@ -44,7 +44,7 @@ func TestRegisterServerLocation(t *testing.T) {
}
func TestObserverEventsDontBlock(t *testing.T) {
observer := NewObserver(&log, false)
observer := NewObserver(&log, &log, false)
var mu sync.Mutex
observer.RegisterSink(EventSinkFunc(func(_ Event) {
// callback will block if lock is already held