tools/monitor_tx/ws_client.go

   1 package main
   2
   3 import (
   4         "encoding/json"
   5         "net/url"
   6
   7         "github.com/gorilla/websocket"
   8 )
   9
  10 type WSClient struct {
  11         Conn *websocket.Conn
  12 }
  13
  14 func (WS *WSClient) New(host string) error {
  15         u := url.URL{Scheme: "ws", Host: host, Path: "/websocket-subscribe"}
  16         c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
  17         if err != nil {
  18                 return err
  19         }
  20         WS.Conn = c
  21         return nil
  22 }
  23
  24 func (WS *WSClient) SendData(req interface{}) {
  25         msg, _ := json.Marshal(req)
  26         WS.Conn.WriteMessage(websocket.TextMessage, msg)
  27 }
  28
  29 func (WS *WSClient) RecvData() ([]byte, error) {
  30         _, msg, err := WS.Conn.ReadMessage()
  31         return msg, err
  32 }