add callbal BeforeRedial in WS interface

client.go

@@ -152,7 +152,8 @@ func (wsConn *WSConn) clientReaderLoop(process func(clientId string, msg WSMsg)
 			if err != ErrorClosed && !wsConn.closing {
 				log.WithFields(log.Fields{"clientID": wsConn.ClientId}).Error("WS client failed to read websocket message", err)
 				if AutoRedial {
-					// wsConn.c.Close() // close the underlying WS; this will stop ping goroutine.
+					wsConn.callback.BeforeRedial()
+
 					wsConn.redialLoop()
 
 					// Run this call back in a goroutine because THIS reader loop must be running to receive responses.

client_test.go

@@ -43,8 +43,12 @@ func (h TestClientWSHandler) Closed(wsConn *WSConn) {
 	log.Infof("WS client %s nclosed %d", wsConn.ClientId, countClosed)
 }
 
-func (h TestClientWSHandler) AfterRedial(wsConn *WSConn) {
+func (h TestClientWSHandler) BeforeRedial() {
 	countRedial++
+	log.Infof("WS client before redial count=%d", countRedial)
+}
+
+func (h TestClientWSHandler) AfterRedial(wsConn *WSConn) {
 	log.Infof("WS client after redial %s count %d", wsConn.ClientId, countRedial)
 }
 

server_test.go

@@ -97,8 +97,11 @@ func (h TestServerWSHandler) Closed(wsConn *WSConn) {
 	log.Infof("server conn %s closed nconns %d", wsConn.ClientId, countConnections)
 }
 
-func (h TestServerWSHandler) AfterRedial(wsConn *WSConn) {
+func (h TestServerWSHandler) BeforeRedial() {
 	h.countRedial++
+	log.Infof("WS client redial %v", h.countRedial)
+}
+func (h TestServerWSHandler) AfterRedial(wsConn *WSConn) {
 	log.Infof("WS client %s redial %d", wsConn.ClientId, h.countRedial)
 }
 

types.go

@@ -51,6 +51,7 @@ type WSServer interface {
 }
 
 type WSClient interface {
+	BeforeRedial()
 	AfterRedial(wsConn *WSConn)
 	Process(clientId string, msg WSMsg) (response WSMsg, err error)
 	Closed(wsConn *WSConn)