YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 06:11:56 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius 2020-07-01 10:41:34 +08:00
parent dade0ab56e
commit 7ccb41b831
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.go
View File

@ -8,6 +8,7 @@ import (
"os" "os"
"time" "time"
"strconv" "strconv"
"math/rand"
) )
var addr = flag.String("addr", "0.0.0.0:8080", "http service address") var addr = flag.String("addr", "0.0.0.0:8080", "http service address")
@ -40,6 +41,7 @@ func serverAPI(w http.ResponseWriter, r *http.Request) {
msgSubmit.Code = 1 msgSubmit.Code = 1
msgSubmit.Error = err.Error() msgSubmit.Error = err.Error()
} else { } else {
job.Name = job.Name + "-" + string(time.Now().Unix()) + string(1000+rand.Intn(8999))
for i := range job.Tasks { for i := range job.Tasks {
job.Tasks[i].ID = job.Name + ":" + job.Tasks[i].Name job.Tasks[i].ID = job.Name + ":" + job.Tasks[i].Name
job.Tasks[i].Job = job.Name job.Tasks[i].Job = job.Name

6
src/optimizer.go
View File

@ -118,7 +118,7 @@ func (optimizer *Optimizer) FeedTime(job Job, stats [][]TaskStatus) {
spider := Spider{} spider := Spider{}
spider.Method = "GET" spider.Method = "GET"
spider.URL = "http://yao-optimizer:8080/feed?job=" + jobName + ":time" + "&features=" + string(features) + "&labels=" + string(labels) spider.URL = "http://yao-optimizer:8080/feed?job=" + jobName + "-time" + "&features=" + string(features) + "&labels=" + string(labels)
err := spider.do() err := spider.do()
if err != nil { if err != nil {
@ -146,7 +146,7 @@ func (optimizer *Optimizer) FeedTime(job Job, stats [][]TaskStatus) {
func (optimizer *Optimizer) trainTime(jobName string) { func (optimizer *Optimizer) trainTime(jobName string) {
spider := Spider{} spider := Spider{}
spider.Method = "GET" spider.Method = "GET"
params := "job=" + jobName + ":time" params := "job=" + jobName + "-time"
spider.URL = "http://yao-optimizer:8080/train?" + params spider.URL = "http://yao-optimizer:8080/train?" + params
err := spider.do() err := spider.do()
@ -213,7 +213,7 @@ func (optimizer *Optimizer) PredictTime(job Job) OptimizerJobExecutionTime {
spider := Spider{} spider := Spider{}
spider.Method = "GET" spider.Method = "GET"
spider.URL = "http://yao-optimizer:8080/predict?job=" + jobName + ":time" + "&features=" + string(features) spider.URL = "http://yao-optimizer:8080/predict?job=" + jobName + "-time" + "&features=" + string(features)
err := spider.do() err := spider.do()
if err != nil { if err != nil {