diff --git a/cmd/fetch.go b/cmd/fetch.go index 5958038..a7c6bba 100644 --- a/cmd/fetch.go +++ b/cmd/fetch.go @@ -4,6 +4,7 @@ import ( "fmt" "os" + "github.com/paypal/gorealis" "github.com/paypal/gorealis/gen-go/apache/aurora" "github.com/spf13/cobra" ) @@ -12,10 +13,15 @@ func init() { rootCmd.AddCommand(fetchCmd) // Sub-commands + + // Fetch Task Config fetchCmd.AddCommand(taskConfigCmd) taskConfigCmd.Flags().StringVarP(&env, "environment", "e", "", "Aurora Environment") taskConfigCmd.Flags().StringVarP(&role, "role", "r", "", "Aurora Role") taskConfigCmd.Flags().StringVarP(&name, "name", "n", "", "Aurora Name") + + // Fetch Leader + fetchCmd.AddCommand(leaderCmd) } var fetchCmd = &cobra.Command{ @@ -24,10 +30,19 @@ var fetchCmd = &cobra.Command{ } var taskConfigCmd = &cobra.Command{ - Use: "config" , + Use: "config", Short: "Fetch a list of task configurations from Aurora.", - Long: `To be written.`, - Run: fetchTasks, + Long: `To be written.`, + Run: fetchTasks, +} + +var leaderCmd = &cobra.Command{ + Use: "leader", + PersistentPreRun: func(cmd *cobra.Command, args []string) {}, //We don't need a realis client for this cmd + PersistentPostRun: func(cmd *cobra.Command, args []string) {}, //We don't need a realis client for this cmd + Short: "Fetch current Aurora leader given Zookeeper nodes. Pass Zookeeper nodes separated by a space as an argument to this command.", + Long: `To be written.`, + Run: fetchLeader, } func fetchTasks(cmd *cobra.Command, args []string) { @@ -46,3 +61,21 @@ func fetchTasks(cmd *cobra.Command, args []string) { fmt.Println(t) } } + +func fetchLeader(cmd *cobra.Command, args []string) { + fmt.Printf("Fetching leader from %v \n", args) + + if len(args) < 1 { + fmt.Println("At least one Zookeper node address must be passed in.") + os.Exit(1) + } + + url, err := realis.LeaderFromZKOpts(realis.ZKEndpoints(args...), realis.ZKPath("/aurora/scheduler")) + + if err != nil { + fmt.Printf("error: %+v\n", err.Error()) + os.Exit(1) + } + + fmt.Print(url) +}