YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-12-15 08:16:43 +00:00
Code Issues Releases Wiki Activity

add dockerfile

Browse Source
This commit is contained in:
Newnius
2019-04-16 15:33:37 +08:00
parent 5877310950
commit e34d75dc7f
5 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Dockerfile Normal file
View File

@@ -0,0 +1,10 @@
FROM ubuntu:16.04
RUN apt update && \
apt install -y wget
RUN wget https://dl.google.com/go/go1.12.4.linux-amd64.tar.gz && \
tar -C /usr/local -xzf go1.12.4.linux-amd64.tar.gz && \
rm go1.12.4.linux-amd64.tar.gz
ENV PATH $PATH:/usr/local/go/bin

2
src/collector.go
View File

@@ -12,7 +12,7 @@ var (
) )
func start(pool *ResourcePool) { func start(pool *ResourcePool) {
consumer, err := sarama.NewConsumer([]string{"kafka_node1:9091", "kafka_node2:9092", "kafka_node3:9093"}, nil) consumer, err := sarama.NewConsumer([]string{"kafka:9092"}, nil)
if err != nil { if err != nil {
panic(err) panic(err)
} }

6
src/job_manager.go
View File

@@ -49,11 +49,11 @@ func (jm *JobManager) start() {
v.Set("workspace", jm.job.Workspace) v.Set("workspace", jm.job.Workspace)
v.Set("gpus", strings.Join(GPUs, ",")) v.Set("gpus", strings.Join(GPUs, ","))
fmt.Print(v.Encode()) fmt.Println(v.Encode())
resp, err := doRequest("POST", "http://kafka_node1:8000/create", strings.NewReader(v.Encode()), "application/x-www-form-urlencoded", "") resp, err := doRequest("POST", "http://kafka:8000/create", strings.NewReader(v.Encode()), "application/x-www-form-urlencoded", "")
if err != nil { if err != nil {
log.Println(err) log.Println(err.Error())
return return
} }
defer resp.Body.Close() defer resp.Body.Close()

11
src/main.go
View File

@@ -37,14 +37,15 @@ func serverAPI(w http.ResponseWriter, r *http.Request) {
case "job_submit": case "job_submit":
var job Job var job Job
fmt.Println("job_submit") fmt.Println("job_submit")
msgSubmit := MsgSubmit{Code: 0}
err := json.Unmarshal([]byte(string(r.PostFormValue("job"))), &job) err := json.Unmarshal([]byte(string(r.PostFormValue("job"))), &job)
if err != nil { if err != nil {
w.Header().Set("Content-Type", "application/json") msgSubmit.Code = 1
w.Write([]byte(err.Error())) msgSubmit.Error = err.Error()
return } else {
allocator.schedule(job)
} }
allocator.schedule(job) js, _ := json.Marshal(msgSubmit)
js, _ := json.Marshal(nodes)
w.Header().Set("Content-Type", "application/json") w.Header().Set("Content-Type", "application/json")
w.Write(js) w.Write(js)
break break

5
src/util.go
View File

@@ -92,8 +92,9 @@ type NodeStatus struct {
} }
type MsgAgent struct { type MsgAgent struct {
ClientID int `json:"code"` ClientID int `json:"code"`
Status []NodeStatus `json:"status"` ClientHost string `json:"host"`
Status []NodeStatus `json:"status"`
} }
type Job struct { type Job struct {