feat: support text ping pong

withchao · Nov 5, 2024 · f185c98 · f185c98
1 parent b49a8dd
commit f185c98
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/msggateway/client.go b/internal/msggateway/client.go
@@ -438,6 +438,9 @@ func (c *Client) handlerTextMessage(b []byte) error {
 		if err != nil {
 			return err
 		}
+		if err := c.conn.SetWriteDeadline(writeWait); err != nil {
+			return err
+		}
 		return c.conn.WriteMessage(MessageText, msgData)
 	default:
 		return fmt.Errorf("not support message type %s", msg.Type)