Moving gen-go out of vendored folder into it's own package so that it can be imported by other projects

This commit is contained in:
Renan DelValle 2016-09-19 13:15:44 -04:00
parent 494f733f4e
commit 4408aefaad
9 changed files with 32167 additions and 2 deletions

File diff suppressed because it is too large Load diff

View file

@ -0,0 +1,799 @@
// Autogenerated by Thrift Compiler (0.9.3)
// DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
package main
import (
"apache/aurora"
"flag"
"fmt"
"git.apache.org/thrift.git/lib/go/thrift"
"math"
"net"
"net/url"
"os"
"strconv"
"strings"
)
func Usage() {
fmt.Fprintln(os.Stderr, "Usage of ", os.Args[0], " [-h host:port] [-u url] [-f[ramed]] function [arg1 [arg2...]]:")
flag.PrintDefaults()
fmt.Fprintln(os.Stderr, "\nFunctions:")
fmt.Fprintln(os.Stderr, " Response createJob(JobConfiguration description)")
fmt.Fprintln(os.Stderr, " Response scheduleCronJob(JobConfiguration description)")
fmt.Fprintln(os.Stderr, " Response descheduleCronJob(JobKey job)")
fmt.Fprintln(os.Stderr, " Response startCronJob(JobKey job)")
fmt.Fprintln(os.Stderr, " Response restartShards(JobKey job, shardIds)")
fmt.Fprintln(os.Stderr, " Response killTasks(JobKey job, instances)")
fmt.Fprintln(os.Stderr, " Response addInstances(InstanceKey key, i32 count)")
fmt.Fprintln(os.Stderr, " Response replaceCronTemplate(JobConfiguration config)")
fmt.Fprintln(os.Stderr, " Response startJobUpdate(JobUpdateRequest request, string message)")
fmt.Fprintln(os.Stderr, " Response pauseJobUpdate(JobUpdateKey key, string message)")
fmt.Fprintln(os.Stderr, " Response resumeJobUpdate(JobUpdateKey key, string message)")
fmt.Fprintln(os.Stderr, " Response abortJobUpdate(JobUpdateKey key, string message)")
fmt.Fprintln(os.Stderr, " Response rollbackJobUpdate(JobUpdateKey key, string message)")
fmt.Fprintln(os.Stderr, " Response pulseJobUpdate(JobUpdateKey key)")
fmt.Fprintln(os.Stderr, " Response getRoleSummary()")
fmt.Fprintln(os.Stderr, " Response getJobSummary(string role)")
fmt.Fprintln(os.Stderr, " Response getTasksStatus(TaskQuery query)")
fmt.Fprintln(os.Stderr, " Response getTasksWithoutConfigs(TaskQuery query)")
fmt.Fprintln(os.Stderr, " Response getPendingReason(TaskQuery query)")
fmt.Fprintln(os.Stderr, " Response getConfigSummary(JobKey job)")
fmt.Fprintln(os.Stderr, " Response getJobs(string ownerRole)")
fmt.Fprintln(os.Stderr, " Response getQuota(string ownerRole)")
fmt.Fprintln(os.Stderr, " Response populateJobConfig(JobConfiguration description)")
fmt.Fprintln(os.Stderr, " Response getJobUpdateSummaries(JobUpdateQuery jobUpdateQuery)")
fmt.Fprintln(os.Stderr, " Response getJobUpdateDetails(JobUpdateKey key)")
fmt.Fprintln(os.Stderr, " Response getJobUpdateDiff(JobUpdateRequest request)")
fmt.Fprintln(os.Stderr, " Response getTierConfigs()")
fmt.Fprintln(os.Stderr)
os.Exit(0)
}
func main() {
flag.Usage = Usage
var host string
var port int
var protocol string
var urlString string
var framed bool
var useHttp bool
var parsedUrl url.URL
var trans thrift.TTransport
_ = strconv.Atoi
_ = math.Abs
flag.Usage = Usage
flag.StringVar(&host, "h", "localhost", "Specify host and port")
flag.IntVar(&port, "p", 9090, "Specify port")
flag.StringVar(&protocol, "P", "binary", "Specify the protocol (binary, compact, simplejson, json)")
flag.StringVar(&urlString, "u", "", "Specify the url")
flag.BoolVar(&framed, "framed", false, "Use framed transport")
flag.BoolVar(&useHttp, "http", false, "Use http")
flag.Parse()
if len(urlString) > 0 {
parsedUrl, err := url.Parse(urlString)
if err != nil {
fmt.Fprintln(os.Stderr, "Error parsing URL: ", err)
flag.Usage()
}
host = parsedUrl.Host
useHttp = len(parsedUrl.Scheme) <= 0 || parsedUrl.Scheme == "http"
} else if useHttp {
_, err := url.Parse(fmt.Sprint("http://", host, ":", port))
if err != nil {
fmt.Fprintln(os.Stderr, "Error parsing URL: ", err)
flag.Usage()
}
}
cmd := flag.Arg(0)
var err error
if useHttp {
trans, err = thrift.NewTHttpClient(parsedUrl.String())
} else {
portStr := fmt.Sprint(port)
if strings.Contains(host, ":") {
host, portStr, err = net.SplitHostPort(host)
if err != nil {
fmt.Fprintln(os.Stderr, "error with host:", err)
os.Exit(1)
}
}
trans, err = thrift.NewTSocket(net.JoinHostPort(host, portStr))
if err != nil {
fmt.Fprintln(os.Stderr, "error resolving address:", err)
os.Exit(1)
}
if framed {
trans = thrift.NewTFramedTransport(trans)
}
}
if err != nil {
fmt.Fprintln(os.Stderr, "Error creating transport", err)
os.Exit(1)
}
defer trans.Close()
var protocolFactory thrift.TProtocolFactory
switch protocol {
case "compact":
protocolFactory = thrift.NewTCompactProtocolFactory()
break
case "simplejson":
protocolFactory = thrift.NewTSimpleJSONProtocolFactory()
break
case "json":
protocolFactory = thrift.NewTJSONProtocolFactory()
break
case "binary", "":
protocolFactory = thrift.NewTBinaryProtocolFactoryDefault()
break
default:
fmt.Fprintln(os.Stderr, "Invalid protocol specified: ", protocol)
Usage()
os.Exit(1)
}
client := aurora.NewAuroraSchedulerManagerClientFactory(trans, protocolFactory)
if err := trans.Open(); err != nil {
fmt.Fprintln(os.Stderr, "Error opening socket to ", host, ":", port, " ", err)
os.Exit(1)
}
switch cmd {
case "createJob":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "CreateJob requires 1 args")
flag.Usage()
}
arg160 := flag.Arg(1)
mbTrans161 := thrift.NewTMemoryBufferLen(len(arg160))
defer mbTrans161.Close()
_, err162 := mbTrans161.WriteString(arg160)
if err162 != nil {
Usage()
return
}
factory163 := thrift.NewTSimpleJSONProtocolFactory()
jsProt164 := factory163.GetProtocol(mbTrans161)
argvalue0 := aurora.NewJobConfiguration()
err165 := argvalue0.Read(jsProt164)
if err165 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.CreateJob(value0))
fmt.Print("\n")
break
case "scheduleCronJob":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "ScheduleCronJob requires 1 args")
flag.Usage()
}
arg166 := flag.Arg(1)
mbTrans167 := thrift.NewTMemoryBufferLen(len(arg166))
defer mbTrans167.Close()
_, err168 := mbTrans167.WriteString(arg166)
if err168 != nil {
Usage()
return
}
factory169 := thrift.NewTSimpleJSONProtocolFactory()
jsProt170 := factory169.GetProtocol(mbTrans167)
argvalue0 := aurora.NewJobConfiguration()
err171 := argvalue0.Read(jsProt170)
if err171 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.ScheduleCronJob(value0))
fmt.Print("\n")
break
case "descheduleCronJob":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "DescheduleCronJob requires 1 args")
flag.Usage()
}
arg172 := flag.Arg(1)
mbTrans173 := thrift.NewTMemoryBufferLen(len(arg172))
defer mbTrans173.Close()
_, err174 := mbTrans173.WriteString(arg172)
if err174 != nil {
Usage()
return
}
factory175 := thrift.NewTSimpleJSONProtocolFactory()
jsProt176 := factory175.GetProtocol(mbTrans173)
argvalue0 := aurora.NewJobKey()
err177 := argvalue0.Read(jsProt176)
if err177 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.DescheduleCronJob(value0))
fmt.Print("\n")
break
case "startCronJob":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "StartCronJob requires 1 args")
flag.Usage()
}
arg178 := flag.Arg(1)
mbTrans179 := thrift.NewTMemoryBufferLen(len(arg178))
defer mbTrans179.Close()
_, err180 := mbTrans179.WriteString(arg178)
if err180 != nil {
Usage()
return
}
factory181 := thrift.NewTSimpleJSONProtocolFactory()
jsProt182 := factory181.GetProtocol(mbTrans179)
argvalue0 := aurora.NewJobKey()
err183 := argvalue0.Read(jsProt182)
if err183 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.StartCronJob(value0))
fmt.Print("\n")
break
case "restartShards":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "RestartShards requires 2 args")
flag.Usage()
}
arg184 := flag.Arg(1)
mbTrans185 := thrift.NewTMemoryBufferLen(len(arg184))
defer mbTrans185.Close()
_, err186 := mbTrans185.WriteString(arg184)
if err186 != nil {
Usage()
return
}
factory187 := thrift.NewTSimpleJSONProtocolFactory()
jsProt188 := factory187.GetProtocol(mbTrans185)
argvalue0 := aurora.NewJobKey()
err189 := argvalue0.Read(jsProt188)
if err189 != nil {
Usage()
return
}
value0 := argvalue0
arg190 := flag.Arg(2)
mbTrans191 := thrift.NewTMemoryBufferLen(len(arg190))
defer mbTrans191.Close()
_, err192 := mbTrans191.WriteString(arg190)
if err192 != nil {
Usage()
return
}
factory193 := thrift.NewTSimpleJSONProtocolFactory()
jsProt194 := factory193.GetProtocol(mbTrans191)
containerStruct1 := aurora.NewAuroraSchedulerManagerRestartShardsArgs()
err195 := containerStruct1.ReadField2(jsProt194)
if err195 != nil {
Usage()
return
}
argvalue1 := containerStruct1.ShardIds
value1 := argvalue1
fmt.Print(client.RestartShards(value0, value1))
fmt.Print("\n")
break
case "killTasks":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "KillTasks requires 2 args")
flag.Usage()
}
arg196 := flag.Arg(1)
mbTrans197 := thrift.NewTMemoryBufferLen(len(arg196))
defer mbTrans197.Close()
_, err198 := mbTrans197.WriteString(arg196)
if err198 != nil {
Usage()
return
}
factory199 := thrift.NewTSimpleJSONProtocolFactory()
jsProt200 := factory199.GetProtocol(mbTrans197)
argvalue0 := aurora.NewJobKey()
err201 := argvalue0.Read(jsProt200)
if err201 != nil {
Usage()
return
}
value0 := argvalue0
arg202 := flag.Arg(2)
mbTrans203 := thrift.NewTMemoryBufferLen(len(arg202))
defer mbTrans203.Close()
_, err204 := mbTrans203.WriteString(arg202)
if err204 != nil {
Usage()
return
}
factory205 := thrift.NewTSimpleJSONProtocolFactory()
jsProt206 := factory205.GetProtocol(mbTrans203)
containerStruct1 := aurora.NewAuroraSchedulerManagerKillTasksArgs()
err207 := containerStruct1.ReadField2(jsProt206)
if err207 != nil {
Usage()
return
}
argvalue1 := containerStruct1.Instances
value1 := argvalue1
fmt.Print(client.KillTasks(value0, value1))
fmt.Print("\n")
break
case "addInstances":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "AddInstances requires 2 args")
flag.Usage()
}
arg208 := flag.Arg(1)
mbTrans209 := thrift.NewTMemoryBufferLen(len(arg208))
defer mbTrans209.Close()
_, err210 := mbTrans209.WriteString(arg208)
if err210 != nil {
Usage()
return
}
factory211 := thrift.NewTSimpleJSONProtocolFactory()
jsProt212 := factory211.GetProtocol(mbTrans209)
argvalue0 := aurora.NewInstanceKey()
err213 := argvalue0.Read(jsProt212)
if err213 != nil {
Usage()
return
}
value0 := argvalue0
tmp1, err214 := (strconv.Atoi(flag.Arg(2)))
if err214 != nil {
Usage()
return
}
argvalue1 := int32(tmp1)
value1 := argvalue1
fmt.Print(client.AddInstances(value0, value1))
fmt.Print("\n")
break
case "replaceCronTemplate":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "ReplaceCronTemplate requires 1 args")
flag.Usage()
}
arg215 := flag.Arg(1)
mbTrans216 := thrift.NewTMemoryBufferLen(len(arg215))
defer mbTrans216.Close()
_, err217 := mbTrans216.WriteString(arg215)
if err217 != nil {
Usage()
return
}
factory218 := thrift.NewTSimpleJSONProtocolFactory()
jsProt219 := factory218.GetProtocol(mbTrans216)
argvalue0 := aurora.NewJobConfiguration()
err220 := argvalue0.Read(jsProt219)
if err220 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.ReplaceCronTemplate(value0))
fmt.Print("\n")
break
case "startJobUpdate":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "StartJobUpdate requires 2 args")
flag.Usage()
}
arg221 := flag.Arg(1)
mbTrans222 := thrift.NewTMemoryBufferLen(len(arg221))
defer mbTrans222.Close()
_, err223 := mbTrans222.WriteString(arg221)
if err223 != nil {
Usage()
return
}
factory224 := thrift.NewTSimpleJSONProtocolFactory()
jsProt225 := factory224.GetProtocol(mbTrans222)
argvalue0 := aurora.NewJobUpdateRequest()
err226 := argvalue0.Read(jsProt225)
if err226 != nil {
Usage()
return
}
value0 := argvalue0
argvalue1 := flag.Arg(2)
value1 := argvalue1
fmt.Print(client.StartJobUpdate(value0, value1))
fmt.Print("\n")
break
case "pauseJobUpdate":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "PauseJobUpdate requires 2 args")
flag.Usage()
}
arg228 := flag.Arg(1)
mbTrans229 := thrift.NewTMemoryBufferLen(len(arg228))
defer mbTrans229.Close()
_, err230 := mbTrans229.WriteString(arg228)
if err230 != nil {
Usage()
return
}
factory231 := thrift.NewTSimpleJSONProtocolFactory()
jsProt232 := factory231.GetProtocol(mbTrans229)
argvalue0 := aurora.NewJobUpdateKey()
err233 := argvalue0.Read(jsProt232)
if err233 != nil {
Usage()
return
}
value0 := argvalue0
argvalue1 := flag.Arg(2)
value1 := argvalue1
fmt.Print(client.PauseJobUpdate(value0, value1))
fmt.Print("\n")
break
case "resumeJobUpdate":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "ResumeJobUpdate requires 2 args")
flag.Usage()
}
arg235 := flag.Arg(1)
mbTrans236 := thrift.NewTMemoryBufferLen(len(arg235))
defer mbTrans236.Close()
_, err237 := mbTrans236.WriteString(arg235)
if err237 != nil {
Usage()
return
}
factory238 := thrift.NewTSimpleJSONProtocolFactory()
jsProt239 := factory238.GetProtocol(mbTrans236)
argvalue0 := aurora.NewJobUpdateKey()
err240 := argvalue0.Read(jsProt239)
if err240 != nil {
Usage()
return
}
value0 := argvalue0
argvalue1 := flag.Arg(2)
value1 := argvalue1
fmt.Print(client.ResumeJobUpdate(value0, value1))
fmt.Print("\n")
break
case "abortJobUpdate":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "AbortJobUpdate requires 2 args")
flag.Usage()
}
arg242 := flag.Arg(1)
mbTrans243 := thrift.NewTMemoryBufferLen(len(arg242))
defer mbTrans243.Close()
_, err244 := mbTrans243.WriteString(arg242)
if err244 != nil {
Usage()
return
}
factory245 := thrift.NewTSimpleJSONProtocolFactory()
jsProt246 := factory245.GetProtocol(mbTrans243)
argvalue0 := aurora.NewJobUpdateKey()
err247 := argvalue0.Read(jsProt246)
if err247 != nil {
Usage()
return
}
value0 := argvalue0
argvalue1 := flag.Arg(2)
value1 := argvalue1
fmt.Print(client.AbortJobUpdate(value0, value1))
fmt.Print("\n")
break
case "rollbackJobUpdate":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "RollbackJobUpdate requires 2 args")
flag.Usage()
}
arg249 := flag.Arg(1)
mbTrans250 := thrift.NewTMemoryBufferLen(len(arg249))
defer mbTrans250.Close()
_, err251 := mbTrans250.WriteString(arg249)
if err251 != nil {
Usage()
return
}
factory252 := thrift.NewTSimpleJSONProtocolFactory()
jsProt253 := factory252.GetProtocol(mbTrans250)
argvalue0 := aurora.NewJobUpdateKey()
err254 := argvalue0.Read(jsProt253)
if err254 != nil {
Usage()
return
}
value0 := argvalue0
argvalue1 := flag.Arg(2)
value1 := argvalue1
fmt.Print(client.RollbackJobUpdate(value0, value1))
fmt.Print("\n")
break
case "pulseJobUpdate":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "PulseJobUpdate requires 1 args")
flag.Usage()
}
arg256 := flag.Arg(1)
mbTrans257 := thrift.NewTMemoryBufferLen(len(arg256))
defer mbTrans257.Close()
_, err258 := mbTrans257.WriteString(arg256)
if err258 != nil {
Usage()
return
}
factory259 := thrift.NewTSimpleJSONProtocolFactory()
jsProt260 := factory259.GetProtocol(mbTrans257)
argvalue0 := aurora.NewJobUpdateKey()
err261 := argvalue0.Read(jsProt260)
if err261 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.PulseJobUpdate(value0))
fmt.Print("\n")
break
case "getRoleSummary":
if flag.NArg()-1 != 0 {
fmt.Fprintln(os.Stderr, "GetRoleSummary requires 0 args")
flag.Usage()
}
fmt.Print(client.GetRoleSummary())
fmt.Print("\n")
break
case "getJobSummary":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobSummary requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.GetJobSummary(value0))
fmt.Print("\n")
break
case "getTasksStatus":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetTasksStatus requires 1 args")
flag.Usage()
}
arg263 := flag.Arg(1)
mbTrans264 := thrift.NewTMemoryBufferLen(len(arg263))
defer mbTrans264.Close()
_, err265 := mbTrans264.WriteString(arg263)
if err265 != nil {
Usage()
return
}
factory266 := thrift.NewTSimpleJSONProtocolFactory()
jsProt267 := factory266.GetProtocol(mbTrans264)
argvalue0 := aurora.NewTaskQuery()
err268 := argvalue0.Read(jsProt267)
if err268 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetTasksStatus(value0))
fmt.Print("\n")
break
case "getTasksWithoutConfigs":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetTasksWithoutConfigs requires 1 args")
flag.Usage()
}
arg269 := flag.Arg(1)
mbTrans270 := thrift.NewTMemoryBufferLen(len(arg269))
defer mbTrans270.Close()
_, err271 := mbTrans270.WriteString(arg269)
if err271 != nil {
Usage()
return
}
factory272 := thrift.NewTSimpleJSONProtocolFactory()
jsProt273 := factory272.GetProtocol(mbTrans270)
argvalue0 := aurora.NewTaskQuery()
err274 := argvalue0.Read(jsProt273)
if err274 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetTasksWithoutConfigs(value0))
fmt.Print("\n")
break
case "getPendingReason":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetPendingReason requires 1 args")
flag.Usage()
}
arg275 := flag.Arg(1)
mbTrans276 := thrift.NewTMemoryBufferLen(len(arg275))
defer mbTrans276.Close()
_, err277 := mbTrans276.WriteString(arg275)
if err277 != nil {
Usage()
return
}
factory278 := thrift.NewTSimpleJSONProtocolFactory()
jsProt279 := factory278.GetProtocol(mbTrans276)
argvalue0 := aurora.NewTaskQuery()
err280 := argvalue0.Read(jsProt279)
if err280 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetPendingReason(value0))
fmt.Print("\n")
break
case "getConfigSummary":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetConfigSummary requires 1 args")
flag.Usage()
}
arg281 := flag.Arg(1)
mbTrans282 := thrift.NewTMemoryBufferLen(len(arg281))
defer mbTrans282.Close()
_, err283 := mbTrans282.WriteString(arg281)
if err283 != nil {
Usage()
return
}
factory284 := thrift.NewTSimpleJSONProtocolFactory()
jsProt285 := factory284.GetProtocol(mbTrans282)
argvalue0 := aurora.NewJobKey()
err286 := argvalue0.Read(jsProt285)
if err286 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetConfigSummary(value0))
fmt.Print("\n")
break
case "getJobs":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobs requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.GetJobs(value0))
fmt.Print("\n")
break
case "getQuota":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetQuota requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.GetQuota(value0))
fmt.Print("\n")
break
case "populateJobConfig":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "PopulateJobConfig requires 1 args")
flag.Usage()
}
arg289 := flag.Arg(1)
mbTrans290 := thrift.NewTMemoryBufferLen(len(arg289))
defer mbTrans290.Close()
_, err291 := mbTrans290.WriteString(arg289)
if err291 != nil {
Usage()
return
}
factory292 := thrift.NewTSimpleJSONProtocolFactory()
jsProt293 := factory292.GetProtocol(mbTrans290)
argvalue0 := aurora.NewJobConfiguration()
err294 := argvalue0.Read(jsProt293)
if err294 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.PopulateJobConfig(value0))
fmt.Print("\n")
break
case "getJobUpdateSummaries":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobUpdateSummaries requires 1 args")
flag.Usage()
}
arg295 := flag.Arg(1)
mbTrans296 := thrift.NewTMemoryBufferLen(len(arg295))
defer mbTrans296.Close()
_, err297 := mbTrans296.WriteString(arg295)
if err297 != nil {
Usage()
return
}
factory298 := thrift.NewTSimpleJSONProtocolFactory()
jsProt299 := factory298.GetProtocol(mbTrans296)
argvalue0 := aurora.NewJobUpdateQuery()
err300 := argvalue0.Read(jsProt299)
if err300 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetJobUpdateSummaries(value0))
fmt.Print("\n")
break
case "getJobUpdateDetails":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobUpdateDetails requires 1 args")
flag.Usage()
}
arg301 := flag.Arg(1)
mbTrans302 := thrift.NewTMemoryBufferLen(len(arg301))
defer mbTrans302.Close()
_, err303 := mbTrans302.WriteString(arg301)
if err303 != nil {
Usage()
return
}
factory304 := thrift.NewTSimpleJSONProtocolFactory()
jsProt305 := factory304.GetProtocol(mbTrans302)
argvalue0 := aurora.NewJobUpdateKey()
err306 := argvalue0.Read(jsProt305)
if err306 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetJobUpdateDetails(value0))
fmt.Print("\n")
break
case "getJobUpdateDiff":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobUpdateDiff requires 1 args")
flag.Usage()
}
arg307 := flag.Arg(1)
mbTrans308 := thrift.NewTMemoryBufferLen(len(arg307))
defer mbTrans308.Close()
_, err309 := mbTrans308.WriteString(arg307)
if err309 != nil {
Usage()
return
}
factory310 := thrift.NewTSimpleJSONProtocolFactory()
jsProt311 := factory310.GetProtocol(mbTrans308)
argvalue0 := aurora.NewJobUpdateRequest()
err312 := argvalue0.Read(jsProt311)
if err312 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetJobUpdateDiff(value0))
fmt.Print("\n")
break
case "getTierConfigs":
if flag.NArg()-1 != 0 {
fmt.Fprintln(os.Stderr, "GetTierConfigs requires 0 args")
flag.Usage()
}
fmt.Print(client.GetTierConfigs())
fmt.Print("\n")
break
case "":
Usage()
break
default:
fmt.Fprintln(os.Stderr, "Invalid function ", cmd)
}
}

