Server: minor fixes to previous commits

Author: luskaner

server/internal/keyRWMutex.go

@@ -14,39 +14,43 @@ func NewKeyRWMutex() *KeyRWMutex {
 }
 
 func (kl *KeyRWMutex) Lock(key interface{}) {
-	ok, lock := kl.lock(key)
+	ok, lock := kl.rlock(key)
 	if !ok {
 		lock = &sync.RWMutex{}
-		kl.mutexes[key] = lock
+		func() {
+			kl.mu.Lock()
+			defer kl.mu.Unlock()
+			kl.mutexes[key] = lock
+		}()
 	}
 	lock.Lock()
 }
 
-func (kl *KeyRWMutex) lock(key interface{}) (ok bool, lock *sync.RWMutex) {
+func (kl *KeyRWMutex) rlock(key interface{}) (ok bool, lock *sync.RWMutex) {
 	kl.mu.RLock()
 	defer kl.mu.RUnlock()
 	lock, ok = kl.mutexes[key]
 	return
 }
 
 func (kl *KeyRWMutex) RLock(key interface{}) {
-	ok, lock := kl.lock(key)
+	ok, lock := kl.rlock(key)
 
 	if ok {
 		lock.RLock()
 	}
 }
 
 func (kl *KeyRWMutex) Unlock(key interface{}) {
-	ok, lock := kl.lock(key)
+	ok, lock := kl.rlock(key)
 
 	if ok {
 		lock.Unlock()
 	}
 }
 
 func (kl *KeyRWMutex) RUnlock(key interface{}) {
-	ok, lock := kl.lock(key)
+	ok, lock := kl.rlock(key)
 
 	if ok {
 		lock.RUnlock()

server/internal/models/session.go

@@ -61,11 +61,11 @@ func (sess *Session) AddMessage(message internal.A) {
 			sess.messagesLock.RLock()
 			defer sess.messagesLock.RUnlock()
 			if sess.messagesIndex == uint8(len(sess.messages)) {
-				time.Sleep(time.Millisecond * 100)
 				wait = true
 			}
 		}()
 		if wait {
+			time.Sleep(time.Millisecond * 100)
 			continue
 		}
 		func() {