Golang Websocket JSONRPC server and client

client.go

package main

import (
	"golang.org/x/net/websocket"
	"log"
	"net/rpc/jsonrpc"
)

// In a real project, these would be defined in a common file
type Args struct {
	A int
	B int
}

type Arith int

func main() {
	origin := "http://localhost/"
	url := "ws://localhost:7000/ws"
	ws, err := websocket.Dial(url, "", origin)
	if err != nil {
		log.Fatal(err)
	}

	defer ws.Close()

	args := Args{7, 8}
	var reply int

	c := jsonrpc.NewClient(ws)
	err = c.Call("Arith.Multiply", args, &reply)
	if err != nil {
		log.Fatal("arith error:", err)
	}
	log.Printf("Arith: %d*%d=%d", args.A, args.B, reply)
}

server.go

package main

import (
	"golang.org/x/net/websocket"
	"log"
	"net/http"
	"net/rpc"
	"net/rpc/jsonrpc"
)
// In a real project, these would be defined in a common file
type Args struct {
	A int
	B int
}

type Arith int

func (t *Arith) Multiply(args *Args, reply *int) error {
	*reply = args.A * args.B
	return nil
}

func main() {
	rpc.Register(new(Arith))

	http.Handle("/ws", websocket.Handler(serve))
	http.ListenAndServe("localhost:7000", nil)
}

func serve(ws *websocket.Conn) {
	log.Printf("Handler starting")
	jsonrpc.ServeConn(ws)
	log.Printf("Handler exiting")
}