feat: v5 ws write deadline for private, trade (#193)

hirokisan · Nov 2, 2024 · 28847ad · 28847ad
1 parent 51de080
commit 28847ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/v5_ws_private.go b/v5_ws_private.go
@@ -132,6 +132,7 @@ func (s *V5WebsocketPrivateService) Start(ctx context.Context, errHandler ErrHan
 		defer close(done)
 		defer s.connection.Close()
 		_ = s.connection.SetReadDeadline(time.Now().Add(60 * time.Second))
+		_ = s.connection.SetWriteDeadline(time.Now().Add(60 * time.Second))
 		s.connection.SetPongHandler(func(string) error {
 			_ = s.connection.SetReadDeadline(time.Now().Add(60 * time.Second))
 			return nil

diff --git a/v5_ws_trade.go b/v5_ws_trade.go
@@ -84,6 +84,7 @@ func (s *V5WebsocketTradeService) Start(ctx context.Context, errHandler ErrHandl
 		defer close(done)
 		defer s.connection.Close()
 		_ = s.connection.SetReadDeadline(time.Now().Add(60 * time.Second))
+		_ = s.connection.SetWriteDeadline(time.Now().Add(60 * time.Second))
 		s.connection.SetPongHandler(func(string) error {
 			_ = s.connection.SetReadDeadline(time.Now().Add(60 * time.Second))
 			return nil