ridv/gorealis-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases 13 Packages Wiki Activity Actions

Thrift API update and Pull Request template. (#43)

Browse source 
* Adding a Pull request template to serve as a reminder of what
to do before creating the pull request.

* Updating our thrift API to match changes made by Aurora.

* Update go bindings to match update thrift API.
This commit is contained in:
Renan DelValle 2017-12-14 14:37:08 -08:00 • committed by GitHub
parent d4027bc95c
commit c338c03355
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 1227 additions and 1729 deletions
Download patch file Download diff file Expand all files Collapse all files

2
.auroraversion
View file

@ -1 +1 @@
0.16.0 0.19.0

11
.github/PULL_REQUEST_TEMPLATE.md vendored Normal file
View file

@ -0,0 +1,11 @@
Before submitting, please make sure you run a vagrant box running Aurora with the latest version shown in .auroraversion
and run go test from the project root.
To run an Aurora Vagrant image, follow the instructions here:
http://aurora.apache.org/documentation/latest/getting-started/vagrant/
* Have you run goformat on the project before submitting?
* Have you run go test on the project before submitting? Do all tests pass?
* Does the Pull Request require a test to be added to the end to end tests? If so, has it been added?

55
auroraAPI.thrift
View file

@ -27,8 +27,8 @@ enum ResponseCode {
ERROR = 2, ERROR = 2,
WARNING = 3, WARNING = 3,
AUTH_FAILED = 4, AUTH_FAILED = 4,
/** Raised when a Lock-protected operation failed due to lock validation. */ /** Raised when an operation was unable to proceed due to an in-progress job update. */
LOCK_ERROR = 5, JOB_UPDATING_ERROR = 5,
/** Raised when a scheduler is transiently unavailable and later retry is recommended. */ /** Raised when a scheduler is transiently unavailable and later retry is recommended. */
ERROR_TRANSIENT = 6 ERROR_TRANSIENT = 6
} }
@ -203,6 +203,8 @@ union Image {
struct MesosContainer { struct MesosContainer {
/** the optional filesystem image to use when launching this task. */ /** the optional filesystem image to use when launching this task. */
1: optional Image image 1: optional Image image
/** the optional list of volumes to mount into the task. */
2: optional list<Volume> volumes
} }
/** Describes a parameter passed to docker cli */ /** Describes a parameter passed to docker cli */
@ -618,6 +620,10 @@ const set<JobUpdateStatus> ACTIVE_JOB_UPDATE_STATES = [JobUpdateStatus.ROLLING_F
JobUpdateStatus.ROLL_FORWARD_AWAITING_PULSE, JobUpdateStatus.ROLL_FORWARD_AWAITING_PULSE,
JobUpdateStatus.ROLL_BACK_AWAITING_PULSE] JobUpdateStatus.ROLL_BACK_AWAITING_PULSE]
/** States the job update can be in while waiting for a pulse. */
const set<JobUpdateStatus> AWAITNG_PULSE_JOB_UPDATE_STATES = [JobUpdateStatus.ROLL_FORWARD_AWAITING_PULSE,
JobUpdateStatus.ROLL_BACK_AWAITING_PULSE]
/** Job update actions that can be applied to job instances. */ /** Job update actions that can be applied to job instances. */
enum JobUpdateAction { enum JobUpdateAction {
/** /**
@ -1073,7 +1079,7 @@ service AuroraSchedulerManager extends ReadOnlyScheduler {
Response restartShards(5: JobKey job, 3: set<i32> shardIds) Response restartShards(5: JobKey job, 3: set<i32> shardIds)
/** Initiates a kill on tasks. */ /** Initiates a kill on tasks. */
Response killTasks(4: JobKey job, 5: set<i32> instances) Response killTasks(4: JobKey job, 5: set<i32> instances, 6: string message)
/** /**
* Adds new instances with the TaskConfig of the existing instance pointed by the key. * Adds new instances with the TaskConfig of the existing instance pointed by the key.
@ -1134,31 +1140,6 @@ service AuroraSchedulerManager extends ReadOnlyScheduler {
Response pulseJobUpdate(1: JobUpdateKey key) Response pulseJobUpdate(1: JobUpdateKey key)
} }
struct InstanceConfigRewrite {
/** Key for the task to rewrite. */
1: InstanceKey instanceKey
/** The original configuration. */
2: TaskConfig oldTask
/** The rewritten configuration. */
3: TaskConfig rewrittenTask
}
struct JobConfigRewrite {
/** The original job configuration. */
1: JobConfiguration oldJob
/** The rewritten job configuration. */
2: JobConfiguration rewrittenJob
}
union ConfigRewrite {
1: JobConfigRewrite jobRewrite
2: InstanceConfigRewrite instanceRewrite
}
struct RewriteConfigsRequest {
1: list<ConfigRewrite> rewriteCommands
}
struct ExplicitReconciliationSettings { struct ExplicitReconciliationSettings {
1: optional i32 batchSize 1: optional i32 batchSize
} }
@ -1214,20 +1195,18 @@ service AuroraAdmin extends AuroraSchedulerManager {
/** Start a storage snapshot and block until it completes. */ /** Start a storage snapshot and block until it completes. */
Response snapshot() Response snapshot()
/**
* Forcibly rewrites the stored definition of user configurations. This is intended to be used
* in a controlled setting, primarily to migrate pieces of configurations that are opaque to the
* scheduler (e.g. executorConfig).
* The scheduler may do some validation of the rewritten configurations, but it is important
* that the caller take care to provide valid input and alter only necessary fields.
*/
Response rewriteConfigs(1: RewriteConfigsRequest request)
/** Tell scheduler to trigger an explicit task reconciliation with the given settings. */ /** Tell scheduler to trigger an explicit task reconciliation with the given settings. */
Response triggerExplicitTaskReconciliation(1: ExplicitReconciliationSettings settings) Response triggerExplicitTaskReconciliation(1: ExplicitReconciliationSettings settings)
/** Tell scheduler to trigger an implicit task reconciliation. */ /** Tell scheduler to trigger an implicit task reconciliation. */
Response triggerImplicitTaskReconciliation() Response triggerImplicitTaskReconciliation()
/**
* Force prune any (terminal) tasks that match the query. If no statuses are supplied with the
* query, it will default to all terminal task states. If statuses are supplied, they must be
* terminal states.
*/
Response pruneTasks(1: TaskQuery query)
} }
// The name of the header that should be sent to bypass leader redirection in the Scheduler. // The name of the header that should be sent to bypass leader redirection in the Scheduler.
@ -1235,4 +1214,4 @@ const string BYPASS_LEADER_REDIRECT_HEADER_NAME = 'Bypass-Leader-Redirect'
// The path under which a task's filesystem should be mounted when using images and the Mesos // The path under which a task's filesystem should be mounted when using images and the Mesos
// unified containerizer. // unified containerizer.
const string TASK_FILESYSTEM_MOUNT_POINT = 'taskfs' const string TASK_FILESYSTEM_MOUNT_POINT = 'taskfs'

642
gen-go/apache/aurora/aurora_admin-remote/aurora_admin-remote.go
View file

