jsonrpc2/rpc/wrapper.go

59 lines
1.7 KiB
Go
Raw Normal View History

2022-01-31 20:55:38 +03:00
//Package rpc provides abstract rpc server
//
//Copyright (C) 2022 Alexander Kiryukhin <i@neonxp.dev>
//
//This file is part of go.neonxp.ru/jsonrpc2 project.
2022-01-31 20:55:38 +03:00
//
//This program is free software: you can redistribute it and/or modify
//it under the terms of the GNU General Public License as published by
//the Free Software Foundation, either version 3 of the License, or
//(at your option) any later version.
//
//This program is distributed in the hope that it will be useful,
//but WITHOUT ANY WARRANTY; without even the implied warranty of
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
//GNU General Public License for more details.
//
//You should have received a copy of the GNU General Public License
//along with this program. If not, see <https://www.gnu.org/licenses/>.
2022-01-31 20:17:31 +03:00
package rpc
2022-01-31 02:31:43 +03:00
import (
"context"
"encoding/json"
)
2022-05-28 16:53:20 +03:00
func H[RQ any, RS any](handler func(context.Context, *RQ) (RS, error)) HandlerFunc {
2022-01-31 02:31:43 +03:00
return func(ctx context.Context, in json.RawMessage) (json.RawMessage, error) {
req := new(RQ)
if err := json.Unmarshal(in, req); err != nil {
2022-05-22 03:27:50 +03:00
return nil, ErrorFromCode(ErrCodeParseError)
2022-01-31 02:31:43 +03:00
}
resp, err := handler(ctx, req)
if err != nil {
return nil, Error{
Code: ErrUser,
Message: err.Error(),
}
}
return json.Marshal(resp)
}
}
// HS is a simple generic wrapper for rpc handlers without any request params.
func HS[RS any](handler func(context.Context) (RS, error)) HandlerFunc {
return func(ctx context.Context, in json.RawMessage) (json.RawMessage, error) {
resp, err := handler(ctx)
if err != nil {
return nil, Error{
Code: ErrUser,
Message: err.Error(),
}
}
return json.Marshal(resp)
}
}
2022-05-28 16:53:20 +03:00
type HandlerFunc func(context.Context, json.RawMessage) (json.RawMessage, error)