58 lines
1.1 KiB
Go
58 lines
1.1 KiB
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(
|
|
rpc.WithLogger(rpc.StdLogger),
|
|
rpc.WithTransport(&transport.HTTP{Bind: ":8000", CORSOrigin: "*"}),
|
|
)
|
|
// Set options after constructor
|
|
s.Use(
|
|
rpc.WithTransport(&transport.TCP{Bind: ":3000"}),
|
|
rpc.WithMiddleware(rpc.LoggerMiddleware(rpc.StdLogger)),
|
|
)
|
|
|
|
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"`
|
|
}
|