conn: attach context to pd client (#287)

Signed-off-by: Neil Shen <[email protected]>
pingcap · May 26, 2020 · bc21c31 · bc21c31
1 parent 0b47b8f
commit bc21c31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/conn/conn.go b/pkg/conn/conn.go
@@ -191,8 +191,8 @@ func NewMgr(
 		grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(maxMsgSize)),
 		grpc.WithDefaultCallOptions(grpc.MaxCallSendMsgSize(maxMsgSize)),
 	}
-	pdClient, err := pd.NewClient(
-		addrs, securityOption, pd.WithGRPCDialOptions(maxCallMsgSize...))
+	pdClient, err := pd.NewClientWithContext(
+		ctx, addrs, securityOption, pd.WithGRPCDialOptions(maxCallMsgSize...))
 	if err != nil {
 		log.Error("fail to create pd client", zap.Error(err))
 		return nil, err