fix: need more time to run task

pull/2/head
Jason Song 2022-10-14 18:30:36 +08:00
parent 7bebd2bbad
commit cce60d466b
1 changed files with 6 additions and 3 deletions

View File

@ -123,12 +123,12 @@ func (p *Poller) poll(ctx context.Context, thread int) error {
l := log.WithField("thread", thread) l := log.WithField("thread", thread)
l.Info("poller: request stage from remote server") l.Info("poller: request stage from remote server")
ctx, cancel := context.WithTimeout(ctx, 30*time.Second) reqCtx, cancel := context.WithTimeout(ctx, 5*time.Second)
defer cancel() defer cancel()
// request a new build stage for execution from the central // request a new build stage for execution from the central
// build server. // build server.
resp, err := p.Client.FetchTask(ctx, connect.NewRequest(&runnerv1.FetchTaskRequest{})) resp, err := p.Client.FetchTask(reqCtx, connect.NewRequest(&runnerv1.FetchTaskRequest{}))
if err == context.Canceled || err == context.DeadlineExceeded { if err == context.Canceled || err == context.DeadlineExceeded {
l.WithError(err).Trace("poller: no stage returned") l.WithError(err).Trace("poller: no stage returned")
p.errorRetryCounter++ p.errorRetryCounter++
@ -155,5 +155,8 @@ func (p *Poller) poll(ctx context.Context, thread int) error {
p.errorRetryCounter = 0 p.errorRetryCounter = 0
return p.Dispatch(ctx, resp.Msg.Task) runCtx, cancel := context.WithTimeout(ctx, time.Hour)
defer cancel()
return p.Dispatch(runCtx, resp.Msg.Task)
} }