Public release of gorealis

This commit is contained in:
Renan DelValle 2016-08-02 11:42:00 -07:00
commit 327c6c369d
82 changed files with 45075 additions and 0 deletions

File diff suppressed because it is too large Load diff

View file

@ -0,0 +1,771 @@
// 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 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()
}
arg158 := flag.Arg(1)
mbTrans159 := thrift.NewTMemoryBufferLen(len(arg158))
defer mbTrans159.Close()
_, err160 := mbTrans159.WriteString(arg158)
if err160 != nil {
Usage()
return
}
factory161 := thrift.NewTSimpleJSONProtocolFactory()
jsProt162 := factory161.GetProtocol(mbTrans159)
argvalue0 := aurora.NewJobConfiguration()
err163 := argvalue0.Read(jsProt162)
if err163 != 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()
}
arg164 := flag.Arg(1)
mbTrans165 := thrift.NewTMemoryBufferLen(len(arg164))
defer mbTrans165.Close()
_, err166 := mbTrans165.WriteString(arg164)
if err166 != nil {
Usage()
return
}
factory167 := thrift.NewTSimpleJSONProtocolFactory()
jsProt168 := factory167.GetProtocol(mbTrans165)
argvalue0 := aurora.NewJobConfiguration()
err169 := argvalue0.Read(jsProt168)
if err169 != 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()
}
arg170 := flag.Arg(1)
mbTrans171 := thrift.NewTMemoryBufferLen(len(arg170))
defer mbTrans171.Close()
_, err172 := mbTrans171.WriteString(arg170)
if err172 != nil {
Usage()
return
}
factory173 := thrift.NewTSimpleJSONProtocolFactory()
jsProt174 := factory173.GetProtocol(mbTrans171)
argvalue0 := aurora.NewJobKey()
err175 := argvalue0.Read(jsProt174)
if err175 != 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()
}
arg176 := flag.Arg(1)
mbTrans177 := thrift.NewTMemoryBufferLen(len(arg176))
defer mbTrans177.Close()
_, err178 := mbTrans177.WriteString(arg176)
if err178 != nil {
Usage()
return
}
factory179 := thrift.NewTSimpleJSONProtocolFactory()
jsProt180 := factory179.GetProtocol(mbTrans177)
argvalue0 := aurora.NewJobKey()
err181 := argvalue0.Read(jsProt180)
if err181 != 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()
}
arg182 := flag.Arg(1)
mbTrans183 := thrift.NewTMemoryBufferLen(len(arg182))
defer mbTrans183.Close()
_, err184 := mbTrans183.WriteString(arg182)
if err184 != nil {
Usage()
return
}
factory185 := thrift.NewTSimpleJSONProtocolFactory()
jsProt186 := factory185.GetProtocol(mbTrans183)
argvalue0 := aurora.NewJobKey()
err187 := argvalue0.Read(jsProt186)
if err187 != nil {
Usage()
return
}
value0 := argvalue0
arg188 := flag.Arg(2)
mbTrans189 := thrift.NewTMemoryBufferLen(len(arg188))
defer mbTrans189.Close()
_, err190 := mbTrans189.WriteString(arg188)
if err190 != nil {
Usage()
return
}
factory191 := thrift.NewTSimpleJSONProtocolFactory()
jsProt192 := factory191.GetProtocol(mbTrans189)
containerStruct1 := aurora.NewAuroraSchedulerManagerRestartShardsArgs()
err193 := containerStruct1.ReadField2(jsProt192)
if err193 != 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()
}
arg194 := flag.Arg(1)
mbTrans195 := thrift.NewTMemoryBufferLen(len(arg194))
defer mbTrans195.Close()
_, err196 := mbTrans195.WriteString(arg194)
if err196 != nil {
Usage()
return
}
factory197 := thrift.NewTSimpleJSONProtocolFactory()
jsProt198 := factory197.GetProtocol(mbTrans195)
argvalue0 := aurora.NewJobKey()
err199 := argvalue0.Read(jsProt198)
if err199 != nil {
Usage()
return
}
value0 := argvalue0
arg200 := flag.Arg(2)
mbTrans201 := thrift.NewTMemoryBufferLen(len(arg200))
defer mbTrans201.Close()
_, err202 := mbTrans201.WriteString(arg200)
if err202 != nil {
Usage()
return
}
factory203 := thrift.NewTSimpleJSONProtocolFactory()
jsProt204 := factory203.GetProtocol(mbTrans201)
containerStruct1 := aurora.NewAuroraSchedulerManagerKillTasksArgs()
err205 := containerStruct1.ReadField2(jsProt204)
if err205 != 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()
}
arg206 := flag.Arg(1)
mbTrans207 := thrift.NewTMemoryBufferLen(len(arg206))
defer mbTrans207.Close()
_, err208 := mbTrans207.WriteString(arg206)
if err208 != nil {
Usage()
return
}
factory209 := thrift.NewTSimpleJSONProtocolFactory()
jsProt210 := factory209.GetProtocol(mbTrans207)
argvalue0 := aurora.NewInstanceKey()
err211 := argvalue0.Read(jsProt210)
if err211 != nil {
Usage()
return
}
value0 := argvalue0
tmp1, err212 := (strconv.Atoi(flag.Arg(2)))
if err212 != 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()
}
arg213 := flag.Arg(1)
mbTrans214 := thrift.NewTMemoryBufferLen(len(arg213))
defer mbTrans214.Close()
_, err215 := mbTrans214.WriteString(arg213)
if err215 != nil {
Usage()
return
}
factory216 := thrift.NewTSimpleJSONProtocolFactory()
jsProt217 := factory216.GetProtocol(mbTrans214)
argvalue0 := aurora.NewJobConfiguration()
err218 := argvalue0.Read(jsProt217)
if err218 != 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()
}
arg219 := flag.Arg(1)
mbTrans220 := thrift.NewTMemoryBufferLen(len(arg219))
defer mbTrans220.Close()
_, err221 := mbTrans220.WriteString(arg219)
if err221 != nil {
Usage()
return
}
factory222 := thrift.NewTSimpleJSONProtocolFactory()
jsProt223 := factory222.GetProtocol(mbTrans220)
argvalue0 := aurora.NewJobUpdateRequest()
err224 := argvalue0.Read(jsProt223)
if err224 != 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()
}
arg226 := flag.Arg(1)
mbTrans227 := thrift.NewTMemoryBufferLen(len(arg226))
defer mbTrans227.Close()
_, err228 := mbTrans227.WriteString(arg226)
if err228 != nil {
Usage()
return
}
factory229 := thrift.NewTSimpleJSONProtocolFactory()
jsProt230 := factory229.GetProtocol(mbTrans227)
argvalue0 := aurora.NewJobUpdateKey()
err231 := argvalue0.Read(jsProt230)
if err231 != 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()
}
arg233 := flag.Arg(1)
mbTrans234 := thrift.NewTMemoryBufferLen(len(arg233))
defer mbTrans234.Close()
_, err235 := mbTrans234.WriteString(arg233)
if err235 != nil {
Usage()
return
}
factory236 := thrift.NewTSimpleJSONProtocolFactory()
jsProt237 := factory236.GetProtocol(mbTrans234)
argvalue0 := aurora.NewJobUpdateKey()
err238 := argvalue0.Read(jsProt237)
if err238 != 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()
}
arg240 := flag.Arg(1)
mbTrans241 := thrift.NewTMemoryBufferLen(len(arg240))
defer mbTrans241.Close()
_, err242 := mbTrans241.WriteString(arg240)
if err242 != nil {
Usage()
return
}
factory243 := thrift.NewTSimpleJSONProtocolFactory()
jsProt244 := factory243.GetProtocol(mbTrans241)
argvalue0 := aurora.NewJobUpdateKey()
err245 := argvalue0.Read(jsProt244)
if err245 != nil {
Usage()
return
}
value0 := argvalue0
argvalue1 := flag.Arg(2)
value1 := argvalue1
fmt.Print(client.AbortJobUpdate(value0, value1))
fmt.Print("\n")
break
case "pulseJobUpdate":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "PulseJobUpdate requires 1 args")
flag.Usage()
}
arg247 := flag.Arg(1)
mbTrans248 := thrift.NewTMemoryBufferLen(len(arg247))
defer mbTrans248.Close()
_, err249 := mbTrans248.WriteString(arg247)
if err249 != nil {
Usage()
return
}
factory250 := thrift.NewTSimpleJSONProtocolFactory()
jsProt251 := factory250.GetProtocol(mbTrans248)
argvalue0 := aurora.NewJobUpdateKey()
err252 := argvalue0.Read(jsProt251)
if err252 != 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()
}
arg254 := flag.Arg(1)
mbTrans255 := thrift.NewTMemoryBufferLen(len(arg254))
defer mbTrans255.Close()
_, err256 := mbTrans255.WriteString(arg254)
if err256 != nil {
Usage()
return
}
factory257 := thrift.NewTSimpleJSONProtocolFactory()
jsProt258 := factory257.GetProtocol(mbTrans255)
argvalue0 := aurora.NewTaskQuery()
err259 := argvalue0.Read(jsProt258)
if err259 != 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()
}
arg260 := flag.Arg(1)
mbTrans261 := thrift.NewTMemoryBufferLen(len(arg260))
defer mbTrans261.Close()
_, err262 := mbTrans261.WriteString(arg260)
if err262 != nil {
Usage()
return
}
factory263 := thrift.NewTSimpleJSONProtocolFactory()
jsProt264 := factory263.GetProtocol(mbTrans261)
argvalue0 := aurora.NewTaskQuery()
err265 := argvalue0.Read(jsProt264)
if err265 != 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()
}
arg266 := flag.Arg(1)
mbTrans267 := thrift.NewTMemoryBufferLen(len(arg266))
defer mbTrans267.Close()
_, err268 := mbTrans267.WriteString(arg266)
if err268 != nil {
Usage()
return
}
factory269 := thrift.NewTSimpleJSONProtocolFactory()
jsProt270 := factory269.GetProtocol(mbTrans267)
argvalue0 := aurora.NewTaskQuery()
err271 := argvalue0.Read(jsProt270)
if err271 != 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()
}
arg272 := flag.Arg(1)
mbTrans273 := thrift.NewTMemoryBufferLen(len(arg272))
defer mbTrans273.Close()
_, err274 := mbTrans273.WriteString(arg272)
if err274 != nil {
Usage()
return
}
factory275 := thrift.NewTSimpleJSONProtocolFactory()
jsProt276 := factory275.GetProtocol(mbTrans273)
argvalue0 := aurora.NewJobKey()
err277 := argvalue0.Read(jsProt276)
if err277 != 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()
}
arg280 := flag.Arg(1)
mbTrans281 := thrift.NewTMemoryBufferLen(len(arg280))
defer mbTrans281.Close()
_, err282 := mbTrans281.WriteString(arg280)
if err282 != nil {
Usage()
return
}
factory283 := thrift.NewTSimpleJSONProtocolFactory()
jsProt284 := factory283.GetProtocol(mbTrans281)
argvalue0 := aurora.NewJobConfiguration()
err285 := argvalue0.Read(jsProt284)
if err285 != 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()
}
arg286 := flag.Arg(1)
mbTrans287 := thrift.NewTMemoryBufferLen(len(arg286))
defer mbTrans287.Close()
_, err288 := mbTrans287.WriteString(arg286)
if err288 != nil {
Usage()
return
}
factory289 := thrift.NewTSimpleJSONProtocolFactory()
jsProt290 := factory289.GetProtocol(mbTrans287)
argvalue0 := aurora.NewJobUpdateQuery()
err291 := argvalue0.Read(jsProt290)
if err291 != 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()
}
arg292 := flag.Arg(1)
mbTrans293 := thrift.NewTMemoryBufferLen(len(arg292))
defer mbTrans293.Close()
_, err294 := mbTrans293.WriteString(arg292)
if err294 != nil {
Usage()
return
}
factory295 := thrift.NewTSimpleJSONProtocolFactory()
jsProt296 := factory295.GetProtocol(mbTrans293)
argvalue0 := aurora.NewJobUpdateKey()
err297 := argvalue0.Read(jsProt296)
if err297 != 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()
}
arg298 := flag.Arg(1)
mbTrans299 := thrift.NewTMemoryBufferLen(len(arg298))
defer mbTrans299.Close()
_, err300 := mbTrans299.WriteString(arg298)
if err300 != nil {
Usage()
return
}
factory301 := thrift.NewTSimpleJSONProtocolFactory()
jsProt302 := factory301.GetProtocol(mbTrans299)
argvalue0 := aurora.NewJobUpdateRequest()
err303 := argvalue0.Read(jsProt302)
if err303 != 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)
}
}

4841
vendor/gen-go/apache/aurora/auroraadmin.go vendored Normal file

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,79 @@
// 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"
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

15610
vendor/gen-go/apache/aurora/ttypes.go vendored Normal file

File diff suppressed because it is too large Load diff