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

fix reconnect close

Browse Source
This commit is contained in:
c9s 2020-07-13 01:06:04 +08:00
parent 906350a5bc
commit a9e25c08d7
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
bbgo/exchange/binance/stream.go
View File

@ -147,12 +147,20 @@ func (s *PrivateStream) read(ctx context.Context, eventC chan interface{}) {
mt, message, err := s.Conn.ReadMessage() mt, message, err := s.Conn.ReadMessage()
if err != nil { if err != nil {
log.WithError(err).Errorf("read error: %s", err.Error()) if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
log.WithError(err).Errorf("read error: %s", err.Error())
}
// reconnect // reconnect
for err != nil { for err != nil {
err = s.connect(ctx) select {
time.Sleep(5 * time.Second) case <-ctx.Done():
return
default:
err = s.connect(ctx)
time.Sleep(5 * time.Second)
}
} }
continue continue