Skip to content

Commit

Permalink
chore: optimize register timeout (#3288)
Browse files Browse the repository at this point in the history
Signed-off-by: Jim Ma <[email protected]>
  • Loading branch information
jim3ma authored May 21, 2024
1 parent f0c4ef0 commit eba0105
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions client/daemon/peer/peertask_conductor.go
Original file line number Diff line number Diff line change
Expand Up @@ -279,17 +279,19 @@ func (pt *peerTaskConductor) register() error {
pt.Errorf("scheduler did not response in %s", pt.SchedulerOption.ScheduleTimeout.Duration)
}
pt.Errorf("step 1: peer %s register failed: %s", pt.request.PeerId, err)
// can not detect source or scheduler error, create a new dummy scheduler client
pt.schedulerClient = &dummySchedulerClient{}
// when peer register failed, some actions need to do with peerPacketStream
pt.peerPacketStream = &dummyPeerPacketStream{}

if pt.SchedulerOption.DisableAutoBackSource {
// when peer register failed, some actions need to do with peerPacketStream
pt.peerPacketStream = &dummyPeerPacketStream{}
pt.Errorf("register peer task failed: %s, peer id: %s, auto back source disabled", err, pt.request.PeerId)
pt.span.RecordError(err)
pt.cancel(commonv1.Code_SchedError, err.Error())
return err
}

needBackSource = true
// can not detect source or scheduler error, create a new dummy scheduler client
pt.schedulerClient = &dummySchedulerClient{}
result = &schedulerv1.RegisterResult{TaskId: pt.taskID}
pt.Warnf("register peer task failed: %s, peer id: %s, try to back source", err, pt.request.PeerId)
} else {
Expand Down

0 comments on commit eba0105

Please sign in to comment.