Question

我有一个前端aplication，它连接到websocket并尝试提供json-rpc2来从服务器调用客户端。在使用＆＃34; github.com/gorilla/websocket"？编写的服务器应用程序中为它创建*rpc.Client的最佳方法是什么？我尝试使用UnderlyingConn()的{{1}}：*websocket.Conn，但它不起作用。

Answer 1

由于NewClient，我向io.ReadWriteCloser提供websocket实施，将读/写转换为powerman操作。

package jsonrpc2ws

import (
    "github.com/powerman/rpc-codec/jsonrpc2"
    "github.com/gorilla/websocket"
    "encoding/json"
    "io"
    "sync"
)

func NewClient(ws *websocket.Conn) *jsonrpc2.Client {
    c := &client{
        ws:ws,
    }
    c.initRead()
    c.initWrite()
    return jsonrpc2.NewClient(c)
}

type client struct {
    w *io.PipeWriter
    r *io.PipeReader
    ws *websocket.Conn

    err error      // io error
    mu sync.Mutex  // protects err
}


func (x *client) error() (err error) {
    x.mu.Lock()
    err = x.err
    x.mu.Unlock()
    return
}

func (x *client) setError (err error) {
    x.mu.Lock()
    if x.err != nil {
        x.err = err
    }
    x.mu.Unlock()
}

func (x *client) initRead(){
    var w *io.PipeWriter
    x.r, w = io.Pipe()
    go func(){
        _,b,err := x.ws.ReadMessage()
        for  ; x.error() == nil && err == nil; _,b,err = x.ws.ReadMessage(){
            _,err = w.Write(b)
        }
        x.setError(err)
        w.Close()
    } ()
}

func (x *client) initWrite(){
    var r *io.PipeReader
    r,x.w = io.Pipe()
    go func(){
        dec := json.NewDecoder(r)
        var b json.RawMessage
        err := dec.Decode(&b)
        for  ; x.error() == nil &&  err == nil; err = dec.Decode(&b){
            err = x.ws.WriteMessage(websocket.TextMessage, b)
        }
        x.setError(err)
        r.Close()
    } ()
}

func (x *client) Write(p []byte) (int, error){
    if x.error() != nil {
        return 0, x.error()
    }
    return x.w.Write(p)
}

func (x *client) Read(p []byte) (int, error){
    if x.error() != nil {
        return 0, x.error()
    }
    return x.r.Read(p)
}

func (x *client) Close() error{
    x.ws.Close()
    x.w.Close()
    x.r.Close()
    return nil
}

如何设置gorilla / websocket连接以充当浏览器应用程序的JSON-RPC客户端？

1 个答案: