Small refactoring

1 files changed, 33 insertions, 0 deletions

diff --git a/http/server.go b/http/server.go
new file mode 100644
index 0000000..5fca66a
--- /dev/null
+++ b/http/server.go
@@ -0,0 +1,33 @@
+package http
+
+import (
+	"bufio"
+	"net/http"
+
+	"github.com/neonxp/jsonrpc2/rpc"
+)
+
+type Server struct {
+	*rpc.RpcServer
+}
+
+func New() *Server {
+	return &Server{RpcServer: rpc.New()}
+}
+
+func (r *Server) ServeHTTP(writer http.ResponseWriter, request *http.Request) {
+	writer.Header().Set("Content-Type", "application/json")
+	reader := bufio.NewReader(request.Body)
+	defer request.Body.Close()
+	firstByte, err := reader.Peek(1)
+	if err != nil {
+		r.Logger.Logf("Can't read body: %v", err)
+		rpc.WriteError(rpc.ErrCodeParseError, writer)
+		return
+	}
+	if string(firstByte) == "[" {
+		r.BatchRequest(request.Context(), reader, writer)
+		return
+	}
+	r.SingleRequest(request.Context(), reader, writer)
+}