gopkg/recws
2
0
Fork 0
Code Activity

make Close more atomic

Browse Source
This commit is contained in:
Nikolay Pavlovich
2022-08-12 01:57:40 +03:00
parent a25330e8f2
commit dd31a10749
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
recws.go
+4 -5
View File
@@ -82,13 +82,12 @@ func (rc *RecConn) getConn() *websocket.Conn {
// Close closes the underlying network connection without // Close closes the underlying network connection without
// sending or waiting for a close frame. // sending or waiting for a close frame.
func (rc *RecConn) Close() { func (rc *RecConn) Close() {
if rc.getConn() != nil { rc.mu.Lock()
rc.mu.Lock() if rc.Conn != nil {
rc.Conn.Close() rc.Conn.Close()
rc.mu.Unlock()
} }
rc.isConnected = false
rc.setIsConnected(false) rc.mu.Unlock()
} }
// Shutdown gracefully closes the connection by sending the websocket.CloseMessage. // Shutdown gracefully closes the connection by sending the websocket.CloseMessage.