jsonrpc2/example/main.go

119 lines
2.1 KiB
Go
Raw Normal View History

2022-01-31 02:31:43 +03:00
package main
import (
"context"
"errors"
2022-05-21 20:38:21 +03:00
"log"
"os"
"os/signal"
2022-01-31 02:31:43 +03:00
"go.neonxp.dev/jsonrpc2/rpc"
2022-05-28 16:53:20 +03:00
"go.neonxp.dev/jsonrpc2/rpc/middleware"
2022-05-21 20:38:21 +03:00
"go.neonxp.dev/jsonrpc2/transport"
2022-01-31 02:31:43 +03:00
)
func main() {
2022-05-22 16:37:48 +03:00
s := rpc.New(
rpc.WithLogger(rpc.StdLogger),
rpc.WithTransport(&transport.HTTP{Bind: ":8000", CORSOrigin: "*"}),
)
2022-05-29 14:18:10 +03:00
2022-05-22 16:37:48 +03:00
// Set options after constructor
2022-05-29 14:18:10 +03:00
serviceSchema := `
{
2022-05-28 16:53:20 +03:00
"divide": {
2022-05-29 14:18:10 +03:00
"request": {
2022-05-28 16:53:20 +03:00
"type": "object",
"properties": {
"a": {
"type": "integer"
},
"b": {
"type": "integer",
"not":{"const":0}
}
},
"required": ["a", "b"]
2022-05-29 14:18:10 +03:00
},
"response": {
2022-05-28 16:53:20 +03:00
"type": "object",
"properties": {
"quo": {
"type": "integer"
},
"rem": {
"type": "integer"
}
},
"required": ["quo", "rem"]
2022-05-29 14:18:10 +03:00
}
2022-05-28 16:53:20 +03:00
},
2022-05-29 14:18:10 +03:00
"multiply": {
"request": {
"type": "object",
"properties": {
"a": {
"type": "integer"
},
"b": {
"type": "integer"
}
},
"required": ["a", "b"]
},
"response": {
"type": "integer"
}
}
}`
validation, err := middleware.Validation(middleware.MustSchema(serviceSchema))
2022-05-28 16:53:20 +03:00
if err != nil {
log.Fatal(err)
}
2022-05-22 16:37:48 +03:00
s.Use(
rpc.WithTransport(&transport.TCP{Bind: ":3000"}),
2022-05-28 16:53:20 +03:00
rpc.WithMiddleware(middleware.Logger(rpc.StdLogger)),
rpc.WithMiddleware(validation),
2022-05-22 16:37:48 +03:00
)
2022-01-31 02:31:43 +03:00
2022-05-21 20:38:21 +03:00
s.Register("multiply", rpc.H(Multiply))
s.Register("divide", rpc.H(Divide))
s.Register("hello", rpc.HS(Hello))
2022-05-21 20:38:21 +03:00
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
defer cancel()
if err := s.Run(ctx); err != nil {
log.Fatal(err)
}
2022-01-31 02:31:43 +03:00
}
func Multiply(ctx context.Context, args *Args) (int, error) {
return args.A * args.B, nil
}
func Divide(ctx context.Context, args *Args) (*Quotient, error) {
if args.B == 0 {
return nil, errors.New("divide by zero")
}
quo := new(Quotient)
quo.Quo = args.A / args.B
quo.Rem = args.A % args.B
return quo, nil
}
func Hello(ctx context.Context) (string, error) {
return "world", nil
}
2022-01-31 02:31:43 +03:00
type Args struct {
A int `json:"a"`
B int `json:"b"`
}
type Quotient struct {
Quo int `json:"quo"`
Rem int `json:"rem"`
}