fix: trigger checkpoint restore in loop (#265)

meguminnnnnnnnn · web-flow · commit 5b1ab13a9324 · 2025-06-03T21:01:14.000+08:00
diff --git a/compose/checkpoint.go b/compose/checkpoint.go
@@ -136,6 +136,7 @@ func forwardCheckPoint(ctx context.Context, nodeKey string) context.Context {
 		return ctx
 	}
 	if subCP, ok := cp.SubGraphs[nodeKey]; ok {
+		delete(cp.SubGraphs, nodeKey) // only forward once
 		return context.WithValue(ctx, checkPointKey{}, subCP)
 	}
 	return context.WithValue(ctx, checkPointKey{}, (*checkpoint)(nil))

Original file line number	Diff line number	Diff line change
`@@ -136,6 +136,7 @@ func forwardCheckPoint(ctx context.Context, nodeKey string) context.Context {`
`136`	`136`	`return ctx`
`137`	`137`	`}`
`138`	`138`	`if subCP, ok := cp.SubGraphs[nodeKey]; ok {`
	`139`	`+ delete(cp.SubGraphs, nodeKey) // only forward once`
`139`	`140`	`return context.WithValue(ctx, checkPointKey{}, subCP)`
`140`	`141`	`}`
`141`	`142`	`return context.WithValue(ctx, checkPointKey{}, (*checkpoint)(nil))`