forked from harness/gitness
-
Notifications
You must be signed in to change notification settings - Fork 0
/
worker.go
91 lines (82 loc) · 2.35 KB
/
worker.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package handler
import (
"encoding/json"
"net/http"
"github.com/drone/drone/server/worker"
"github.com/drone/drone/server/worker/director"
"github.com/drone/drone/server/worker/docker"
"github.com/drone/drone/server/worker/pool"
"github.com/goji/context"
"github.com/zenazn/goji/web"
)
// GetWorkers accepts a request to retrieve the list
// of registered workers and return the results
// in JSON format.
//
// GET /api/workers
//
func GetWorkers(c web.C, w http.ResponseWriter, r *http.Request) {
ctx := context.FromC(c)
workers := pool.FromContext(ctx).List()
json.NewEncoder(w).Encode(workers)
}
// PostWorker accepts a request to allocate a new
// worker to the pool.
//
// POST /api/workers
//
func PostWorker(c web.C, w http.ResponseWriter, r *http.Request) {
ctx := context.FromC(c)
pool := pool.FromContext(ctx)
node := r.FormValue("address")
pool.Allocate(docker.NewHost(node))
w.WriteHeader(http.StatusOK)
}
// Delete accepts a request to delete a worker
// from the pool.
//
// DELETE /api/workers
//
func DelWorker(c web.C, w http.ResponseWriter, r *http.Request) {
ctx := context.FromC(c)
pool := pool.FromContext(ctx)
uuid := r.FormValue("id")
for _, worker := range pool.List() {
if worker.(*docker.Docker).UUID != uuid {
pool.Deallocate(worker)
w.WriteHeader(http.StatusNoContent)
return
}
}
w.WriteHeader(http.StatusNotFound)
}
// GetWorkPending accepts a request to retrieve the list
// of pending work and returns in JSON format.
//
// GET /api/work/pending
//
func GetWorkPending(c web.C, w http.ResponseWriter, r *http.Request) {
ctx := context.FromC(c)
d := worker.FromContext(ctx).(*director.Director)
json.NewEncoder(w).Encode(d.GetPending())
}
// GetWorkStarted accepts a request to retrieve the list
// of started work and returns in JSON format.
//
// GET /api/work/started
//
func GetWorkStarted(c web.C, w http.ResponseWriter, r *http.Request) {
ctx := context.FromC(c)
d := worker.FromContext(ctx).(*director.Director)
json.NewEncoder(w).Encode(d.GetStarted())
}
// GetWorkAssigned accepts a request to retrieve the list
// of started work and returns in JSON format.
//
// GET /api/work/assignments
//
func GetWorkAssigned(c web.C, w http.ResponseWriter, r *http.Request) {
ctx := context.FromC(c)
d := worker.FromContext(ctx).(*director.Director)
json.NewEncoder(w).Encode(d.GetAssignemnts())
}