You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.3 KiB
69 lines
1.3 KiB
package core_client
|
|
|
|
import (
|
|
"github.com/gorilla/websocket"
|
|
"github.com/tendermint/tendermint/rpc"
|
|
"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(rpc.WSRequest{
|
|
Type: "subscribe",
|
|
Event: eventid,
|
|
})
|
|
}
|
|
|
|
// unsubscribe from an event
|
|
func (wsc *WSClient) Unsubscribe(eventid string) error {
|
|
return wsc.conn.WriteJSON(rpc.WSRequest{
|
|
Type: "unsubscribe",
|
|
Event: 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
|
|
}
|