memory transport: use write mutex lock when close

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2019-07-12 12:11:08 +03:00 · 2019-07-12 12:11:08 +03:00 · a72e1185da
commit a72e1185da
parent 5157241c88
1 changed files with 2 additions and 2 deletions
--- a/transport/memory/memory.go
+++ b/transport/memory/memory.go
@ -84,8 +84,8 @@ func (ms *memorySocket) Send(m *transport.Message) error {
 }

 func (ms *memorySocket) Close() error {
-	ms.RLock()
-	defer ms.RUnlock()
+	ms.Lock()
+	defer ms.Unlock()
 	select {
 	case <-ms.exit:
 		return nil