benchmark: Log TChannel warnings (or higher) to stderr

uber · Jun 22, 2016 · 2dc86c1 · 2dc86c1
1 parent e978a5a
commit 2dc86c1
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/benchmark/internal_client.go b/benchmark/internal_client.go
@@ -23,6 +23,7 @@ package benchmark
 import (
 	"bytes"
 	"fmt"
+	"os"
 	"time"
 
 	"github.com/uber/tchannel-go"
@@ -66,7 +67,9 @@ func newClient(hosts []string, opts *options) inProcClient {
 }
 
 func newInternalClient(hosts []string, opts *options) inProcClient {
-	ch, err := tchannel.NewChannel(opts.svcName, nil)
+	ch, err := tchannel.NewChannel(opts.svcName, &tchannel.ChannelOptions{
+		Logger: tchannel.NewLevelLogger(tchannel.NewLogger(os.Stderr), tchannel.LogLevelWarn),
+	})
 	if err != nil {
 		panic("failed to create channel: " + err.Error())
 	}

diff --git a/benchmark/internal_server.go b/benchmark/internal_server.go
@@ -22,6 +22,7 @@ package benchmark
 
 import (
 	"fmt"
+	"os"
 
 	"github.com/uber/tchannel-go"
 	"github.com/uber/tchannel-go/hyperbahn"
@@ -48,7 +49,9 @@ func NewServer(optFns ...Option) Server {
 		return newExternalServer(opts)
 	}
 
-	ch, err := tchannel.NewChannel(opts.svcName, nil)
+	ch, err := tchannel.NewChannel(opts.svcName, &tchannel.ChannelOptions{
+		Logger: tchannel.NewLevelLogger(tchannel.NewLogger(os.Stderr), tchannel.LogLevelWarn),
+	})
 	if err != nil {
 		panic("failed to create channel: " + err.Error())
 	}

diff --git a/benchmark/real_relay.go b/benchmark/real_relay.go
@@ -21,6 +21,8 @@
 package benchmark
 
 import (
+	"os"
+
 	"github.com/uber/tchannel-go"
 	"github.com/uber/tchannel-go/relay"
 
@@ -52,6 +54,7 @@ func NewRealRelay(services map[string][]string) (Relay, error) {
 	hosts := &fixedHosts{hosts: services}
 	ch, err := tchannel.NewChannel("relay", &tchannel.ChannelOptions{
 		RelayHosts: hosts,
+		Logger:     tchannel.NewLevelLogger(tchannel.NewLogger(os.Stderr), tchannel.LogLevelWarn),
 	})
 	if err != nil {
 		return nil, err