@ -34,15 +34,15 @@ func Usage() {
fmt.Fprintln(os.Stderr, " Response maintenanceStatus(Hosts hosts)") fmt.Fprintln(os.Stderr, " Response maintenanceStatus(Hosts hosts)")
fmt.Fprintln(os.Stderr, " Response endMaintenance(Hosts hosts)") fmt.Fprintln(os.Stderr, " Response endMaintenance(Hosts hosts)")
fmt.Fprintln(os.Stderr, " Response snapshot()") fmt.Fprintln(os.Stderr, " Response snapshot()")
fmt.Fprintln(os.Stderr, " Response rewriteConfigs(RewriteConfigsRequest request)")
fmt.Fprintln(os.Stderr, " Response triggerExplicitTaskReconciliation(ExplicitReconciliationSettings settings)") fmt.Fprintln(os.Stderr, " Response triggerExplicitTaskReconciliation(ExplicitReconciliationSettings settings)")
fmt.Fprintln(os.Stderr, " Response triggerImplicitTaskReconciliation()") fmt.Fprintln(os.Stderr, " Response triggerImplicitTaskReconciliation()")
fmt.Fprintln(os.Stderr, " Response pruneTasks(TaskQuery query)")
fmt.Fprintln(os.Stderr, " Response createJob(JobConfiguration description)") fmt.Fprintln(os.Stderr, " Response createJob(JobConfiguration description)")
fmt.Fprintln(os.Stderr, " Response scheduleCronJob(JobConfiguration description)") fmt.Fprintln(os.Stderr, " Response scheduleCronJob(JobConfiguration description)")
fmt.Fprintln(os.Stderr, " Response descheduleCronJob(JobKey job)") fmt.Fprintln(os.Stderr, " Response descheduleCronJob(JobKey job)")
fmt.Fprintln(os.Stderr, " Response startCronJob(JobKey job)") fmt.Fprintln(os.Stderr, " Response startCronJob(JobKey job)")
fmt.Fprintln(os.Stderr, " Response restartShards(JobKey job, shardIds)") fmt.Fprintln(os.Stderr, " Response restartShards(JobKey job, shardIds)")
fmt.Fprintln(os.Stderr, " Response killTasks(JobKey job, instances)") fmt.Fprintln(os.Stderr, " Response killTasks(JobKey job, instances, string message)")
fmt.Fprintln(os.Stderr, " Response addInstances(InstanceKey key, i32 count)") fmt.Fprintln(os.Stderr, " Response addInstances(InstanceKey key, i32 count)")
fmt.Fprintln(os.Stderr, " Response replaceCronTemplate(JobConfiguration config)") fmt.Fprintln(os.Stderr, " Response replaceCronTemplate(JobConfiguration config)")
fmt.Fprintln(os.Stderr, " Response startJobUpdate(JobUpdateRequest request, string message)") fmt.Fprintln(os.Stderr, " Response startJobUpdate(JobUpdateRequest request, string message)")
@ -165,19 +165,19 @@ func main() {
} }
argvalue0 := flag.Arg(1) argvalue0 := flag.Arg(1)
value0 := argvalue0 value0 := argvalue0
arg352 := flag.Arg(2) arg353 := flag.Arg(2)
mbTrans353 := thrift.NewTMemoryBufferLen(len(arg352)) mbTrans354 := thrift.NewTMemoryBufferLen(len(arg353))
defer mbTrans353.Close() defer mbTrans354.Close()
_, err354 := mbTrans353.WriteString(arg352) _, err355 := mbTrans354.WriteString(arg353)
if err354 != nil { if err355 != nil {
Usage() Usage()
return return
} }
factory355 := thrift.NewTSimpleJSONProtocolFactory() factory356 := thrift.NewTSimpleJSONProtocolFactory()
jsProt356 := factory355.GetProtocol(mbTrans353) jsProt357 := factory356.GetProtocol(mbTrans354)
argvalue1 := aurora.NewResourceAggregate() argvalue1 := aurora.NewResourceAggregate()
err357 := argvalue1.Read(jsProt356) err358 := argvalue1.Read(jsProt357)
if err357 != nil { if err358 != nil {
Usage() Usage()
return return
} }
@ -233,19 +233,19 @@ func main() {
fmt.Fprintln(os.Stderr, "QueryRecovery requires 1 args") fmt.Fprintln(os.Stderr, "QueryRecovery requires 1 args")
flag.Usage() flag.Usage()
} }
arg360 := flag.Arg(1) arg361 := flag.Arg(1)
mbTrans361 := thrift.NewTMemoryBufferLen(len(arg360)) mbTrans362 := thrift.NewTMemoryBufferLen(len(arg361))
defer mbTrans361.Close() defer mbTrans362.Close()
_, err362 := mbTrans361.WriteString(arg360) _, err363 := mbTrans362.WriteString(arg361)
if err362 != nil { if err363 != nil {
Usage() Usage()
return return
} }
factory363 := thrift.NewTSimpleJSONProtocolFactory() factory364 := thrift.NewTSimpleJSONProtocolFactory()
jsProt364 := factory363.GetProtocol(mbTrans361) jsProt365 := factory364.GetProtocol(mbTrans362)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err365 := argvalue0.Read(jsProt364) err366 := argvalue0.Read(jsProt365)
if err365 != nil { if err366 != nil {
Usage() Usage()
return return
} }
@ -258,19 +258,19 @@ func main() {
fmt.Fprintln(os.Stderr, "DeleteRecoveryTasks requires 1 args") fmt.Fprintln(os.Stderr, "DeleteRecoveryTasks requires 1 args")
flag.Usage() flag.Usage()
} }
arg366 := flag.Arg(1) arg367 := flag.Arg(1)
mbTrans367 := thrift.NewTMemoryBufferLen(len(arg366)) mbTrans368 := thrift.NewTMemoryBufferLen(len(arg367))
defer mbTrans367.Close() defer mbTrans368.Close()
_, err368 := mbTrans367.WriteString(arg366) _, err369 := mbTrans368.WriteString(arg367)
if err368 != nil { if err369 != nil {
Usage() Usage()
return return
} }
factory369 := thrift.NewTSimpleJSONProtocolFactory() factory370 := thrift.NewTSimpleJSONProtocolFactory()
jsProt370 := factory369.GetProtocol(mbTrans367) jsProt371 := factory370.GetProtocol(mbTrans368)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err371 := argvalue0.Read(jsProt370) err372 := argvalue0.Read(jsProt371)
if err371 != nil { if err372 != nil {
Usage() Usage()
return return
} }
@ -299,19 +299,19 @@ func main() {
fmt.Fprintln(os.Stderr, "StartMaintenance requires 1 args") fmt.Fprintln(os.Stderr, "StartMaintenance requires 1 args")
flag.Usage() flag.Usage()
} }
arg372 := flag.Arg(1) arg373 := flag.Arg(1)
mbTrans373 := thrift.NewTMemoryBufferLen(len(arg372)) mbTrans374 := thrift.NewTMemoryBufferLen(len(arg373))
defer mbTrans373.Close() defer mbTrans374.Close()
_, err374 := mbTrans373.WriteString(arg372) _, err375 := mbTrans374.WriteString(arg373)
if err374 != nil { if err375 != nil {
Usage() Usage()
return return
} }
factory375 := thrift.NewTSimpleJSONProtocolFactory() factory376 := thrift.NewTSimpleJSONProtocolFactory()
jsProt376 := factory375.GetProtocol(mbTrans373) jsProt377 := factory376.GetProtocol(mbTrans374)
argvalue0 := aurora.NewHosts() argvalue0 := aurora.NewHosts()
err377 := argvalue0.Read(jsProt376) err378 := argvalue0.Read(jsProt377)
if err377 != nil { if err378 != nil {
Usage() Usage()
return return
} }
@ -324,19 +324,19 @@ func main() {
fmt.Fprintln(os.Stderr, "DrainHosts requires 1 args") fmt.Fprintln(os.Stderr, "DrainHosts requires 1 args")
flag.Usage() flag.Usage()
} }
arg378 := flag.Arg(1) arg379 := flag.Arg(1)
mbTrans379 := thrift.NewTMemoryBufferLen(len(arg378)) mbTrans380 := thrift.NewTMemoryBufferLen(len(arg379))
defer mbTrans379.Close() defer mbTrans380.Close()
_, err380 := mbTrans379.WriteString(arg378) _, err381 := mbTrans380.WriteString(arg379)
if err380 != nil { if err381 != nil {
Usage() Usage()
return return
} }
factory381 := thrift.NewTSimpleJSONProtocolFactory() factory382 := thrift.NewTSimpleJSONProtocolFactory()
jsProt382 := factory381.GetProtocol(mbTrans379) jsProt383 := factory382.GetProtocol(mbTrans380)
argvalue0 := aurora.NewHosts() argvalue0 := aurora.NewHosts()
err383 := argvalue0.Read(jsProt382) err384 := argvalue0.Read(jsProt383)
if err383 != nil { if err384 != nil {
Usage() Usage()
return return
} }
@ -349,19 +349,19 @@ func main() {
fmt.Fprintln(os.Stderr, "MaintenanceStatus requires 1 args") fmt.Fprintln(os.Stderr, "MaintenanceStatus requires 1 args")
flag.Usage() flag.Usage()
} }
arg384 := flag.Arg(1) arg385 := flag.Arg(1)
mbTrans385 := thrift.NewTMemoryBufferLen(len(arg384)) mbTrans386 := thrift.NewTMemoryBufferLen(len(arg385))
defer mbTrans385.Close() defer mbTrans386.Close()
_, err386 := mbTrans385.WriteString(arg384) _, err387 := mbTrans386.WriteString(arg385)
if err386 != nil { if err387 != nil {
Usage() Usage()
return return
} }
factory387 := thrift.NewTSimpleJSONProtocolFactory() factory388 := thrift.NewTSimpleJSONProtocolFactory()
jsProt388 := factory387.GetProtocol(mbTrans385) jsProt389 := factory388.GetProtocol(mbTrans386)
argvalue0 := aurora.NewHosts() argvalue0 := aurora.NewHosts()
err389 := argvalue0.Read(jsProt388) err390 := argvalue0.Read(jsProt389)
if err389 != nil { if err390 != nil {
Usage() Usage()
return return
} }
@ -374,19 +374,19 @@ func main() {
fmt.Fprintln(os.Stderr, "EndMaintenance requires 1 args") fmt.Fprintln(os.Stderr, "EndMaintenance requires 1 args")
flag.Usage() flag.Usage()
} }
arg390 := flag.Arg(1) arg391 := flag.Arg(1)
mbTrans391 := thrift.NewTMemoryBufferLen(len(arg390)) mbTrans392 := thrift.NewTMemoryBufferLen(len(arg391))
defer mbTrans391.Close() defer mbTrans392.Close()
_, err392 := mbTrans391.WriteString(arg390) _, err393 := mbTrans392.WriteString(arg391)
if err392 != nil { if err393 != nil {
Usage() Usage()
return return
} }
factory393 := thrift.NewTSimpleJSONProtocolFactory() factory394 := thrift.NewTSimpleJSONProtocolFactory()
jsProt394 := factory393.GetProtocol(mbTrans391) jsProt395 := factory394.GetProtocol(mbTrans392)
argvalue0 := aurora.NewHosts() argvalue0 := aurora.NewHosts()
err395 := argvalue0.Read(jsProt394) err396 := argvalue0.Read(jsProt395)
if err395 != nil { if err396 != nil {
Usage() Usage()
return return
} }
@ -402,49 +402,24 @@ func main() {
fmt.Print(client.Snapshot()) fmt.Print(client.Snapshot())
fmt.Print("\n") fmt.Print("\n")
break break
case "rewriteConfigs":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "RewriteConfigs requires 1 args")
flag.Usage()
}
arg396 := flag.Arg(1)
mbTrans397 := thrift.NewTMemoryBufferLen(len(arg396))
defer mbTrans397.Close()
_, err398 := mbTrans397.WriteString(arg396)
if err398 != nil {
Usage()
return
}
factory399 := thrift.NewTSimpleJSONProtocolFactory()
jsProt400 := factory399.GetProtocol(mbTrans397)
argvalue0 := aurora.NewRewriteConfigsRequest()
err401 := argvalue0.Read(jsProt400)
if err401 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.RewriteConfigs(value0))
fmt.Print("\n")
break
case "triggerExplicitTaskReconciliation": case "triggerExplicitTaskReconciliation":
if flag.NArg()-1 != 1 { if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TriggerExplicitTaskReconciliation requires 1 args") fmt.Fprintln(os.Stderr, "TriggerExplicitTaskReconciliation requires 1 args")
flag.Usage() flag.Usage()
} }
arg402 := flag.Arg(1) arg397 := flag.Arg(1)
mbTrans403 := thrift.NewTMemoryBufferLen(len(arg402)) mbTrans398 := thrift.NewTMemoryBufferLen(len(arg397))
defer mbTrans403.Close() defer mbTrans398.Close()
_, err404 := mbTrans403.WriteString(arg402) _, err399 := mbTrans398.WriteString(arg397)
if err404 != nil { if err399 != nil {
Usage() Usage()
return return
} }
factory405 := thrift.NewTSimpleJSONProtocolFactory() factory400 := thrift.NewTSimpleJSONProtocolFactory()
jsProt406 := factory405.GetProtocol(mbTrans403) jsProt401 := factory400.GetProtocol(mbTrans398)
argvalue0 := aurora.NewExplicitReconciliationSettings() argvalue0 := aurora.NewExplicitReconciliationSettings()
err407 := argvalue0.Read(jsProt406) err402 := argvalue0.Read(jsProt401)
if err407 != nil { if err402 != nil {
Usage() Usage()
return return
} }
@ -460,24 +435,49 @@ func main() {
fmt.Print(client.TriggerImplicitTaskReconciliation()) fmt.Print(client.TriggerImplicitTaskReconciliation())
fmt.Print("\n") fmt.Print("\n")
break break
case "pruneTasks":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "PruneTasks requires 1 args")
flag.Usage()
}
arg403 := flag.Arg(1)
mbTrans404 := thrift.NewTMemoryBufferLen(len(arg403))
defer mbTrans404.Close()
_, err405 := mbTrans404.WriteString(arg403)
if err405 != nil {
Usage()
return
}
factory406 := thrift.NewTSimpleJSONProtocolFactory()
jsProt407 := factory406.GetProtocol(mbTrans404)
argvalue0 := aurora.NewTaskQuery()
err408 := argvalue0.Read(jsProt407)
if err408 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.PruneTasks(value0))
fmt.Print("\n")
break
case "createJob": case "createJob":
if flag.NArg()-1 != 1 { if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "CreateJob requires 1 args") fmt.Fprintln(os.Stderr, "CreateJob requires 1 args")
flag.Usage() flag.Usage()
} }
arg408 := flag.Arg(1) arg409 := flag.Arg(1)
mbTrans409 := thrift.NewTMemoryBufferLen(len(arg408)) mbTrans410 := thrift.NewTMemoryBufferLen(len(arg409))
defer mbTrans409.Close() defer mbTrans410.Close()
_, err410 := mbTrans409.WriteString(arg408) _, err411 := mbTrans410.WriteString(arg409)
if err410 != nil { if err411 != nil {
Usage() Usage()
return return
} }
factory411 := thrift.NewTSimpleJSONProtocolFactory() factory412 := thrift.NewTSimpleJSONProtocolFactory()
jsProt412 := factory411.GetProtocol(mbTrans409) jsProt413 := factory412.GetProtocol(mbTrans410)
argvalue0 := aurora.NewJobConfiguration() argvalue0 := aurora.NewJobConfiguration()
err413 := argvalue0.Read(jsProt412) err414 := argvalue0.Read(jsProt413)
if err413 != nil { if err414 != nil {
Usage() Usage()
return return
} }
@ -490,19 +490,19 @@ func main() {
fmt.Fprintln(os.Stderr, "ScheduleCronJob requires 1 args") fmt.Fprintln(os.Stderr, "ScheduleCronJob requires 1 args")
flag.Usage() flag.Usage()
} }
arg414 := flag.Arg(1) arg415 := flag.Arg(1)
mbTrans415 := thrift.NewTMemoryBufferLen(len(arg414)) mbTrans416 := thrift.NewTMemoryBufferLen(len(arg415))
defer mbTrans415.Close() defer mbTrans416.Close()
_, err416 := mbTrans415.WriteString(arg414) _, err417 := mbTrans416.WriteString(arg415)
if err416 != nil { if err417 != nil {
Usage() Usage()
return return
} }
factory417 := thrift.NewTSimpleJSONProtocolFactory() factory418 := thrift.NewTSimpleJSONProtocolFactory()
jsProt418 := factory417.GetProtocol(mbTrans415) jsProt419 := factory418.GetProtocol(mbTrans416)
argvalue0 := aurora.NewJobConfiguration() argvalue0 := aurora.NewJobConfiguration()
err419 := argvalue0.Read(jsProt418) err420 := argvalue0.Read(jsProt419)
if err419 != nil { if err420 != nil {
Usage() Usage()
return return
} }
@ -515,19 +515,19 @@ func main() {
fmt.Fprintln(os.Stderr, "DescheduleCronJob requires 1 args") fmt.Fprintln(os.Stderr, "DescheduleCronJob requires 1 args")
flag.Usage() flag.Usage()
} }
arg420 := flag.Arg(1) arg421 := flag.Arg(1)
mbTrans421 := thrift.NewTMemoryBufferLen(len(arg420)) mbTrans422 := thrift.NewTMemoryBufferLen(len(arg421))
defer mbTrans421.Close() defer mbTrans422.Close()
_, err422 := mbTrans421.WriteString(arg420) _, err423 := mbTrans422.WriteString(arg421)
if err422 != nil { if err423 != nil {
Usage() Usage()
return return
} }
factory423 := thrift.NewTSimpleJSONProtocolFactory() factory424 := thrift.NewTSimpleJSONProtocolFactory()
jsProt424 := factory423.GetProtocol(mbTrans421) jsProt425 := factory424.GetProtocol(mbTrans422)
argvalue0 := aurora.NewJobKey() argvalue0 := aurora.NewJobKey()
err425 := argvalue0.Read(jsProt424) err426 := argvalue0.Read(jsProt425)
if err425 != nil { if err426 != nil {
Usage() Usage()
return return
} }
@ -540,19 +540,19 @@ func main() {
fmt.Fprintln(os.Stderr, "StartCronJob requires 1 args") fmt.Fprintln(os.Stderr, "StartCronJob requires 1 args")
flag.Usage() flag.Usage()
} }
arg426 := flag.Arg(1) arg427 := flag.Arg(1)
mbTrans427 := thrift.NewTMemoryBufferLen(len(arg426)) mbTrans428 := thrift.NewTMemoryBufferLen(len(arg427))
defer mbTrans427.Close() defer mbTrans428.Close()
_, err428 := mbTrans427.WriteString(arg426) _, err429 := mbTrans428.WriteString(arg427)
if err428 != nil { if err429 != nil {
Usage() Usage()
return return
} }
factory429 := thrift.NewTSimpleJSONProtocolFactory() factory430 := thrift.NewTSimpleJSONProtocolFactory()
jsProt430 := factory429.GetProtocol(mbTrans427) jsProt431 := factory430.GetProtocol(mbTrans428)
argvalue0 := aurora.NewJobKey() argvalue0 := aurora.NewJobKey()
err431 := argvalue0.Read(jsProt430) err432 := argvalue0.Read(jsProt431)
if err431 != nil { if err432 != nil {
Usage() Usage()
return return
} }
@ -565,36 +565,36 @@ func main() {
fmt.Fprintln(os.Stderr, "RestartShards requires 2 args") fmt.Fprintln(os.Stderr, "RestartShards requires 2 args")
flag.Usage() flag.Usage()
} }
arg432 := flag.Arg(1) arg433 := flag.Arg(1)
mbTrans433 := thrift.NewTMemoryBufferLen(len(arg432)) mbTrans434 := thrift.NewTMemoryBufferLen(len(arg433))
defer mbTrans433.Close() defer mbTrans434.Close()
_, err434 := mbTrans433.WriteString(arg432) _, err435 := mbTrans434.WriteString(arg433)
if err434 != nil { if err435 != nil {
Usage() Usage()
return return
} }
factory435 := thrift.NewTSimpleJSONProtocolFactory() factory436 := thrift.NewTSimpleJSONProtocolFactory()
jsProt436 := factory435.GetProtocol(mbTrans433) jsProt437 := factory436.GetProtocol(mbTrans434)
argvalue0 := aurora.NewJobKey() argvalue0 := aurora.NewJobKey()
err437 := argvalue0.Read(jsProt436) err438 := argvalue0.Read(jsProt437)
if err437 != nil { if err438 != nil {
Usage() Usage()
return return
} }
value0 := argvalue0 value0 := argvalue0
arg438 := flag.Arg(2) arg439 := flag.Arg(2)
mbTrans439 := thrift.NewTMemoryBufferLen(len(arg438)) mbTrans440 := thrift.NewTMemoryBufferLen(len(arg439))
defer mbTrans439.Close() defer mbTrans440.Close()
_, err440 := mbTrans439.WriteString(arg438) _, err441 := mbTrans440.WriteString(arg439)
if err440 != nil { if err441 != nil {
Usage() Usage()
return return
} }
factory441 := thrift.NewTSimpleJSONProtocolFactory() factory442 := thrift.NewTSimpleJSONProtocolFactory()
jsProt442 := factory441.GetProtocol(mbTrans439) jsProt443 := factory442.GetProtocol(mbTrans440)
containerStruct1 := aurora.NewAuroraAdminRestartShardsArgs() containerStruct1 := aurora.NewAuroraAdminRestartShardsArgs()
err443 := containerStruct1.ReadField2(jsProt442) err444 := containerStruct1.ReadField2(jsProt443)
if err443 != nil { if err444 != nil {
Usage() Usage()
return return
} }
@ -604,46 +604,48 @@ func main() {
fmt.Print("\n") fmt.Print("\n")
break break
case "killTasks": case "killTasks":
if flag.NArg()-1 != 2 { if flag.NArg()-1 != 3 {
fmt.Fprintln(os.Stderr, "KillTasks requires 2 args") fmt.Fprintln(os.Stderr, "KillTasks requires 3 args")
flag.Usage() flag.Usage()
} }
arg444 := flag.Arg(1) arg445 := flag.Arg(1)
mbTrans445 := thrift.NewTMemoryBufferLen(len(arg444)) mbTrans446 := thrift.NewTMemoryBufferLen(len(arg445))
defer mbTrans445.Close() defer mbTrans446.Close()
_, err446 := mbTrans445.WriteString(arg444) _, err447 := mbTrans446.WriteString(arg445)
if err446 != nil { if err447 != nil {
Usage() Usage()
return return
} }
factory447 := thrift.NewTSimpleJSONProtocolFactory() factory448 := thrift.NewTSimpleJSONProtocolFactory()
jsProt448 := factory447.GetProtocol(mbTrans445) jsProt449 := factory448.GetProtocol(mbTrans446)
argvalue0 := aurora.NewJobKey() argvalue0 := aurora.NewJobKey()
err449 := argvalue0.Read(jsProt448) err450 := argvalue0.Read(jsProt449)
if err449 != nil { if err450 != nil {
Usage() Usage()
return return
} }
value0 := argvalue0 value0 := argvalue0
arg450 := flag.Arg(2) arg451 := flag.Arg(2)
mbTrans451 := thrift.NewTMemoryBufferLen(len(arg450)) mbTrans452 := thrift.NewTMemoryBufferLen(len(arg451))
defer mbTrans451.Close() defer mbTrans452.Close()
_, err452 := mbTrans451.WriteString(arg450) _, err453 := mbTrans452.WriteString(arg451)
if err452 != nil { if err453 != nil {
Usage() Usage()
return return
} }
factory453 := thrift.NewTSimpleJSONProtocolFactory() factory454 := thrift.NewTSimpleJSONProtocolFactory()
jsProt454 := factory453.GetProtocol(mbTrans451) jsProt455 := factory454.GetProtocol(mbTrans452)
containerStruct1 := aurora.NewAuroraAdminKillTasksArgs() containerStruct1 := aurora.NewAuroraAdminKillTasksArgs()
err455 := containerStruct1.ReadField2(jsProt454) err456 := containerStruct1.ReadField2(jsProt455)
if err455 != nil { if err456 != nil {
Usage() Usage()
return return
} }
argvalue1 := containerStruct1.Instances argvalue1 := containerStruct1.Instances
value1 := argvalue1 value1 := argvalue1
fmt.Print(client.KillTasks(value0, value1)) argvalue2 := flag.Arg(3)
value2 := argvalue2
fmt.Print(client.KillTasks(value0, value1, value2))
fmt.Print("\n") fmt.Print("\n")
break break
case "addInstances": case "addInstances":
@ -651,25 +653,25 @@ func main() {
fmt.Fprintln(os.Stderr, "AddInstances requires 2 args") fmt.Fprintln(os.Stderr, "AddInstances requires 2 args")
flag.Usage() flag.Usage()
} }
arg456 := flag.Arg(1) arg458 := flag.Arg(1)
mbTrans457 := thrift.NewTMemoryBufferLen(len(arg456)) mbTrans459 := thrift.NewTMemoryBufferLen(len(arg458))
defer mbTrans457.Close() defer mbTrans459.Close()
_, err458 := mbTrans457.WriteString(arg456) _, err460 := mbTrans459.WriteString(arg458)
if err458 != nil { if err460 != nil {
Usage() Usage()
return return
} }
factory459 := thrift.NewTSimpleJSONProtocolFactory() factory461 := thrift.NewTSimpleJSONProtocolFactory()
jsProt460 := factory459.GetProtocol(mbTrans457) jsProt462 := factory461.GetProtocol(mbTrans459)
argvalue0 := aurora.NewInstanceKey() argvalue0 := aurora.NewInstanceKey()
err461 := argvalue0.Read(jsProt460) err463 := argvalue0.Read(jsProt462)
if err461 != nil { if err463 != nil {
Usage() Usage()
return return
} }
value0 := argvalue0 value0 := argvalue0
tmp1, err462 := (strconv.Atoi(flag.Arg(2))) tmp1, err464 := (strconv.Atoi(flag.Arg(2)))
if err462 != nil { if err464 != nil {
Usage() Usage()
return return
} }
@ -683,19 +685,19 @@ func main() {
fmt.Fprintln(os.Stderr, "ReplaceCronTemplate requires 1 args") fmt.Fprintln(os.Stderr, "ReplaceCronTemplate requires 1 args")
flag.Usage() flag.Usage()
} }
arg463 := flag.Arg(1) arg465 := flag.Arg(1)
mbTrans464 := thrift.NewTMemoryBufferLen(len(arg463)) mbTrans466 := thrift.NewTMemoryBufferLen(len(arg465))
defer mbTrans464.Close() defer mbTrans466.Close()
_, err465 := mbTrans464.WriteString(arg463) _, err467 := mbTrans466.WriteString(arg465)
if err465 != nil { if err467 != nil {
Usage() Usage()
return return
} }
factory466 := thrift.NewTSimpleJSONProtocolFactory() factory468 := thrift.NewTSimpleJSONProtocolFactory()
jsProt467 := factory466.GetProtocol(mbTrans464) jsProt469 := factory468.GetProtocol(mbTrans466)
argvalue0 := aurora.NewJobConfiguration() argvalue0 := aurora.NewJobConfiguration()
err468 := argvalue0.Read(jsProt467) err470 := argvalue0.Read(jsProt469)
if err468 != nil { if err470 != nil {
Usage() Usage()
return return
} }
@ -708,19 +710,19 @@ func main() {
fmt.Fprintln(os.Stderr, "StartJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "StartJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg469 := flag.Arg(1) arg471 := flag.Arg(1)
mbTrans470 := thrift.NewTMemoryBufferLen(len(arg469)) mbTrans472 := thrift.NewTMemoryBufferLen(len(arg471))
defer mbTrans470.Close() defer mbTrans472.Close()
_, err471 := mbTrans470.WriteString(arg469) _, err473 := mbTrans472.WriteString(arg471)
if err471 != nil { if err473 != nil {
Usage() Usage()
return return
} }
factory472 := thrift.NewTSimpleJSONProtocolFactory() factory474 := thrift.NewTSimpleJSONProtocolFactory()
jsProt473 := factory472.GetProtocol(mbTrans470) jsProt475 := factory474.GetProtocol(mbTrans472)
argvalue0 := aurora.NewJobUpdateRequest() argvalue0 := aurora.NewJobUpdateRequest()
err474 := argvalue0.Read(jsProt473) err476 := argvalue0.Read(jsProt475)
if err474 != nil { if err476 != nil {
Usage() Usage()
return return
} }
@ -735,19 +737,19 @@ func main() {
fmt.Fprintln(os.Stderr, "PauseJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "PauseJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg476 := flag.Arg(1) arg478 := flag.Arg(1)
mbTrans477 := thrift.NewTMemoryBufferLen(len(arg476)) mbTrans479 := thrift.NewTMemoryBufferLen(len(arg478))
defer mbTrans477.Close() defer mbTrans479.Close()
_, err478 := mbTrans477.WriteString(arg476) _, err480 := mbTrans479.WriteString(arg478)
if err478 != nil { if err480 != nil {
Usage() Usage()
return return
} }
factory479 := thrift.NewTSimpleJSONProtocolFactory() factory481 := thrift.NewTSimpleJSONProtocolFactory()
jsProt480 := factory479.GetProtocol(mbTrans477) jsProt482 := factory481.GetProtocol(mbTrans479)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err481 := argvalue0.Read(jsProt480) err483 := argvalue0.Read(jsProt482)
if err481 != nil { if err483 != nil {
Usage() Usage()
return return
} }
@ -762,19 +764,19 @@ func main() {
fmt.Fprintln(os.Stderr, "ResumeJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "ResumeJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg483 := flag.Arg(1) arg485 := flag.Arg(1)
mbTrans484 := thrift.NewTMemoryBufferLen(len(arg483)) mbTrans486 := thrift.NewTMemoryBufferLen(len(arg485))
defer mbTrans484.Close() defer mbTrans486.Close()
_, err485 := mbTrans484.WriteString(arg483) _, err487 := mbTrans486.WriteString(arg485)
if err485 != nil { if err487 != nil {
Usage() Usage()
return return
} }
factory486 := thrift.NewTSimpleJSONProtocolFactory() factory488 := thrift.NewTSimpleJSONProtocolFactory()
jsProt487 := factory486.GetProtocol(mbTrans484) jsProt489 := factory488.GetProtocol(mbTrans486)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err488 := argvalue0.Read(jsProt487) err490 := argvalue0.Read(jsProt489)
if err488 != nil { if err490 != nil {
Usage() Usage()
return return
} }
@ -789,19 +791,19 @@ func main() {
fmt.Fprintln(os.Stderr, "AbortJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "AbortJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg490 := flag.Arg(1) arg492 := flag.Arg(1)
mbTrans491 := thrift.NewTMemoryBufferLen(len(arg490)) mbTrans493 := thrift.NewTMemoryBufferLen(len(arg492))
defer mbTrans491.Close() defer mbTrans493.Close()
_, err492 := mbTrans491.WriteString(arg490) _, err494 := mbTrans493.WriteString(arg492)
if err492 != nil { if err494 != nil {
Usage() Usage()
return return
} }
factory493 := thrift.NewTSimpleJSONProtocolFactory() factory495 := thrift.NewTSimpleJSONProtocolFactory()
jsProt494 := factory493.GetProtocol(mbTrans491) jsProt496 := factory495.GetProtocol(mbTrans493)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err495 := argvalue0.Read(jsProt494) err497 := argvalue0.Read(jsProt496)
if err495 != nil { if err497 != nil {
Usage() Usage()
return return
} }
@ -816,19 +818,19 @@ func main() {
fmt.Fprintln(os.Stderr, "RollbackJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "RollbackJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg497 := flag.Arg(1) arg499 := flag.Arg(1)
mbTrans498 := thrift.NewTMemoryBufferLen(len(arg497)) mbTrans500 := thrift.NewTMemoryBufferLen(len(arg499))
defer mbTrans498.Close() defer mbTrans500.Close()
_, err499 := mbTrans498.WriteString(arg497) _, err501 := mbTrans500.WriteString(arg499)
if err499 != nil { if err501 != nil {
Usage() Usage()
return return
} }
factory500 := thrift.NewTSimpleJSONProtocolFactory() factory502 := thrift.NewTSimpleJSONProtocolFactory()
jsProt501 := factory500.GetProtocol(mbTrans498) jsProt503 := factory502.GetProtocol(mbTrans500)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err502 := argvalue0.Read(jsProt501) err504 := argvalue0.Read(jsProt503)
if err502 != nil { if err504 != nil {
Usage() Usage()
return return
} }
@ -843,19 +845,19 @@ func main() {
fmt.Fprintln(os.Stderr, "PulseJobUpdate requires 1 args") fmt.Fprintln(os.Stderr, "PulseJobUpdate requires 1 args")
flag.Usage() flag.Usage()
} }
arg504 := flag.Arg(1) arg506 := flag.Arg(1)
mbTrans505 := thrift.NewTMemoryBufferLen(len(arg504)) mbTrans507 := thrift.NewTMemoryBufferLen(len(arg506))
defer mbTrans505.Close() defer mbTrans507.Close()
_, err506 := mbTrans505.WriteString(arg504) _, err508 := mbTrans507.WriteString(arg506)
if err506 != nil { if err508 != nil {
Usage() Usage()
return return
} }
factory507 := thrift.NewTSimpleJSONProtocolFactory() factory509 := thrift.NewTSimpleJSONProtocolFactory()
jsProt508 := factory507.GetProtocol(mbTrans505) jsProt510 := factory509.GetProtocol(mbTrans507)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err509 := argvalue0.Read(jsProt508) err511 := argvalue0.Read(jsProt510)
if err509 != nil { if err511 != nil {
Usage() Usage()
return return
} }
@ -886,19 +888,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetTasksStatus requires 1 args") fmt.Fprintln(os.Stderr, "GetTasksStatus requires 1 args")
flag.Usage() flag.Usage()
} }
arg511 := flag.Arg(1) arg513 := flag.Arg(1)
mbTrans512 := thrift.NewTMemoryBufferLen(len(arg511)) mbTrans514 := thrift.NewTMemoryBufferLen(len(arg513))
defer mbTrans512.Close() defer mbTrans514.Close()
_, err513 := mbTrans512.WriteString(arg511) _, err515 := mbTrans514.WriteString(arg513)
if err513 != nil { if err515 != nil {
Usage() Usage()
return return
} }
factory514 := thrift.NewTSimpleJSONProtocolFactory() factory516 := thrift.NewTSimpleJSONProtocolFactory()
jsProt515 := factory514.GetProtocol(mbTrans512) jsProt517 := factory516.GetProtocol(mbTrans514)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err516 := argvalue0.Read(jsProt515) err518 := argvalue0.Read(jsProt517)
if err516 != nil { if err518 != nil {
Usage() Usage()
return return
} }
@ -911,19 +913,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetTasksWithoutConfigs requires 1 args") fmt.Fprintln(os.Stderr, "GetTasksWithoutConfigs requires 1 args")
flag.Usage() flag.Usage()
} }
arg517 := flag.Arg(1) arg519 := flag.Arg(1)
mbTrans518 := thrift.NewTMemoryBufferLen(len(arg517)) mbTrans520 := thrift.NewTMemoryBufferLen(len(arg519))
defer mbTrans518.Close() defer mbTrans520.Close()
_, err519 := mbTrans518.WriteString(arg517) _, err521 := mbTrans520.WriteString(arg519)
if err519 != nil { if err521 != nil {
Usage() Usage()
return return
} }
factory520 := thrift.NewTSimpleJSONProtocolFactory() factory522 := thrift.NewTSimpleJSONProtocolFactory()
jsProt521 := factory520.GetProtocol(mbTrans518) jsProt523 := factory522.GetProtocol(mbTrans520)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err522 := argvalue0.Read(jsProt521) err524 := argvalue0.Read(jsProt523)
if err522 != nil { if err524 != nil {
Usage() Usage()
return return
} }
@ -936,19 +938,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetPendingReason requires 1 args") fmt.Fprintln(os.Stderr, "GetPendingReason requires 1 args")
flag.Usage() flag.Usage()
} }
arg523 := flag.Arg(1) arg525 := flag.Arg(1)
mbTrans524 := thrift.NewTMemoryBufferLen(len(arg523)) mbTrans526 := thrift.NewTMemoryBufferLen(len(arg525))
defer mbTrans524.Close() defer mbTrans526.Close()
_, err525 := mbTrans524.WriteString(arg523) _, err527 := mbTrans526.WriteString(arg525)
if err525 != nil { if err527 != nil {
Usage() Usage()
return return
} }
factory526 := thrift.NewTSimpleJSONProtocolFactory() factory528 := thrift.NewTSimpleJSONProtocolFactory()
jsProt527 := factory526.GetProtocol(mbTrans524) jsProt529 := factory528.GetProtocol(mbTrans526)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err528 := argvalue0.Read(jsProt527) err530 := argvalue0.Read(jsProt529)
if err528 != nil { if err530 != nil {
Usage() Usage()
return return
} }
@ -961,19 +963,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetConfigSummary requires 1 args") fmt.Fprintln(os.Stderr, "GetConfigSummary requires 1 args")
flag.Usage() flag.Usage()
} }
arg529 := flag.Arg(1) arg531 := flag.Arg(1)
mbTrans530 := thrift.NewTMemoryBufferLen(len(arg529)) mbTrans532 := thrift.NewTMemoryBufferLen(len(arg531))
defer mbTrans530.Close() defer mbTrans532.Close()
_, err531 := mbTrans530.WriteString(arg529) _, err533 := mbTrans532.WriteString(arg531)
if err531 != nil { if err533 != nil {
Usage() Usage()
return return
} }
factory532 := thrift.NewTSimpleJSONProtocolFactory() factory534 := thrift.NewTSimpleJSONProtocolFactory()
jsProt533 := factory532.GetProtocol(mbTrans530) jsProt535 := factory534.GetProtocol(mbTrans532)
argvalue0 := aurora.NewJobKey() argvalue0 := aurora.NewJobKey()
err534 := argvalue0.Read(jsProt533) err536 := argvalue0.Read(jsProt535)
if err534 != nil { if err536 != nil {
Usage() Usage()
return return
} }
@ -1006,19 +1008,19 @@ func main() {
fmt.Fprintln(os.Stderr, "PopulateJobConfig requires 1 args") fmt.Fprintln(os.Stderr, "PopulateJobConfig requires 1 args")
flag.Usage() flag.Usage()
} }
arg537 := flag.Arg(1) arg539 := flag.Arg(1)
mbTrans538 := thrift.NewTMemoryBufferLen(len(arg537)) mbTrans540 := thrift.NewTMemoryBufferLen(len(arg539))
defer mbTrans538.Close() defer mbTrans540.Close()
_, err539 := mbTrans538.WriteString(arg537) _, err541 := mbTrans540.WriteString(arg539)
if err539 != nil { if err541 != nil {
Usage() Usage()
return return
} }
factory540 := thrift.NewTSimpleJSONProtocolFactory() factory542 := thrift.NewTSimpleJSONProtocolFactory()
jsProt541 := factory540.GetProtocol(mbTrans538) jsProt543 := factory542.GetProtocol(mbTrans540)
argvalue0 := aurora.NewJobConfiguration() argvalue0 := aurora.NewJobConfiguration()
err542 := argvalue0.Read(jsProt541) err544 := argvalue0.Read(jsProt543)
if err542 != nil { if err544 != nil {
Usage() Usage()
return return
} }
@ -1031,19 +1033,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetJobUpdateSummaries requires 1 args") fmt.Fprintln(os.Stderr, "GetJobUpdateSummaries requires 1 args")
flag.Usage() flag.Usage()
} }
arg543 := flag.Arg(1) arg545 := flag.Arg(1)
mbTrans544 := thrift.NewTMemoryBufferLen(len(arg543)) mbTrans546 := thrift.NewTMemoryBufferLen(len(arg545))
defer mbTrans544.Close() defer mbTrans546.Close()
_, err545 := mbTrans544.WriteString(arg543) _, err547 := mbTrans546.WriteString(arg545)
if err545 != nil { if err547 != nil {
Usage() Usage()
return return
} }
factory546 := thrift.NewTSimpleJSONProtocolFactory() factory548 := thrift.NewTSimpleJSONProtocolFactory()
jsProt547 := factory546.GetProtocol(mbTrans544) jsProt549 := factory548.GetProtocol(mbTrans546)
argvalue0 := aurora.NewJobUpdateQuery() argvalue0 := aurora.NewJobUpdateQuery()
err548 := argvalue0.Read(jsProt547) err550 := argvalue0.Read(jsProt549)
if err548 != nil { if err550 != nil {
Usage() Usage()
return return
} }
@ -1056,19 +1058,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetJobUpdateDetails requires 1 args") fmt.Fprintln(os.Stderr, "GetJobUpdateDetails requires 1 args")
flag.Usage() flag.Usage()
} }
arg549 := flag.Arg(1) arg551 := flag.Arg(1)
mbTrans550 := thrift.NewTMemoryBufferLen(len(arg549)) mbTrans552 := thrift.NewTMemoryBufferLen(len(arg551))
defer mbTrans550.Close() defer mbTrans552.Close()
_, err551 := mbTrans550.WriteString(arg549) _, err553 := mbTrans552.WriteString(arg551)
if err551 != nil { if err553 != nil {
Usage() Usage()
return return
} }
factory552 := thrift.NewTSimpleJSONProtocolFactory() factory554 := thrift.NewTSimpleJSONProtocolFactory()
jsProt553 := factory552.GetProtocol(mbTrans550) jsProt555 := factory554.GetProtocol(mbTrans552)
argvalue0 := aurora.NewJobUpdateQuery() argvalue0 := aurora.NewJobUpdateQuery()
err554 := argvalue0.Read(jsProt553) err556 := argvalue0.Read(jsProt555)
if err554 != nil { if err556 != nil {
Usage() Usage()
return return
} }
@ -1081,19 +1083,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetJobUpdateDiff requires 1 args") fmt.Fprintln(os.Stderr, "GetJobUpdateDiff requires 1 args")
flag.Usage() flag.Usage()
} }
arg555 := flag.Arg(1) arg557 := flag.Arg(1)
mbTrans556 := thrift.NewTMemoryBufferLen(len(arg555)) mbTrans558 := thrift.NewTMemoryBufferLen(len(arg557))
defer mbTrans556.Close() defer mbTrans558.Close()
_, err557 := mbTrans556.WriteString(arg555) _, err559 := mbTrans558.WriteString(arg557)
if err557 != nil { if err559 != nil {
Usage() Usage()
return return
} }
factory558 := thrift.NewTSimpleJSONProtocolFactory() factory560 := thrift.NewTSimpleJSONProtocolFactory()
jsProt559 := factory558.GetProtocol(mbTrans556) jsProt561 := factory560.GetProtocol(mbTrans558)
argvalue0 := aurora.NewJobUpdateRequest() argvalue0 := aurora.NewJobUpdateRequest()
err560 := argvalue0.Read(jsProt559) err562 := argvalue0.Read(jsProt561)
if err560 != nil { if err562 != nil {
Usage() Usage()
return return
} }

