Merge pull request #54 from libp2p/fix/graceful-close

fix: close gracefully
libp2p · Sep 30, 2019 · 4b1b551 · 4b1b551
2 parents ac7f16b + 7f87ef8
commit 4b1b551
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/p2p/transport/websocket/conn.go b/p2p/transport/websocket/conn.go
@@ -84,7 +84,11 @@ func (c *Conn) Write(b []byte) (n int, err error) {
 func (c *Conn) Close() error {
 	var err error
 	c.closeOnce.Do(func() {
-		err1 := c.Conn.WriteControl(ws.CloseMessage, nil, time.Now().Add(GracefulCloseTimeout))
+		err1 := c.Conn.WriteControl(
+			ws.CloseMessage,
+			ws.FormatCloseMessage(ws.CloseNormalClosure, "closed"),
+			time.Now().Add(GracefulCloseTimeout),
+		)
 		err2 := c.Conn.Close()
 		switch {
 		case err1 != nil: