From f1dec253fc6965e97256566ba50f1e0f15d33f82 Mon Sep 17 00:00:00 2001 From: Gaius Date: Fri, 6 Sep 2024 11:18:37 +0800 Subject: [PATCH] feat: use context.Background for ScheduleCandidateParents to prevent stream breaking (#3485) Signed-off-by: Gaius --- client-rs | 2 +- scheduler/service/service_v2.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/client-rs b/client-rs index 2fe844ac2..232ee8ab4 160000 --- a/client-rs +++ b/client-rs @@ -1 +1 @@ -Subproject commit 2fe844ac26afadbdee1b35f3fb7cd8e091929385 +Subproject commit 232ee8ab4631f6781ee39ff36580db0426a2c801 diff --git a/scheduler/service/service_v2.go b/scheduler/service/service_v2.go index 39d20edc3..c082c57bf 100644 --- a/scheduler/service/service_v2.go +++ b/scheduler/service/service_v2.go @@ -963,7 +963,7 @@ func (v *V2) handleRegisterPeerRequest(ctx context.Context, stream schedulerv2.S // Record the start time. start := time.Now() - if err := v.scheduling.ScheduleCandidateParents(ctx, peer, peer.BlockParents); err != nil { + if err := v.scheduling.ScheduleCandidateParents(context.Background(), peer, peer.BlockParents); err != nil { // Collect RegisterPeerFailureCount metrics. metrics.RegisterPeerFailureCount.WithLabelValues(priority.String(), peer.Task.Type.String(), peer.Host.Type.Name()).Inc() @@ -1042,7 +1042,7 @@ func (v *V2) handleRescheduleRequest(ctx context.Context, peerID string, candida // Record the start time. start := time.Now() - if err := v.scheduling.ScheduleCandidateParents(ctx, peer, peer.BlockParents); err != nil { + if err := v.scheduling.ScheduleCandidateParents(context.Background(), peer, peer.BlockParents); err != nil { return status.Error(codes.FailedPrecondition, err.Error()) }