gopkg/recws
2
0
Fork 0
Code Activity

Merge pull request #3 from nikepan/atomic-close

Browse Source 
make Close more atomic
This commit is contained in:
Nikolay Pavlovich
2022-08-12 01:58:26 +03:00
committed by GitHub
parent a25330e8f2 dd31a10749
commit 3e74d640ea
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.