1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
package main
import (
"context"
"errors"
"log"
"os"
"os/signal"
"go.neonxp.dev/jsonrpc2/rpc"
"go.neonxp.dev/jsonrpc2/rpc/middleware"
"go.neonxp.dev/jsonrpc2/transport"
)
func main() {
s := rpc.New(
rpc.WithLogger(rpc.StdLogger),
rpc.WithTransport(&transport.HTTP{Bind: ":8000", CORSOrigin: "*"}),
)
// Set options after constructor
serviceSchema := `
{
"divide": {
"request": {
"type": "object",
"properties": {
"a": {
"type": "integer"
},
"b": {
"type": "integer",
"not":{"const":0}
}
},
"required": ["a", "b"]
},
"response": {
"type": "object",
"properties": {
"quo": {
"type": "integer"
},
"rem": {
"type": "integer"
}
},
"required": ["quo", "rem"]
}
},
"multiply": {
"request": {
"type": "object",
"properties": {
"a": {
"type": "integer"
},
"b": {
"type": "integer"
}
},
"required": ["a", "b"]
},
"response": {
"type": "integer"
}
}
}`
validation, err := middleware.Validation(middleware.MustSchema(serviceSchema))
if err != nil {
log.Fatal(err)
}
s.Use(
rpc.WithTransport(&transport.TCP{Bind: ":3000"}),
rpc.WithMiddleware(middleware.Logger(rpc.StdLogger)),
rpc.WithMiddleware(validation),
)
s.Register("multiply", rpc.H(Multiply))
s.Register("divide", rpc.H(Divide))
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
defer cancel()
if err := s.Run(ctx); err != nil {
log.Fatal(err)
}
}
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
}
type Args struct {
A int `json:"a"`
B int `json:"b"`
}
type Quotient struct {
Quo int `json:"quo"`
Rem int `json:"rem"`
}
|
