2
0

recommended code

This commit is contained in:
Roman Vasilyev
2019-02-11 12:12:24 -08:00
parent d61b9bc78b
commit e4150e8085
+10 -3
View File
@@ -63,13 +63,20 @@ func (rc *RecConn) setIsConnected(state bool) {
rc.isConnected = state rc.isConnected = state
} }
func (rc *RecConn) getConn() *websocket.Conn {
rc.mu.RLock()
defer rc.mu.RUnlock()
return rc.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.Conn != nil { if rc.getConn() != nil {
rc.mu.RLock() rc.mu.Lock()
rc.Conn.Close() rc.Conn.Close()
rc.mu.RUnlock() rc.mu.Unlock()
} }
rc.setIsConnected(false) rc.setIsConnected(false)