File diff suppressed because it is too large Load diff

File diff suppressed because it is too large Load diff

View file

@ -0,0 +1,80 @@
// Autogenerated by Thrift Compiler (0.9.3)
// DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
package aurora
import (
"bytes"
"fmt"
"git.apache.org/thrift.git/lib/go/thrift"
)
// (needed to ensure safety because of naive import list construction.)
var _ = thrift.ZERO
var _ = fmt.Printf
var _ = bytes.Equal
const AURORA_EXECUTOR_NAME = "AuroraExecutor"
var ACTIVE_STATES map[ScheduleStatus]bool
var SLAVE_ASSIGNED_STATES map[ScheduleStatus]bool
var LIVE_STATES map[ScheduleStatus]bool
var TERMINAL_STATES map[ScheduleStatus]bool
const GOOD_IDENTIFIER_PATTERN = "^[\\w\\-\\.]+$"
const GOOD_IDENTIFIER_PATTERN_JVM = "^[\\w\\-\\.]+$"
const GOOD_IDENTIFIER_PATTERN_PYTHON = "^[\\w\\-\\.]+$"
var ACTIVE_JOB_UPDATE_STATES map[JobUpdateStatus]bool
const BYPASS_LEADER_REDIRECT_HEADER_NAME = "Bypass-Leader-Redirect"
const TASK_FILESYSTEM_MOUNT_POINT = "taskfs"
func init() {
ACTIVE_STATES = map[ScheduleStatus]bool{
9: true,
17: true,
6: true,
0: true,
13: true,
12: true,
2: true,
1: true,
16: true,
}
SLAVE_ASSIGNED_STATES = map[ScheduleStatus]bool{
9: true,
17: true,
6: true,
13: true,
12: true,
2: true,
1: true,
}
LIVE_STATES = map[ScheduleStatus]bool{
6: true,
13: true,
12: true,
17: true,
2: true,
}
TERMINAL_STATES = map[ScheduleStatus]bool{
4: true,
3: true,
5: true,
7: true,
}
ACTIVE_JOB_UPDATE_STATES = map[JobUpdateStatus]bool{
0: true,
1: true,
2: true,
3: true,
9: true,
10: true,
}
}

