Skip to content

Commit

Permalink
Apply suggestions from code review via local.
Browse files Browse the repository at this point in the history
  • Loading branch information
jrasell committed Jun 14, 2022
1 parent ca36c54 commit 9177f83
Show file tree
Hide file tree
Showing 4 changed files with 37 additions and 36 deletions.
54 changes: 27 additions & 27 deletions command/operator_scheduler_set_config.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,14 +19,14 @@ type OperatorSchedulerSetConfig struct {
// The scheduler configuration flags allow us to tell whether the user set
// a value or not. This means we can safely merge the current configuration
// with user supplied, selective updates.
schedulerAlgorithm string
memoryOversubscription flagHelper.BoolValue
rejectJobRegistration flagHelper.BoolValue
pauseEvalBroker flagHelper.BoolValue
preemptionBatchScheduler flagHelper.BoolValue
preemptionServiceScheduler flagHelper.BoolValue
preemptionSysBatchScheduler flagHelper.BoolValue
preemptionSystemScheduler flagHelper.BoolValue
schedulerAlgorithm string
memoryOversubscription flagHelper.BoolValue
rejectJobRegistration flagHelper.BoolValue
pauseEvalBroker flagHelper.BoolValue
preemptBatchScheduler flagHelper.BoolValue
preemptServiceScheduler flagHelper.BoolValue
preemptSysBatchScheduler flagHelper.BoolValue
preemptSystemScheduler flagHelper.BoolValue
}

