address the comment

Signed-off-by: Ryan Leung <[email protected]>
rleungx · Sep 14, 2023 · 97fb042 · 97fb042
1 parent 4fc1586
commit 97fb042
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/grpc_service.go b/server/grpc_service.go
@@ -1032,7 +1032,7 @@ func (s *GrpcServer) StoreHeartbeat(ctx context.Context, request *pdpb.StoreHear
 func (s *GrpcServer) updateSchedulingClient(ctx context.Context) {
 	forwardedHost, _ := s.GetServicePrimaryAddr(ctx, utils.SchedulingServiceName)
 	pre := s.schedulingClient.Load()
-	if forwardedHost != "" && (pre != nil && forwardedHost != pre.(*schedulingClient).getPrimaryAddr()) {
+	if forwardedHost != "" && ((pre == nil) || (pre != nil && forwardedHost != pre.(*schedulingClient).getPrimaryAddr())) {
 		client, err := s.getDelegateClient(ctx, forwardedHost)
 		if err != nil {
 			log.Error("get delegate client failed", zap.Error(err))