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

skip monitoring create/kill jobs

Browse source
This commit is contained in:
Nhat Tan Le 2020-09-30 16:15:22 -07:00
parent 82fe22e013
commit 81c05fb700
3 changed files with 17 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

20
cmd/create.go
View file

@ -21,13 +21,14 @@ import (
func init() { func init() {
rootCmd.AddCommand(createCmd) rootCmd.AddCommand(createCmd)
createCmd.Flags().BoolVarP(&monitor, "monitor", "m", false, "monitor the result after sending the command")
} }
var createCmd = &cobra.Command{ var createCmd = &cobra.Command{
Use: "create", Use: "create",
Short: "Create an Aurora Job", Short: "Create an Aurora Job",
Run: createJob, Run: createJob,
Args: cobra.ExactArgs(1), Args: cobra.RangeArgs(1, 2),
} }
func createJob(cmd *cobra.Command, args []string) { func createJob(cmd *cobra.Command, args []string) {
@ -46,14 +47,15 @@ func createJob(cmd *cobra.Command, args []string) {
log.Fatal("unable to create Aurora job: ", err) log.Fatal("unable to create Aurora job: ", err)
} }
if ok, monitorErr := client.MonitorInstances(auroraJob.JobKey(), if monitor {
auroraJob.GetInstanceCount(), if ok, monitorErr := client.MonitorInstances(auroraJob.JobKey(),
5, auroraJob.GetInstanceCount(),
50); !ok || monitorErr != nil { 5,
if err := client.KillJob(auroraJob.JobKey()); err != nil { 50); !ok || monitorErr != nil {
log.Fatal(monitorErr, err) if err := client.KillJob(auroraJob.JobKey()); err != nil {
log.Fatal(monitorErr, err)
}
log.Fatal(monitorErr)
} }
log.Fatal(monitorErr)
} }
} }

8
cmd/kill.go
View file

@ -30,6 +30,7 @@ func init() {
killJobCmd.Flags().StringVarP(env, "environment", "e", "", "Aurora Environment") killJobCmd.Flags().StringVarP(env, "environment", "e", "", "Aurora Environment")
killJobCmd.Flags().StringVarP(role, "role", "r", "", "Aurora Role") killJobCmd.Flags().StringVarP(role, "role", "r", "", "Aurora Role")
killJobCmd.Flags().StringVarP(name, "name", "n", "", "Aurora Name") killJobCmd.Flags().StringVarP(name, "name", "n", "", "Aurora Name")
killJobCmd.Flags().BoolVarP(&monitor, "monitor", "m", false, "monitor the result after sending the command")
killJobCmd.MarkFlagRequired("environment") killJobCmd.MarkFlagRequired("environment")
killJobCmd.MarkFlagRequired("role") killJobCmd.MarkFlagRequired("role")
killJobCmd.MarkFlagRequired("name") killJobCmd.MarkFlagRequired("name")
@ -57,8 +58,9 @@ func killJob(cmd *cobra.Command, args []string) {
if err != nil { if err != nil {
log.Fatalln(err) log.Fatalln(err)
} }
if monitor {
if ok, err := client.MonitorInstances(job.JobKey(), 0, 5, 50); !ok || err != nil { if ok, err := client.MonitorInstances(job.JobKey(), 0, 5, 50); !ok || err != nil {
log.Fatalln("Unable to kill all instances of job") log.Fatalln("Unable to kill all instances of job")
}
} }
} }

1
cmd/root.go
View file

@ -46,6 +46,7 @@ var count int64
var filename string var filename string
var message = new(string) var message = new(string)
var updateID string var updateID string
var monitor bool
var log = logrus.New() var log = logrus.New()
const australisVer = "v0.22.0" const australisVer = "v0.22.0"