fix: refactor read message function

internal/delivery/websocket/client.go

@@ -66,7 +66,7 @@ func (c *client) Run(ctx context.Context) error {
 		return err
 	}
 
-	message, err := c.readMessage()
+	message, err := c.conn.ReadData()
 	if err != nil {
 		c.logger.Error(err)
 		return err
@@ -97,25 +97,13 @@ func (c *client) Run(ctx context.Context) error {
 	return nil
 }
 
-// readMessage from socket response
-func (c *client) readMessage() ([]byte, error) {
-	var message = make([]byte, 1024*1024) //TODO need change global? 1MB
-
-	n, err := c.conn.ReadData(message)
-	if err != nil {
-		return nil, err
-	}
-
-	return message[:n], nil
-}
-
 // responseReader write to symbol channel from response socket data
 func (c *client) responseReader(symbol string, hMap map[string]chan entity.Ticker) error {
 	var mu = sync.Mutex{}
 	var tickData *coinbase.Response
 
 	for {
-		message, err := c.readMessage()
+		message, err := c.conn.ReadData()
 		if err != nil {
 			if errors.Is(err, net.ErrClosed) {
 				break

pkg/exchange/coinbase/client.go

@@ -37,8 +37,15 @@ func (c *client) WriteData(message []byte) (int, error) {
 	return c.Write(message)
 }
 
-func (c *client) ReadData(data []byte) (int, error) {
-	return c.Read(data)
+func (c *client) ReadData() ([]byte, error) {
+	var message = make([]byte, 512) //TODO need change global? 1MB
+
+	n, err := c.Read(message)
+	if err != nil {
+		return nil, err
+	}
+
+	return message[:n], nil
 }
 
 func (c *client) CloseConnection() error {

pkg/exchange/manager.go

@@ -7,5 +7,5 @@ type Manager interface {
 	// WriteData command write data to exchange connection
 	WriteData(message []byte) (int, error)
 	// ReadData command is reading from receiver data
-	ReadData(data []byte) (int, error)
+	ReadData() ([]byte, error)
 }