ridv/australis
1
0
Fork 0
Code Issues Pull requests Projects Releases 16 Packages Wiki Activity Actions

Adding interval and timeout flags for start and stop drain.

Browse source
This commit is contained in:
Renan DelValle 2018-05-30 18:15:16 -07:00
parent afa7a5d14f
commit 51daad88eb
No known key found for this signature in database
GPG key ID: C240AD6D6F443EC9
2 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
cmd/root.go
View file

@ -28,6 +28,8 @@ var insecureSkipVerify bool
var caCertsPath string
var clientKey, clientCert string
var monitorInterval, monitorTimeout int
func init() {
rootCmd.PersistentFlags().StringVarP(&zkAddr, "zookeeper", "z", "", "Zookeeper node(s) where Aurora stores information.")
rootCmd.PersistentFlags().StringVarP(&username, "username", "u", "", "Username to use for API authentication")

11
cmd/start.go
View file

@ -11,8 +11,13 @@ import (
func init() {
rootCmd.AddCommand(startCmd)
// Sub-commands
startCmd.AddCommand(startMaintCmd)
startMaintCmd.Flags().IntVar(&monitorInterval,"interval", 5, "Interval at which to poll scheduler.")
startMaintCmd.Flags().IntVar(&monitorTimeout,"timeout", 50, "Time after which the monitor will stop polling and throw an error.")
}
var startCmd = &cobra.Command{
@ -44,8 +49,8 @@ func drain(cmd *cobra.Command, args []string) {
hostResult, err := monitor.HostMaintenance(
args,
[]aurora.MaintenanceMode{aurora.MaintenanceMode_DRAINED},
5,
10)
monitorInterval,
monitorTimeout)
if err != nil {
for host, ok := range hostResult {
if !ok {
@ -57,5 +62,5 @@ func drain(cmd *cobra.Command, args []string) {
return
}
fmt.Print(result.String())
fmt.Println(result.String())
}