Small refactoring

1 files changed, 25 insertions, 0 deletions

diff --git a/rpc/wrapper.go b/rpc/wrapper.go
new file mode 100644
index 0000000..0bc0a5c
--- /dev/null
+++ b/rpc/wrapper.go
@@ -0,0 +1,25 @@
+package rpc
+
+import (
+	"context"
+	"encoding/json"
+)
+
+func Wrap[RQ any, RS any](handler func(context.Context, *RQ) (RS, error)) Handler {
+	return func(ctx context.Context, in json.RawMessage) (json.RawMessage, error) {
+		req := new(RQ)
+		if err := json.Unmarshal(in, req); err != nil {
+			return nil, NewError(ErrCodeParseError)
+		}
+		resp, err := handler(ctx, req)
+		if err != nil {
+			return nil, Error{
+				Code:    ErrUser,
+				Message: err.Error(),
+			}
+		}
+		return json.Marshal(resp)
+	}
+}
+
+type Handler func(context.Context, json.RawMessage) (json.RawMessage, error)