53 lines
951 B
Go
53 lines
951 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
|
|
"go.neonxp.dev/jsonrpc2/rpc"
|
|
"go.neonxp.dev/jsonrpc2/transport"
|
|
)
|
|
|
|
func main() {
|
|
s := rpc.New()
|
|
|
|
s.AddTransport(&transport.HTTP{Bind: ":8000"})
|
|
s.AddTransport(&transport.TCP{Bind: ":3000"})
|
|
|
|
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"`
|
|
}
|