mirror of https://github.com/databricks/cli.git
38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package run
|
|
|
|
import (
|
|
"github.com/databricks/cli/clis"
|
|
"github.com/databricks/cli/libs/cmdgroup"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type Options struct {
|
|
Job JobOptions
|
|
Pipeline PipelineOptions
|
|
NoWait bool
|
|
}
|
|
|
|
func (o *Options) Define(cmd *cobra.Command, cliType clis.CLIType) {
|
|
if cliType == clis.DLT {
|
|
// Only show the DLT flags, and don't group them
|
|
o.Pipeline.Define(cmd.Flags())
|
|
return
|
|
}
|
|
|
|
jobGroup := cmdgroup.NewFlagGroup("Job")
|
|
o.Job.DefineJobOptions(jobGroup.FlagSet())
|
|
|
|
jobTaskGroup := cmdgroup.NewFlagGroup("Job Task")
|
|
jobTaskGroup.SetDescription(`Note: please prefer use of job-level parameters (--param) over task-level parameters.
|
|
For more information, see https://docs.databricks.com/en/workflows/jobs/create-run-jobs.html#pass-parameters-to-a-databricks-job-task`)
|
|
o.Job.DefineTaskOptions(jobTaskGroup.FlagSet())
|
|
|
|
pipelineGroup := cmdgroup.NewFlagGroup("DLT")
|
|
o.Pipeline.Define(pipelineGroup.FlagSet())
|
|
|
|
wrappedCmd := cmdgroup.NewCommandWithGroupFlag(cmd)
|
|
wrappedCmd.AddFlagGroup(jobGroup)
|
|
wrappedCmd.AddFlagGroup(jobTaskGroup)
|
|
wrappedCmd.AddFlagGroup(pipelineGroup)
|
|
}
|