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
169 lines
5.3 KiB
Go
169 lines
5.3 KiB
Go
package engine
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"sync"
|
|
|
|
"github.com/gorilla/mux"
|
|
gws "github.com/gorilla/websocket"
|
|
"github.com/thrasher-corp/gocryptotrader/config"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/account"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/orderbook"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/ticker"
|
|
)
|
|
|
|
// Const vars for websocket
|
|
const (
|
|
WebsocketResponseSuccess = "OK"
|
|
restIndexResponse = "<html>GoCryptoTrader RESTful interface. For the web GUI, please visit the <a href=https://github.com/thrasher-corp/gocryptotrader/blob/master/web/README.md>web GUI readme.</a></html>"
|
|
DeprecatedName = "deprecated_rpc"
|
|
WebsocketName = "websocket_rpc"
|
|
)
|
|
|
|
var (
|
|
wsHub *websocketHub
|
|
wsHubStarted bool
|
|
errNilRemoteConfig = errors.New("received nil remote config")
|
|
errNilPProfConfig = errors.New("received nil pprof config")
|
|
errNilBot = errors.New("received nil engine bot")
|
|
errEmptyConfigPath = errors.New("received empty config path")
|
|
errServerDisabled = errors.New("server disabled")
|
|
errAlreadyRunning = errors.New("already running")
|
|
// ErrWebsocketServiceNotRunning occurs when a message is sent to be broadcast via websocket
|
|
// and its not running
|
|
ErrWebsocketServiceNotRunning = errors.New("websocket service not started")
|
|
)
|
|
|
|
// apiServerManager holds all relevant fields to manage both REST and websocket
|
|
// api servers
|
|
type apiServerManager struct {
|
|
restStarted int32
|
|
websocketStarted int32
|
|
restListenAddress string
|
|
websocketListenAddress string
|
|
gctConfigPath string
|
|
restHTTPServer *http.Server
|
|
websocketHTTPServer *http.Server
|
|
wgRest sync.WaitGroup
|
|
wgWebsocket sync.WaitGroup
|
|
|
|
restRouter *mux.Router
|
|
websocketRouter *mux.Router
|
|
websocketHub *websocketHub
|
|
|
|
remoteConfig *config.RemoteControlConfig
|
|
pprofConfig *config.Profiler
|
|
exchangeManager iExchangeManager
|
|
bot iBot
|
|
portfolioManager iPortfolioManager
|
|
}
|
|
|
|
// websocketClient stores information related to the websocket client
|
|
type websocketClient struct {
|
|
Hub *websocketHub
|
|
Conn *gws.Conn
|
|
Authenticated bool
|
|
authFailures int
|
|
Send chan []byte
|
|
username string
|
|
password string
|
|
maxAuthFailures int
|
|
exchangeManager iExchangeManager
|
|
bot iBot
|
|
portfolioManager iPortfolioManager
|
|
configPath string
|
|
}
|
|
|
|
// websocketHub stores the data for managing websocket clients
|
|
type websocketHub struct {
|
|
Clients map[*websocketClient]bool
|
|
Broadcast chan []byte
|
|
Register chan *websocketClient
|
|
Unregister chan *websocketClient
|
|
}
|
|
|
|
// WebsocketEvent is the struct used for websocket events
|
|
type WebsocketEvent struct {
|
|
Exchange string `json:"exchange,omitempty"`
|
|
AssetType string `json:"assetType,omitempty"`
|
|
Event string
|
|
Data any
|
|
}
|
|
|
|
// WebsocketEventResponse is the struct used for websocket event responses
|
|
type WebsocketEventResponse struct {
|
|
Event string `json:"event"`
|
|
Data any `json:"data"`
|
|
Error string `json:"error"`
|
|
}
|
|
|
|
// WebsocketOrderbookTickerRequest is a struct used for ticker and orderbook
|
|
// requests
|
|
type WebsocketOrderbookTickerRequest struct {
|
|
Exchange string `json:"exchangeName"`
|
|
Currency string `json:"currency"`
|
|
AssetType string `json:"assetType"`
|
|
}
|
|
|
|
// WebsocketAuth is a struct used for
|
|
type WebsocketAuth struct {
|
|
Username string `json:"username"`
|
|
Password string `json:"password"`
|
|
}
|
|
|
|
// Route is a sub type that holds the request routes
|
|
type Route struct {
|
|
Name string
|
|
Method string
|
|
Pattern string
|
|
HandlerFunc http.HandlerFunc
|
|
}
|
|
|
|
// AllEnabledExchangeOrderbooks holds the enabled exchange orderbooks
|
|
type AllEnabledExchangeOrderbooks struct {
|
|
Data []EnabledExchangeOrderbooks `json:"data"`
|
|
}
|
|
|
|
// EnabledExchangeOrderbooks is a sub type for singular exchanges and respective
|
|
// orderbooks
|
|
type EnabledExchangeOrderbooks struct {
|
|
ExchangeName string `json:"exchangeName"`
|
|
ExchangeValues []orderbook.Base `json:"exchangeValues"`
|
|
}
|
|
|
|
// AllEnabledExchangeCurrencies holds the enabled exchange currencies
|
|
type AllEnabledExchangeCurrencies struct {
|
|
Data []EnabledExchangeCurrencies `json:"data"`
|
|
}
|
|
|
|
// EnabledExchangeCurrencies is a sub type for singular exchanges and respective
|
|
// currencies
|
|
type EnabledExchangeCurrencies struct {
|
|
ExchangeName string `json:"exchangeName"`
|
|
ExchangeValues []*ticker.Price `json:"exchangeValues"`
|
|
}
|
|
|
|
// AllEnabledExchangeAccounts holds all enabled accounts info
|
|
type AllEnabledExchangeAccounts struct {
|
|
Data []account.Holdings `json:"data"`
|
|
}
|
|
|
|
var wsHandlers = map[string]wsCommandHandler{
|
|
"auth": {authRequired: false, handler: wsAuth},
|
|
"getconfig": {authRequired: true, handler: wsGetConfig},
|
|
"saveconfig": {authRequired: true, handler: wsSaveConfig},
|
|
"getaccountinfo": {authRequired: true, handler: wsGetAccountInfo},
|
|
"gettickers": {authRequired: false, handler: wsGetTickers},
|
|
"getticker": {authRequired: false, handler: wsGetTicker},
|
|
"getorderbooks": {authRequired: false, handler: wsGetOrderbooks},
|
|
"getorderbook": {authRequired: false, handler: wsGetOrderbook},
|
|
"getexchangerates": {authRequired: false, handler: wsGetExchangeRates},
|
|
"getportfolio": {authRequired: true, handler: wsGetPortfolio},
|
|
}
|
|
|
|
type wsCommandHandler struct {
|
|
authRequired bool
|
|
handler func(client *websocketClient, data any) error
|
|
}
|