lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-26 08:45:16 +00:00
Code Issues Actions 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()
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
for err != nil {
err = s.connect(ctx)
time.Sleep(5 * time.Second)
select {
case <-ctx.Done():
return
default:
err = s.connect(ctx)
time.Sleep(5 * time.Second)
}
}
continue