From 7ccb41b831c4934f7e5e07e38bbcef2987b09a26 Mon Sep 17 00:00:00 2001 From: Newnius Date: Wed, 1 Jul 2020 10:41:34 +0800 Subject: [PATCH] update --- src/main.go | 2 ++ src/optimizer.go | 6 +++--- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/main.go b/src/main.go index a5311bd..c72027d 100644 --- a/src/main.go +++ b/src/main.go @@ -8,6 +8,7 @@ import ( "os" "time" "strconv" + "math/rand" ) 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.Error = err.Error() } else { + job.Name = job.Name + "-" + string(time.Now().Unix()) + string(1000+rand.Intn(8999)) for i := range job.Tasks { job.Tasks[i].ID = job.Name + ":" + job.Tasks[i].Name job.Tasks[i].Job = job.Name diff --git a/src/optimizer.go b/src/optimizer.go index ae2deb4..c53c8ac 100644 --- a/src/optimizer.go +++ b/src/optimizer.go @@ -118,7 +118,7 @@ func (optimizer *Optimizer) FeedTime(job Job, stats [][]TaskStatus) { spider := Spider{} 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() if err != nil { @@ -146,7 +146,7 @@ func (optimizer *Optimizer) FeedTime(job Job, stats [][]TaskStatus) { func (optimizer *Optimizer) trainTime(jobName string) { spider := Spider{} spider.Method = "GET" - params := "job=" + jobName + ":time" + params := "job=" + jobName + "-time" spider.URL = "http://yao-optimizer:8080/train?" + params err := spider.do() @@ -213,7 +213,7 @@ func (optimizer *Optimizer) PredictTime(job Job) OptimizerJobExecutionTime { spider := Spider{} 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() if err != nil {