feat: add wrapper supporting no request params

2 files changed, 19 insertions, 0 deletions

diff --git a/example/main.go b/example/main.go
index 37b9037..1c9fa41 100644
--- a/example/main.go
+++ b/example/main.go
@@ -79,6 +79,7 @@ func main() {
 
 	s.Register("multiply", rpc.H(Multiply))
 	s.Register("divide", rpc.H(Divide))
+	s.Register("hello", rpc.HS(Hello))
 
 	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
 	defer cancel()
@@ -102,6 +103,10 @@ func Divide(ctx context.Context, args *Args) (*Quotient, error) {
 	return quo, nil
 }
 
+func Hello(ctx context.Context) (string, error) {
+	return "world", nil
+}
+
 type Args struct {
 	A int `json:"a"`
 	B int `json:"b"`
diff --git a/rpc/wrapper.go b/rpc/wrapper.go
index 8aa9556..f16833f 100644
--- a/rpc/wrapper.go
+++ b/rpc/wrapper.go
@@ -41,4 +41,18 @@ func H[RQ any, RS any](handler func(context.Context, *RQ) (RS, error)) HandlerFu
 	}
 }
 
+// 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)
+	}
+}
+
 type HandlerFunc func(context.Context, json.RawMessage) (json.RawMessage, error)