jsonrpc.DefaultErrorEncoder: add RequestID in error body

transport/http/jsonrpc/server.go

@@ -11,6 +11,8 @@ import (
 	httptransport "github.com/go-kit/kit/transport/http"
 )
 
+const requestIDKey = "request-id"
+
 // Server wraps an endpoint and implements http.Handler.
 type Server struct {
 	ecm          EndpointCodecMap
@@ -105,6 +107,8 @@ func (s Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	ctx = context.WithValue(ctx, requestIDKey, req.ID)
+
 	// Get the endpoint and codecs from the map using the method
 	// defined in the JSON  object
 	ecm, ok := s.ecm[req.Method]
@@ -160,7 +164,7 @@ func (s Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 // If the error implements ErrorCoder, the provided code will be set on the
 // response error.
 // If the error implements Headerer, the given headers will be set.
-func DefaultErrorEncoder(_ context.Context, err error, w http.ResponseWriter) {
+func DefaultErrorEncoder(ctx context.Context, err error, w http.ResponseWriter) {
 	w.Header().Set("Content-Type", ContentType)
 	if headerer, ok := err.(httptransport.Headerer); ok {
 		for k := range headerer.Headers() {
@@ -177,7 +181,13 @@ func DefaultErrorEncoder(_ context.Context, err error, w http.ResponseWriter) {
 	}
 
 	w.WriteHeader(http.StatusOK)
+
+	var requestID *RequestID
+	if ctx.Value(requestIDKey) != nil {
+		requestID = ctx.Value(requestIDKey).(*RequestID)
+	}
 	_ = json.NewEncoder(w).Encode(Response{
+		ID:      requestID,
 		JSONRPC: Version,
 		Error:   &e,
 	})

transport/http/jsonrpc/server_test.go

@@ -24,9 +24,14 @@ func body(in string) io.Reader {
 	return strings.NewReader(in)
 }
 
-func expectErrorCode(t *testing.T, want int, body []byte) {
+func unmarshalResponse(body []byte) (jsonrpc.Response, error) {
 	var r jsonrpc.Response
 	err := json.Unmarshal(body, &r)
+	return r, err
+}
+
+func expectErrorCode(t *testing.T, want int, body []byte) {
+	r, err := unmarshalResponse(body)
 	if err != nil {
 		t.Fatalf("Cant' decode response. err=%s, body=%s", err, body)
 	}
@@ -38,6 +43,30 @@ func expectErrorCode(t *testing.T, want int, body []byte) {
 	}
 }
 
+func expectValidRequestID(t *testing.T, want int, body []byte) {
+	r, err := unmarshalResponse(body)
+	if err != nil {
+		t.Fatalf("Cant' decode response. err=%s, body=%s", err, body)
+	}
+	have, err := r.ID.Int()
+	if err != nil {
+		t.Fatalf("Cant' get requestID in response. err=%s, body=%s", err, body)
+	}
+	if want != have {
+		t.Fatalf("Unexpected request ID. Want %d, have %d: %s", want, have, body)
+	}
+}
+
+func expectNilRequestID(t *testing.T, body []byte) {
+	r, err := unmarshalResponse(body)
+	if err != nil {
+		t.Fatalf("Cant' decode response. err=%s, body=%s", err, body)
+	}
+	if nil != r.ID {
+		t.Fatalf("Unexpected request ID. Want nil, have %v", r.ID)
+	}
+}
+
 func nopDecoder(context.Context, json.RawMessage) (interface{}, error) { return struct{}{}, nil }
 func nopEncoder(context.Context, interface{}) (json.RawMessage, error) { return []byte("[]"), nil }
 
@@ -92,6 +121,7 @@ func TestServerBadEndpoint(t *testing.T) {
 	}
 	buf, _ := ioutil.ReadAll(resp.Body)
 	expectErrorCode(t, jsonrpc.InternalError, buf)
+	expectValidRequestID(t, 1, buf)
 }
 
 func TestServerBadEncode(t *testing.T) {
@@ -111,6 +141,7 @@ func TestServerBadEncode(t *testing.T) {
 	}
 	buf, _ := ioutil.ReadAll(resp.Body)
 	expectErrorCode(t, jsonrpc.InternalError, buf)
+	expectValidRequestID(t, 1, buf)
 }
 
 func TestServerErrorEncoder(t *testing.T) {
@@ -162,6 +193,7 @@ func TestCanRejectInvalidJSON(t *testing.T) {
 	}
 	buf, _ := ioutil.ReadAll(resp.Body)
 	expectErrorCode(t, jsonrpc.ParseError, buf)
+	expectNilRequestID(t, buf)
 }
 
 func TestServerUnregisteredMethod(t *testing.T) {