mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
b4e45e9a1b
* Websocket: Rename stream package * Websocket: Rename Websocket to Manager * Websocket: Replace explicit errs with common.NilGuard * Websocket: Move websocket_types.go to types.go * Websocket: Minor field comment and alignment in types * Webosocket: Rename WebsocketConnection to Connection * Alphapoint: Make gorilla ws import explicit Just to avoid confusion with our own packages. * Websocket: Move stream_match to match * Websocket: Move websocket_connection to connection * Websocket: Move websocket.go to manager.go * Websocket: Break out all subscription methods into subscriptions.go * Websocket: Move connection type into its file * Websocket: Remove PositionUpdated Type is not used anywhere * Kraken: Use local constant for pong Was the only use of websocket.Pong and doesn't really feel right to represent kraken's api resp in one of our packages * Websocket: Move connection sub-types to connection package * Websocket: Move manager types into manager * Websocket: Move ConnectionWrapper into manager * Websocket: Move websocket_test to manager_test * Websocket: Privatise connectionWrapper * Websocket: Remaining types into types.go These really belong somewhere else mostly, but this will do for now * Websocket: Tidy up connection method vars * Gofumpt: Moving package imports around * Websocket: Rename errDuplicateConnectionSetup * Websocket: Fix duplicate import of gws * Websocket: Fix gofumpt -extra * Websocket: Standardise import of gws across other pkgs * Kraken: Remove unused sub conf consts These were replaced by the generic Levels and Depth fields on all subs * Websocket: Privitise ConnectioWrapper fields * Websocket: inline single use var WebsocketNotAuthenticatedUsingRest * Websocket: Move documentation to template * Bithumb: Assertify TestWsHandleData
77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
package alphapoint
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
gws "github.com/gorilla/websocket"
|
|
"github.com/thrasher-corp/gocryptotrader/encoding/json"
|
|
exchange "github.com/thrasher-corp/gocryptotrader/exchanges"
|
|
"github.com/thrasher-corp/gocryptotrader/log"
|
|
)
|
|
|
|
const (
|
|
alphapointDefaultWebsocketURL = "wss://sim3.alphapoint.com:8401/v1/GetTicker/"
|
|
)
|
|
|
|
// WebsocketClient starts a new webstocket connection
|
|
func (a *Alphapoint) WebsocketClient() {
|
|
for a.Enabled {
|
|
var dialer gws.Dialer
|
|
var err error
|
|
var httpResp *http.Response
|
|
endpoint, err := a.API.Endpoints.GetURL(exchange.WebsocketSpot)
|
|
if err != nil {
|
|
log.Errorln(log.WebsocketMgr, err)
|
|
}
|
|
a.WebsocketConn, httpResp, err = dialer.Dial(endpoint, http.Header{})
|
|
httpResp.Body.Close() // not used, so safely free the body
|
|
|
|
if err != nil {
|
|
log.Errorf(log.ExchangeSys, "%s Unable to connect to Websocket. Error: %s\n", a.Name, err)
|
|
continue
|
|
}
|
|
|
|
if a.Verbose {
|
|
log.Debugf(log.ExchangeSys, "%s Connected to Websocket.\n", a.Name)
|
|
}
|
|
|
|
err = a.WebsocketConn.WriteMessage(gws.TextMessage, []byte(`{"messageType": "logon"}`))
|
|
if err != nil {
|
|
log.Errorln(log.ExchangeSys, err)
|
|
return
|
|
}
|
|
|
|
for a.Enabled {
|
|
msgType, resp, err := a.WebsocketConn.ReadMessage()
|
|
if err != nil {
|
|
log.Errorln(log.ExchangeSys, err)
|
|
break
|
|
}
|
|
|
|
if msgType == gws.TextMessage {
|
|
type MsgType struct {
|
|
MessageType string `json:"messageType"`
|
|
}
|
|
|
|
msgType := MsgType{}
|
|
err := json.Unmarshal(resp, &msgType)
|
|
if err != nil {
|
|
log.Errorln(log.ExchangeSys, err)
|
|
continue
|
|
}
|
|
|
|
if msgType.MessageType == "Ticker" {
|
|
ticker := WebsocketTicker{}
|
|
err = json.Unmarshal(resp, &ticker)
|
|
if err != nil {
|
|
log.Errorln(log.ExchangeSys, err)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
a.WebsocketConn.Close()
|
|
log.Debugf(log.ExchangeSys, "%s Websocket client disconnected.", a.Name)
|
|
}
|
|
}
|