return an error instead of panicking when failing to get edge

Signed-off-by: Maxime Lagresle <[email protected]>
moby · Sep 30, 2021 · 0e7de55 · 0e7de55
1 parent e839ff7
commit 0e7de55
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/solver/scheduler.go b/solver/scheduler.go
@@ -351,7 +351,9 @@ type pipeFactory struct {
 func (pf *pipeFactory) NewInputRequest(ee Edge, req *edgeRequest) pipe.Receiver {
 	target := pf.s.ef.getEdge(ee)
 	if target == nil {
-		panic("failed to get edge") // TODO: return errored pipe
+		return pf.NewFuncRequest(func(_ context.Context) (interface{}, error) {
+			return req, errors.Errorf("failed to get edge")
+		})
 	}
 	p := pf.s.newPipe(target, pf.e, pipe.Request{Payload: req})
 	if debugScheduler {