jsonrpc2/example/main.go

54 lines
951 B
Go
Raw Permalink 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-21 20:38:21 +03:00
"go.neonxp.dev/jsonrpc2/transport"
2022-01-31 02:31:43 +03:00
)
func main() {
2022-05-21 20:38:21 +03:00
s := rpc.New()
2022-01-31 20:17:31 +03:00
2022-05-21 20:38:21 +03:00
s.AddTransport(&transport.HTTP{Bind: ":8000"})
s.AddTransport(&transport.TCP{Bind: ":3000"})
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))
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
}
type Args struct {
A int `json:"a"`
B int `json:"b"`
}
type Quotient struct {
Quo int `json:"quo"`
Rem int `json:"rem"`
}