mirror of
https://github.com/fluencelabs/tendermint
synced 2025-06-28 04:01:40 +00:00
ws_client fixes
This commit is contained in:
@ -1,73 +0,0 @@
|
||||
package core_client
|
||||
|
||||
import (
|
||||
"github.com/tendermint/tendermint/Godeps/_workspace/src/github.com/gorilla/websocket"
|
||||
rpctypes "github.com/tendermint/tendermint/rpc/types"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// A websocket client subscribes and unsubscribes to events
|
||||
type WSClient struct {
|
||||
host string
|
||||
conn *websocket.Conn
|
||||
}
|
||||
|
||||
// create a new connection
|
||||
func NewWSClient(addr string) *WSClient {
|
||||
return &WSClient{
|
||||
host: addr,
|
||||
}
|
||||
}
|
||||
|
||||
func (wsc *WSClient) Dial() (*http.Response, error) {
|
||||
dialer := websocket.DefaultDialer
|
||||
rHeader := http.Header{}
|
||||
conn, r, err := dialer.Dial(wsc.host, rHeader)
|
||||
if err != nil {
|
||||
return r, err
|
||||
}
|
||||
wsc.conn = conn
|
||||
return r, nil
|
||||
}
|
||||
|
||||
// subscribe to an event
|
||||
func (wsc *WSClient) Subscribe(eventid string) error {
|
||||
return wsc.conn.WriteJSON(rpctypes.RPCRequest{
|
||||
JSONRPC: "2.0",
|
||||
Id: "",
|
||||
Method: "subscribe",
|
||||
Params: []interface{}{eventid},
|
||||
})
|
||||
}
|
||||
|
||||
// unsubscribe from an event
|
||||
func (wsc *WSClient) Unsubscribe(eventid string) error {
|
||||
return wsc.conn.WriteJSON(rpctypes.RPCRequest{
|
||||
JSONRPC: "2.0",
|
||||
Id: "",
|
||||
Method: "unsubscribe",
|
||||
Params: []interface{}{eventid},
|
||||
})
|
||||
}
|
||||
|
||||
type WSMsg struct {
|
||||
Data []byte
|
||||
Error error
|
||||
}
|
||||
|
||||
// returns a channel from which messages can be pulled
|
||||
// from a go routine that reads the socket.
|
||||
// if the ws returns an error (eg. closes), we return
|
||||
func (wsc *WSClient) Read() chan *WSMsg {
|
||||
ch := make(chan *WSMsg)
|
||||
go func() {
|
||||
for {
|
||||
_, p, err := wsc.conn.ReadMessage()
|
||||
ch <- &WSMsg{p, err}
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
return ch
|
||||
}
|
Reference in New Issue
Block a user