ability to add and remove workers

server/handler/worker.go

@@ -31,12 +31,32 @@ func GetWorkers(c web.C, w http.ResponseWriter, r *http.Request) {
 //
 func PostWorker(c web.C, w http.ResponseWriter, r *http.Request) {
 	ctx := context.FromC(c)
-	workers := pool.FromContext(ctx)
+	pool := pool.FromContext(ctx)
-	node := r.FormValue("node")
+	node := r.FormValue("address")
-	workers.Allocate(docker.NewHost(node))
+	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.
 //

server/main.go

@@ -163,6 +163,8 @@ func main() {
 	work.Get("/api/work/pending", handler.GetWorkPending)
 	work.Get("/api/work/assignments", handler.GetWorkAssigned)
 	work.Get("/api/workers", handler.GetWorkers)
+	work.Post("/api/workers", handler.PostWorker)
+	work.Delete("/api/workers", handler.DelWorker)
 	goji.Handle("/api/work*", work)
 
 	// Include static resources