Merge pull request #77 from yaohcn/timeout-ctx

add RPC timeout
2020-07-23 22:09:50 +02:00 · 2020-07-23 22:09:50 +02:00 · ed2e57dde6
commit ed2e57dde6
parent 7fafaf81c0 fafecdfd17
1 changed files with 3 additions and 1 deletions
--- a/sched.go
+++ b/sched.go
@ -258,7 +258,9 @@ func (sh *scheduler) trySched() {
 				continue
 			}

-			ok, err := task.sel.Ok(task.ctx, task.taskType, sh.spt, worker)
+			rpcCtx, cancel := context.WithTimeout(task.ctx, SelectorTimeout)
+			ok, err := task.sel.Ok(rpcCtx, task.taskType, sh.spt, worker)
+			cancel()
 			if err != nil {
 				log.Errorf("trySched(1) req.sel.Ok error: %+v", err)
 				continue