bybit: enable multiconnection handling across websocket endpoints (#1670)

* glorious: whooops

* gk: nits

* Leak issue and edge case

* Websocket: Add SendMessageReturnResponses

* whooooooopsie

* gk: nitssssss

* Update exchanges/stream/stream_match.go

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* Update exchanges/stream/stream_match_test.go

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* linter: appease the linter gods

* gk: nits

* gk: drain brain

* started

* more changes before merge match pr

* gateio: still building out

* gateio: finish spot

* fix up tests in gateio

* Add tests for stream package

* rm unused field

* glorious: nits

* rn files, specifically set function names to asset and offload routing to websocket type.

* linter: fix

* Add futures websocket request support

* gateio: integrate with IBOTExchange (cherry pick my nose)

* linter: fix

* glorious: nits

* add counter and update gateio

* fix collision issue

* Update exchanges/stream/websocket.go

Co-authored-by: Scott <gloriousCode@users.noreply.github.com>

* glorious: nits

* add tests

* linter: fix

* After merge

* Add error connection info

* upgrade to upstream merge

* Fix edge case where it does not reconnect made by an already closed connection

* stream coverage

* glorious: nits

* glorious: nits removed asset error handling in stream package

* linter: fix

* rm block

* Add basic readme

* fix asset enabled flush cycle for multi connection

* spella: fix

* linter: fix

* Add glorious suggestions, fix some race thing

* reinstate name before any routine gets spawned

* stop on error in mock tests

* glorious: nits

* Set correct price

* glorious: nits found in CI build

* Add test for drain, bumped wait times as there seems to be something happening on macos CI builds, used context.WithTimeout because its instant.

* mutex across shutdown and connect for protection

* lint: fix

* test time withoffset, reinstate stop

* fix whoops

* const trafficCheckInterval; rm testmain

* y

* fix lint

* bump time check window

* stream: fix intermittant test failures while testing routines and remove code that is not needed.

* spells

* cant do what I did

* protect race due to routine.

* update testURL

* use mock websocket connection instead of test URL's

* linter: fix

* remove url because its throwing errors on CI builds

* connections drop all the time, don't need to worry about not being able to echo back ws data as it can be easily reviewed _test file side.

* remove another superfluous url thats not really set up for this

* spawn overwatch routine when there is no errors, inline checker instead of waiting for a time period, add sleep inline with echo handler as this is really quick and wanted to ensure that latency is handing correctly

* linter: fixerino uperino

* fix ID bug, why I do this, I don't know.

* glorious: panix

* linter: things

* whoops

* dont need to make consecutive Unix() calls

* websocket: fix potential panic on error and no responses and adding waitForResponses

* bybit: enable multiconnection handling across websocket endpoints

* rm debug lines

* rm json parser and handle in json package instead

* in favour of json package unmarshalling

* fix processing issues with tickers

* linter: fix

* linter: fix again

* * change field name OutboundRequestSignature to WrapperDefinedConnectionSignature for agnostic inbound and outbound connections.
* change method name GetOutboundConnection to GetConnection for agnostic inbound and outbound connections.
* drop outbound field map for improved performance just using a range and field check (less complex as well)
* change field name connections to connectionToWrapper for better clarity

* spells and magic and wands

* merge: fixup

* linter: fix

* spelling: fix

* glorious: nits

* comparable check for signature

* mv err var

* glorious: nits and stuff

* attempt to fix race

* linter: fix

* fix tests

* types/time: strict usage of time type for usage with unix timestamps

* fix tests etc

* glorious: nits

* gk: nits; engine log cleanup

* gk: nits; OCD

* gk: nits; move function change file names

* gk: nits; 🚀

* gk: nits; convert variadic function and message inspection to interface and include a specific function for that handling so as to not need nil on every call

* gk: nits; continued

* gk: engine nits; rm loaded exchange

* gk: nits; drop WebsocketLoginResponse

* stream: Add match method EnsureMatchWithData

* gk: nits; rn Inspect to IsFinal

* gk: nits; rn to MessageFilter

* linter: fix

* gateio: update rate limit definitions (cherry-pick)

* Add test and missing

* Shared REST rate limit definitions with Websocket service, set lookup item to nil for systems that do not require rate limiting; add glorious nit

* integrate rate limits for websocket trading spot

* bybit: split public and private processing to dedicated handler add supporting function and tests

* use correct handler for private inbound connection

* conform to match upstream changes

* standardise names to upstream style

* fix wrapper standards test when sending a auth request through a websocket connection

* whoops

* Update exchanges/gateio/gateio_types.go

Co-authored-by: Scott <gloriousCode@users.noreply.github.com>

* glorious: nits

* linter: fix

* linter: overload

* whoops

* spelling fixes on recent merge

* glorious: nits

* linter: fix?

* glorious: nits

* gk: assert errors touched

* gk: unexport derive functions

* gk: nitssssssss

* fix test

* gk: nitters v1

* gk: http status

* gk/nits: Add getAssetFromFuturesPair

* gk: nits single response when submitting

* gk: new pair with delimiter in tests

* gk: param update slice to slice of pointers

* gk: add asset type in params, includes t.Context() for tests

* linter: fix

* linter: fix

* fix merge whoopsie

* glorious: nits

* gk: nit

* linter: fix

* glorious: nits

* linter/misc: fix and remove meows

* okx: update requestID gen func without func wrapping

* RM: functions not needed

* Update docs/ADD_NEW_EXCHANGE.md

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* gk: nitsssssss

* linter: fix

* Update exchanges/bybit/bybit_test.go

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* Update exchanges/bybit/bybit_test.go

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* gk: nit words

* cranktakular: nits

* linter: fix

* cranktakular: nits and expand coverage

* linter: fix?

* misc fix

* cranktakular: missing nit which I thumbed up but did not do. Sillllllly billlyyyy nilllyyy

* cranktakular: nits

* cranktakular: purge DCP ref/handling and add another TODO

* Update exchanges/bybit/bybit_websocket.go

Co-authored-by: Scott <gloriousCode@users.noreply.github.com>

* glorious: nits

* fix test

* fix alignment issue and rm println

* Update exchanges/bybit/bybit_websocket.go

Co-authored-by: Scott <gloriousCode@users.noreply.github.com>

* Update exchanges/bybit/bybit_websocket.go

Co-authored-by: Scott <gloriousCode@users.noreply.github.com>

* glorious: fix

* Update exchanges/bybit/bybit_websocket.go

Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>

* Update common/common.go

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* Update common/common_test.go

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* Update exchanges/bybit/bybit_test.go

Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>

* gk: nits

* gk: nit with test

---------

Co-authored-by: Ryan O'Hara-Reid <ryan.oharareid@thrasher.io>
Co-authored-by: Gareth Kirwan <gbjkirwan@gmail.com>
Co-authored-by: Scott <gloriousCode@users.noreply.github.com>
Co-authored-by: Adrian Gallagher <adrian.gallagher@thrasher.io>
This commit is contained in:
Ryan O'Hara-Reid
2025-08-08 14:22:29 +10:00
committed by GitHub
parent ba92ba3254
commit dcf596c72b
32 changed files with 1475 additions and 939 deletions

View File

@@ -1692,50 +1692,39 @@ func TestAddTradesToBuffer(t *testing.T) {
}
func TestString(t *testing.T) {
if RestSpot.String() != restSpotURL {
t.Errorf("received '%v' expected '%v'", RestSpot, restSpotURL)
}
if RestSpotSupplementary.String() != restSpotSupplementaryURL {
t.Errorf("received '%v' expected '%v'", RestSpotSupplementary, restSpotSupplementaryURL)
}
if RestUSDTMargined.String() != "RestUSDTMarginedFuturesURL" {
t.Errorf("received '%v' expected '%v'", RestUSDTMargined, "RestUSDTMarginedFuturesURL")
}
if RestCoinMargined.String() != restCoinMarginedFuturesURL {
t.Errorf("received '%v' expected '%v'", RestCoinMargined, restCoinMarginedFuturesURL)
}
if RestFutures.String() != restFuturesURL {
t.Errorf("received '%v' expected '%v'", RestFutures, restFuturesURL)
}
if RestFuturesSupplementary.String() != restFuturesSupplementaryURL {
t.Errorf("received '%v' expected '%v'", RestFutures, restFuturesSupplementaryURL)
}
if RestUSDCMargined.String() != restUSDCMarginedFuturesURL {
t.Errorf("received '%v' expected '%v'", RestUSDCMargined, restUSDCMarginedFuturesURL)
}
if RestSandbox.String() != restSandboxURL {
t.Errorf("received '%v' expected '%v'", RestSandbox, restSandboxURL)
}
if RestSwap.String() != restSwapURL {
t.Errorf("received '%v' expected '%v'", RestSwap, restSwapURL)
}
if WebsocketSpot.String() != websocketSpotURL {
t.Errorf("received '%v' expected '%v'", WebsocketSpot, websocketSpotURL)
}
if WebsocketSpotSupplementary.String() != websocketSpotSupplementaryURL {
t.Errorf("received '%v' expected '%v'", WebsocketSpotSupplementary, websocketSpotSupplementaryURL)
}
if ChainAnalysis.String() != chainAnalysisURL {
t.Errorf("received '%v' expected '%v'", ChainAnalysis, chainAnalysisURL)
}
if EdgeCase1.String() != edgeCase1URL {
t.Errorf("received '%v' expected '%v'", EdgeCase1, edgeCase1URL)
}
if EdgeCase2.String() != edgeCase2URL {
t.Errorf("received '%v' expected '%v'", EdgeCase2, edgeCase2URL)
}
if EdgeCase3.String() != edgeCase3URL {
t.Errorf("received '%v' expected '%v'", EdgeCase3, edgeCase3URL)
t.Parallel()
for _, tc := range []struct {
url URL
expected string
}{
{0, ""},
{RestSpot, restSpotURL},
{RestSpotSupplementary, restSpotSupplementaryURL},
{RestUSDTMargined, restUSDTMarginedFuturesURL},
{RestCoinMargined, restCoinMarginedFuturesURL},
{RestFutures, restFuturesURL},
{RestFuturesSupplementary, restFuturesSupplementaryURL},
{RestUSDCMargined, restUSDCMarginedFuturesURL},
{RestSandbox, restSandboxURL},
{RestSwap, restSwapURL},
{WebsocketSpot, websocketSpotURL},
{WebsocketCoinMargined, websocketCoinMarginedURL},
{WebsocketUSDTMargined, websocketUSDTMarginedURL},
{WebsocketUSDCMargined, websocketUSDCMarginedURL},
{WebsocketOptions, websocketOptionsURL},
{WebsocketPrivate, websocketPrivateURL},
{WebsocketSpotSupplementary, websocketSpotSupplementaryURL},
{ChainAnalysis, chainAnalysisURL},
{EdgeCase1, edgeCase1URL},
{EdgeCase2, edgeCase2URL},
{EdgeCase3, edgeCase3URL},
{420, ""},
} {
t.Run(tc.url.String(), func(t *testing.T) {
t.Parallel()
assert.Equal(t, tc.expected, tc.url.String(), "String() should return the expected URL")
})
}
}
@@ -1886,20 +1875,26 @@ func TestGetGetURLTypeFromString(t *testing.T) {
Expected URL
Error error
}{
{Endpoint: "RestSpotURL", Expected: RestSpot},
{Endpoint: "RestSpotSupplementaryURL", Expected: RestSpotSupplementary},
{Endpoint: "RestUSDTMarginedFuturesURL", Expected: RestUSDTMargined},
{Endpoint: "RestCoinMarginedFuturesURL", Expected: RestCoinMargined},
{Endpoint: "RestFuturesURL", Expected: RestFutures},
{Endpoint: "RestUSDCMarginedFuturesURL", Expected: RestUSDCMargined},
{Endpoint: "RestSandboxURL", Expected: RestSandbox},
{Endpoint: "RestSwapURL", Expected: RestSwap},
{Endpoint: "WebsocketSpotURL", Expected: WebsocketSpot},
{Endpoint: "WebsocketSpotSupplementaryURL", Expected: WebsocketSpotSupplementary},
{Endpoint: "ChainAnalysisURL", Expected: ChainAnalysis},
{Endpoint: "EdgeCase1URL", Expected: EdgeCase1},
{Endpoint: "EdgeCase2URL", Expected: EdgeCase2},
{Endpoint: "EdgeCase3URL", Expected: EdgeCase3},
{Endpoint: restSpotURL, Expected: RestSpot},
{Endpoint: restSpotSupplementaryURL, Expected: RestSpotSupplementary},
{Endpoint: restUSDTMarginedFuturesURL, Expected: RestUSDTMargined},
{Endpoint: restCoinMarginedFuturesURL, Expected: RestCoinMargined},
{Endpoint: restFuturesURL, Expected: RestFutures},
{Endpoint: restFuturesSupplementaryURL, Expected: RestFuturesSupplementary},
{Endpoint: restUSDCMarginedFuturesURL, Expected: RestUSDCMargined},
{Endpoint: restSandboxURL, Expected: RestSandbox},
{Endpoint: restSwapURL, Expected: RestSwap},
{Endpoint: websocketSpotURL, Expected: WebsocketSpot},
{Endpoint: websocketCoinMarginedURL, Expected: WebsocketCoinMargined},
{Endpoint: websocketUSDTMarginedURL, Expected: WebsocketUSDTMargined},
{Endpoint: websocketUSDCMarginedURL, Expected: WebsocketUSDCMargined},
{Endpoint: websocketOptionsURL, Expected: WebsocketOptions},
{Endpoint: websocketPrivateURL, Expected: WebsocketPrivate},
{Endpoint: websocketSpotSupplementaryURL, Expected: WebsocketSpotSupplementary},
{Endpoint: chainAnalysisURL, Expected: ChainAnalysis},
{Endpoint: edgeCase1URL, Expected: EdgeCase1},
{Endpoint: edgeCase2URL, Expected: EdgeCase2},
{Endpoint: edgeCase3URL, Expected: EdgeCase3},
{Endpoint: "sillyMcSillyBilly", Expected: 0, Error: errEndpointStringNotFound},
}