adjust bitmart scraper for watchdog restart.

diadata-org · Aug 15, 2024 · fa7df4e · fa7df4e
1 parent d7efa31
commit fa7df4e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cmd/exchange-scrapers/collector/go.mod b/cmd/exchange-scrapers/collector/go.mod
@@ -3,7 +3,7 @@ module github.com/diadata-org/diadata/exchange-scrapers/collector
 go 1.19
 
 require (
-	github.com/diadata-org/diadata v1.4.518
+	github.com/diadata-org/diadata v1.4.519
 	github.com/segmentio/kafka-go v0.4.35
 	github.com/sirupsen/logrus v1.9.3
 )

diff --git a/pkg/dia/scraper/exchange-scrapers/BitMartScraper.go b/pkg/dia/scraper/exchange-scrapers/BitMartScraper.go
@@ -309,7 +309,7 @@ func (s *BitMartScraper) mainLoop() {
 							return
 						}
 						if err := s.retryConnection(idx); err != nil || ws.IsCloseError(err, ws.CloseAbnormalClosure) {
-							continue
+							return
 						}
 					}
 
@@ -330,7 +330,7 @@ func (s *BitMartScraper) mainLoop() {
 					case ws.TextMessage:
 						if string(msg) == bitMartPongMessage {
 							s.errCount[idx] = 0
-							continue
+							return
 						}
 						output = msg
 					}
@@ -342,7 +342,7 @@ func (s *BitMartScraper) mainLoop() {
 						s.setError(err)
 						s.errCount[idx]++
 						if err := s.retryConnection(idx); err != nil {
-							continue
+							return
 						}
 					}
 					if subResults.ErrorCode != "" {