From 8b34d3695e52ae2e935edd59fc5dce5cb480405b Mon Sep 17 00:00:00 2001
From: Gordon <46924906+FGadvancer@users.noreply.github.com>
Date: Fri, 8 Sep 2023 21:17:39 +0800
Subject: [PATCH] test: for pressure test.

Signed-off-by: Gordon <46924906+FGadvancer@users.noreply.github.com>
---
 internal/msggateway/client.go    | 6 +++---
 internal/msggateway/long_conn.go | 8 ++++++--
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/internal/msggateway/client.go b/internal/msggateway/client.go
index 96e4b02c6..6b6380d0b 100644
--- a/internal/msggateway/client.go
+++ b/internal/msggateway/client.go
@@ -59,7 +59,7 @@ const (
 	PongMessage = 10
 )
 
-type PongHandler func(string) error
+type PingPongHandler func(string) error
 
 type Client struct {
 	w              *sync.Mutex
@@ -107,7 +107,7 @@ func (c *Client) ResetClient(
 	c.token = token
 }
 
-func (c *Client) pongHandler(_ string) error {
+func (c *Client) pingHandler(_ string) error {
 	c.conn.SetReadDeadline(pongWait)
 	return nil
 }
@@ -122,7 +122,7 @@ func (c *Client) readMessage() {
 	}()
 	c.conn.SetReadLimit(maxMessageSize)
 	_ = c.conn.SetReadDeadline(pongWait)
-	c.conn.SetPongHandler(c.pongHandler)
+	c.conn.SetPingHandler(c.pingHandler)
 	for {
 		messageType, message, returnErr := c.conn.ReadMessage()
 		if returnErr != nil {
diff --git a/internal/msggateway/long_conn.go b/internal/msggateway/long_conn.go
index 59e4d5b45..f7676aec1 100644
--- a/internal/msggateway/long_conn.go
+++ b/internal/msggateway/long_conn.go
@@ -41,7 +41,8 @@ type LongConn interface {
 	SetConnNil()
 	// SetReadLimit sets the maximum size for a message read from the peer.bytes
 	SetReadLimit(limit int64)
-	SetPongHandler(handler PongHandler)
+	SetPongHandler(handler PingPongHandler)
+	SetPingHandler(handler PingPongHandler)
 	// GenerateLongConn Check the connection of the current and when it was sent are the same
 	GenerateLongConn(w http.ResponseWriter, r *http.Request) error
 }
@@ -116,9 +117,12 @@ func (d *GWebSocket) SetReadLimit(limit int64) {
 	d.conn.SetReadLimit(limit)
 }
 
-func (d *GWebSocket) SetPongHandler(handler PongHandler) {
+func (d *GWebSocket) SetPongHandler(handler PingPongHandler) {
 	d.conn.SetPongHandler(handler)
 }
+func (d *GWebSocket) SetPingHandler(handler PingPongHandler) {
+	d.conn.SetPingHandler(handler)
+}
 
 //func (d *GWebSocket) CheckSendConnDiffNow() bool {
 //	return d.conn == d.sendConn