View file

@ -0,0 +1,382 @@
// Autogenerated by Thrift Compiler (0.9.3)
// DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
package main
import (
"apache/aurora"
"flag"
"fmt"
"git.apache.org/thrift.git/lib/go/thrift"
"math"
"net"
"net/url"
"os"
"strconv"
"strings"
)
func Usage() {
fmt.Fprintln(os.Stderr, "Usage of ", os.Args[0], " [-h host:port] [-u url] [-f[ramed]] function [arg1 [arg2...]]:")
flag.PrintDefaults()
fmt.Fprintln(os.Stderr, "\nFunctions:")
fmt.Fprintln(os.Stderr, " Response getRoleSummary()")
fmt.Fprintln(os.Stderr, " Response getJobSummary(string role)")
fmt.Fprintln(os.Stderr, " Response getTasksStatus(TaskQuery query)")
fmt.Fprintln(os.Stderr, " Response getTasksWithoutConfigs(TaskQuery query)")
fmt.Fprintln(os.Stderr, " Response getPendingReason(TaskQuery query)")
fmt.Fprintln(os.Stderr, " Response getConfigSummary(JobKey job)")
fmt.Fprintln(os.Stderr, " Response getJobs(string ownerRole)")
fmt.Fprintln(os.Stderr, " Response getQuota(string ownerRole)")
fmt.Fprintln(os.Stderr, " Response populateJobConfig(JobConfiguration description)")
fmt.Fprintln(os.Stderr, " Response getJobUpdateSummaries(JobUpdateQuery jobUpdateQuery)")
fmt.Fprintln(os.Stderr, " Response getJobUpdateDetails(JobUpdateKey key)")
fmt.Fprintln(os.Stderr, " Response getJobUpdateDiff(JobUpdateRequest request)")
fmt.Fprintln(os.Stderr, " Response getTierConfigs()")
fmt.Fprintln(os.Stderr)
os.Exit(0)
}
func main() {
flag.Usage = Usage
var host string
var port int
var protocol string
var urlString string
var framed bool
var useHttp bool
var parsedUrl url.URL
var trans thrift.TTransport
_ = strconv.Atoi
_ = math.Abs
flag.Usage = Usage
flag.StringVar(&host, "h", "localhost", "Specify host and port")
flag.IntVar(&port, "p", 9090, "Specify port")
flag.StringVar(&protocol, "P", "binary", "Specify the protocol (binary, compact, simplejson, json)")
flag.StringVar(&urlString, "u", "", "Specify the url")
flag.BoolVar(&framed, "framed", false, "Use framed transport")
flag.BoolVar(&useHttp, "http", false, "Use http")
flag.Parse()
if len(urlString) > 0 {
parsedUrl, err := url.Parse(urlString)
if err != nil {
fmt.Fprintln(os.Stderr, "Error parsing URL: ", err)
flag.Usage()
}
host = parsedUrl.Host
useHttp = len(parsedUrl.Scheme) <= 0 || parsedUrl.Scheme == "http"
} else if useHttp {
_, err := url.Parse(fmt.Sprint("http://", host, ":", port))
if err != nil {
fmt.Fprintln(os.Stderr, "Error parsing URL: ", err)
flag.Usage()
}
}
cmd := flag.Arg(0)
var err error
if useHttp {
trans, err = thrift.NewTHttpClient(parsedUrl.String())
} else {
portStr := fmt.Sprint(port)
if strings.Contains(host, ":") {
host, portStr, err = net.SplitHostPort(host)
if err != nil {
fmt.Fprintln(os.Stderr, "error with host:", err)
os.Exit(1)
}
}
trans, err = thrift.NewTSocket(net.JoinHostPort(host, portStr))
if err != nil {
fmt.Fprintln(os.Stderr, "error resolving address:", err)
os.Exit(1)
}
if framed {
trans = thrift.NewTFramedTransport(trans)
}
}
if err != nil {
fmt.Fprintln(os.Stderr, "Error creating transport", err)
os.Exit(1)
}
defer trans.Close()
var protocolFactory thrift.TProtocolFactory
switch protocol {
case "compact":
protocolFactory = thrift.NewTCompactProtocolFactory()
break
case "simplejson":
protocolFactory = thrift.NewTSimpleJSONProtocolFactory()
break
case "json":
protocolFactory = thrift.NewTJSONProtocolFactory()
break
case "binary", "":
protocolFactory = thrift.NewTBinaryProtocolFactoryDefault()
break
default:
fmt.Fprintln(os.Stderr, "Invalid protocol specified: ", protocol)
Usage()
os.Exit(1)
}
client := aurora.NewReadOnlySchedulerClientFactory(trans, protocolFactory)
if err := trans.Open(); err != nil {
fmt.Fprintln(os.Stderr, "Error opening socket to ", host, ":", port, " ", err)
os.Exit(1)
}
switch cmd {
case "getRoleSummary":
if flag.NArg()-1 != 0 {
fmt.Fprintln(os.Stderr, "GetRoleSummary requires 0 args")
flag.Usage()
}
fmt.Print(client.GetRoleSummary())
fmt.Print("\n")
break
case "getJobSummary":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobSummary requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.GetJobSummary(value0))
fmt.Print("\n")
break
case "getTasksStatus":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetTasksStatus requires 1 args")
flag.Usage()
}
arg79 := flag.Arg(1)
mbTrans80 := thrift.NewTMemoryBufferLen(len(arg79))
defer mbTrans80.Close()
_, err81 := mbTrans80.WriteString(arg79)
if err81 != nil {
Usage()
return
}
factory82 := thrift.NewTSimpleJSONProtocolFactory()
jsProt83 := factory82.GetProtocol(mbTrans80)
argvalue0 := aurora.NewTaskQuery()
err84 := argvalue0.Read(jsProt83)
if err84 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetTasksStatus(value0))
fmt.Print("\n")
break
case "getTasksWithoutConfigs":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetTasksWithoutConfigs requires 1 args")
flag.Usage()
}
arg85 := flag.Arg(1)
mbTrans86 := thrift.NewTMemoryBufferLen(len(arg85))
defer mbTrans86.Close()
_, err87 := mbTrans86.WriteString(arg85)
if err87 != nil {
Usage()
return
}
factory88 := thrift.NewTSimpleJSONProtocolFactory()
jsProt89 := factory88.GetProtocol(mbTrans86)
argvalue0 := aurora.NewTaskQuery()
err90 := argvalue0.Read(jsProt89)
if err90 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetTasksWithoutConfigs(value0))
fmt.Print("\n")
break
case "getPendingReason":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetPendingReason requires 1 args")
flag.Usage()
}
arg91 := flag.Arg(1)
mbTrans92 := thrift.NewTMemoryBufferLen(len(arg91))
defer mbTrans92.Close()
_, err93 := mbTrans92.WriteString(arg91)
if err93 != nil {
Usage()
return
}
factory94 := thrift.NewTSimpleJSONProtocolFactory()
jsProt95 := factory94.GetProtocol(mbTrans92)
argvalue0 := aurora.NewTaskQuery()
err96 := argvalue0.Read(jsProt95)
if err96 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetPendingReason(value0))
fmt.Print("\n")
break
case "getConfigSummary":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetConfigSummary requires 1 args")
flag.Usage()
}
arg97 := flag.Arg(1)
mbTrans98 := thrift.NewTMemoryBufferLen(len(arg97))
defer mbTrans98.Close()
_, err99 := mbTrans98.WriteString(arg97)
if err99 != nil {
Usage()
return
}
factory100 := thrift.NewTSimpleJSONProtocolFactory()
jsProt101 := factory100.GetProtocol(mbTrans98)
argvalue0 := aurora.NewJobKey()
err102 := argvalue0.Read(jsProt101)
if err102 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetConfigSummary(value0))
fmt.Print("\n")
break
case "getJobs":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobs requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.GetJobs(value0))
fmt.Print("\n")
break
case "getQuota":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetQuota requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.GetQuota(value0))
fmt.Print("\n")
break
case "populateJobConfig":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "PopulateJobConfig requires 1 args")
flag.Usage()
}
arg105 := flag.Arg(1)
mbTrans106 := thrift.NewTMemoryBufferLen(len(arg105))
defer mbTrans106.Close()
_, err107 := mbTrans106.WriteString(arg105)
if err107 != nil {
Usage()
return
}
factory108 := thrift.NewTSimpleJSONProtocolFactory()
jsProt109 := factory108.GetProtocol(mbTrans106)
argvalue0 := aurora.NewJobConfiguration()
err110 := argvalue0.Read(jsProt109)
if err110 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.PopulateJobConfig(value0))
fmt.Print("\n")
break
case "getJobUpdateSummaries":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobUpdateSummaries requires 1 args")
flag.Usage()
}
arg111 := flag.Arg(1)
mbTrans112 := thrift.NewTMemoryBufferLen(len(arg111))
defer mbTrans112.Close()
_, err113 := mbTrans112.WriteString(arg111)
if err113 != nil {
Usage()
return
}
factory114 := thrift.NewTSimpleJSONProtocolFactory()
jsProt115 := factory114.GetProtocol(mbTrans112)
argvalue0 := aurora.NewJobUpdateQuery()
err116 := argvalue0.Read(jsProt115)
if err116 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetJobUpdateSummaries(value0))
fmt.Print("\n")
break
case "getJobUpdateDetails":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobUpdateDetails requires 1 args")
flag.Usage()
}
arg117 := flag.Arg(1)
mbTrans118 := thrift.NewTMemoryBufferLen(len(arg117))
defer mbTrans118.Close()
_, err119 := mbTrans118.WriteString(arg117)
if err119 != nil {
Usage()
return
}
factory120 := thrift.NewTSimpleJSONProtocolFactory()
jsProt121 := factory120.GetProtocol(mbTrans118)
argvalue0 := aurora.NewJobUpdateKey()
err122 := argvalue0.Read(jsProt121)
if err122 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetJobUpdateDetails(value0))
fmt.Print("\n")
break
case "getJobUpdateDiff":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "GetJobUpdateDiff requires 1 args")
flag.Usage()
}
arg123 := flag.Arg(1)
mbTrans124 := thrift.NewTMemoryBufferLen(len(arg123))
defer mbTrans124.Close()
_, err125 := mbTrans124.WriteString(arg123)
if err125 != nil {
Usage()
return
}
factory126 := thrift.NewTSimpleJSONProtocolFactory()
jsProt127 := factory126.GetProtocol(mbTrans124)
argvalue0 := aurora.NewJobUpdateRequest()
err128 := argvalue0.Read(jsProt127)
if err128 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.GetJobUpdateDiff(value0))
fmt.Print("\n")
break
case "getTierConfigs":
if flag.NArg()-1 != 0 {
fmt.Fprintln(os.Stderr, "GetTierConfigs requires 0 args")
flag.Usage()
}
fmt.Print(client.GetTierConfigs())
fmt.Print("\n")
break
case "":
Usage()
break
default:
fmt.Fprintln(os.Stderr, "Invalid function ", cmd)
}
}

File diff suppressed because it is too large Load diff

15611
gen-go/apache/aurora/ttypes.go Normal file

File diff suppressed because it is too large Load diff