Merge pull request #77 from yaohcn/timeout-ctx

add RPC timeout
2020-07-23 22:09:50 +02:00
parent 7fafaf81c0 fafecdfd17
commit ed2e57dde6
1 changed files with 3 additions and 1 deletions
@@ -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