Merge pull request #21 from sunilmvn/master

Updating the client to not use instanceId 0 by default
This commit is contained in:
kkrishna 2017-04-18 19:06:52 -07:00 committed by GitHub
commit e12370ffbb
2 changed files with 31 additions and 6 deletions

View file

@ -316,7 +316,12 @@ func main() {
} }
currInstances := int32(len(live)) currInstances := int32(len(live))
fmt.Println("Current num of instances: ", currInstances) fmt.Println("Current num of instances: ", currInstances)
resp, err := r.AddInstances(aurora.InstanceKey{job.JobKey(), 0}, numOfInstances) var instId int32
for k := range live{
instId = k
break
}
resp, err := r.AddInstances(aurora.InstanceKey{job.JobKey(), instId}, numOfInstances)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)
@ -358,7 +363,17 @@ func main() {
break break
case "update": case "update":
fmt.Println("Updating a job with with more RAM and to 5 instances") fmt.Println("Updating a job with with more RAM and to 5 instances")
taskConfig, err := r.FetchTaskConfig(aurora.InstanceKey{job.JobKey(), 0}) live, err := r.GetInstanceIds(job.JobKey(), aurora.ACTIVE_STATES)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
var instId int32
for k := range live{
instId = k
break
}
taskConfig, err := r.FetchTaskConfig(aurora.InstanceKey{job.JobKey(), instId})
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)
@ -406,7 +421,17 @@ func main() {
break break
case "taskConfig": case "taskConfig":
fmt.Println("Getting job info") fmt.Println("Getting job info")
config, err := r.FetchTaskConfig(aurora.InstanceKey{job.JobKey(), 0}) live, err := r.GetInstanceIds(job.JobKey(), aurora.ACTIVE_STATES)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
var instId int32
for k := range live{
instId = k
break
}
config, err := r.FetchTaskConfig(aurora.InstanceKey{job.JobKey(), instId})
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)

View file

@ -208,7 +208,7 @@ func NewRealisClient(options ...option) (Realis, error) {
fmt.Printf(" updating default backoff : %+v\n", *config.backoff) fmt.Printf(" updating default backoff : %+v\n", *config.backoff)
} }
fmt.Printf("gorealis config: %+v\n", config) fmt.Printf("gorealis config url: %+v\n", config.url)
return &realisClient{ return &realisClient{
config: config, config: config,
@ -488,7 +488,7 @@ func (r *realisClient) ReestablishConn() error {
fmt.Println(" r.config.url: ", r.config.url) fmt.Println(" r.config.url: ", r.config.url)
return errors.New(" Missing Data for ReestablishConn ") return errors.New(" Missing Data for ReestablishConn ")
} }
fmt.Printf(" config before return: %+v\n", r.config) fmt.Printf(" config url before return: %+v\n", r.config.url)
return nil return nil
} }
@ -934,7 +934,7 @@ func (r *realisClient) AddInstances(instKey aurora.InstanceKey, count int32) (*a
return nil, errors.Wrap(err, "Error sending AddInstances command to Aurora Scheduler") return nil, errors.Wrap(err, "Error sending AddInstances command to Aurora Scheduler")
} }
//Scale down the number of instances under a job configuration using the configuratipn of a specific instance //Scale down the number of instances under a job configuration using the configuration of a specific instance
func (r *realisClient) RemoveInstances(key *aurora.JobKey, count int32) (*aurora.Response, error) { func (r *realisClient) RemoveInstances(key *aurora.JobKey, count int32) (*aurora.Response, error) {
instanceIds, err := r.GetInstanceIds(key, aurora.ACTIVE_STATES) instanceIds, err := r.GetInstanceIds(key, aurora.ACTIVE_STATES)
if err != nil { if err != nil {