in progress

rest_websocket/client.go

@@ -7,5 +7,5 @@ func NewClient(addr string) (*Socket, error) {
 	if err != nil {
 		return nil, err
 	}
-	return NewSocket(conn), nil
+	return NewSocket(conn, false), nil
 }

rest_websocket/rest.go

@@ -46,7 +46,7 @@ func (s *Rest) handle(w http.ResponseWriter, r *http.Request) {
 		log.Println(err)
 	}
 	socket := NewSocketServer(
-		NewSocket(ws),
+		NewSocket(ws, true),
 		s.appConf,
 	)
 	s.appConf.app.Connect() <- socket

rest_websocket/socket.go

@@ -13,7 +13,7 @@ import (
 	"github.com/gorilla/websocket"
 )
 
-func NewSocket(conn *websocket.Conn) *Socket {
+func NewSocket(conn *websocket.Conn, pingEnable bool) *Socket {
 	ctx, cancel := context.WithCancel(context.Background())
 	ws := &Socket{
 		conn:        conn,
@@ -22,7 +22,7 @@ func NewSocket(conn *websocket.Conn) *Socket {
 		cancel:      cancel,
 		writeLocker: &sync.Mutex{},
 		chIn:        make(chan *rest.RequestStream, 10),
-		pingEnable:  true,
+		pingEnable:  pingEnable,
 	}
 	ws.lastWrite.Store(time.Now().Unix())
 	go ws.read()