diff options
Diffstat (limited to 'rpc/wrapper.go')
-rw-r--r-- | rpc/wrapper.go | 25 |
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) |