agent: Fix decoding of checks. Fixes #60

command/agent/agent_endpoint.go

@@ -55,7 +55,11 @@ func (s *HTTPServer) AgentForceLeave(resp http.ResponseWriter, req *http.Request
 
 func (s *HTTPServer) AgentRegisterCheck(resp http.ResponseWriter, req *http.Request) (interface{}, error) {
 	var args CheckDefinition
-	if err := decodeBody(req, &args); err != nil {
+	// Fixup the type decode of TTL or Interval
+	decodeCB := func(raw interface{}) error {
+		return FixupCheckType(raw)
+	}
+	if err := decodeBody(req, &args, decodeCB); err != nil {
 		resp.WriteHeader(400)
 		resp.Write([]byte(fmt.Sprintf("Request decode failed: %v", err)))
 		return nil, nil
@@ -108,7 +112,19 @@ func (s *HTTPServer) AgentCheckFail(resp http.ResponseWriter, req *http.Request)
 
 func (s *HTTPServer) AgentRegisterService(resp http.ResponseWriter, req *http.Request) (interface{}, error) {
 	var args ServiceDefinition
-	if err := decodeBody(req, &args); err != nil {
+	// Fixup the type decode of TTL or Interval if a check if provided
+	decodeCB := func(raw interface{}) error {
+		rawMap, ok := raw.(map[string]interface{})
+		if !ok {
+			return nil
+		}
+		check, ok := rawMap["check"]
+		if !ok {
+			return nil
+		}
+		return FixupCheckType(check)
+	}
+	if err := decodeBody(req, &args, decodeCB); err != nil {
 		resp.WriteHeader(400)
 		resp.Write([]byte(fmt.Sprintf("Request decode failed: %v", err)))
 		return nil, nil

command/agent/catalog_endpoint.go

@@ -9,7 +9,7 @@ import (
 
 func (s *HTTPServer) CatalogRegister(resp http.ResponseWriter, req *http.Request) (interface{}, error) {
 	var args structs.RegisterRequest
-	if err := decodeBody(req, &args); err != nil {
+	if err := decodeBody(req, &args, nil); err != nil {
 		resp.WriteHeader(400)
 		resp.Write([]byte(fmt.Sprintf("Request decode failed: %v", err)))
 		return nil, nil
@@ -30,7 +30,7 @@ func (s *HTTPServer) CatalogRegister(resp http.ResponseWriter, req *http.Request
 
 func (s *HTTPServer) CatalogDeregister(resp http.ResponseWriter, req *http.Request) (interface{}, error) {
 	var args structs.DeregisterRequest
-	if err := decodeBody(req, &args); err != nil {
+	if err := decodeBody(req, &args, nil); err != nil {
 		resp.WriteHeader(400)
 		resp.Write([]byte(fmt.Sprintf("Request decode failed: %v", err)))
 		return nil, nil

command/agent/http.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"encoding/json"
 	"github.com/hashicorp/consul/consul/structs"
+	"github.com/mitchellh/mapstructure"
 	"io"
 	"log"
 	"net"
@@ -151,9 +152,20 @@ func (s *HTTPServer) Index(resp http.ResponseWriter, req *http.Request) {
 }
 
 // decodeBody is used to decode a JSON request body
-func decodeBody(req *http.Request, out interface{}) error {
+func decodeBody(req *http.Request, out interface{}, cb func(interface{}) error) error {
+	var raw interface{}
 	dec := json.NewDecoder(req.Body)
-	return dec.Decode(out)
+	if err := dec.Decode(&raw); err != nil {
+		return err
+	}
+
+	// Invoke the callback prior to decode
+	if cb != nil {
+		if err := cb(raw); err != nil {
+			return err
+		}
+	}
+	return mapstructure.Decode(raw, out)
 }
 
 // setIndex is used to set the index response header