From 4cf58de9bb7b109dddbc80adefe52cdf61328d0d Mon Sep 17 00:00:00 2001
From: Alexander Kiryukhin <a.kiryukhin@mail.ru>
Date: Mon, 31 Jan 2022 20:17:31 +0300
Subject: Small refactoring

---
 http/server.go | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 http/server.go

(limited to 'http/server.go')

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)
+}
-- 
cgit v1.2.3