add RPC timeout

2020-07-23 20:34:27 +08:00 · 2020-07-23 20:34:27 +08:00 · fafecdfd17
commit fafecdfd17
parent 7c19c5b229
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