302
gen-go/apache/aurora/aurora_scheduler_manager-remote/aurora_scheduler_manager-remote.go
View file

@ -25,7 +25,7 @@ func Usage() {
fmt.Fprintln(os.Stderr, " Response descheduleCronJob(JobKey job)") fmt.Fprintln(os.Stderr, " Response descheduleCronJob(JobKey job)")
fmt.Fprintln(os.Stderr, " Response startCronJob(JobKey job)") fmt.Fprintln(os.Stderr, " Response startCronJob(JobKey job)")
fmt.Fprintln(os.Stderr, " Response restartShards(JobKey job, shardIds)") fmt.Fprintln(os.Stderr, " Response restartShards(JobKey job, shardIds)")
fmt.Fprintln(os.Stderr, " Response killTasks(JobKey job, instances)") fmt.Fprintln(os.Stderr, " Response killTasks(JobKey job, instances, string message)")
fmt.Fprintln(os.Stderr, " Response addInstances(InstanceKey key, i32 count)") fmt.Fprintln(os.Stderr, " Response addInstances(InstanceKey key, i32 count)")
fmt.Fprintln(os.Stderr, " Response replaceCronTemplate(JobConfiguration config)") fmt.Fprintln(os.Stderr, " Response replaceCronTemplate(JobConfiguration config)")
fmt.Fprintln(os.Stderr, " Response startJobUpdate(JobUpdateRequest request, string message)") fmt.Fprintln(os.Stderr, " Response startJobUpdate(JobUpdateRequest request, string message)")
@ -285,8 +285,8 @@ func main() {
fmt.Print("\n") fmt.Print("\n")
break break
case "killTasks": case "killTasks":
if flag.NArg()-1 != 2 { if flag.NArg()-1 != 3 {
fmt.Fprintln(os.Stderr, "KillTasks requires 2 args") fmt.Fprintln(os.Stderr, "KillTasks requires 3 args")
flag.Usage() flag.Usage()
} }
arg199 := flag.Arg(1) arg199 := flag.Arg(1)
@ -324,7 +324,9 @@ func main() {
} }
argvalue1 := containerStruct1.Instances argvalue1 := containerStruct1.Instances
value1 := argvalue1 value1 := argvalue1
fmt.Print(client.KillTasks(value0, value1)) argvalue2 := flag.Arg(3)
value2 := argvalue2
fmt.Print(client.KillTasks(value0, value1, value2))
fmt.Print("\n") fmt.Print("\n")
break break
case "addInstances": case "addInstances":
@ -332,25 +334,25 @@ func main() {
fmt.Fprintln(os.Stderr, "AddInstances requires 2 args") fmt.Fprintln(os.Stderr, "AddInstances requires 2 args")
flag.Usage() flag.Usage()
} }
arg211 := flag.Arg(1) arg212 := flag.Arg(1)
mbTrans212 := thrift.NewTMemoryBufferLen(len(arg211)) mbTrans213 := thrift.NewTMemoryBufferLen(len(arg212))
defer mbTrans212.Close() defer mbTrans213.Close()
_, err213 := mbTrans212.WriteString(arg211) _, err214 := mbTrans213.WriteString(arg212)
if err213 != nil { if err214 != nil {
Usage() Usage()
return return
} }
factory214 := thrift.NewTSimpleJSONProtocolFactory() factory215 := thrift.NewTSimpleJSONProtocolFactory()
jsProt215 := factory214.GetProtocol(mbTrans212) jsProt216 := factory215.GetProtocol(mbTrans213)
argvalue0 := aurora.NewInstanceKey() argvalue0 := aurora.NewInstanceKey()
err216 := argvalue0.Read(jsProt215) err217 := argvalue0.Read(jsProt216)
if err216 != nil { if err217 != nil {
Usage() Usage()
return return
} }
value0 := argvalue0 value0 := argvalue0
tmp1, err217 := (strconv.Atoi(flag.Arg(2))) tmp1, err218 := (strconv.Atoi(flag.Arg(2)))
if err217 != nil { if err218 != nil {
Usage() Usage()
return return
} }
@ -364,19 +366,19 @@ func main() {
fmt.Fprintln(os.Stderr, "ReplaceCronTemplate requires 1 args") fmt.Fprintln(os.Stderr, "ReplaceCronTemplate requires 1 args")
flag.Usage() flag.Usage()
} }
arg218 := flag.Arg(1) arg219 := flag.Arg(1)
mbTrans219 := thrift.NewTMemoryBufferLen(len(arg218)) mbTrans220 := thrift.NewTMemoryBufferLen(len(arg219))
defer mbTrans219.Close() defer mbTrans220.Close()
_, err220 := mbTrans219.WriteString(arg218) _, err221 := mbTrans220.WriteString(arg219)
if err220 != nil { if err221 != nil {
Usage() Usage()
return return
} }
factory221 := thrift.NewTSimpleJSONProtocolFactory() factory222 := thrift.NewTSimpleJSONProtocolFactory()
jsProt222 := factory221.GetProtocol(mbTrans219) jsProt223 := factory222.GetProtocol(mbTrans220)
argvalue0 := aurora.NewJobConfiguration() argvalue0 := aurora.NewJobConfiguration()
err223 := argvalue0.Read(jsProt222) err224 := argvalue0.Read(jsProt223)
if err223 != nil { if err224 != nil {
Usage() Usage()
return return
} }
@ -389,19 +391,19 @@ func main() {
fmt.Fprintln(os.Stderr, "StartJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "StartJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg224 := flag.Arg(1) arg225 := flag.Arg(1)
mbTrans225 := thrift.NewTMemoryBufferLen(len(arg224)) mbTrans226 := thrift.NewTMemoryBufferLen(len(arg225))
defer mbTrans225.Close() defer mbTrans226.Close()
_, err226 := mbTrans225.WriteString(arg224) _, err227 := mbTrans226.WriteString(arg225)
if err226 != nil { if err227 != nil {
Usage() Usage()
return return
} }
factory227 := thrift.NewTSimpleJSONProtocolFactory() factory228 := thrift.NewTSimpleJSONProtocolFactory()
jsProt228 := factory227.GetProtocol(mbTrans225) jsProt229 := factory228.GetProtocol(mbTrans226)
argvalue0 := aurora.NewJobUpdateRequest() argvalue0 := aurora.NewJobUpdateRequest()
err229 := argvalue0.Read(jsProt228) err230 := argvalue0.Read(jsProt229)
if err229 != nil { if err230 != nil {
Usage() Usage()
return return
} }
@ -416,19 +418,19 @@ func main() {
fmt.Fprintln(os.Stderr, "PauseJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "PauseJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg231 := flag.Arg(1) arg232 := flag.Arg(1)
mbTrans232 := thrift.NewTMemoryBufferLen(len(arg231)) mbTrans233 := thrift.NewTMemoryBufferLen(len(arg232))
defer mbTrans232.Close() defer mbTrans233.Close()
_, err233 := mbTrans232.WriteString(arg231) _, err234 := mbTrans233.WriteString(arg232)
if err233 != nil { if err234 != nil {
Usage() Usage()
return return
} }
factory234 := thrift.NewTSimpleJSONProtocolFactory() factory235 := thrift.NewTSimpleJSONProtocolFactory()
jsProt235 := factory234.GetProtocol(mbTrans232) jsProt236 := factory235.GetProtocol(mbTrans233)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err236 := argvalue0.Read(jsProt235) err237 := argvalue0.Read(jsProt236)
if err236 != nil { if err237 != nil {
Usage() Usage()
return return
} }
@ -443,19 +445,19 @@ func main() {
fmt.Fprintln(os.Stderr, "ResumeJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "ResumeJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg238 := flag.Arg(1) arg239 := flag.Arg(1)
mbTrans239 := thrift.NewTMemoryBufferLen(len(arg238)) mbTrans240 := thrift.NewTMemoryBufferLen(len(arg239))
defer mbTrans239.Close() defer mbTrans240.Close()
_, err240 := mbTrans239.WriteString(arg238) _, err241 := mbTrans240.WriteString(arg239)
if err240 != nil { if err241 != nil {
Usage() Usage()
return return
} }
factory241 := thrift.NewTSimpleJSONProtocolFactory() factory242 := thrift.NewTSimpleJSONProtocolFactory()
jsProt242 := factory241.GetProtocol(mbTrans239) jsProt243 := factory242.GetProtocol(mbTrans240)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err243 := argvalue0.Read(jsProt242) err244 := argvalue0.Read(jsProt243)
if err243 != nil { if err244 != nil {
Usage() Usage()
return return
} }
@ -470,19 +472,19 @@ func main() {
fmt.Fprintln(os.Stderr, "AbortJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "AbortJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg245 := flag.Arg(1) arg246 := flag.Arg(1)
mbTrans246 := thrift.NewTMemoryBufferLen(len(arg245)) mbTrans247 := thrift.NewTMemoryBufferLen(len(arg246))
defer mbTrans246.Close() defer mbTrans247.Close()
_, err247 := mbTrans246.WriteString(arg245) _, err248 := mbTrans247.WriteString(arg246)
if err247 != nil { if err248 != nil {
Usage() Usage()
return return
} }
factory248 := thrift.NewTSimpleJSONProtocolFactory() factory249 := thrift.NewTSimpleJSONProtocolFactory()
jsProt249 := factory248.GetProtocol(mbTrans246) jsProt250 := factory249.GetProtocol(mbTrans247)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err250 := argvalue0.Read(jsProt249) err251 := argvalue0.Read(jsProt250)
if err250 != nil { if err251 != nil {
Usage() Usage()
return return
} }
@ -497,19 +499,19 @@ func main() {
fmt.Fprintln(os.Stderr, "RollbackJobUpdate requires 2 args") fmt.Fprintln(os.Stderr, "RollbackJobUpdate requires 2 args")
flag.Usage() flag.Usage()
} }
arg252 := flag.Arg(1) arg253 := flag.Arg(1)
mbTrans253 := thrift.NewTMemoryBufferLen(len(arg252)) mbTrans254 := thrift.NewTMemoryBufferLen(len(arg253))
defer mbTrans253.Close() defer mbTrans254.Close()
_, err254 := mbTrans253.WriteString(arg252) _, err255 := mbTrans254.WriteString(arg253)
if err254 != nil { if err255 != nil {
Usage() Usage()
return return
} }
factory255 := thrift.NewTSimpleJSONProtocolFactory() factory256 := thrift.NewTSimpleJSONProtocolFactory()
jsProt256 := factory255.GetProtocol(mbTrans253) jsProt257 := factory256.GetProtocol(mbTrans254)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err257 := argvalue0.Read(jsProt256) err258 := argvalue0.Read(jsProt257)
if err257 != nil { if err258 != nil {
Usage() Usage()
return return
} }
@ -524,19 +526,19 @@ func main() {
fmt.Fprintln(os.Stderr, "PulseJobUpdate requires 1 args") fmt.Fprintln(os.Stderr, "PulseJobUpdate requires 1 args")
flag.Usage() flag.Usage()
} }
arg259 := flag.Arg(1) arg260 := flag.Arg(1)
mbTrans260 := thrift.NewTMemoryBufferLen(len(arg259)) mbTrans261 := thrift.NewTMemoryBufferLen(len(arg260))
defer mbTrans260.Close() defer mbTrans261.Close()
_, err261 := mbTrans260.WriteString(arg259) _, err262 := mbTrans261.WriteString(arg260)
if err261 != nil { if err262 != nil {
Usage() Usage()
return return
} }
factory262 := thrift.NewTSimpleJSONProtocolFactory() factory263 := thrift.NewTSimpleJSONProtocolFactory()
jsProt263 := factory262.GetProtocol(mbTrans260) jsProt264 := factory263.GetProtocol(mbTrans261)
argvalue0 := aurora.NewJobUpdateKey() argvalue0 := aurora.NewJobUpdateKey()
err264 := argvalue0.Read(jsProt263) err265 := argvalue0.Read(jsProt264)
if err264 != nil { if err265 != nil {
Usage() Usage()
return return
} }
@ -567,19 +569,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetTasksStatus requires 1 args") fmt.Fprintln(os.Stderr, "GetTasksStatus requires 1 args")
flag.Usage() flag.Usage()
} }
arg266 := flag.Arg(1) arg267 := flag.Arg(1)
mbTrans267 := thrift.NewTMemoryBufferLen(len(arg266)) mbTrans268 := thrift.NewTMemoryBufferLen(len(arg267))
defer mbTrans267.Close() defer mbTrans268.Close()
_, err268 := mbTrans267.WriteString(arg266) _, err269 := mbTrans268.WriteString(arg267)
if err268 != nil { if err269 != nil {
Usage() Usage()
return return
} }
factory269 := thrift.NewTSimpleJSONProtocolFactory() factory270 := thrift.NewTSimpleJSONProtocolFactory()
jsProt270 := factory269.GetProtocol(mbTrans267) jsProt271 := factory270.GetProtocol(mbTrans268)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err271 := argvalue0.Read(jsProt270) err272 := argvalue0.Read(jsProt271)
if err271 != nil { if err272 != nil {
Usage() Usage()
return return
} }
@ -592,19 +594,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetTasksWithoutConfigs requires 1 args") fmt.Fprintln(os.Stderr, "GetTasksWithoutConfigs requires 1 args")
flag.Usage() flag.Usage()
} }
arg272 := flag.Arg(1) arg273 := flag.Arg(1)
mbTrans273 := thrift.NewTMemoryBufferLen(len(arg272)) mbTrans274 := thrift.NewTMemoryBufferLen(len(arg273))
defer mbTrans273.Close() defer mbTrans274.Close()
_, err274 := mbTrans273.WriteString(arg272) _, err275 := mbTrans274.WriteString(arg273)
if err274 != nil { if err275 != nil {
Usage() Usage()
return return
} }
factory275 := thrift.NewTSimpleJSONProtocolFactory() factory276 := thrift.NewTSimpleJSONProtocolFactory()
jsProt276 := factory275.GetProtocol(mbTrans273) jsProt277 := factory276.GetProtocol(mbTrans274)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err277 := argvalue0.Read(jsProt276) err278 := argvalue0.Read(jsProt277)
if err277 != nil { if err278 != nil {
Usage() Usage()
return return
} }
@ -617,19 +619,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetPendingReason requires 1 args") fmt.Fprintln(os.Stderr, "GetPendingReason requires 1 args")
flag.Usage() flag.Usage()
} }
arg278 := flag.Arg(1) arg279 := flag.Arg(1)
mbTrans279 := thrift.NewTMemoryBufferLen(len(arg278)) mbTrans280 := thrift.NewTMemoryBufferLen(len(arg279))
defer mbTrans279.Close() defer mbTrans280.Close()
_, err280 := mbTrans279.WriteString(arg278) _, err281 := mbTrans280.WriteString(arg279)
if err280 != nil { if err281 != nil {
Usage() Usage()
return return
} }
factory281 := thrift.NewTSimpleJSONProtocolFactory() factory282 := thrift.NewTSimpleJSONProtocolFactory()
jsProt282 := factory281.GetProtocol(mbTrans279) jsProt283 := factory282.GetProtocol(mbTrans280)
argvalue0 := aurora.NewTaskQuery() argvalue0 := aurora.NewTaskQuery()
err283 := argvalue0.Read(jsProt282) err284 := argvalue0.Read(jsProt283)
if err283 != nil { if err284 != nil {
Usage() Usage()
return return
} }
@ -642,19 +644,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetConfigSummary requires 1 args") fmt.Fprintln(os.Stderr, "GetConfigSummary requires 1 args")
flag.Usage() flag.Usage()
} }
arg284 := flag.Arg(1) arg285 := flag.Arg(1)
mbTrans285 := thrift.NewTMemoryBufferLen(len(arg284)) mbTrans286 := thrift.NewTMemoryBufferLen(len(arg285))
defer mbTrans285.Close() defer mbTrans286.Close()
_, err286 := mbTrans285.WriteString(arg284) _, err287 := mbTrans286.WriteString(arg285)
if err286 != nil { if err287 != nil {
Usage() Usage()
return return
} }
factory287 := thrift.NewTSimpleJSONProtocolFactory() factory288 := thrift.NewTSimpleJSONProtocolFactory()
jsProt288 := factory287.GetProtocol(mbTrans285) jsProt289 := factory288.GetProtocol(mbTrans286)
argvalue0 := aurora.NewJobKey() argvalue0 := aurora.NewJobKey()
err289 := argvalue0.Read(jsProt288) err290 := argvalue0.Read(jsProt289)
if err289 != nil { if err290 != nil {
Usage() Usage()
return return
} }
@ -687,19 +689,19 @@ func main() {
fmt.Fprintln(os.Stderr, "PopulateJobConfig requires 1 args") fmt.Fprintln(os.Stderr, "PopulateJobConfig requires 1 args")
flag.Usage() flag.Usage()
} }
arg292 := flag.Arg(1) arg293 := flag.Arg(1)
mbTrans293 := thrift.NewTMemoryBufferLen(len(arg292)) mbTrans294 := thrift.NewTMemoryBufferLen(len(arg293))
defer mbTrans293.Close() defer mbTrans294.Close()
_, err294 := mbTrans293.WriteString(arg292) _, err295 := mbTrans294.WriteString(arg293)
if err294 != nil { if err295 != nil {
Usage() Usage()
return return
} }
factory295 := thrift.NewTSimpleJSONProtocolFactory() factory296 := thrift.NewTSimpleJSONProtocolFactory()
jsProt296 := factory295.GetProtocol(mbTrans293) jsProt297 := factory296.GetProtocol(mbTrans294)
argvalue0 := aurora.NewJobConfiguration() argvalue0 := aurora.NewJobConfiguration()
err297 := argvalue0.Read(jsProt296) err298 := argvalue0.Read(jsProt297)
if err297 != nil { if err298 != nil {
Usage() Usage()
return return
} }
@ -712,19 +714,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetJobUpdateSummaries requires 1 args") fmt.Fprintln(os.Stderr, "GetJobUpdateSummaries requires 1 args")
flag.Usage() flag.Usage()
} }
arg298 := flag.Arg(1) arg299 := flag.Arg(1)
mbTrans299 := thrift.NewTMemoryBufferLen(len(arg298)) mbTrans300 := thrift.NewTMemoryBufferLen(len(arg299))
defer mbTrans299.Close() defer mbTrans300.Close()
_, err300 := mbTrans299.WriteString(arg298) _, err301 := mbTrans300.WriteString(arg299)
if err300 != nil { if err301 != nil {
Usage() Usage()
return return
} }
factory301 := thrift.NewTSimpleJSONProtocolFactory() factory302 := thrift.NewTSimpleJSONProtocolFactory()
jsProt302 := factory301.GetProtocol(mbTrans299) jsProt303 := factory302.GetProtocol(mbTrans300)
argvalue0 := aurora.NewJobUpdateQuery() argvalue0 := aurora.NewJobUpdateQuery()
err303 := argvalue0.Read(jsProt302) err304 := argvalue0.Read(jsProt303)
if err303 != nil { if err304 != nil {
Usage() Usage()
return return
} }
@ -737,19 +739,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetJobUpdateDetails requires 1 args") fmt.Fprintln(os.Stderr, "GetJobUpdateDetails requires 1 args")
flag.Usage() flag.Usage()
} }
arg304 := flag.Arg(1) arg305 := flag.Arg(1)
mbTrans305 := thrift.NewTMemoryBufferLen(len(arg304)) mbTrans306 := thrift.NewTMemoryBufferLen(len(arg305))
defer mbTrans305.Close() defer mbTrans306.Close()
_, err306 := mbTrans305.WriteString(arg304) _, err307 := mbTrans306.WriteString(arg305)
if err306 != nil { if err307 != nil {
Usage() Usage()
return return
} }
factory307 := thrift.NewTSimpleJSONProtocolFactory() factory308 := thrift.NewTSimpleJSONProtocolFactory()
jsProt308 := factory307.GetProtocol(mbTrans305) jsProt309 := factory308.GetProtocol(mbTrans306)
argvalue0 := aurora.NewJobUpdateQuery() argvalue0 := aurora.NewJobUpdateQuery()
err309 := argvalue0.Read(jsProt308) err310 := argvalue0.Read(jsProt309)
if err309 != nil { if err310 != nil {
Usage() Usage()
return return
} }
@ -762,19 +764,19 @@ func main() {
fmt.Fprintln(os.Stderr, "GetJobUpdateDiff requires 1 args") fmt.Fprintln(os.Stderr, "GetJobUpdateDiff requires 1 args")
flag.Usage() flag.Usage()
} }
arg310 := flag.Arg(1) arg311 := flag.Arg(1)
mbTrans311 := thrift.NewTMemoryBufferLen(len(arg310)) mbTrans312 := thrift.NewTMemoryBufferLen(len(arg311))
defer mbTrans311.Close() defer mbTrans312.Close()
_, err312 := mbTrans311.WriteString(arg310) _, err313 := mbTrans312.WriteString(arg311)
if err312 != nil { if err313 != nil {
Usage() Usage()
return return
} }
factory313 := thrift.NewTSimpleJSONProtocolFactory() factory314 := thrift.NewTSimpleJSONProtocolFactory()
jsProt314 := factory313.GetProtocol(mbTrans311) jsProt315 := factory314.GetProtocol(mbTrans312)
argvalue0 := aurora.NewJobUpdateRequest() argvalue0 := aurora.NewJobUpdateRequest()
err315 := argvalue0.Read(jsProt314) err316 := argvalue0.Read(jsProt315)
if err315 != nil { if err316 != nil {
Usage() Usage()
return return
} }

