68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package jsonrpc2
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"net/http"
|
|
)
|
|
|
|
type httpHandler struct {
|
|
handler Handler
|
|
}
|
|
|
|
// NewHTTPHandler returns an http.Handler which will decode RPC calls out of the
|
|
// HTTP request body, and return the result as the response body. It will try to
|
|
// intelligently set the status code on the response as well.
|
|
func NewHTTPHandler(h Handler) http.Handler {
|
|
return httpHandler{h}
|
|
}
|
|
|
|
func (h httpHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
rw.WriteHeader(http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
var (
|
|
ctx = r.Context()
|
|
enc = json.NewEncoder(rw)
|
|
dec = json.NewDecoder(r.Body)
|
|
)
|
|
|
|
rw.Header().Set("Content-Type", "application/json")
|
|
|
|
req, decErr := decodeRequest(dec)
|
|
if decErr.Code != 0 {
|
|
rw.WriteHeader(http.StatusBadRequest)
|
|
_ = encodeErrorResponse(enc, req.ID, decErr)
|
|
return
|
|
}
|
|
|
|
var rpcErr Error
|
|
res, err := h.handler.ServeRPC(ctx, req)
|
|
if err != nil && !errors.As(err, &rpcErr) {
|
|
// We deliberately don't include the actual error in the response. If
|
|
// doing so is desired this can be achieved using a middleware which
|
|
// translates the non-Error into an Error prior to it reaching this
|
|
// point.
|
|
rpcErr = NewError(errCodeServerError, "unexpected server-side error")
|
|
}
|
|
|
|
switch rpcErr.Code {
|
|
case 0: // no error
|
|
_ = encodeResponse(enc, req.ID, res)
|
|
return
|
|
|
|
case errCodeMethodNotFound:
|
|
rw.WriteHeader(http.StatusNotFound)
|
|
|
|
case errCodeServerError:
|
|
rw.WriteHeader(http.StatusInternalServerError)
|
|
|
|
default:
|
|
rw.WriteHeader(http.StatusBadRequest)
|
|
}
|
|
|
|
_ = encodeErrorResponse(enc, req.ID, rpcErr)
|
|
}
|