lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-22 14:55:16 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

binance: add listenKeyExpired callback

Browse Source
This commit is contained in:
c9s 2022-09-11 14:10:30 +08:00
parent 4890d19ebf
commit ff635195fb
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/exchange/binance/stream.go
View File

@ -62,6 +62,8 @@ type Stream struct {
accountUpdateEventCallbacks []func(e *AccountUpdateEvent)
accountConfigUpdateEventCallbacks []func(e *AccountConfigUpdateEvent)
listenKeyExpiredCallbacks []func(e *ListenKeyExpired)
depthBuffers map[string]*depth.Buffer
}
@ -363,6 +365,10 @@ func (s *Stream) dispatchEvent(e interface{}) {
case *AccountConfigUpdateEvent:
s.EmitAccountConfigUpdateEvent(e)
case *ListenKeyExpired:
s.EmitListenKeyExpired(e)
}
}

12
pkg/exchange/binance/stream_callbacks.go
View File

@ -154,6 +154,16 @@ func (s *Stream) EmitAccountConfigUpdateEvent(e *AccountConfigUpdateEvent) {
}
}
func (s *Stream) OnListenKeyExpired(cb func(e *ListenKeyExpired)) {
s.listenKeyExpiredCallbacks = append(s.listenKeyExpiredCallbacks, cb)
}
func (s *Stream) EmitListenKeyExpired(e *ListenKeyExpired) {
for _, cb := range s.listenKeyExpiredCallbacks {
cb(e)
}
}
type StreamEventHub interface {
OnDepthEvent(cb func(e *DepthEvent))
@ -184,4 +194,6 @@ type StreamEventHub interface {
OnAccountUpdateEvent(cb func(e *AccountUpdateEvent))
OnAccountConfigUpdateEvent(cb func(e *AccountConfigUpdateEvent))
OnListenKeyExpired(cb func(e *ListenKeyExpired))
}