d0zingcat/gocryptotrader
1
0
Fork 0
mirror of https://github.com/d0zingcat/gocryptotrader.git synced 2026-05-16 15:09:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reconnect Huobi Websocket on disconnection.

Browse Source
This commit is contained in:
Adrian Gallagher
2015-04-12 23:24:58 +10:00
parent 32c1252283
commit c6bfa15e86
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
btcchinahttp.go
View File

@@ -203,6 +203,10 @@ func (b *BTCChina) Run() {
log.Printf("%s polling delay: %ds.\n", b.GetName(), b.PollingDelay)
}
if b.Websocket {
go b.WebsocketClient()
}
b.GetAccountInfo("all")
for b.Enabled {

14
huobiwebsocket.go
View File

@@ -130,7 +130,12 @@ func (h *HUOBI) OnConnect(output chan socketio.Message) {
}
func (h *HUOBI) OnDisconnect(output chan socketio.Message) {
log.Println("Disconnected from websocket client.. Reconnecting")
log.Printf("%s Disconnected from websocket server.. Reconnecting.\n", h.GetName())
h.WebsocketClient()
}
func (h *HUOBI) OnError() {
log.Printf("%s Error with Websocket connection.. Reconnecting.\n", h.GetName())
h.WebsocketClient()
}
@@ -154,15 +159,16 @@ func (h *HUOBI) WebsocketClient() {
HuobiSocket = &socketio.SocketIO{
OnConnect: h.OnConnect,
OnDisconnect: h.OnDisconnect,
OnEvent: events,
OnError: h.OnError,
OnDisconnect: h.OnDisconnect,
}
err := socketio.ConnectToSocket(HUOBI_SOCKETIO_ADDRESS, HuobiSocket)
if err != nil {
fmt.Println(err)
return
}
log.Printf("%s Websocket client disconnected.", h.GetName())
log.Printf("%s Websocket client disconnected.. Reconnecting.", h.GetName())
h.WebsocketClient()
}