Skip to content

Remove the 'all-priority-classes' flag from armadactl commands #5135

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
JamesMurkin wants to merge 2 commits into
base: master
Choose a base branch
from
+468 −511
Open

Remove the 'all-priority-classes' flag from armadactl commands #5135

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5c24ec6
Remove the 'all-priority-classes' flag from armadactl commands
JamesMurkin Aug 27, 2026
cef6269
More robust tests
JamesMurkin Aug 27, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
78 changes: 9 additions & 69 deletions cmd/armadactl/cmd/cancel.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,9 @@ func cancelCmd() *cobra.Command {
cmd.AddCommand(
cancelJobCmd(),
cancelJobSetCmd(),
cancelExecutorCmd(),
cancelNodeCmd(),
cancelQueueCmd(),
cancelExecutorCmd(armadactl.New()),
cancelNodeCmd(armadactl.New()),
cancelQueueCmd(armadactl.New()),
)
return cmd
}
Expand Down Expand Up @@ -66,23 +66,13 @@ func cancelJobSetCmd() *cobra.Command {
return cmd
}

func cancelExecutorCmd() *cobra.Command {
a := armadactl.New()
func cancelExecutorCmd(a *armadactl.App) *cobra.Command {
cmd := &cobra.Command{
Use: "executor <executor>",
Short: "Cancels jobs on executor.",
Long: `Cancels jobs on executor with provided executor name, priority classes, queues, and pools.`,
Args: cobra.ExactArgs(1),
PreRunE: func(cmd *cobra.Command, args []string) error {
all, err := cmd.Flags().GetBool("all-priority-classes")
if err != nil {
return fmt.Errorf("error reading all-priority-classes flag: %s", err)
}
if !all {
if err := cmd.MarkFlagRequired("priority-classes"); err != nil {
return fmt.Errorf("error marking priority-class flag as required: %s", err)
}
}
return initParams(cmd, a.Params)
},
RunE: func(cmd *cobra.Command, args []string) error {
Expand All @@ -92,10 +82,6 @@ func cancelExecutorCmd() *cobra.Command {
if err != nil {
return fmt.Errorf("error reading priority-class selection: %s", err)
}
allPriorityClasses, _ := cmd.Flags().GetBool("all-priority-classes")
if allPriorityClasses {
priorityClasses = nil
}

queues, err := cmd.Flags().GetStringSlice("queues")
if err != nil {
Expand All @@ -121,13 +107,7 @@ func cancelExecutorCmd() *cobra.Command {
"priority-classes",
"p",
[]string{},
"Cancel jobs on executor matching the specified priority classes. Provided priority classes should be comma separated, as in the following example: armada-default,armada-preemptible.",
)
cmd.Flags().BoolP(
"all-priority-classes",
"a",
false,
"Cancel jobs on executor for all priority classes.",
"Cancel jobs on executor matching the specified priority classes, comma separated (e.g. armada-default,armada-preemptible). If no priority classes are provided, jobs across all priority classes will be cancelled.",
)
cmd.Flags().StringSlice(
"pools",
Expand All @@ -137,23 +117,13 @@ func cancelExecutorCmd() *cobra.Command {
return cmd
}

func cancelNodeCmd() *cobra.Command {
a := armadactl.New()
func cancelNodeCmd(a *armadactl.App) *cobra.Command {
cmd := &cobra.Command{
Use: "node <name>",
Short: "Cancels jobs on node for specified executor.",
Long: `Cancels jobs on node for executor with provided executor name, priority classes and queues.`,
Args: cobra.ExactArgs(1),
PreRunE: func(cmd *cobra.Command, args []string) error {
all, err := cmd.Flags().GetBool("all-priority-classes")
if err != nil {
return fmt.Errorf("error reading all-priority-classes flag: %s", err)
}
if !all {
if err := cmd.MarkFlagRequired("priority-classes"); err != nil {
return fmt.Errorf("error marking priority-class flag as required: %s", err)
}
}
if err := cmd.MarkFlagRequired("executor"); err != nil {
return fmt.Errorf("error marking executor flag as required: %s", err)
}
Expand All @@ -166,10 +136,6 @@ func cancelNodeCmd() *cobra.Command {
if err != nil {
return fmt.Errorf("error reading priority-class selection: %s", err)
}
allPriorityClasses, _ := cmd.Flags().GetBool("all-priority-classes")
if allPriorityClasses {
priorityClasses = nil
}

queues, err := cmd.Flags().GetStringSlice("queues")
if err != nil {
Expand All @@ -195,13 +161,7 @@ func cancelNodeCmd() *cobra.Command {
"priority-classes",
"p",
[]string{},
"Cancel jobs on node for specified executor matching the specified priority classes. Provided priority classes should be comma separated, as in the following example: armada-default,armada-preemptible.",
)
cmd.Flags().BoolP(
"all-priority-classes",
"a",
false,
"Preempt jobs on executor for all priority classes.",
"Cancel jobs on node for specified executor matching the specified priority classes, comma separated (e.g. armada-default,armada-preemptible). If no priority classes are provided, jobs across all priority classes will be cancelled.",
)
cmd.Flags().StringP(
"executor",
Expand All @@ -212,8 +172,7 @@ func cancelNodeCmd() *cobra.Command {
return cmd
}

func cancelQueueCmd() *cobra.Command {
a := armadactl.New()
func cancelQueueCmd(a *armadactl.App) *cobra.Command {
cmd := &cobra.Command{
Use: "queues <queue_1> <queue_2> <queue_3> ...",
Short: "Cancels jobs on queues.",
Expand All @@ -223,15 +182,6 @@ func cancelQueueCmd() *cobra.Command {
if err := cmd.MarkFlagRequired("job-states"); err != nil {
return err
}
all, err := cmd.Flags().GetBool("all-priority-classes")
if err != nil {
return fmt.Errorf("error reading all-priority-classes flag: %s", err)
}
if !all {
if err := cmd.MarkFlagRequired("priority-classes"); err != nil {
return fmt.Errorf("error marking priority-class flag as required: %s", err)
}
}
return initParams(cmd, a.Params)
},
RunE: func(cmd *cobra.Command, queues []string) error {
Expand Down Expand Up @@ -273,10 +223,6 @@ func cancelQueueCmd() *cobra.Command {
if err != nil {
return fmt.Errorf("error reading priority-classes flag: %s", err)
}
allPriorityClasses, _ := cmd.Flags().GetBool("all-priority-classes")
if allPriorityClasses {
priorityClasses = nil
}

pools, err := cmd.Flags().GetStringSlice("pools")
if err != nil {
Expand Down Expand Up @@ -313,13 +259,7 @@ func cancelQueueCmd() *cobra.Command {
"priority-classes",
"p",
[]string{},
"Jobs matching the provided priority classes will be cancelled.",
)
cmd.Flags().BoolP(
"all-priority-classes",
"a",
false,
"Preempt jobs on executor for all priority classes.",
"Jobs matching the provided priority classes will be cancelled. If no priority classes are provided, jobs across all priority classes will be cancelled.",
)
cmd.Flags().StringSlice(
"pools",
Expand Down
Loading
Loading