mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-06-01 15:10:44 +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
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package websocket
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"sync"
|
|
)
|
|
|
|
// ErrSignatureNotMatched is returned when a signature does not match a request
|
|
var ErrSignatureNotMatched = errors.New("websocket response to request signature not matched")
|
|
|
|
var (
|
|
errSignatureCollision = errors.New("signature collision")
|
|
errInvalidBufferSize = errors.New("buffer size must be positive")
|
|
)
|
|
|
|
// NewMatch returns a new Match
|
|
func NewMatch() *Match {
|
|
return &Match{m: make(map[any]*incoming)}
|
|
}
|
|
|
|
// Match is a distributed subtype that handles the matching of requests and
|
|
// responses in a timely manner, reducing the need to differentiate between
|
|
// connections. Stream systems fan in all incoming payloads to one routine for
|
|
// processing.
|
|
type Match struct {
|
|
m map[any]*incoming
|
|
mu sync.Mutex
|
|
}
|
|
|
|
type incoming struct {
|
|
expected int
|
|
c chan<- []byte
|
|
}
|
|
|
|
// IncomingWithData matches with requests and takes in the returned payload, to
|
|
// be processed outside of a stream processing routine and returns true if a handler was found
|
|
func (m *Match) IncomingWithData(signature any, data []byte) bool {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
ch, ok := m.m[signature]
|
|
if !ok {
|
|
return false
|
|
}
|
|
ch.c <- data
|
|
ch.expected--
|
|
if ch.expected == 0 {
|
|
close(ch.c)
|
|
delete(m.m, signature)
|
|
}
|
|
return true
|
|
}
|
|
|
|
// RequireMatchWithData validates that incoming data matches a request's signature.
|
|
// If a match is found, the data is processed; otherwise, it returns an error.
|
|
func (m *Match) RequireMatchWithData(signature any, data []byte) error {
|
|
if m.IncomingWithData(signature, data) {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("'%v' %w with data %v", signature, ErrSignatureNotMatched, string(data))
|
|
}
|
|
|
|
// Set the signature response channel for incoming data
|
|
func (m *Match) Set(signature any, bufSize int) (<-chan []byte, error) {
|
|
if bufSize <= 0 {
|
|
return nil, errInvalidBufferSize
|
|
}
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
if _, ok := m.m[signature]; ok {
|
|
return nil, errSignatureCollision
|
|
}
|
|
ch := make(chan []byte, bufSize)
|
|
m.m[signature] = &incoming{expected: bufSize, c: ch}
|
|
return ch, nil
|
|
}
|
|
|
|
// RemoveSignature removes the signature response from map and closes the channel.
|
|
func (m *Match) RemoveSignature(signature any) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
if ch, ok := m.m[signature]; ok {
|
|
close(ch.c)
|
|
delete(m.m, signature)
|
|
}
|
|
}
|