jsonrpc2/wrapper.go
2022-01-31 02:31:43 +03:00

25 lines
578 B
Go

package jsonrpc2
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)