Support for partially running a DAG

Signed-off-by: Andrea Luzzardi <aluzzardi@gmail.com>
This commit is contained in:
Andrea Luzzardi 2021-12-23 16:09:01 +01:00
parent 2f92d8db4e
commit 2b30580217
2 changed files with 17 additions and 8 deletions

View File

@ -145,8 +145,8 @@ func europaUp(ctx context.Context, cl *client.Client, args ...string) error {
p, err := plan.Load(ctx, plan.Config{ p, err := plan.Load(ctx, plan.Config{
Args: args, Args: args,
With: viper.GetStringSlice("with"), With: viper.GetStringSlice("with"),
Target: viper.GetString("target"),
}) })
if err != nil { if err != nil {
lg.Fatal().Err(err).Msg("failed to load plan") lg.Fatal().Err(err).Msg("failed to load plan")
} }
@ -225,6 +225,7 @@ func init() {
upCmd.Flags().BoolP("force", "f", false, "Force up, disable inputs check") upCmd.Flags().BoolP("force", "f", false, "Force up, disable inputs check")
upCmd.Flags().String("output", "", "Write computed output. Prints on stdout if set to-") upCmd.Flags().String("output", "", "Write computed output. Prints on stdout if set to-")
upCmd.Flags().StringArrayP("with", "w", []string{}, "") upCmd.Flags().StringArrayP("with", "w", []string{}, "")
upCmd.Flags().StringP("target", "t", "", "Run a single target of the DAG (for debugging only)")
if err := viper.BindPFlags(upCmd.Flags()); err != nil { if err := viper.BindPFlags(upCmd.Flags()); err != nil {
panic(err) panic(err)

View File

@ -28,6 +28,7 @@ type Plan struct {
type Config struct { type Config struct {
Args []string Args []string
With []string With []string
Target string
} }
func Load(ctx context.Context, cfg Config) (*Plan, error) { func Load(ctx context.Context, cfg Config) (*Plan, error) {
@ -149,10 +150,17 @@ func (p *Plan) Up(ctx context.Context, s solver.Solver) (*compiler.Value, error)
computed := compiler.NewValue() computed := compiler.NewValue()
flow := cueflow.New( cfg := &cueflow.Config{
&cueflow.Config{
FindHiddenTasks: true, FindHiddenTasks: true,
}, }
if p.config.Target != "" {
cfg.Root = cue.ParsePath(p.config.Target)
// The target may reference dependencies outside of the target path.
// InferTasks will include them in the workflow.
cfg.InferTasks = true
}
flow := cueflow.New(
cfg,
p.source.Cue(), p.source.Cue(),
newRunner(p.context, s, computed), newRunner(p.context, s, computed),
) )