gopkg/recws
2
0
Fork 0
Code Activity

critical for writing from mutiple threads

Browse Source
This commit is contained in:
Roman Vasilyev
2019-01-31 23:10:02 -08:00
parent e4b6961e7e
commit a570815616
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
recws.go
+4 -1
View File
@@ -98,7 +98,9 @@ func (rc *RecConn) ReadMessage() (messageType int, message []byte, err error) {
func (rc *RecConn) WriteMessage(messageType int, data []byte) error { func (rc *RecConn) WriteMessage(messageType int, data []byte) error {
err := ErrNotConnected err := ErrNotConnected
if rc.IsConnected() { if rc.IsConnected() {
rc.mu.Lock()
err = rc.Conn.WriteMessage(messageType, data) err = rc.Conn.WriteMessage(messageType, data)
rc.mu.Unlock()
if err != nil { if err != nil {
rc.closeAndReconnect() rc.closeAndReconnect()
} }
@@ -116,7 +118,9 @@ func (rc *RecConn) WriteMessage(messageType int, data []byte) error {
func (rc *RecConn) WriteJSON(v interface{}) error { func (rc *RecConn) WriteJSON(v interface{}) error {
err := ErrNotConnected err := ErrNotConnected
if rc.IsConnected() { if rc.IsConnected() {
rc.mu.Lock()
err = rc.Conn.WriteJSON(v) err = rc.Conn.WriteJSON(v)
rc.mu.Unlock()
if err != nil { if err != nil {
rc.closeAndReconnect() rc.closeAndReconnect()
} }
@@ -210,7 +214,6 @@ func (rc *RecConn) setDefaultHandshakeTimeout() {
} }
} }
func (rc *RecConn) setDefaultDialer(handshakeTimeout time.Duration) { func (rc *RecConn) setDefaultDialer(handshakeTimeout time.Duration) {
rc.mu.Lock() rc.mu.Lock()
defer rc.mu.Unlock() defer rc.mu.Unlock()