Browse Source

Use parsed struct in OnMessage

Volodymyr Tkach 2 years ago
parent
commit
760d31ea15
2 changed files with 4 additions and 4 deletions
  1. 3 3
      pubsub/connection.go
  2. 1 1
      pubsub/go_reader.go

+ 3 - 3
pubsub/connection.go

@@ -36,7 +36,7 @@ type Connection struct {
 	eventOnDisconnect func(*Connection)
 	eventOnError      func(*Connection, error)
 	eventOnInfo       func(*Connection, string)
-	eventOnMessage    func(*Connection, []byte)
+	eventOnMessage    func(*Connection, *Answer)
 	eventOnPing       func(*Connection, time.Time)
 	eventOnPong       func(*Connection, time.Time, time.Time)
 }
@@ -92,7 +92,7 @@ func (c *Connection) onInfo(str string) {
 	}
 }
 
-func (c *Connection) onMessage(msg []byte) {
+func (c *Connection) onMessage(msg *Answer) {
 	if c.eventOnMessage != nil {
 		c.eventOnMessage(c, msg)
 	}
@@ -195,7 +195,7 @@ func (c *Connection) OnInfo(fn func(*Connection, string)) {
 	c.eventOnInfo = fn
 }
 
-func (c *Connection) OnMessage(fn func(*Connection, []byte)) {
+func (c *Connection) OnMessage(fn func(*Connection, *Answer)) {
 	c.eventOnMessage = fn
 }
 

+ 1 - 1
pubsub/go_reader.go

@@ -52,7 +52,7 @@ func go_reader(c *Connection) {
 									c.onInfo(fmt.Sprintf("type: %s, data: %#v", answer.Type, answer.Data))
 								}
 							} else {
-								c.onMessage(msg)
+								c.onMessage(&answer)
 							}
 						}
 					}