fix stream panic

Signed-off-by: Ryan Leung <[email protected]>
tikv · Oct 23, 2023 · 55dd28e · 55dd28e
1 parent e19265f
commit 55dd28e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/server/grpc_service.go b/server/grpc_service.go
@@ -2529,6 +2529,9 @@ func forwardRegionHeartbeatClientToServer(forwardStream pdpb.PD_RegionHeartbeatC
 }
 
 func (s *GrpcServer) createSchedulingStream(client *grpc.ClientConn) (schedulingpb.Scheduling_RegionHeartbeatClient, context.CancelFunc, error) {
+	if client == nil {
+		return nil, nil, errors.New("connection is not set")
+	}
 	done := make(chan struct{})
 	ctx, cancel := context.WithCancel(s.ctx)
 	go grpcutil.CheckStream(ctx, cancel, done)