Method Serve

rpc/jsonrpc/websocket.go:24–60

Serve accepts incoming connections and serve each.

()

// Serve accepts incoming connections and serve each.
func (ws *WebsocketServer) Serve() error {
	var (
		mux      = http.NewServeMux()
		upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}
		handler  = ws.RPCHandler
	)

	if handler == nil {
		handler = defaultHandler
	}

	mux.HandleFunc("/", func(rw http.ResponseWriter, r *http.Request) {
		conn, err := upgrader.Upgrade(rw, r, nil)
		if err != nil {
			log.WithError(err).Error("jsonrpc: upgrade http connection to websocket failed")
			http.Error(rw, errors.WithMessage(err, "could not upgrade to websocket").Error(), http.StatusBadRequest)
			return
		}
		defer conn.Close()

		// TODO: add metric for the connections
		<-jsonrpc2.NewConn(
			context.Background(),
			wsstream.NewObjectStream(conn),
			handler,
		).DisconnectNotify()
	})

	addr := ws.Addr
	listener, err := net.Listen("tcp", addr)
	if err != nil {
		return errors.Wrapf(err, "couldn't bind to address %q", addr)
	}

	ws.Handler = mux
	return ws.Server.Serve(listener)
}

// Stop stops the server and returns a channel indicating server is stopped.
func (ws *WebsocketServer) Stop() {