From 63ae02a2f911f0212d5e0bfcb9c6e8b816879da2 Mon Sep 17 00:00:00 2001 From: Armon Dadgar Date: Sun, 6 Sep 2015 19:35:05 -0700 Subject: [PATCH] http: parsing flags for node drain --- command/agent/node_endpoint.go | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/command/agent/node_endpoint.go b/command/agent/node_endpoint.go index 0817b4eec..4d326a98c 100644 --- a/command/agent/node_endpoint.go +++ b/command/agent/node_endpoint.go @@ -2,6 +2,7 @@ package agent import ( "net/http" + "strconv" "strings" "github.com/hashicorp/nomad/nomad/structs" @@ -88,6 +89,16 @@ func (s *HTTPServer) nodeToggleDrain(resp http.ResponseWriter, req *http.Request return nil, CodedError(405, ErrInvalidMethod) } + // Get the enable value + enableRaw := req.URL.Query().Get("enable") + if enableRaw == "" { + return nil, CodedError(400, "missing enable value") + } + enable, err := strconv.ParseBool(enableRaw) + if err != nil { + return nil, CodedError(400, "invalid enable value") + } + // TODO return nil, nil }