71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
// Program wshttp demonstrates how to set up a JSON-RPC 20 server using
|
|
// the github.com/creachadair/jrpc2 package with a Websocket transport.
|
|
//
|
|
// Usage:
|
|
// go build github.com/creachadair/jrpc2/tools/examples/wshttp
|
|
// ./wshttp -listen :8080
|
|
//
|
|
// The server accepts RPC connections on ws://<address>/rpc.
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/creachadair/jrpc2"
|
|
"github.com/creachadair/jrpc2/channel"
|
|
"github.com/creachadair/jrpc2/handler"
|
|
"github.com/creachadair/jrpc2/server"
|
|
"github.com/creachadair/wschannel"
|
|
)
|
|
|
|
var listenAddr = flag.String("listen", "", "Service address")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if *listenAddr == "" {
|
|
log.Fatal("You must provide a non-empty -listen address")
|
|
}
|
|
|
|
lst := wschannel.NewListener(nil)
|
|
hs := &http.Server{Addr: *listenAddr, Handler: http.DefaultServeMux}
|
|
http.Handle("/rpc", lst)
|
|
go hs.ListenAndServe()
|
|
|
|
acc := accepter{
|
|
Listener: lst,
|
|
ctx: context.Background(),
|
|
}
|
|
svc := handler.Map{"Reverse": handler.New(reverse)}
|
|
|
|
log.Printf("Listing at ws://%s/rpc", *listenAddr)
|
|
err := server.Loop(acc, server.Static(svc), &server.LoopOptions{
|
|
ServerOptions: &jrpc2.ServerOptions{
|
|
Logger: jrpc2.StdLogger(nil),
|
|
},
|
|
})
|
|
hs.Shutdown(acc.ctx)
|
|
if err != nil {
|
|
log.Fatalf("Loop exited: %v", err)
|
|
}
|
|
}
|
|
|
|
func reverse(_ context.Context, ss []string) []string {
|
|
for i, j := 0, len(ss)-1; i < j; i++ {
|
|
ss[i], ss[j] = ss[j], ss[i]
|
|
j--
|
|
}
|
|
return ss
|
|
}
|
|
|
|
type accepter struct {
|
|
*wschannel.Listener
|
|
ctx context.Context
|
|
}
|
|
|
|
func (a accepter) Accept() (channel.Channel, error) {
|
|
return a.Listener.Accept(a.ctx)
|
|
}
|