842
gen-go/apache/aurora/auroraadmin.go
View file

File diff suppressed because it is too large Load diff

48
gen-go/apache/aurora/auroraschedulermanager.go
View file

@ -53,7 +53,8 @@ type AuroraSchedulerManager interface {
// Parameters: // Parameters:
// - Job // - Job
// - Instances // - Instances
KillTasks(job *JobKey, instances map[int32]bool) (r *Response, err error) // - Message
KillTasks(job *JobKey, instances map[int32]bool, message string) (r *Response, err error)
// Adds new instances with the TaskConfig of the existing instance pointed by the key. // Adds new instances with the TaskConfig of the existing instance pointed by the key.
// //
// Parameters: // Parameters:
@ -524,14 +525,15 @@ func (p *AuroraSchedulerManagerClient) recvRestartShards() (value *Response, err
// Parameters: // Parameters:
// - Job // - Job
// - Instances // - Instances
func (p *AuroraSchedulerManagerClient) KillTasks(job *JobKey, instances map[int32]bool) (r *Response, err error) { // - Message
if err = p.sendKillTasks(job, instances); err != nil { func (p *AuroraSchedulerManagerClient) KillTasks(job *JobKey, instances map[int32]bool, message string) (r *Response, err error) {
if err = p.sendKillTasks(job, instances, message); err != nil {
return return
} }
return p.recvKillTasks() return p.recvKillTasks()
} }
func (p *AuroraSchedulerManagerClient) sendKillTasks(job *JobKey, instances map[int32]bool) (err error) { func (p *AuroraSchedulerManagerClient) sendKillTasks(job *JobKey, instances map[int32]bool, message string) (err error) {
oprot := p.OutputProtocol oprot := p.OutputProtocol
if oprot == nil { if oprot == nil {
oprot = p.ProtocolFactory.GetProtocol(p.Transport) oprot = p.ProtocolFactory.GetProtocol(p.Transport)
@ -544,6 +546,7 @@ func (p *AuroraSchedulerManagerClient) sendKillTasks(job *JobKey, instances map[
args := AuroraSchedulerManagerKillTasksArgs{ args := AuroraSchedulerManagerKillTasksArgs{
Job: job, Job: job,
Instances: instances, Instances: instances,
Message: message,
} }
if err = args.Write(oprot); err != nil { if err = args.Write(oprot); err != nil {
return return
@ -1530,7 +1533,7 @@ func (p *auroraSchedulerManagerProcessorKillTasks) Process(seqId int32, iprot, o
result := AuroraSchedulerManagerKillTasksResult{} result := AuroraSchedulerManagerKillTasksResult{}
var retval *Response var retval *Response
var err2 error var err2 error
if retval, err2 = p.handler.KillTasks(args.Job, args.Instances); err2 != nil { if retval, err2 = p.handler.KillTasks(args.Job, args.Instances, args.Message); err2 != nil {
x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing killTasks: "+err2.Error()) x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing killTasks: "+err2.Error())
oprot.WriteMessageBegin("killTasks", thrift.EXCEPTION, seqId) oprot.WriteMessageBegin("killTasks", thrift.EXCEPTION, seqId)
x.Write(oprot) x.Write(oprot)
@ -3007,10 +3010,12 @@ func (p *AuroraSchedulerManagerRestartShardsResult) String() string {
// Attributes: // Attributes:
// - Job // - Job
// - Instances // - Instances
// - Message
type AuroraSchedulerManagerKillTasksArgs struct { type AuroraSchedulerManagerKillTasksArgs struct {
// unused fields # 1 to 3 // unused fields # 1 to 3
Job *JobKey `thrift:"job,4" json:"job"` Job *JobKey `thrift:"job,4" json:"job"`
Instances map[int32]bool `thrift:"instances,5" json:"instances"` Instances map[int32]bool `thrift:"instances,5" json:"instances"`
Message string `thrift:"message,6" json:"message"`
} }
func NewAuroraSchedulerManagerKillTasksArgs() *AuroraSchedulerManagerKillTasksArgs { func NewAuroraSchedulerManagerKillTasksArgs() *AuroraSchedulerManagerKillTasksArgs {
@ -3029,6 +3034,10 @@ func (p *AuroraSchedulerManagerKillTasksArgs) GetJob() *JobKey {
func (p *AuroraSchedulerManagerKillTasksArgs) GetInstances() map[int32]bool { func (p *AuroraSchedulerManagerKillTasksArgs) GetInstances() map[int32]bool {
return p.Instances return p.Instances
} }
func (p *AuroraSchedulerManagerKillTasksArgs) GetMessage() string {
return p.Message
}
func (p *AuroraSchedulerManagerKillTasksArgs) IsSetJob() bool { func (p *AuroraSchedulerManagerKillTasksArgs) IsSetJob() bool {
return p.Job != nil return p.Job != nil
} }
@ -3055,6 +3064,10 @@ func (p *AuroraSchedulerManagerKillTasksArgs) Read(iprot thrift.TProtocol) error
if err := p.readField5(iprot); err != nil { if err := p.readField5(iprot); err != nil {
return err return err
} }
case 6:
if err := p.readField6(iprot); err != nil {
return err
}
default: default:
if err := iprot.Skip(fieldTypeId); err != nil { if err := iprot.Skip(fieldTypeId); err != nil {
return err return err
@ -3100,6 +3113,15 @@ func (p *AuroraSchedulerManagerKillTasksArgs) readField5(iprot thrift.TProtocol)
return nil return nil
} }
func (p *AuroraSchedulerManagerKillTasksArgs) readField6(iprot thrift.TProtocol) error {
if v, err := iprot.ReadString(); err != nil {
return thrift.PrependError("error reading field 6: ", err)
} else {
p.Message = v
}
return nil
}
func (p *AuroraSchedulerManagerKillTasksArgs) Write(oprot thrift.TProtocol) error { func (p *AuroraSchedulerManagerKillTasksArgs) Write(oprot thrift.TProtocol) error {
if err := oprot.WriteStructBegin("killTasks_args"); err != nil { if err := oprot.WriteStructBegin("killTasks_args"); err != nil {
return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err) return thrift.PrependError(fmt.Sprintf("%T write struct begin error: ", p), err)
@ -3110,6 +3132,9 @@ func (p *AuroraSchedulerManagerKillTasksArgs) Write(oprot thrift.TProtocol) erro
if err := p.writeField5(oprot); err != nil { if err := p.writeField5(oprot); err != nil {
return err return err
} }
if err := p.writeField6(oprot); err != nil {
return err
}
if err := oprot.WriteFieldStop(); err != nil { if err := oprot.WriteFieldStop(); err != nil {
return thrift.PrependError("write field stop error: ", err) return thrift.PrependError("write field stop error: ", err)
} }
@ -3153,6 +3178,19 @@ func (p *AuroraSchedulerManagerKillTasksArgs) writeField5(oprot thrift.TProtocol
return err return err
} }
func (p *AuroraSchedulerManagerKillTasksArgs) writeField6(oprot thrift.TProtocol) (err error) {
if err := oprot.WriteFieldBegin("message", thrift.STRING, 6); err != nil {
return thrift.PrependError(fmt.Sprintf("%T write field begin error 6:message: ", p), err)
}
if err := oprot.WriteString(string(p.Message)); err != nil {
return thrift.PrependError(fmt.Sprintf("%T.message (6) field write error: ", p), err)
}
if err := oprot.WriteFieldEnd(); err != nil {
return thrift.PrependError(fmt.Sprintf("%T write field end error 6:message: ", p), err)
}
return err
}
func (p *AuroraSchedulerManagerKillTasksArgs) String() string { func (p *AuroraSchedulerManagerKillTasksArgs) String() string {
if p == nil { if p == nil {
return "<nil>" return "<nil>"

6
gen-go/apache/aurora/constants.go
View file

@ -26,6 +26,7 @@ const GOOD_IDENTIFIER_PATTERN_JVM = "^[\\w\\-\\.]+$"
const GOOD_IDENTIFIER_PATTERN_PYTHON = "^[\\w\\-\\.]+$" const GOOD_IDENTIFIER_PATTERN_PYTHON = "^[\\w\\-\\.]+$"
var ACTIVE_JOB_UPDATE_STATES map[JobUpdateStatus]bool var ACTIVE_JOB_UPDATE_STATES map[JobUpdateStatus]bool
var AWAITNG_PULSE_JOB_UPDATE_STATES map[JobUpdateStatus]bool
const BYPASS_LEADER_REDIRECT_HEADER_NAME = "Bypass-Leader-Redirect" const BYPASS_LEADER_REDIRECT_HEADER_NAME = "Bypass-Leader-Redirect"
const TASK_FILESYSTEM_MOUNT_POINT = "taskfs" const TASK_FILESYSTEM_MOUNT_POINT = "taskfs"
@ -77,4 +78,9 @@ func init() {
10: true, 10: true,
} }
AWAITNG_PULSE_JOB_UPDATE_STATES = map[JobUpdateStatus]bool{
9: true,
10: true,
}
} }

1042
gen-go/apache/aurora/ttypes.go
View file

File diff suppressed because it is too large Load diff

6
realis.go
View file

@ -267,7 +267,7 @@ func NewRealisClient(options ...ClientOption) (Realis, error) {
config.protoFactory = thrift.NewTBinaryProtocolFactoryDefault() config.protoFactory = thrift.NewTBinaryProtocolFactoryDefault()
} }
config.logger.Printf("gorealis config url: %+v\n", config.url) config.logger.Printf("gorealis config url: %+v\n", url)
//Basic Authentication. //Basic Authentication.
if config.username != "" && config.password != "" { if config.username != "" && config.password != "" {
@ -579,7 +579,7 @@ func (r *realisClient) KillInstances(key *aurora.JobKey, instances ...int32) (*a
retryErr := ExponentialBackoff(*r.config.backoff, func() (bool, error) { retryErr := ExponentialBackoff(*r.config.backoff, func() (bool, error) {
resp, clientErr = CheckAndRetryConn(r, func() (*aurora.Response, error) { resp, clientErr = CheckAndRetryConn(r, func() (*aurora.Response, error) {
return r.client.KillTasks(key, instanceIds) return r.client.KillTasks(key, instanceIds, "")
}) })
if clientErr != nil && clientErr.Error() == RetryConnErr.Error() { if clientErr != nil && clientErr.Error() == RetryConnErr.Error() {
return false, nil return false, nil
@ -614,7 +614,7 @@ func (r *realisClient) KillJob(key *aurora.JobKey) (*aurora.Response, error) {
if len(instanceIds) > 0 { if len(instanceIds) > 0 {
retryErr := ExponentialBackoff(*r.config.backoff, func() (bool, error) { retryErr := ExponentialBackoff(*r.config.backoff, func() (bool, error) {
resp, clientErr = CheckAndRetryConn(r, func() (*aurora.Response, error) { resp, clientErr = CheckAndRetryConn(r, func() (*aurora.Response, error) {
return r.client.KillTasks(key, instanceIds) return r.client.KillTasks(key, instanceIds, "")
}) })
if clientErr != nil && clientErr.Error() == RetryConnErr.Error() { if clientErr != nil && clientErr.Error() == RetryConnErr.Error() {
return false, nil return false, nil