Added LakeBTC Websocket support.

README.md

@@ -16,7 +16,7 @@ A cryptocurrency trading bot supporting multiple exchanges written in Golang.
 | Huobi | Yes | No |No
 | ItBit | Yes | NA | NA |
 | Kraken | Yes | NA | NA
-| LakeBTC | Yes | No | NA
+| LakeBTC | Yes | Yes | NA
 |OKCoin (both) | Yes | Yes | No
 
 ** NA means not applicable as the Exchange does not support the feature.

config_example.json

@@ -131,7 +131,7 @@
       "BaseCurrencies": "CNY",
       "Enabled": true,
       "Verbose": false,
-      "Websocket": false,
+      "Websocket": true,
       "PollingDelay": 10
     },
     {

lakebtchttp.go

@@ -43,6 +43,11 @@ type LakeBTCTicker struct {
 	Volume float64
 }
 
+type LakeBTCOrderbook struct {
+	Bids [][]float64 `json:"asks"`
+	Asks [][]float64 `json:"bids"`
+}
+
 type LakeBTCTickerResponse struct {
 	USD LakeBTCTicker
 	CNY LakeBTCTicker
@@ -85,8 +90,14 @@ func (l *LakeBTC) GetFee(maker bool) (float64) {
 
 func (l *LakeBTC) Run() {
 	if l.Verbose {
+		log.Printf("%s Websocket: %s.", l.GetName(), IsEnabled(l.Websocket))
 		log.Printf("%s polling delay: %ds.\n", l.GetName(), l.PollingDelay)
 	}
+
+	if l.Websocket {
+		l.WebsocketClient()
+	}
+
 	for l.Enabled {
 		go func() {
 			LakeBTCTickerResponse := l.GetTicker()

lakebtcwebsocket.go

@@ -0,0 +1,119 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+	"github.com/gorilla/websocket"
+)
+
+const (
+	LAKEBTC_WEBSOCKET_URL = "wss://www.LakeBTC.com/websocket"
+)
+
+func WSRailsSubscribe(channel string, conn *websocket.Conn) {
+	data := fmt.Sprintf(`["websocket_rails.subscribe", {"data":{"channel": "%s" }}]`, channel)
+	err := conn.WriteMessage(websocket.TextMessage, []byte(data))
+
+	if err != nil {
+		log.Println(err)
+		return
+	}
+}
+
+func WSRailsUnsubscribe(channel string, conn *websocket.Conn) {
+	data := fmt.Sprintf(`["websocket_rails.unsubscribe", {"data":{"channel": "%s" }}]`, channel)
+	err := conn.WriteMessage(websocket.TextMessage, []byte(data))
+
+	if err != nil {
+		log.Println(err)
+		return
+	}
+}
+
+func WSRailsPong(id string, conn *websocket.Conn) {
+	data := fmt.Sprintf(`["websocket_rails.pong", {"data":{"connection_id": %s}}]`, id)
+	err := conn.WriteMessage(websocket.TextMessage, []byte(data))
+
+	if err != nil {
+		log.Println(err)
+		return
+	}
+}
+
+func (l *LakeBTC) WebsocketClient() {
+	var Dialer websocket.Dialer
+	conn, resp, err := Dialer.Dial(LAKEBTC_WEBSOCKET_URL, http.Header{})
+
+	if err != nil {
+		log.Println(err)
+		return
+	}
+
+	if l.Verbose {
+		log.Printf("%s Connected to Websocket.", l.GetName())
+		log.Println(resp)
+	}
+
+	for {
+		msgType, resp, err := conn.ReadMessage()
+		if err != nil {
+			log.Println(err)
+			break
+		}
+
+		response := [][]interface{}{}
+		err = JSONDecode(resp, &response)
+
+		if err != nil {
+			log.Println(err)
+			break
+		}
+
+		if msgType == websocket.TextMessage {
+			event := response[0][0]
+			data := response[0][1]
+
+			switch event {
+			case "client_connected":
+				WSRailsSubscribe("ticker", conn)
+				WSRailsSubscribe("orderbook_CNY", conn)
+				WSRailsSubscribe("orderbook_USD", conn)
+			case "websocket_rails.subscribe":
+			case "websocket_rails.ping":
+				WSRailsPong("null", conn)
+			case "update":
+				update := data.(map[string]interface{})
+				channel := update["channel"]
+				data = update["data"]
+				dataJSON, err := JSONEncode(data)
+
+				if err != nil {
+					log.Println(err)
+					continue
+				}
+
+				switch channel {
+				case "ticker":
+					ticker := LakeBTCTickerResponse{}
+					err = JSONDecode(dataJSON, &ticker)
+
+					if err != nil {
+						log.Println(err)
+						continue
+					}
+				case "orderbook_USD", "orderbook_CNY":
+					orderbook := LakeBTCOrderbook{}
+					err = JSONDecode(dataJSON, &orderbook)
+
+					if err != nil {
+						log.Println(err)
+						continue
+					}
+				}
+			}
+		}
+	}
+	conn.Close()
+	log.Printf("%s Websocket client disconnected.", l.GetName())
+}