Updating Aurora thrift api and regenerating stubs.

This commit is contained in:
Renan DelValle 2017-02-22 19:08:48 -05:00
parent ac40251f00
commit 47236ed3fd
8 changed files with 1132 additions and 1026 deletions

View file

@ -53,7 +53,8 @@ type AuroraSchedulerManager interface {
// Parameters:
// - Job
// - 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.
//
// Parameters:
@ -170,16 +171,16 @@ func (p *AuroraSchedulerManagerClient) recvCreateJob() (value *Response, err err
return
}
if mTypeId == thrift.EXCEPTION {
error132 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error133 error
error133, err = error132.Read(iprot)
error133 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error134 error
error134, err = error133.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error133
err = error134
return
}
if mTypeId != thrift.REPLY {
@ -251,16 +252,16 @@ func (p *AuroraSchedulerManagerClient) recvScheduleCronJob() (value *Response, e
return
}
if mTypeId == thrift.EXCEPTION {
error134 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error135 error
error135, err = error134.Read(iprot)
error135 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error136 error
error136, err = error135.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error135
err = error136
return
}
if mTypeId != thrift.REPLY {
@ -331,16 +332,16 @@ func (p *AuroraSchedulerManagerClient) recvDescheduleCronJob() (value *Response,
return
}
if mTypeId == thrift.EXCEPTION {
error136 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error137 error
error137, err = error136.Read(iprot)
error137 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error138 error
error138, err = error137.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error137
err = error138
return
}
if mTypeId != thrift.REPLY {
@ -411,16 +412,16 @@ func (p *AuroraSchedulerManagerClient) recvStartCronJob() (value *Response, err
return
}
if mTypeId == thrift.EXCEPTION {
error138 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error139 error
error139, err = error138.Read(iprot)
error139 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error140 error
error140, err = error139.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error139
err = error140
return
}
if mTypeId != thrift.REPLY {
@ -492,16 +493,16 @@ func (p *AuroraSchedulerManagerClient) recvRestartShards() (value *Response, err
return
}
if mTypeId == thrift.EXCEPTION {
error140 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error141 error
error141, err = error140.Read(iprot)
error141 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error142 error
error142, err = error141.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error141
err = error142
return
}
if mTypeId != thrift.REPLY {
@ -524,14 +525,15 @@ func (p *AuroraSchedulerManagerClient) recvRestartShards() (value *Response, err
// Parameters:
// - Job
// - Instances
func (p *AuroraSchedulerManagerClient) KillTasks(job *JobKey, instances map[int32]bool) (r *Response, err error) {
if err = p.sendKillTasks(job, instances); err != nil {
// - Message
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 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
if oprot == nil {
oprot = p.ProtocolFactory.GetProtocol(p.Transport)
@ -544,6 +546,7 @@ func (p *AuroraSchedulerManagerClient) sendKillTasks(job *JobKey, instances map[
args := AuroraSchedulerManagerKillTasksArgs{
Job: job,
Instances: instances,
Message: message,
}
if err = args.Write(oprot); err != nil {
return
@ -573,16 +576,16 @@ func (p *AuroraSchedulerManagerClient) recvKillTasks() (value *Response, err err
return
}
if mTypeId == thrift.EXCEPTION {
error142 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error143 error
error143, err = error142.Read(iprot)
error143 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error144 error
error144, err = error143.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error143
err = error144
return
}
if mTypeId != thrift.REPLY {
@ -654,16 +657,16 @@ func (p *AuroraSchedulerManagerClient) recvAddInstances() (value *Response, err
return
}
if mTypeId == thrift.EXCEPTION {
error144 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error145 error
error145, err = error144.Read(iprot)
error145 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error146 error
error146, err = error145.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error145
err = error146
return
}
if mTypeId != thrift.REPLY {
@ -734,16 +737,16 @@ func (p *AuroraSchedulerManagerClient) recvReplaceCronTemplate() (value *Respons
return
}
if mTypeId == thrift.EXCEPTION {
error146 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error147 error
error147, err = error146.Read(iprot)
error147 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error148 error
error148, err = error147.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error147
err = error148
return
}
if mTypeId != thrift.REPLY {
@ -815,16 +818,16 @@ func (p *AuroraSchedulerManagerClient) recvStartJobUpdate() (value *Response, er
return
}
if mTypeId == thrift.EXCEPTION {
error148 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error149 error
error149, err = error148.Read(iprot)
error149 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error150 error
error150, err = error149.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error149
err = error150
return
}
if mTypeId != thrift.REPLY {
@ -896,16 +899,16 @@ func (p *AuroraSchedulerManagerClient) recvPauseJobUpdate() (value *Response, er
return
}
if mTypeId == thrift.EXCEPTION {
error150 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error151 error
error151, err = error150.Read(iprot)
error151 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error152 error
error152, err = error151.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error151
err = error152
return
}
if mTypeId != thrift.REPLY {
@ -977,16 +980,16 @@ func (p *AuroraSchedulerManagerClient) recvResumeJobUpdate() (value *Response, e
return
}
if mTypeId == thrift.EXCEPTION {
error152 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error153 error
error153, err = error152.Read(iprot)
error153 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error154 error
error154, err = error153.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error153
err = error154
return
}
if mTypeId != thrift.REPLY {
@ -1058,16 +1061,16 @@ func (p *AuroraSchedulerManagerClient) recvAbortJobUpdate() (value *Response, er
return
}
if mTypeId == thrift.EXCEPTION {
error154 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error155 error
error155, err = error154.Read(iprot)
error155 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error156 error
error156, err = error155.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error155
err = error156
return
}
if mTypeId != thrift.REPLY {
@ -1139,16 +1142,16 @@ func (p *AuroraSchedulerManagerClient) recvRollbackJobUpdate() (value *Response,
return
}
if mTypeId == thrift.EXCEPTION {
error156 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error157 error
error157, err = error156.Read(iprot)
error157 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error158 error
error158, err = error157.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error157
err = error158
return
}
if mTypeId != thrift.REPLY {
@ -1220,16 +1223,16 @@ func (p *AuroraSchedulerManagerClient) recvPulseJobUpdate() (value *Response, er
return
}
if mTypeId == thrift.EXCEPTION {
error158 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error159 error
error159, err = error158.Read(iprot)
error159 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error160 error
error160, err = error159.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error159
err = error160
return
}
if mTypeId != thrift.REPLY {
@ -1252,22 +1255,22 @@ type AuroraSchedulerManagerProcessor struct {
}
func NewAuroraSchedulerManagerProcessor(handler AuroraSchedulerManager) *AuroraSchedulerManagerProcessor {
self160 := &AuroraSchedulerManagerProcessor{NewReadOnlySchedulerProcessor(handler)}
self160.AddToProcessorMap("createJob", &auroraSchedulerManagerProcessorCreateJob{handler: handler})
self160.AddToProcessorMap("scheduleCronJob", &auroraSchedulerManagerProcessorScheduleCronJob{handler: handler})
self160.AddToProcessorMap("descheduleCronJob", &auroraSchedulerManagerProcessorDescheduleCronJob{handler: handler})
self160.AddToProcessorMap("startCronJob", &auroraSchedulerManagerProcessorStartCronJob{handler: handler})
self160.AddToProcessorMap("restartShards", &auroraSchedulerManagerProcessorRestartShards{handler: handler})
self160.AddToProcessorMap("killTasks", &auroraSchedulerManagerProcessorKillTasks{handler: handler})
self160.AddToProcessorMap("addInstances", &auroraSchedulerManagerProcessorAddInstances{handler: handler})
self160.AddToProcessorMap("replaceCronTemplate", &auroraSchedulerManagerProcessorReplaceCronTemplate{handler: handler})
self160.AddToProcessorMap("startJobUpdate", &auroraSchedulerManagerProcessorStartJobUpdate{handler: handler})
self160.AddToProcessorMap("pauseJobUpdate", &auroraSchedulerManagerProcessorPauseJobUpdate{handler: handler})
self160.AddToProcessorMap("resumeJobUpdate", &auroraSchedulerManagerProcessorResumeJobUpdate{handler: handler})
self160.AddToProcessorMap("abortJobUpdate", &auroraSchedulerManagerProcessorAbortJobUpdate{handler: handler})
self160.AddToProcessorMap("rollbackJobUpdate", &auroraSchedulerManagerProcessorRollbackJobUpdate{handler: handler})
self160.AddToProcessorMap("pulseJobUpdate", &auroraSchedulerManagerProcessorPulseJobUpdate{handler: handler})
return self160
self161 := &AuroraSchedulerManagerProcessor{NewReadOnlySchedulerProcessor(handler)}
self161.AddToProcessorMap("createJob", &auroraSchedulerManagerProcessorCreateJob{handler: handler})
self161.AddToProcessorMap("scheduleCronJob", &auroraSchedulerManagerProcessorScheduleCronJob{handler: handler})
self161.AddToProcessorMap("descheduleCronJob", &auroraSchedulerManagerProcessorDescheduleCronJob{handler: handler})
self161.AddToProcessorMap("startCronJob", &auroraSchedulerManagerProcessorStartCronJob{handler: handler})
self161.AddToProcessorMap("restartShards", &auroraSchedulerManagerProcessorRestartShards{handler: handler})
self161.AddToProcessorMap("killTasks", &auroraSchedulerManagerProcessorKillTasks{handler: handler})
self161.AddToProcessorMap("addInstances", &auroraSchedulerManagerProcessorAddInstances{handler: handler})
self161.AddToProcessorMap("replaceCronTemplate", &auroraSchedulerManagerProcessorReplaceCronTemplate{handler: handler})
self161.AddToProcessorMap("startJobUpdate", &auroraSchedulerManagerProcessorStartJobUpdate{handler: handler})
self161.AddToProcessorMap("pauseJobUpdate", &auroraSchedulerManagerProcessorPauseJobUpdate{handler: handler})
self161.AddToProcessorMap("resumeJobUpdate", &auroraSchedulerManagerProcessorResumeJobUpdate{handler: handler})
self161.AddToProcessorMap("abortJobUpdate", &auroraSchedulerManagerProcessorAbortJobUpdate{handler: handler})
self161.AddToProcessorMap("rollbackJobUpdate", &auroraSchedulerManagerProcessorRollbackJobUpdate{handler: handler})
self161.AddToProcessorMap("pulseJobUpdate", &auroraSchedulerManagerProcessorPulseJobUpdate{handler: handler})
return self161
}
type auroraSchedulerManagerProcessorCreateJob struct {
@ -1530,7 +1533,7 @@ func (p *auroraSchedulerManagerProcessorKillTasks) Process(seqId int32, iprot, o
result := AuroraSchedulerManagerKillTasksResult{}
var retval *Response
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())
oprot.WriteMessageBegin("killTasks", thrift.EXCEPTION, seqId)
x.Write(oprot)
@ -2829,13 +2832,13 @@ func (p *AuroraSchedulerManagerRestartShardsArgs) readField3(iprot thrift.TProto
tSet := make(map[int32]bool, size)
p.ShardIds = tSet
for i := 0; i < size; i++ {
var _elem161 int32
var _elem162 int32
if v, err := iprot.ReadI32(); err != nil {
return thrift.PrependError("error reading field 0: ", err)
} else {
_elem161 = v
_elem162 = v
}
p.ShardIds[_elem161] = true
p.ShardIds[_elem162] = true
}
if err := iprot.ReadSetEnd(); err != nil {
return thrift.PrependError("error reading set end: ", err)
@ -3007,10 +3010,12 @@ func (p *AuroraSchedulerManagerRestartShardsResult) String() string {
// Attributes:
// - Job
// - Instances
// - Message
type AuroraSchedulerManagerKillTasksArgs struct {
// unused fields # 1 to 3
Job *JobKey `thrift:"job,4" json:"job"`
Instances map[int32]bool `thrift:"instances,5" json:"instances"`
Message string `thrift:"message,6" json:"message"`
}
func NewAuroraSchedulerManagerKillTasksArgs() *AuroraSchedulerManagerKillTasksArgs {
@ -3029,6 +3034,10 @@ func (p *AuroraSchedulerManagerKillTasksArgs) GetJob() *JobKey {
func (p *AuroraSchedulerManagerKillTasksArgs) GetInstances() map[int32]bool {
return p.Instances
}
func (p *AuroraSchedulerManagerKillTasksArgs) GetMessage() string {
return p.Message
}
func (p *AuroraSchedulerManagerKillTasksArgs) IsSetJob() bool {
return p.Job != nil
}
@ -3055,6 +3064,10 @@ func (p *AuroraSchedulerManagerKillTasksArgs) Read(iprot thrift.TProtocol) error
if err := p.readField5(iprot); err != nil {
return err
}
case 6:
if err := p.readField6(iprot); err != nil {
return err
}
default:
if err := iprot.Skip(fieldTypeId); err != nil {
return err
@ -3086,13 +3099,13 @@ func (p *AuroraSchedulerManagerKillTasksArgs) readField5(iprot thrift.TProtocol)
tSet := make(map[int32]bool, size)
p.Instances = tSet
for i := 0; i < size; i++ {
var _elem162 int32
var _elem163 int32
if v, err := iprot.ReadI32(); err != nil {
return thrift.PrependError("error reading field 0: ", err)
} else {
_elem162 = v
_elem163 = v
}
p.Instances[_elem162] = true
p.Instances[_elem163] = true
}
if err := iprot.ReadSetEnd(); err != nil {
return thrift.PrependError("error reading set end: ", err)
@ -3100,6 +3113,15 @@ func (p *AuroraSchedulerManagerKillTasksArgs) readField5(iprot thrift.TProtocol)
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 {
if err := oprot.WriteStructBegin("killTasks_args"); err != nil {
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 {
return err
}
if err := p.writeField6(oprot); err != nil {
return err
}
if err := oprot.WriteFieldStop(); err != nil {
return thrift.PrependError("write field stop error: ", err)
}
@ -3153,6 +3178,19 @@ func (p *AuroraSchedulerManagerKillTasksArgs) writeField5(oprot thrift.TProtocol
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 {
if p == nil {
return "<nil>"