func getlog(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
jobid := ps.ByName("jobid")
env := gomr.NewEnvironment()
logger := env.GetLogger([]string{})
defer logger.Close()
results := logger.Fetch(jobid, 50)
b, err := json.MarshalIndent(results, "", " ")
if err != nil {
http.Error(w, err.Error(), 500)
return
}
w.Header().Set("Content-Type", "application/json")
w.Write(b)
}
func execute(binpath, jobname, bucketname string) {
log.Println("TASK", binpath, jobname, bucketname)
dir := os.TempDir() + "/gomrbin"
//log.Println(dir)
_, err := os.Stat(dir)
if err != nil {
if os.IsNotExist(err) {
err = os.Mkdir(dir, 0776)
if err != nil {
log.Println(err)
return
}
} else {
log.Println(err)
return
}
}
log.Println("Temp dir:", dir)
//Check if binary exists
bin := dir + "/" + strings.Split(binpath, "/")[1]
log.Println(bin)
_, err = os.Stat(bin)
if err != nil {
if os.IsNotExist(err) {
log.Println("Downloading binary from S3")
env := gomr.NewEnvironment()
s3bucket, err := env.GetS3Bucket(bucketname)
if err != nil {
log.Println(err)
return
}
raw, err := s3bucket.Get(binpath)
if err != nil {
log.Println(err)
return
}
//Uncompress the contents
var rawb bytes.Buffer
var datab bytes.Buffer
_, err = rawb.Write(raw)
if err != nil {
log.Println(err)
return
}
rd, err := gzip.NewReader(&rawb)
if err != nil {
log.Println(err)
return
}
io.Copy(&datab, rd)
rd.Close()
data := datab.Bytes()
f, err := os.Create(bin)
if err != nil {
log.Println(err)
return
}
_, err = f.Write(data)
if err != nil {
log.Println(err)
f.Close()
os.Remove(bin)
return
}
err = f.Chmod(100)
if err != nil {
log.Println(err)
f.Close()
os.Remove(bin)
return
}
f.Close()
} else {
log.Println(err)
return
}
}
//Now execute...
cmd := exec.Command(bin, jobname)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
log.Println(cmd.Run())
}
func execute(binpath, jobname, bucketname string) {
log.Println("TASK", binpath, jobname, bucketname)
dir := os.TempDir() + "/gomrbin"
//log.Println(dir)
_, err := os.Stat(dir)
if err != nil {
if os.IsNotExist(err) {
err = os.Mkdir(dir, 0776)
if err != nil {
log.Println(err)
return
}
} else {
log.Println(err)
return
}
}
log.Println("Temp dir:", dir)
//Check if binary exists
bin := dir + "/" + strings.Split(binpath, "/")[1]
log.Println(bin)
_, err = os.Stat(bin)
if err != nil {
if os.IsNotExist(err) {
log.Println("Downloading binary from S3")
env := gomr.NewEnvironment()
s3bucket, err := env.GetS3Bucket(bucketname)
if err != nil {
log.Println(err)
return
}
data, err := s3bucket.Get(binpath)
if err != nil {
log.Println(err)
return
}
f, err := os.Create(bin)
if err != nil {
log.Println(err)
return
}
_, err = f.Write(data)
if err != nil {
log.Println(err)
f.Close()
os.Remove(bin)
return
}
err = f.Chmod(100)
if err != nil {
log.Println(err)
f.Close()
os.Remove(bin)
return
}
f.Close()
} else {
log.Println(err)
return
}
}
//Now execute...
cmd := exec.Command(bin, jobname)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
log.Println(cmd.Run())
}
