Allocate subchannel for new client

uber · Apr 3, 2017 · 1faa5ab · 1faa5ab
1 parent be64aa8
commit 1faa5ab
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/example-gateway/clients/baz/baz.go b/examples/example-gateway/clients/baz/baz.go
@@ -28,13 +28,13 @@ func NewClient(config *zanzibar.StaticConfig, gateway *zanzibar.Gateway) *BazCli
 	// TODO: (lu) get serviceName from thrift
 	serviceName := "SimpleService"
 
-	ch := gateway.Channel
+	sc := gateway.Channel.GetSubChannel(serviceName)
 
 	ip := config.MustGetString("clients.baz.ip")
 	port := config.MustGetInt("clients.baz.port")
-	ch.Peers().Add(ip + ":" + strconv.Itoa(int(port)))
+	sc.Peers().Add(ip + ":" + strconv.Itoa(int(port)))
 
-	client := zt.NewClient(ch, serviceName)
+	client := zt.NewClient(gateway.Channel, serviceName)
 
 	return &BazClient{
 		thriftService: serviceName,