Added Websocket (pusher) support for Bitstamp.

This commit is contained in:
Adrian Gallagher
2015-03-23 18:07:51 +11:00
parent 6cdd6e01d1
commit f62d2acd9d
3 changed files with 77 additions and 1 deletions

View File

@@ -114,11 +114,16 @@ func (b *Bitstamp) SetAPIKeys(clientID, apiKey, apiSecret string) {
func (b *Bitstamp) Run() {
if b.Verbose {
log.Printf("%s Websocket: %s.", b.GetName(), IsEnabled(b.Websocket))
log.Printf("%s polling delay: %ds.\n", b.GetName(), b.PollingDelay)
}
b.GetBalance()
if b.Websocket {
go b.PusherClient()
}
for b.Enabled {
go func() {
BitstampBTC := b.GetTicker()

71
bitstampwebsocket.go Normal file
View File

@@ -0,0 +1,71 @@
package main
import (
"github.com/toorop/go-pusher"
"log"
)
type BitstampPusherOrderbook struct {
Asks[][] string `json:"asks"`
Bids[][] string `json:"bids"`
}
type BitstampPusherTrade struct {
Price float64 `json:"price"`
Amount float64 `json:"amount"`
ID int64 `json:"id"`
}
const (
BITSTAMP_PUSHER_KEY = "de504dc5763aeef9ff52"
)
func (b *Bitstamp) PusherClient() {
pusherClient, err := pusher.NewClient(BITSTAMP_PUSHER_KEY)
if err != nil {
log.Fatalln(err)
return
}
err = pusherClient.Subscribe("live_trades")
if err != nil {
log.Println("Subscription error : ", err)
return
}
err = pusherClient.Subscribe("order_book")
if err != nil {
log.Println("Subscription error : ", err)
return
}
dataChannelTrade, err := pusherClient.Bind("data")
if err != nil {
log.Println("Bind error: ", err)
return
}
tradeChannelTrade, err := pusherClient.Bind("trade")
if err != nil {
log.Println("Bind error: ", err)
return
}
log.Printf("%s Pusher client ready.\n", b.GetName())
for b.Websocket {
select {
case data := <-dataChannelTrade:
result := BitstampPusherOrderbook{}
err := JSONDecode([]byte(data.Data), &result)
if err != nil {
log.Println(err)
}
case trade := <-tradeChannelTrade:
result := BitstampPusherTrade{}
err := JSONDecode([]byte(trade.Data), &result)
if err != nil {
log.Println(err)
}
log.Printf("%s Pusher trade: Price: %f Amount: %f\n", b.GetName(), result.Price, result.Amount)
}
}
}

View File

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