Merge pull request #3 from jon4hz/feat-no-params
feat: add wrapper supporting no request params
This commit is contained in:
commit
3d69d1e7ea
3 changed files with 39 additions and 3 deletions
23
README.md
23
README.md
|
@ -38,18 +38,30 @@ Go 1.18+ required
|
||||||
)
|
)
|
||||||
```
|
```
|
||||||
|
|
||||||
3. Write handler:
|
3. Write handlers:
|
||||||
```go
|
```go
|
||||||
|
|
||||||
|
// This handler supports request parameters
|
||||||
func Multiply(ctx context.Context, args *Args) (int, error) {
|
func Multiply(ctx context.Context, args *Args) (int, error) {
|
||||||
return args.A * args.B, nil
|
return args.A * args.B, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// This handler has no request parameters
|
||||||
|
func Hello(ctx context.Context) (string, error) {
|
||||||
|
return "World", nil
|
||||||
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
Handler must have exact two arguments (context and input of any json serializable type) and exact two return values (output of any json serializable type and error)
|
A handler must have a context as first parameter and may have a second parameter, representing request paramters (input of any json serializable type). A handler always returns exactly two values (output of any json serializable type and error).
|
||||||
|
|
||||||
|
4. Wrap the handler using one of the two functions `rpc.H` (supporting req params) or `rpc.HS` (no params) and register it with the server:
|
||||||
|
|
||||||
4. Wrap handler with `rpc.H` method and register it in server:
|
|
||||||
```go
|
```go
|
||||||
|
// handler has params
|
||||||
s.Register("multiply", rpc.H(Multiply))
|
s.Register("multiply", rpc.H(Multiply))
|
||||||
|
|
||||||
|
// handler has no params
|
||||||
|
s.Register("hello", rpc.HS(Hello))
|
||||||
```
|
```
|
||||||
|
|
||||||
5. Run RPC server:
|
5. Run RPC server:
|
||||||
|
@ -96,6 +108,7 @@ func main() {
|
||||||
|
|
||||||
s.Register("multiply", rpc.H(Multiply))
|
s.Register("multiply", rpc.H(Multiply))
|
||||||
s.Register("divide", rpc.H(Divide))
|
s.Register("divide", rpc.H(Divide))
|
||||||
|
s.Register("hello", rpc.HS(Hello))
|
||||||
|
|
||||||
s.Run(context.Background())
|
s.Run(context.Background())
|
||||||
}
|
}
|
||||||
|
@ -108,6 +121,10 @@ func Divide(ctx context.Context, args *Args) (*Quotient, error) {
|
||||||
//...
|
//...
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Hello(ctx context.Context) (string, error) {
|
||||||
|
// ...
|
||||||
|
}
|
||||||
|
|
||||||
type Args struct {
|
type Args struct {
|
||||||
A int `json:"a"`
|
A int `json:"a"`
|
||||||
B int `json:"b"`
|
B int `json:"b"`
|
||||||
|
|
|
@ -79,6 +79,7 @@ func main() {
|
||||||
|
|
||||||
s.Register("multiply", rpc.H(Multiply))
|
s.Register("multiply", rpc.H(Multiply))
|
||||||
s.Register("divide", rpc.H(Divide))
|
s.Register("divide", rpc.H(Divide))
|
||||||
|
s.Register("hello", rpc.HS(Hello))
|
||||||
|
|
||||||
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
|
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
|
@ -102,6 +103,10 @@ func Divide(ctx context.Context, args *Args) (*Quotient, error) {
|
||||||
return quo, nil
|
return quo, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Hello(ctx context.Context) (string, error) {
|
||||||
|
return "world", nil
|
||||||
|
}
|
||||||
|
|
||||||
type Args struct {
|
type Args struct {
|
||||||
A int `json:"a"`
|
A int `json:"a"`
|
||||||
B int `json:"b"`
|
B int `json:"b"`
|
||||||
|
|
|
@ -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)
|
type HandlerFunc func(context.Context, json.RawMessage) (json.RawMessage, error)
|
||||||
|
|
Loading…
Reference in a new issue