func (o *OperatorSchedulerSetConfig) AutocompleteFlags() complete.Flags {
Expand All @@ -36,13 +36,13 @@ func (o *OperatorSchedulerSetConfig) AutocompleteFlags() complete.Flags {
string(api.SchedulerAlgorithmBinpack),
string(api.SchedulerAlgorithmSpread),
),
"-memory-oversubscription": complete.PredictSet("true", "false"),
"-reject-job-registration": complete.PredictSet("true", "false"),
"-pause-eval-broker": complete.PredictSet("true", "false"),
"-preemption-batch-scheduler": complete.PredictSet("true", "false"),
"-preemption-service-scheduler": complete.PredictSet("true", "false"),
"-preemption-sysbatch-scheduler": complete.PredictSet("true", "false"),
"-preemption-system-scheduler": complete.PredictSet("true", "false"),
"-memory-oversubscription": complete.PredictSet("true", "false"),
"-reject-job-registration": complete.PredictSet("true", "false"),
"-pause-eval-broker": complete.PredictSet("true", "false"),
"-preempt-batch-scheduler": complete.PredictSet("true", "false"),
"-preempt-service-scheduler": complete.PredictSet("true", "false"),
"-preempt-sysbatch-scheduler": complete.PredictSet("true", "false"),
"-preempt-system-scheduler": complete.PredictSet("true", "false"),
},
)
}
Expand All @@ -62,10 +62,10 @@ func (o *OperatorSchedulerSetConfig) Run(args []string) int {
flags.Var(&o.memoryOversubscription, "memory-oversubscription", "")
flags.Var(&o.rejectJobRegistration, "reject-job-registration", "")
flags.Var(&o.pauseEvalBroker, "pause-eval-broker", "")
flags.Var(&o.preemptionBatchScheduler, "preemption-batch-scheduler", "")
flags.Var(&o.preemptionServiceScheduler, "preemption-service-scheduler", "")
flags.Var(&o.preemptionSysBatchScheduler, "preemption-sysbatch-scheduler", "")
flags.Var(&o.preemptionSystemScheduler, "preemption-system-scheduler", "")
flags.Var(&o.preemptBatchScheduler, "preempt-batch-scheduler", "")
flags.Var(&o.preemptServiceScheduler, "preempt-service-scheduler", "")
flags.Var(&o.preemptSysBatchScheduler, "preempt-sysbatch-scheduler", "")
flags.Var(&o.preemptSystemScheduler, "preempt-system-scheduler", "")

if err := flags.Parse(args); err != nil {
return 1
Expand Down Expand Up @@ -94,10 +94,10 @@ func (o *OperatorSchedulerSetConfig) Run(args []string) int {
o.memoryOversubscription.Merge(&schedulerConfig.MemoryOversubscriptionEnabled)
o.rejectJobRegistration.Merge(&schedulerConfig.RejectJobRegistration)
o.pauseEvalBroker.Merge(&schedulerConfig.PauseEvalBroker)
o.preemptionBatchScheduler.Merge(&schedulerConfig.PreemptionConfig.BatchSchedulerEnabled)
o.preemptionServiceScheduler.Merge(&schedulerConfig.PreemptionConfig.ServiceSchedulerEnabled)
o.preemptionSysBatchScheduler.Merge(&schedulerConfig.PreemptionConfig.SysBatchSchedulerEnabled)
o.preemptionSystemScheduler.Merge(&schedulerConfig.PreemptionConfig.SystemSchedulerEnabled)
o.preemptBatchScheduler.Merge(&schedulerConfig.PreemptionConfig.BatchSchedulerEnabled)
o.preemptServiceScheduler.Merge(&schedulerConfig.PreemptionConfig.ServiceSchedulerEnabled)
o.preemptSysBatchScheduler.Merge(&schedulerConfig.PreemptionConfig.SysBatchSchedulerEnabled)
o.preemptSystemScheduler.Merge(&schedulerConfig.PreemptionConfig.SystemSchedulerEnabled)

// Check-and-set the new configuration.
result, _, err := client.Operator().SchedulerCASConfiguration(schedulerConfig, nil)
Expand Down Expand Up @@ -152,19 +152,19 @@ Scheduler Set Config Options:
When set to true, the eval broker which usually runs on the leader will be
disabled. This will prevent the scheduler workers from receiving new work.
-preemption-batch-scheduler=[true|false]
-preempt-batch-scheduler=[true|false]
Specifies whether preemption for batch jobs is enabled. Note that if this
is set to true, then batch jobs can preempt any other jobs.
-preemption-service-scheduler=[true|false]
-preempt-service-scheduler=[true|false]
Specifies whether preemption for service jobs is enabled. Note that if this
is set to true, then service jobs can preempt any other jobs.
-preemption-sysbatch-scheduler=[true|false]
-preempt-sysbatch-scheduler=[true|false]
Specifies whether preemption for system batch jobs is enabled. Note that if
this is set to true, then system batch jobs can preempt any other jobs.
-preemption-system-scheduler=[true|false]
-preempt-system-scheduler=[true|false]
Specifies whether preemption for system jobs is enabled. Note that if this
is set to true, then system jobs can preempt any other jobs.
`
Expand Down
9 changes: 4 additions & 5 deletions command/operator_scheduler_set_config_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -41,13 +41,12 @@ func TestOperatorSchedulerSetConfig_Run(t *testing.T) {
"-address=" + addr,
"-scheduler-algorithm=spread",
"-pause-eval-broker=true",
"-preemption-sysbatch-scheduler=true",
"-memory-oversubscription=true",
"-reject-job-registration=true",
"-preemption-batch-scheduler=true",
"-preemption-service-scheduler=true",
"-preemption-sysbatch-scheduler=true",
"-preemption-system-scheduler=false",
"-preempt-batch-scheduler=true",
"-preempt-service-scheduler=true",
"-preempt-sysbatch-scheduler=true",
"-preempt-system-scheduler=false",
}
require.EqualValues(t, 0, c.Run(modifyingArgs))
s := ui.OutputWriter.String()
Expand Down
2 changes: 2 additions & 0 deletions nomad/leader_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -1676,12 +1676,14 @@ func TestServer_handleEvalBrokerStateChange(t *testing.T) {
name string
}{
{
startValue: false,
testServerCallBackConfig: func(c *Config) { c.DefaultSchedulerConfig.PauseEvalBroker = false },
inputSchedulerConfiguration: nil,
expectedOutput: true,
name: "bootstrap un-paused",
},
{
startValue: false,
testServerCallBackConfig: func(c *Config) { c.DefaultSchedulerConfig.PauseEvalBroker = true },
inputSchedulerConfiguration: nil,
expectedOutput: false,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -42,19 +42,19 @@ capability.
the leader will be disabled. This will prevent the scheduler workers from
receiving new work. Must be one of `[true|false]`.

- `-preemption-batch-scheduler` - Specifies whether preemption for batch jobs
- `-preempt-batch-scheduler` - Specifies whether preemption for batch jobs
is enabled. Note that if this is set to true, then batch jobs can preempt any
other jobs. Must be one of `[true|false]`.

- `-preemption-service-scheduler` - Specifies whether preemption for service jobs
- `-preempt-service-scheduler` - Specifies whether preemption for service jobs
is enabled. Note that if this is set to true, then service jobs can preempt any
other jobs. Must be one of `[true|false]`.

- `-preemption-sysbatch-scheduler` - Specifies whether preemption for system batch
- `-preempt-sysbatch-scheduler` - Specifies whether preemption for system batch
jobs is enabled. Note that if this is set to true, then system batch jobs can
preempt any other jobs. Must be one of `[true|false]`.

- `-preemption-system-scheduler` - Specifies whether preemption for system jobs
- `-preempt-system-scheduler` - Specifies whether preemption for system jobs
is enabled. Note that if this is set to true, then system jobs can preempt any
other jobs. Must be one of `[true|false]`.

Expand Down

0 comments on commit 9177f83

Please sign in to comment.